Improve RF cavity treatment of endpoints.

BLAST-ImpactX · Apr 29, 2024 · f66a081 · f66a081
1 parent 53a00d7
commit f66a081
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/particles/elements/RFCavity.H b/src/particles/elements/RFCavity.H
@@ -377,6 +377,22 @@ namespace RFCavityData
                       zlen*sin_data[j]*cos(j*2*pi*z/zlen)/(j*2*pi);
                }
             }
+            else if (z < -zmid)
+            {
+               efieldint = -zmid*efield;
+               for (int j=1; j < m_ncoef; ++j)
+               {
+                 efieldint = efieldint - zlen*sin_data[j]*cos(j*pi)/(j*2*pi);
+               }
+            }
+            else
+            {
+               efieldint = zmid*efield;
+               for (int j=1; j < m_ncoef; ++j)
+               {
+                 efieldint = efieldint - zlen*sin_data[j]*cos(j*pi)/(j*2*pi);
+               }
+            }
             return std::make_tuple(efield, efieldp, efieldint);
         }