Update model.py

Adding a method to input mediator and dark-matter masses

Author: yhaddad

src/lhctodd/model.py

@@ -112,6 +112,9 @@ def __init__(self, g_chi=1.0, g_quark=0.25, g_lepton=0.0, label=None):
         self.label = label
 
     def from_csv(self, filename, delimiter=","):
+        """
+        filename: txt/csv file with mediator and dark matter mass the first and second column
+        """
         print("filename : ", filename, delimiter)
         my_data = np.genfromtxt(filename, delimiter=delimiter)
         self.med_mass = my_data[:,0] # mediator
@@ -120,6 +123,18 @@ def from_csv(self, filename, delimiter=","):
             self.sigma(self.med_mass, self.chi_mass),
             self.chi_mass
         ]).T
+    
+    def from_array(self, med_mass, chi_mass, delimiter=","):
+        """
+        med_mass: array-like mediator mass
+        chi_mass: array-like dark matter candidate mass
+        """
+        self.med_mass = med_mass # mediator
+        self.chi_mass = chi_mass # darkmatter
+        return np.vstack([
+            self.sigma(self.med_mass, self.chi_mass),
+            self.chi_mass
+        ]).T
 
     def sigma(self, med_mass, chi_mass):
         raise NotImplementedError("sigma not implemented!")