Skip to content

Commit 0a85ba3

Browse files
committed
add TrajectStyle
1 parent fb23bee commit 0a85ba3

File tree

1 file changed

+14
-1
lines changed

1 file changed

+14
-1
lines changed

src/trajectory.jl

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,10 @@
1-
export Trajectory
1+
export Trajectory, TrajectoryStyle, SyncTrajectoryStyle, AsyncTrajectoryStyle
22

33
using Base.Threads
44

5+
struct AsyncTrajectoryStyle end
6+
struct SyncTrajectoryStyle end
7+
58

69
"""
710
Trajectory(container, sampler, controler)
@@ -53,6 +56,16 @@ Base.@kwdef struct Trajectory{C,S,T}
5356
end
5457
end
5558

59+
TrajectoryStyle(::Trajectory) = SyncTrajectoryStyle()
60+
TrajectoryStyle(::Trajectory{<:Any,<:Any,<:AsyncInsertSampleRatioControler}) = AsyncTrajectoryStyle()
61+
62+
Base.bind(::Trajectory, ::Task) = nothing
63+
64+
function Base.bind(t::Trajectory{<:Any,<:Any,<:AsyncInsertSampleRatioControler}, tassk)
65+
bind(t.controler.ch_in, task)
66+
bind(t.controler.ch_out, task)
67+
end
68+
5669

5770
Base.push!(t::Trajectory; kw...) = push!(t, values(kw))
5871

0 commit comments

Comments
 (0)