adding helper function to return electric charge properties

meta/TreeMasses.m

@@ -155,6 +155,7 @@
 IsSMDownQuark::usage="";
 IsSMQuark::usage="";
 IsSMParticle::usage="";
+IsElectricallyCharged::usage="";
 ContainsGoldstone::usage="";
 
 GetSMChargedLeptons::usage="";
@@ -173,6 +174,8 @@
 
 GetMass::usage="wraps M[] head around particle";
 
+GetElectricCharge::usage="Returns electric charge";
+
 StripGenerators::usage="removes all generators Lam, Sig, fSU2, fSU3
 and removes Delta with the given indices";
 
@@ -284,6 +287,8 @@
 IsChargino[p_] :=
     p === Parameters`GetParticleFromDescription["Charginos"];
 
+IsElectricallyCharged[par_] := GetElectricCharge[par] != 0;
+
 ContainsGoldstone[sym_] := MemberQ[GetGoldstoneBosons[] /. a_[{idx__}] :> a, sym];
 
 ContainsGoldstone[sym_[__]] := MemberQ[GetGoldstoneBosons[] /. a_[{idx__}] :> a, sym];
@@ -441,6 +446,8 @@
 GetMass[particle_[idx__]] := GetMass[particle][idx];
 GetMass[particle_Symbol] := FlexibleSUSY`M[particle];
 
+GetElectricCharge[par_] := SARAH`getElectricCharge[par];
+
 (* Returns list of pairs {p,v}, where p is the given golstone
    boson and v is the corresponding vector boson.