From aec96a08041e422f015da6ef049739ede17a02ee Mon Sep 17 00:00:00 2001
From: "Christopher M. Pierce" <chris@chris-pierce.com>
Date: Tue, 2 Jul 2024 19:36:07 -0700
Subject: [PATCH] add muon and antimuon to `CHARGE_OF`, `CHARGE_STATE`, and
 `MASS_OF`

---
 pmd_beamphysics/species.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/pmd_beamphysics/species.py b/pmd_beamphysics/species.py
index 964c025..cdab34c 100644
--- a/pmd_beamphysics/species.py
+++ b/pmd_beamphysics/species.py
@@ -8,9 +8,8 @@
 
 mec2 = scipy.constants.value('electron mass energy equivalent in MeV')*1e6
 mpc2 = scipy.constants.value('proton mass energy equivalent in MeV')*1e6
-
+mmc2 = scipy.constants.value('muon mass energy equivalent in MeV')*1e6
 mhmc2 = mpc2 + mec2 * 2 # H- mass energy equivalent in MeV
-
 mH2pc2 = 2*mpc2 + mec2 # Molecular Hydrogen Ion H2+
 
 e_charge = scipy.constants.e
@@ -21,6 +20,8 @@
             'proton': e_charge,
             'H-': -e_charge,
             'H2+': e_charge,
+            'muon': -e_charge,
+            'antimuon': e_charge,
             }
 
 CHARGE_STATE = {
@@ -29,6 +30,8 @@
     'proton': 1,
     'H-': -1,
     'H2+': +1,
+    'muon': -1,
+    'antimuon': 1,
     }
 
 
@@ -37,6 +40,8 @@
            'proton': mpc2,
            'H-': mhmc2,
            'H2+': mH2pc2,
+           'muon': mmc2,
+           'antimuon': mmc2,
            }