Exemplo n.º 1
0
 def setUp(self):
     """Set up decay model"""
     # Read the full SM
     sm_path = import_ufo.find_ufo_path("sm")
     self.base_model = import_ufo.import_full_model(sm_path)
     model = copy.deepcopy(self.base_model)
     self.model = import_ufo.RestrictModel(model)
     self.restrict_file = os.path.join(_file_path, os.path.pardir, "input_files", "restrict_sm.dat")
Exemplo n.º 2
0
 def setUp(self):
     """Set up decay model"""
     #Read the full SM
     sm_path = import_ufo.find_ufo_path('sm')
     self.base_model = import_ufo.import_full_model(sm_path)
     model = copy.deepcopy(self.base_model)
     self.model = import_ufo.RestrictModel(model)
     self.restrict_file = os.path.join(_file_path, os.path.pardir,
                                  'input_files', 'restrict_sm.dat')
Exemplo n.º 3
0
    def make_UFO_pkl():
        """ """
        file_cond = lambda p: os.path.exists(
            os.path.join(MG5DIR, 'models', p, 'particles.py'))
        #1. find UFO model:
        ufo_model = [
            os.path.join(MG5DIR, 'models', p)
            for p in os.listdir(os.path.join(MG5DIR, 'models')) if file_cond(p)
        ]
        # model.pkl
        for model_path in ufo_model:
            start = time.time()
            print 'make model.pkl for %s :' % os.path.basename(model_path),
            #remove old pkl
            try:
                os.remove(os.path.join(model_path, 'model.pkl'))
            except:
                pass
            import_ufo.import_full_model(model_path)
            print '%2fs' % (time.time() - start)

        return
        # aloha routine
        for model_path in ufo_model:
            start = time.time()
            print 'make ALOHA for %s' % os.path.basename(model_path)
            #remove old pkl
            try:
                os.remove(os.path.join(model_path, 'aloha.pkl'))
            except:
                pass
            try:
                os.system('rm -rf %s &> /dev/null' %
                          os.path.join(model_path, 'Fortran'))
            except:
                pass

            ufo_path, ufo_name = os.path.split(model_path)
            sys.path.insert(0, ufo_path)
            output_dir = os.path.join(model_path, 'Fortran')
            create_aloha.AbstractALOHAModel(ufo_name,
                                            write_dir=output_dir,
                                            format='Fortran')
            print 'done in %2fs' % (time.time() - start)
Exemplo n.º 4
0
    def setUp(self):
        """Set up decay model"""
        #Read the full SM
        sm_path = import_ufo.find_ufo_path('sm')
        self.base_model = import_ufo.import_full_model(sm_path)

        model = copy.deepcopy(self.base_model)
        self.model = import_ufo.RestrictModel(model)
        self.restrict_file = os.path.join(_file_path, os.path.pardir,
                                     'input_files', 'restrict_sm.dat')
        self.model.set_parameters_and_couplings(self.restrict_file)
Exemplo n.º 5
0
 def make_UFO_pkl():
     """ """
     file_cond = lambda p : os.path.exists(os.path.join(MG5DIR,'models',p,'particles.py'))
     #1. find UFO model:
     ufo_model = [os.path.join(MG5DIR,'models',p) 
                     for p in os.listdir(os.path.join(MG5DIR,'models')) 
                         if file_cond(p)]
     # model.pkl
     for model_path in ufo_model:
         start = time.time()
         print 'make model.pkl for %s :' % os.path.basename(model_path),
         #remove old pkl
         try:
             os.remove(os.path.join(model_path,'model.pkl'))
         except:
             pass
         import_ufo.import_full_model(model_path)
         print '%2fs' % (time.time() - start)
     
     return
     # aloha routine 
     for model_path in ufo_model:
         start = time.time()
         print 'make ALOHA for %s' % os.path.basename(model_path)
         #remove old pkl
         try:
             os.remove(os.path.join(model_path,'aloha.pkl'))
         except:
             pass    
         try:
             os.system('rm -rf %s &> /dev/null' % os.path.join(model_path,'Fortran'))
         except:
             pass            
         
         ufo_path, ufo_name =os.path.split(model_path)
         sys.path.insert(0, ufo_path)
         output_dir = os.path.join(model_path, 'Fortran')
         create_aloha.AbstractALOHAModel(ufo_name, write_dir=output_dir, format='Fortran')
         print 'done in %2fs' % (time.time() - start)
