adding helper function to extract masses of unrotated particles

FlexibleSUSY · Apr 2, 2016 · ff2a6a4 · ff2a6a4
1 parent 13dc9fc
commit ff2a6a4
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/meta/TreeMasses.m b/meta/TreeMasses.m
@@ -6,6 +6,9 @@
 ConvertSarahMassMatrices::usage="creates list of mass matrices using
 SARAH's MassMatrix[] function";
 
+GetUnmixedParticleMasses::usage="returns list of masses of unmixed
+ particles";
+
 CreateMassGetter::usage="creates a C function for
 the mass getter";
 
@@ -776,6 +779,18 @@
            Return[result];
           ];
 
+(* returns masses of unmixed particles *)
+GetUnmixedParticleMasses[] :=
+    Module[{particles = {}, result = {}, eigenstateName, massMatrix, k},
+           particles = Select[GetParticles[], (!IsGhost[#] && IsUnmixed[#])&];
+           For[k = 1, k <= Length[particles], k++,
+               eigenstateName = particles[[k]];
+               massMatrix = { ReplaceDependencies[GetMassOfUnmixedParticle[eigenstateName]] };
+               AppendTo[result, TreeMasses`FSMassMatrix[massMatrix, eigenstateName, Null]];
+              ];
+           result
+          ];
+
 GetMixingMatrixSymbol[massMatrix_TreeMasses`FSMassMatrix] := massMatrix[[3]];
 
 GetMassEigenstate[massMatrix_TreeMasses`FSMassMatrix] := massMatrix[[2]];