Fix ohmienvelope with wigglers (#870)

* Typo in GWigSymplecticRadPass * Added EnergyLossRadPass * fix Matlab bug: string -> char array
atcollab · Dec 11, 2024 · 3777663 · 3777663
1 parent 4a92361
commit 3777663
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/atmat/atphysics/Radiation/atdiffmat.m b/atmat/atphysics/Radiation/atdiffmat.m
@@ -15,7 +15,9 @@
 
 newmethods = {'BndMPoleSymplectic4RadPass', ...
               'StrMPoleSymplectic4RadPass', ...
-              'ExactMultipoleRadPass'};
+              'ExactMultipoleRadPass',...
+              'GWigSymplecticRadPass',...
+              'EnergyLossRadPass'};
 
 NumElements=length(ring);
 
@@ -62,7 +64,7 @@
 
     function elem=substitute(elem)
         if ~any(strcmp(elem.PassMethod, newmethods))
-            elem.PassMethod = "BndMPoleSymplectic4RadPass";
+            elem.PassMethod = 'BndMPoleSymplectic4RadPass';
         end
     end
 end

diff --git a/pyat/at/physics/radiation.py b/pyat/at/physics/radiation.py
@@ -31,7 +31,8 @@
     "BndMPoleSymplectic4RadPass",
     "StrMPoleSymplectic4RadPass",
     "ExactMultipoleRadPass",
-    "GWigSymplectic4RadPass",
+    "GWigSymplecticRadPass",
+    "EnergyLossRadPass",
 }
 
 _NSTEP = 60  # nb slices in a wiggler period