Add derivative for GuidedAmplitude

Author: goerz

src/derivs.jl

@@ -1,5 +1,5 @@
 using QuantumPropagators.Generators: Generator, Operator, _make_generator, evaluate
-using QuantumPropagators.Amplitudes: LockedAmplitude, ShapedAmplitude
+using QuantumPropagators.Amplitudes: LockedAmplitude, ShapedAmplitude, GuidedAmplitude
 
 
 """Get a vector of the derivatives of `generator` w.r.t. each control.
@@ -102,3 +102,6 @@ get_control_deriv(ampl::LockedAmplitude, control) = 0.0
 
 get_control_deriv(ampl::ShapedAmplitude, control) =
     (control ≡ ampl.control) ? LockedAmplitude(ampl.shape) : 0.0
+
+get_control_deriv(ampl::GuidedAmplitude, control) =
+    (control ≡ ampl.control) ? LockedAmplitude(ampl.shape) : 0.0

test/test_derivs.jl

@@ -1,12 +1,11 @@
 using Test
 using LinearAlgebra
 using QuantumControl.Controls: get_control_deriv, get_control_derivs
-using QuantumPropagators.Generators
-using QuantumPropagators.Controls
 using QuantumPropagators: Generator, Operator
+using QuantumPropagators.Controls: evaluate, get_controls
 using QuantumControlTestUtils.RandomObjects: random_matrix, random_state_vector
 using QuantumControl.Interfaces: check_generator, check_amplitude
-using QuantumPropagators.Amplitudes: LockedAmplitude, ShapedAmplitude
+using QuantumPropagators.Amplitudes: LockedAmplitude, ShapedAmplitude, GuidedAmplitude
 import QuantumPropagators
 import QuantumControl
 
@@ -166,3 +165,29 @@ end
     @test get_control_deriv(ampl2, control2) == LockedAmplitude(ampl2.shape)
 
 end
+
+
+@testset "GuidedAmplitude  get_control_derivs" begin
+
+    shape(t) = 1.0
+    guide(t) = 0.1
+    control(t) = 0.5
+    tlist = [0.0, 0.5, 1.0]
+
+    ampl1 = GuidedAmplitude(control; shape, guide)
+    ampl2 = GuidedAmplitude(control, tlist; shape, guide)
+
+    @test get_control_deriv(ampl1, t -> 0.0) == 0.0
+    @test get_control_deriv(ampl1, shape) == 0.0
+    @test get_control_deriv(ampl1, control) == LockedAmplitude(shape)
+
+    @test get_control_deriv(ampl2, t -> 0.0) == 0.0
+    @test get_control_deriv(ampl2, shape) == 0.0
+    control2 = get_controls(ampl2)[1]
+    @test control2 ≢ control  # should have been discretized
+    @test ampl2.shape ≢ shape  # should have been discretized
+    @test control2 isa Vector{Float64}
+    @test ampl2.shape isa Vector{Float64}
+    @test get_control_deriv(ampl2, control2) == LockedAmplitude(ampl2.shape)
+
+end