ParseFluxesFile.py

Go to the documentation of this file.

#!/usr/bin/env python
import SupernovaUtilities as utils
import ROOT
ROOT.gROOT.SetBatch(1)
import math

# Begin
print "\nrun  : --- Parse and a flux file into a 2-d histogram"

# Parse command line options
from optparse import OptionParser
parser = OptionParser()
parser.add_option("-m", "--model",
                  help="which model to run, options are garching or livermore",
                  action="store",
                  type="str",
                  dest="model",
                  default="garching")
                  
parser.add_option("-f", "--fermi_dirac",
                  help="use a fermi-dirac parameterisaion",
                  action="store_true",
                  dest="fermi",
                  default=False)
                  
parser.add_option("-v", "--verbose_mode",
                  help="turn on verbose mode",
                  action="store_true",
                  dest="verbose",
                  default=False)
                  
(options, args) = parser.parse_args()

print "run  : --- Options:"
print "run  :     model:        ", options.model
print "run  :     fermi-dirac:  ", options.fermi
print "run  :     verbose mode: ", options.verbose
assert (options.model in ["garching", "livermore"]) or (
    "SND" in options.model), "Unconfigured mode requested: %s" % options.model

# Read logs
# Livermore and Garching data files contain similar information but in different formats
# therefore the first thing we do is to to consolodate these into a common format
#   - a dictionary where the keys are the times and the value is an array of data:
#       [L nue, L anti-nue, L nux, E nue, E anti-nue, E nux, alpha nue, alpha anti-nue, alpha nux]
# L = Luminosity, E = Energy
if options.model == "garching":
    raw_data = utils.ReadGarchingFluxes()
    
    param = "alpha"
    if options.fermi:
        param = "fermi-dirac"
    
    histograms = utils.ParameterisedDataToHistograms(raw_data,
                                                     parameterisation=param,
                                                     verbose=options.verbose)
    
elif options.model == "livermore":
    raw_data = utils.ReadLivermoreFluxes()
    histograms = utils.ParameterisedDataToHistograms(raw_data,
                                                     verbose=options.verbose)

# Supernova database files come in a different format again. Rather than go via the intermediate
# dictionary step as we do above, just skip straight to filling the histograms.
elif "SND" in options.model:
    if "integ" in options.model:
        # These are time integrated so are only really useful as a cross check
        # also, I haven't written the code for this so you'll get an assert
        # fail here.
        histograms = utils.ParseIntegratedSNDFile(options.model)
    else:
        histograms = utils.ParseSNDFile(options.model, verbose=options.verbose)

# Write histograms to file
if options.fermi:
    root_output_name = "./data/%s_FD_fluxes.root" % options.model
else:
    root_output_name = "./data/%s_fluxes.root" % options.model
print "run  : Writing histograms to file: %s" % root_output_name
root_file = ROOT.TFile(root_output_name, "RECREATE")

for i in range(3):
    # histograms["Time_Energy_Fluence_%i"%i].GetXaxis().SetRange(0,-1)
    histograms["Time_Energy_Fluence_%i" % i].Write()
    histograms["lumi_%i" % i].Write()
    histograms["lumi_mev_%i" % i].Write()
    histograms["E_%i" % i].Write()
    histograms["alpha_%i" % i].Write()
    histograms["eta_%i" % i].Write()
    histograms["T_%i" % i].Write()
root_file.Close()
print "run  : Done\n"