From 16bab2fedde012b0894de4edb5005e38e474dbe5 Mon Sep 17 00:00:00 2001
From: Michiel Cottaar <michiel.cottaar@ndcn.ox.ac.uk>
Date: Thu, 25 Apr 2024 13:46:36 +0100
Subject: [PATCH] Make instant gradient adjustable
---
src/components/instant_gradients.jl | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/src/components/instant_gradients.jl b/src/components/instant_gradients.jl
index ba0eddc..a167872 100644
--- a/src/components/instant_gradients.jl
+++ b/src/components/instant_gradients.jl
@@ -1,7 +1,7 @@
module InstantGradients
import StaticArrays: SVector, SMatrix
import JuMP: @constraint
-import ...Variables: VariableType, duration, qval, bmat_gradient, get_free_variable, set_simple_constraints!, effective_time, make_generic, adjust_internal
+import ...Variables: VariableType, duration, qval, bmat_gradient, get_free_variable, set_simple_constraints!, effective_time, make_generic, adjust_internal, adjustable
import ...BuildSequences: global_model
import ..AbstractTypes: EventComponent, GradientWaveform
@@ -70,6 +70,8 @@ bmat_gradient(::InstantGradient, qstart=nothing) = zero(SMatrix{3, 3, Float64, 3
make_generic(ig::InstantGradient) = ig
+adjustable(::InstantGradient) = :gradient
+
function adjust_internal(ig::InstantGradient1D; orientation=nothing, scale=1., rotation=nothing)
if !isnothing(orientation) && !isnothing(rotation)
error("Cannot set both the gradient orientation and rotation.")
--
GitLab