change jump_times and jump_which definitions

Author: albertomercurio

src/time_evolution/callback_helpers/mcsolve_callback_helpers.jl

@@ -28,9 +28,9 @@ struct LindbladJump{
     cache_mc::CT
     weights_mc::WT
     cumsum_weights_mc::WT
-    jump_times::JTT
-    jump_which::JWT
-    jump_times_which_idx::JTWIT
+    col_times::JTT
+    col_which::JWT
+    col_times_which_idx::JTWIT
 end
 
 (f::LindbladJump)(integrator) = _lindblad_jump_affect!(
@@ -42,9 +42,9 @@ end
     f.cache_mc,
     f.weights_mc,
     f.cumsum_weights_mc,
-    f.jump_times,
-    f.jump_which,
-    f.jump_times_which_idx,
+    f.col_times,
+    f.col_which,
+    f.col_times_which_idx,
 )
 
 ##
@@ -71,9 +71,9 @@ function _generate_mcsolve_kwargs(ψ0, T, e_ops, tlist, c_ops, jump_callback, rn
     weights_mc = Vector{Float64}(undef, length(c_ops))
     cumsum_weights_mc = similar(weights_mc)
 
-    jump_times = Vector{Float64}(undef, JUMP_TIMES_WHICH_INIT_SIZE)
-    jump_which = Vector{Int}(undef, JUMP_TIMES_WHICH_INIT_SIZE)
-    jump_times_which_idx = Ref(1)
+    col_times = Vector{Float64}(undef, COL_TIMES_WHICH_INIT_SIZE)
+    col_which = Vector{Int}(undef, COL_TIMES_WHICH_INIT_SIZE)
+    col_times_which_idx = Ref(1)
 
     random_n = Ref(rand(rng))
 
@@ -85,9 +85,9 @@ function _generate_mcsolve_kwargs(ψ0, T, e_ops, tlist, c_ops, jump_callback, rn
         cache_mc,
         weights_mc,
         cumsum_weights_mc,
-        jump_times,
-        jump_which,
-        jump_times_which_idx,
+        col_times,
+        col_which,
+        col_times_which_idx,
     )
 
     if jump_callback isa DiscreteLindbladJumpCallback
@@ -129,9 +129,9 @@ function _lindblad_jump_affect!(
     cache_mc,
     weights_mc,
     cumsum_weights_mc,
-    jump_times,
-    jump_which,
-    jump_times_which_idx,
+    col_times,
+    col_which,
+    col_times_which_idx,
 )
     ψ = integrator.u
 
@@ -147,13 +147,13 @@ function _lindblad_jump_affect!(
 
     random_n[] = rand(traj_rng)
 
-    idx = jump_times_which_idx[]
-    @inbounds jump_times[idx] = integrator.t
-    @inbounds jump_which[idx] = collapse_idx
-    jump_times_which_idx[] += 1
-    if jump_times_which_idx[] > length(jump_times)
-        resize!(jump_times, length(jump_times) + JUMP_TIMES_WHICH_INIT_SIZE)
-        resize!(jump_which, length(jump_which) + JUMP_TIMES_WHICH_INIT_SIZE)
+    idx = col_times_which_idx[]
+    @inbounds col_times[idx] = integrator.t
+    @inbounds col_which[idx] = collapse_idx
+    col_times_which_idx[] += 1
+    if col_times_which_idx[] > length(col_times)
+        resize!(col_times, length(col_times) + COL_TIMES_WHICH_INIT_SIZE)
+        resize!(col_which, length(col_which) + COL_TIMES_WHICH_INIT_SIZE)
     end
     u_modified!(integrator, true)
     return nothing
@@ -309,9 +309,9 @@ function _similar_affect!(affect::LindbladJump, traj_rng)
     cache_mc = similar(affect.cache_mc)
     weights_mc = similar(affect.weights_mc)
     cumsum_weights_mc = similar(affect.cumsum_weights_mc)
-    jump_times = similar(affect.jump_times)
-    jump_which = similar(affect.jump_which)
-    jump_times_which_idx = Ref(1)
+    col_times = similar(affect.col_times)
+    col_which = similar(affect.col_which)
+    col_times_which_idx = Ref(1)
 
     return LindbladJump(
         affect.c_ops,
@@ -321,9 +321,9 @@ function _similar_affect!(affect::LindbladJump, traj_rng)
         cache_mc,
         weights_mc,
         cumsum_weights_mc,
-        jump_times,
-        jump_which,
-        jump_times_which_idx,
+        col_times,
+        col_which,
+        col_times_which_idx,
     )
 end
 

src/time_evolution/mcsolve.jl

@@ -14,9 +14,9 @@ end
 
 # Standard output function
 function _mcsolve_output_func(sol, i)
-    idx = _mc_get_jump_callback(sol).affect!.jump_times_which_idx[]
-    resize!(_mc_get_jump_callback(sol).affect!.jump_times, idx - 1)
-    resize!(_mc_get_jump_callback(sol).affect!.jump_which, idx - 1)
+    idx = _mc_get_jump_callback(sol).affect!.col_times_which_idx[]
+    resize!(_mc_get_jump_callback(sol).affect!.col_times, idx - 1)
+    resize!(_mc_get_jump_callback(sol).affect!.col_which, idx - 1)
     return (sol, false)
 end
 
@@ -403,8 +403,8 @@ function mcsolve(
     _expvals_all = _expvals_sol_1 isa Nothing ? nothing : map(i -> _mcsolve_get_expvals(sol[:, i]), eachindex(sol))
     expvals_all = _expvals_all isa Nothing ? nothing : stack(_expvals_all)
     states = map(i -> _normalize_state!.(sol[:, i].u, Ref(dims), normalize_states), eachindex(sol))
-    jump_times = map(i -> _mc_get_jump_callback(sol[:, i]).affect!.jump_times, eachindex(sol))
-    jump_which = map(i -> _mc_get_jump_callback(sol[:, i]).affect!.jump_which, eachindex(sol))
+    col_times = map(i -> _mc_get_jump_callback(sol[:, i]).affect!.col_times, eachindex(sol))
+    col_which = map(i -> _mc_get_jump_callback(sol[:, i]).affect!.col_which, eachindex(sol))
 
     expvals = _expvals_sol_1 isa Nothing ? nothing : dropdims(sum(expvals_all, dims = 3), dims = 3) ./ length(sol)
 
@@ -414,8 +414,8 @@ function mcsolve(
         states,
         expvals,
         expvals_all,
-        jump_times,
-        jump_which,
+        col_times,
+        col_which,
         sol.converged,
         _sol_1.alg,
         NamedTuple(_sol_1.prob.kwargs).abstol,

src/time_evolution/time_evolution.jl

@@ -4,7 +4,7 @@ export liouvillian_floquet, liouvillian_generalized
 
 const DEFAULT_ODE_SOLVER_OPTIONS = (abstol = 1e-8, reltol = 1e-6, save_everystep = false, save_end = true)
 const DEFAULT_SDE_SOLVER_OPTIONS = (abstol = 1e-2, reltol = 1e-2, save_everystep = false, save_end = true)
-const JUMP_TIMES_WHICH_INIT_SIZE = 200
+const COL_TIMES_WHICH_INIT_SIZE = 200
 
 @doc raw"""
     struct TimeEvolutionProblem
@@ -100,8 +100,8 @@ A structure storing the results and some information from solving quantum trajec
 - `states::Vector{Vector{QuantumObject}}`: The list of result states in each trajectory.
 - `expect::Union{AbstractMatrix,Nothing}`: The expectation values (averaging all trajectories) corresponding to each time point in `times`.
 - `expect_all::Union{AbstractMatrix,Nothing}`: The expectation values corresponding to each trajectory and each time point in `times`
-- `jump_times::Vector{Vector{Real}}`: The time records of every quantum jump occurred in each trajectory.
-- `jump_which::Vector{Vector{Int}}`: The indices of the jump operators in `c_ops` that describe the corresponding quantum jumps occurred in each trajectory.
+- `col_times::Vector{Vector{Real}}`: The time records of every quantum jump occurred in each trajectory.
+- `col_which::Vector{Vector{Int}}`: The indices of which collapse operator was responsible for each quantum jump in `col_times`.
 - `converged::Bool`: Whether the solution is converged or not.
 - `alg`: The algorithm which is used during the solving process.
 - `abstol::Real`: The absolute tolerance which is used during the solving process.
@@ -123,8 +123,8 @@ struct TimeEvolutionMCSol{
     states::TS
     expect::TE
     expect_all::TEA
-    jump_times::TJT
-    jump_which::TJW
+    col_times::TJT
+    col_which::TJW
     converged::Bool
     alg::AlgT
     abstol::AT

test/core-test/time_evolution.jl

@@ -661,8 +661,8 @@
 
             @test sol_mc1.expect ≈ sol_mc2.expect atol = 1e-10
             @test sol_mc1.expect_all ≈ sol_mc2.expect_all atol = 1e-10
-            @test sol_mc1.jump_times ≈ sol_mc2.jump_times atol = 1e-10
-            @test sol_mc1.jump_which ≈ sol_mc2.jump_which atol = 1e-10
+            @test sol_mc1.col_times ≈ sol_mc2.col_times atol = 1e-10
+            @test sol_mc1.col_which ≈ sol_mc2.col_which atol = 1e-10
 
             @test sol_mc1.expect_all ≈ sol_mc3.expect_all[:, :, 1:500] atol = 1e-10