Exemplo n.º 6
0
    def setUp(self):
        """load the hardcoded NLO toy model to compare against the imported 
            one"""

        self.hardcoded_loopmodel = loadLoopToyModel()
        # Make sure to move the pickle first in order not to load it
        if os.path.exists(os.path.join(_input_file_path, "loop_ToyModel", "model.pkl")):
            os.system(
                "mv -f "
                + str(os.path.join(_input_file_path, "loop_ToyModel", "model.pkl"))
                + " "
                + str(os.path.join(_input_file_path, "loop_ToyModel", "model_old.pkl"))
            )
        #        self.imported_loopmodel = models.import_full_model(os.path.join(\
        #            _input_file_path,'loop_ToyModel'))
        self.imported_loopmodel = models.import_full_model(os.path.join(_input_file_path, "LoopSMTest"))
        self.imported_loopmodel.actualize_dictionaries()
        self.hardcoded_loopmodel.actualize_dictionaries()
    def setUp(self):
        """load the hardcoded NLO toy model to compare against the imported 
            one"""
        
        self.hardcoded_loopmodel = loadLoopToyModel() 
        # Make sure to move the pickle first in order not to load it
        if os.path.exists(os.path.join(\
            _input_file_path,'loop_ToyModel','model.pkl')):
            os.system("mv -f "+str(os.path.join(\
                _input_file_path,'loop_ToyModel','model.pkl'))+" "+\
                str(os.path.join(_input_file_path,'loop_ToyModel',\
                'model_old.pkl')))
#        self.imported_loopmodel = models.import_full_model(os.path.join(\
#            _input_file_path,'loop_ToyModel'))
        self.imported_loopmodel = models.import_full_model(os.path.join(\
            _input_file_path,'LoopSMTest'))
        self.imported_loopmodel.actualize_dictionaries()
        self.hardcoded_loopmodel.actualize_dictionaries()
Exemplo n.º 8
0
    outPath = pjoin(os.path.dirname(os.path.realpath(eventFile)),
                    '%s_rwgt_%i.%s' % (evtFileName, i, evtFileExt))

try:
    evtBanner = banner.Banner(eventFile)
    evtFile = lhe_parser.EventFile(eventFile)
except:
    logging.error("Could not read event file %s." % eventFile)

# Start by loading the model of the banner
try:
    model_name = evtBanner.get_detail('model')
    model_name = model_name.split('-')[0]
    # For now use the model 'sm' no matter what since it is only used for the particle naming scheme
    model_name = 'sm'
    user_model = import_ufo.import_full_model(
        pjoin(proc_path, os.pardir, 'models', model_name))
except:
    logging.error("Could not load the model %s used for generating the event file"\
                 %evtBanner.get_detail('model'))
    exit()

# Detect the total number of event in an efficient way
n_evts = int(
    subprocess.Popen('grep -rin "<event>" %s | wc -l' %
                     os.path.realpath(eventFile),
                     stdout=subprocess.PIPE,
                     shell=True).communicate()[0])

logging.info("Writing out the reweighted event file on\n   %s" % outPath)
# Write out the output event file
outputEvtFile = open(outPath, 'w')
Exemplo n.º 9
0
 def test_get_nflav_sm(self):
     """Tests the get_nflav_function for the full SM.
     Here b and c quark are massive"""
     sm_path = import_ufo.find_ufo_path('sm')
     model = import_ufo.import_full_model(sm_path)
     self.assertEqual(model.get_nflav(), 3)
Exemplo n.º 10
0
 def setUp(self):
     """Set up decay model"""
     #Read the full SM
     sm_path = import_ufo.find_ufo_path('heft')
     self.base_model = import_ufo.import_full_model(sm_path)
Exemplo n.º 11
0
 def test_get_nflav_sm(self):
     """Tests the get_nflav_function for the full SM.
     Here b and c quark are massive"""
     sm_path = import_ufo.find_ufo_path('sm')
     model = import_ufo.import_full_model(sm_path)
     self.assertEqual(model.get_nflav(), 3)
Exemplo n.º 12
0
 def setUp(self):
     """Set up decay model"""
     #Read the full SM
     sm_path = import_ufo.find_ufo_path('heft')
     self.base_model = import_ufo.import_full_model(sm_path)
Exemplo n.º 13
0
    outPath=pjoin(os.path.dirname(os.path.realpath(eventFile)),
                  '%s_rwgt_%i.%s'%(evtFileName,i,evtFileExt))

try:
    evtBanner = banner.Banner(eventFile)    
    evtFile = lhe_parser.EventFile(eventFile)
except:
    logging.error("Could not read event file %s."%eventFile)

# Start by loading the model of the banner
try:
    model_name=evtBanner.get_detail('model')
    model_name=model_name.split('-')[0]
    # For now use the model 'sm' no matter what since it is only used for the particle naming scheme
    model_name='sm'
    user_model = import_ufo.import_full_model(pjoin(proc_path,os.pardir,'models',model_name))
except:
    logging.error("Could not load the model %s used for generating the event file"\
                 %evtBanner.get_detail('model'))
    exit()

# Detect the total number of event in an efficient way
n_evts = int(subprocess.Popen('grep -rin "<event>" %s | wc -l'%os.path.realpath(eventFile),
                        stdout=subprocess.PIPE, shell=True).communicate()[0])

logging.info("Writing out the reweighted event file on\n   %s"%outPath)
# Write out the output event file
outputEvtFile = open(outPath,'w')
outputEvtFile.write(evtFile.banner)

# List the channel present in the process output