Tue Apr 24 11:22:19 CDT 2018

1 files changed, 31 insertions, 0 deletions

diff --git a/utils/gf.py b/utils/gf.py
index 20afc75..59d1033 100644
--- a/utils/gf.py
+++ b/utils/gf.py
@@ -18,6 +18,37 @@ from utils.enums import DataType, SteeringCateg
 from utils.misc import enum_parse, thread_factors
 
 
+def fit_flags(llh_paramset):
+    default_flags = {
+        # False means it's not fixed in minimization
+        'astroFlavorAngle1'         : True,
+        'astroFlavorAngle2'         : True,
+        'astroENorm'                : True,
+        'astroMuNorm'               : True,
+        'astroTauNorm'              : True,
+        'convNorm'                  : True,
+        'promptNorm'                : True,
+        'muonNorm'                  : True,
+        'astroNorm'                 : True,
+        'astroParticleBalance'      : True,
+        'astroDeltaGamma'           : True,
+        'cutoffEnergy'              : True,
+        'CRDeltaGamma'              : True,
+        'piKRatio'                  : True,
+        'NeutrinoAntineutrinoRatio' : True,
+        'darkNorm'                  : True,
+        'domEfficiency'             : True,
+        'holeiceForward'            : True,
+        'anisotropyScale'           : True,
+        'astroNormSec'              : True,
+        'astroDeltaGammaSec'        : True
+    }
+    flags = gf.FitParametersFlag()
+    for param in llh_paramset:
+        flags.__setattr__(param.name, False)
+    return flags
+
+
 def steering_params(args):
     steering_categ = args.ast
     params = gf.SteeringParams()