Check elem track needs energy

atintegrators/BeamLoadingCavityPass.c

@@ -223,7 +223,14 @@ ExportMode struct elem *trackFunction(const atElem *ElemData,struct elem *Elem,
         Energy=atGetOptionalDouble(ElemData,"Energy",Param->energy); check_error();
         z_cuts=atGetOptionalDoubleArray(ElemData,"ZCuts"); check_error();
         feedback_angle_offset=atGetOptionalDouble(ElemData,"feedback_angle_offset", 0.0); check_error();
-
+
+        /* Check energy */
+        Energy = atEnergy(Param->energy, Energy);
+        if (Energy == 0) {
+            atError("Energy needs to be defined. Check lattice parameters or pass method options.\n");
+            check_error();
+        }
+
         int dimsth[] = {Param->nbunch*nslice*nturns, 4};
         atCheckArrayDims(ElemData,"_turnhistory", 2, dimsth); check_error();
         int dimsvb[] = {Param->nbunch, 2};

atintegrators/BndMPoleSymplectic4E2RadPass.c

@@ -232,6 +232,13 @@ ExportMode struct elem *trackFunction(const atElem *ElemData,struct elem *Elem,
         RApertures=atGetOptionalDoubleArray(ElemData,"RApertures"); check_error();
         KickAngle=atGetOptionalDoubleArray(ElemData,"KickAngle"); check_error();
 
+        /* Check energy */
+        Energy = atEnergy(Param->energy, Energy);
+        if (Energy == 0) {
+            atError("Energy needs to be defined. Check lattice parameters or pass method options.\n");
+            check_error();
+        }
+
         Elem = (struct elem*)atMalloc(sizeof(struct elem));
         Elem->Length=Length;
         Elem->PolynomA=PolynomA;

atintegrators/BndMPoleSymplectic4QuantPass.c

@@ -215,6 +215,13 @@ ExportMode struct elem *trackFunction(const atElem *ElemData,struct elem *Elem,
         RApertures=atGetOptionalDoubleArray(ElemData,"RApertures"); check_error();
         KickAngle=atGetOptionalDoubleArray(ElemData,"KickAngle"); check_error();
 
+        /* Check energy */
+        Energy = atEnergy(Param->energy, Energy);
+        if (Energy == 0) {
+            atError("Energy needs to be defined. Check lattice parameters or pass method options.\n");
+            check_error();
+        }
+
         Elem = (struct elem*)atMalloc(sizeof(struct elem));
         Elem->Length=Length;
         Elem->PolynomA=PolynomA;

atintegrators/BndMPoleSymplectic4RadPass.c

@@ -168,6 +168,13 @@ ExportMode struct elem *trackFunction(const atElem *ElemData,struct elem *Elem,
         RApertures=atGetOptionalDoubleArray(ElemData,"RApertures"); check_error();
         KickAngle=atGetOptionalDoubleArray(ElemData,"KickAngle"); check_error();
 
+        /* Check energy */
+        Energy = atEnergy(Param->energy, Energy);
+        if (Energy == 0) {
+            atError("Energy needs to be defined. Check lattice parameters or pass method options.\n");
+            check_error();
+        }
+
         Elem = (struct elem*)atMalloc(sizeof(struct elem));
         Elem->Length=Length;
         Elem->PolynomA=PolynomA;

atintegrators/CrabCavityPass.c

@@ -101,6 +101,13 @@ ExportMode struct elem *trackFunction(const atElem *ElemData,struct elem *Elem,
 		SigPhi = atGetOptionalDouble(ElemData,"SigPhi",0.0); check_error();
 		SigVV = atGetOptionalDouble(ElemData,"SigVV",0.0); check_error();
 
+        /* Check energy */
+        Energy = atEnergy(Param->energy, Energy);
+        if (Energy == 0) {
+            atError("Energy needs to be defined. Check lattice parameters or pass method options.\n");
+            check_error();
+        }
+
         Elem = (struct elem*)atMalloc(sizeof(struct elem));
         Elem->Length=Length;
         Elem->Vx=Voltages[0];

atintegrators/ExactMultipoleRadPass.c

@@ -138,6 +138,13 @@ ExportMode struct elem *trackFunction(const atElem *ElemData, struct elem *Elem,
         atError("NumIntSteps must be positive"); check_error();
     }
 
+    /* Check energy */
+    Energy = atEnergy(Param->energy, Energy);
+    if (Energy == 0) {
+        atError("Energy needs to be defined. Check lattice parameters or pass method options.\n");
+        check_error();
+    }
+
     Elem = (struct elem *)atMalloc(sizeof(struct elem));
     Elem->Length = Length;
     Elem->PolynomA = PolynomA;

atintegrators/ExactRectBendRadPass.c

@@ -173,6 +173,13 @@ ExportMode struct elem *trackFunction(const atElem *ElemData,struct elem *Elem,
             atError("NumIntSteps == 0 not allowed with radiation"); check_error();
         }
 
+        /* Check energy */
+        Energy = atEnergy(Param->energy, Energy);
+        if (Energy == 0) {
+            atError("Energy needs to be defined. Check lattice parameters or pass method options.\n");
+            check_error();
+        }
+
         Elem = (struct elem*)atMalloc(sizeof(struct elem));
         Elem->Length=Length;
         Elem->PolynomA=PolynomA;

atintegrators/ExactRectangularBendRadPass.c

@@ -175,6 +175,13 @@ ExportMode struct elem *trackFunction(const atElem *ElemData,struct elem *Elem,
             atError("NumIntSteps must be positive"); check_error();
         }
 
+        /* Check energy */
+        Energy = atEnergy(Param->energy, Energy);
+        if (Energy == 0) {
+            atError("Energy needs to be defined. Check lattice parameters or pass method options.\n");
+            check_error();
+        }
+
         Elem = (struct elem*)atMalloc(sizeof(struct elem));
         Elem->Length=Length;
         Elem->PolynomA=PolynomA;

atintegrators/ExactSectorBendRadPass.c

@@ -160,6 +160,13 @@ ExportMode struct elem *trackFunction(const atElem *ElemData,struct elem *Elem,
             atError("NumIntSteps == 0 not allowed with radiation"); check_error();
         }
 
+        /* Check energy */
+        Energy = atEnergy(Param->energy, Energy);
+        if (Energy == 0) {
+            atError("Energy needs to be defined. Check lattice parameters or pass method options.\n");
+            check_error();
+        }
+
         Elem = (struct elem*)atMalloc(sizeof(struct elem));
         Elem->Length=Length;
         Elem->PolynomA=PolynomA;

atintegrators/GWigSymplecticPass.c

@@ -144,6 +144,13 @@ ExportMode struct elem *trackFunction(const atElem *ElemData,struct elem *Elem,
         T1 = atGetOptionalDoubleArray(ElemData, "T1"); check_error();
         T2 = atGetOptionalDoubleArray(ElemData, "T2"); check_error();
 
+        /* Check energy */
+        Energy = atEnergy(Param->energy, Energy);
+        if (Energy == 0) {
+            atError("Energy needs to be defined. Check lattice parameters or pass method options.\n");
+            check_error();
+        }
+
         Elem = (struct elem*)atMalloc(sizeof(struct elem));
         Elem->Energy=Energy;
         Elem->Length=Ltot;

atintegrators/GWigSymplecticRadPass.c

@@ -270,6 +270,13 @@ ExportMode struct elem *trackFunction(const atElem *ElemData,struct elem *Elem,
         T1 = atGetOptionalDoubleArray(ElemData, "T1"); check_error();
         T2 = atGetOptionalDoubleArray(ElemData, "T2"); check_error();
 
+        /* Check energy */
+        Energy = atEnergy(Param->energy, Energy);
+        if (Energy == 0) {
+            atError("Energy needs to be defined. Check lattice parameters or pass method options.\n");
+            check_error();
+        }
+
         Elem = (struct elem*)atMalloc(sizeof(struct elem));
         Elem->Energy=Energy;
         Elem->Length=Ltot;

atintegrators/RFCavityPass.c

@@ -47,6 +47,14 @@ ExportMode struct elem *trackFunction(const atElem *ElemData,struct elem *Elem,
         Energy=atGetOptionalDouble(ElemData,"Energy",energy); check_error();
         TimeLag=atGetOptionalDouble(ElemData,"TimeLag",0); check_error();
         PhaseLag=atGetOptionalDouble(ElemData,"PhaseLag",0); check_error();
+
+        /* Check energy */
+        Energy = atEnergy(Param->energy, Energy);
+        if (Energy == 0) {
+            atError("Energy needs to be defined. Check lattice parameters or pass method options.\n");
+            check_error();
+        }
+
         Elem = (struct elem*)atMalloc(sizeof(struct elem));
         Elem->Length=Length;
         Elem->Voltage=Voltage;
@@ -56,7 +64,7 @@ ExportMode struct elem *trackFunction(const atElem *ElemData,struct elem *Elem,
         Elem->TimeLag=TimeLag;
         Elem->PhaseLag=PhaseLag;
     }
-    if (energy == 0.0) energy = Elem->Energy;
+    energy = atEnergy(energy, Elem->Energy);
 
     RFCavityPass(r_in, Elem->Length, Elem->Voltage/energy, Elem->Frequency, Elem->HarmNumber, Elem->TimeLag,
                  Elem->PhaseLag, nturn, T0, num_particles);

atintegrators/StrMPoleSymplectic4QuantPass.c

@@ -184,6 +184,13 @@ ExportMode struct elem *trackFunction(const atElem *ElemData,struct elem *Elem,
         RApertures=atGetOptionalDoubleArray(ElemData,"RApertures"); check_error();
         KickAngle=atGetOptionalDoubleArray(ElemData,"KickAngle"); check_error();
 
+        /* Check energy */
+        Energy = atEnergy(Param->energy, Energy);
+        if (Energy == 0) {
+            atError("Energy needs to be defined. Check lattice parameters or pass method options.\n");
+            check_error();
+        }
+
         Elem = (struct elem*)atMalloc(sizeof(struct elem));
         Elem->Length=Length;
         Elem->PolynomA=PolynomA;

atintegrators/StrMPoleSymplectic4RadPass.c

@@ -144,6 +144,13 @@ ExportMode struct elem *trackFunction(const atElem *ElemData,struct elem *Elem,
         RApertures=atGetOptionalDoubleArray(ElemData,"RApertures"); check_error();
         KickAngle=atGetOptionalDoubleArray(ElemData,"KickAngle"); check_error();
 
+        /* Check energy */
+        Energy = atEnergy(Param->energy, Energy);
+        if (Energy == 0) {
+            atError("Energy needs to be defined. Check lattice parameters or pass method options.\n");
+            check_error();
+        }
+
         Elem = (struct elem*)atMalloc(sizeof(struct elem));
         Elem->Length=Length;
         Elem->PolynomA=PolynomA;

atmat/atphysics/Radiation/atdiffmat.m

@@ -56,7 +56,7 @@
         end
         % Calculate 6-by-6 linear transfer matrix in each element
         % near the equilibrium orbit
-        m=findelemm66(elem,passm,orbit);
+        m=findelemm66(elem,passm,orbit,'Energy',energy);
         % Cumulative diffusion matrix of the entire ring
         BCUM = m*BCUM*m' + bdiff;
         btx=BCUM;

atmat/attrack/elempass.m

@@ -26,6 +26,28 @@
     props.Particle=particle;
 end
 
+method_req_energy = { 'BeamLoadingCavityPass' , ...
+                      'BndMPoleSymplectic4E2RadPass', ...
+                      'BndMPoleSymplectic4QuantPass', ...
+                      'BndMPoleSymplectic4RadPass', ...
+                      'CrabCavityPass', ...
+                      'ExactMultipoleRadPass', ...
+                      'ExactRectangularBendRadPass', ...
+                      'ExactRectBendRadPass', ...
+                      'ExactSectorBendRadPass', ...
+                      'GWigSymplecticPass', ...
+                      'GWigSymplecticRadPass', ...
+                      'RFCavityPass', ...
+                      'StrMPoleSymplectic4QuantPass', ...
+                      'StrMPoleSymplectic4RadPass' ...
+                      };
+
+if any(strcmp(method_req_energy, methodname))
+    if props.Energy <= 0
+        error("Energy parameter must be defined.");
+    end
+end
+
 rout = feval(methodname,elem,rin,props);
 
 end