desy-ml · austin-hoover · Sep 28, 2025 · Sep 28, 2025 · Oct 7, 2025 · Oct 7, 2025
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -4,8 +4,12 @@
 
 ### 🚨 Breaking Changes
 
+- The `SpaceChargeKick` elements has been renamed to `SpaceChargeKick3D` in order to reflect the difference to the new `SpaceChargeKick2D` element (see #576) (@austin-hoover, @RemiLehe, @jank324)
+
 ### 🚀 Features
 
+- Add a `SpaceChargeKick2D` element that applies a space charge kick using the 2D integrated Green function method (see #576) (@austin-hoover, @RemiLehe, @jank324)
+
 ### 🐛 Bug fixes
 
 - RPN parsing was made more robust to missing whitespace in expressions, and arc trig expressions were added to infix parsing. (see #585) (@amylizzle)

diff --git a/cheetah/__init__.py b/cheetah/__init__.py
@@ -17,7 +17,8 @@
     Segment,
     Sextupole,
     Solenoid,
-    SpaceChargeKick,
+    SpaceChargeKick2D,
+    SpaceChargeKick3D,
     TransverseDeflectingCavity,
     Undulator,
     VerticalCorrector,

diff --git a/cheetah/accelerator/__init__.py b/cheetah/accelerator/__init__.py
@@ -14,7 +14,8 @@
 from .segment import Segment  # noqa: F401
 from .sextupole import Sextupole  # noqa: F401
 from .solenoid import Solenoid  # noqa: F401
-from .space_charge_kick import SpaceChargeKick  # noqa: F401
+from .space_charge_kick_2d import SpaceChargeKick2D  # noqa: F401
+from .space_charge_kick_3d import SpaceChargeKick3D  # noqa: F401
 from .transverse_deflecting_cavity import TransverseDeflectingCavity  # noqa: F401
 from .undulator import Undulator  # noqa: F401
 from .vertical_corrector import VerticalCorrector  # noqa: F401