In This Package:

Sandglass.py

Go to the documentation of this file.

#!/usr/bin/env python
'''
Mix the geant4-pre-simulated events, and do the following full dayabay nearsite simulation

usage:
    nuwa.py --history off -n50 -o fifteen.root -m "Sandglass" > log

    More examples:
      nuwa.py -R 1 -n2000000 -o DYB_hour1.root -m "Sandglass -w 2011-02-03T00:00:00" > log

    Nov. 26, 2010 Created by Zhe Wang

    **********************

    (1) Updated the mixing rates using ZHAN Liang's recommendations in DOCDB#5962
    (2) For FMCP11a: -f should run from 0 to 4999.
    (3) Increase AD multiplicity trigger threshold to > 30
    Mar. 29, 2011  Modified by C.-J. Lin
    
'''

import GaudiKernel.SystemOfUnits as units
from LoadingProc.Assembler import Assembler

class ConfigureSandglass:

    '''
    Configure a testing ground for LoadingProc to mix SimHeader and ReadoutHeader input.
    '''

    def __init__(self,argv):
        self.parse_args(argv)
        return
        
    def parse_args(self,argv):
        from optparse import OptionParser
        import time
        parser = OptionParser(usage=self.__doc__)
        default_time_format = '%Y-%m-%dT%H:%M:%S'
        parser.add_option("-w","--start-time",
                          help="Date string to set simulation start, assumed UTC",
                          default=time.strftime(default_time_format,time.localtime()))
        parser.add_option("-F","--time-format",
                          help="Format for parsing time (see man date), " \
                          + "default is '%s'"%default_time_format \
                          + " ('[YYYY]-[MM]-[DD]T[HH]:[MM]:[SS]')",
                          default=default_time_format)
        parser.add_option("-g","--generator",
                          help="A python module with a function of GetGnrtr returning a Gnrtr object.",
                          default="1223")
        parser.add_option("-T","--top-stage",
                          help="Kinematic, Detector, Electronic, TrigRead or SingleLoader",
                          default="SingleLoader")
        parser.add_option("-s","--seed",
                          help="Random seed for generators",
                          default=1234567)
        parser.add_option("-f","--frac",
                          help="Starting position of input streams",
                          default=0.0)

        (options,args) = parser.parse_args(args=argv)
        self.opts = options
        self.args = args
        
        # print "Your random seed =", self.opts.seed
        timeformat = self.opts.time_format
        print "Using time format =",timeformat

        try:
            datestring = self.opts.start_time
        except AttributeError:
            self.start_time_seconds = 0
        else:
            # This may raise ValueError if string and format don't
            # match.  Just let it get thrown.
            t = time.strptime(datestring,timeformat)
            self.start_time_seconds = time.mktime(t) - time.timezone
            print 'Start time in human readable format',self.opts.start_time
            print 'Start time in seconds UTC =',self.start_time_seconds
            print 'Top stage =',self.opts.top_stage
            
        return

        
    def load(self,
             datapath = "data/",
             filename = "IBD_DYB_AD1_acrylic.root",  # or "IBD_DYB_AD1_acrylic.list"
             rate = 1.0,
             MyMode = "random",
             MyStart = 0,
             MyStep = 1):

        extension = filename[-5:]
        modulename = filename[:-5]
        if extension != ".list" and extension != ".root":
            print "Invalid input data file ", filename

        from LoadingProc.LoadingProcConf import LoadingProc  
        load = LoadingProc("LoadingProc."+modulename)
        load.StartSec = self.start_time_seconds
        load.StartNano = 0
        load.Distribution = "Exponential"
        load.Rate = rate
        load.HsAssembler = modulename+"Assem"
        #load.OutputLevel = 2
        assem = Assembler(toolname = modulename+"Assem",
                          filename = datapath+filename,
                          mode = MyMode,
                          start = MyStart,
                          step = MyStep)

        # This and lower stage
        load.ThisStageName = "Detector"
        load.LowerStageName = ""
        # Add this processor to Gaudi sequencer
        self.stage_cfg.DetectorSequence.Members.append(load)

    def configureLoadingProc(self):
        #frac = (int(self.opts.frac)-1)/24.0
        frac = int(self.opts.frac)
        '''Configure the LoadingProc stage'''

        self.load( datapath = "data/", filename = "IBD_DayaBayAD1.list",      rate = 4334.0/(24.*60.*60.) , MyMode = "random")
        self.load( datapath = "data/", filename = "IBD_DayaBayAD2.list",      rate = 4334.0/(24.*60.*60.) , MyMode = "random")

        self.load( datapath = "data/", filename = "U238-gds_DayaBayAD1.list", rate = 3.2    , MyMode = "sequential", MyStart=40*frac)    
        self.load( datapath = "data/", filename = "U238-lso_DayaBayAD1.list", rate = 3.4    , MyMode = "sequential", MyStart=50*frac)    
        self.load( datapath = "data/", filename = "U238-PMT_DayaBayAD1.list", rate = 3978.0 , MyMode = "sequential", MyStart=17000*frac)
        self.load( datapath = "data/", filename = "U238-sst_DayaBayAD1.list", rate = 376.0  , MyMode = "sequential", MyStart=1600*frac)
        self.load( datapath = "data/", filename = "U238-gds_DayaBayAD2.list", rate = 3.2    , MyMode = "sequential", MyStart=40*frac)    
        self.load( datapath = "data/", filename = "U238-lso_DayaBayAD2.list", rate = 3.4    , MyMode = "sequential", MyStart=50*frac)    
        self.load( datapath = "data/", filename = "U238-PMT_DayaBayAD2.list", rate = 3978.0 , MyMode = "sequential", MyStart=17000*frac)
        self.load( datapath = "data/", filename = "U238-sst_DayaBayAD2.list", rate = 376.0  , MyMode = "sequential", MyStart=1600*frac)

        self.load( datapath = "data/", filename = "Th232-gds_DayaBayAD1.list",rate = 0.76   , MyMode = "sequential", MyStart=10*frac)   
        self.load( datapath = "data/", filename = "Th232-lso_DayaBayAD1.list",rate = 0.81   , MyMode = "sequential", MyStart=11*frac)   
        self.load( datapath = "data/", filename = "Th232-PMT_DayaBayAD1.list",rate = 2018.0 , MyMode = "sequential", MyStart=8800*frac)
        self.load( datapath = "data/", filename = "Th232-sst_DayaBayAD1.list",rate = 1332.0 , MyMode = "sequential", MyStart=6000*frac)
        self.load( datapath = "data/", filename = "Th232-gds_DayaBayAD2.list",rate = 0.76   , MyMode = "sequential", MyStart=10*frac)   
        self.load( datapath = "data/", filename = "Th232-lso_DayaBayAD2.list",rate = 0.81   , MyMode = "sequential", MyStart=11*frac)   
        self.load( datapath = "data/", filename = "Th232-PMT_DayaBayAD2.list",rate = 2018.0 , MyMode = "sequential", MyStart=8800*frac)
        self.load( datapath = "data/", filename = "Th232-sst_DayaBayAD2.list",rate = 1332.0 , MyMode = "sequential", MyStart=6000*frac)

        self.load( datapath = "data/", filename = "K40-gds_DayaBayAD1.list",  rate = 5.2    , MyMode = "sequential", MyStart=70*frac)    
        self.load( datapath = "data/", filename = "K40-lso_DayaBayAD1.list",  rate = 5.6    , MyMode = "sequential", MyStart=80*frac)    
        self.load( datapath = "data/", filename = "K40-PMT_DayaBayAD1.list",  rate = 817.0  , MyMode = "sequential", MyStart=3600*frac)
        self.load( datapath = "data/", filename = "K40-sst_DayaBayAD1.list",  rate = 311.0  , MyMode = "sequential", MyStart=1400*frac)
        self.load( datapath = "data/", filename = "K40-gds_DayaBayAD2.list",  rate = 5.2    , MyMode = "sequential", MyStart=70*frac)    
        self.load( datapath = "data/", filename = "K40-lso_DayaBayAD2.list",  rate = 5.6    , MyMode = "sequential", MyStart=80*frac)    
        self.load( datapath = "data/", filename = "K40-PMT_DayaBayAD2.list",  rate = 817.0  , MyMode = "sequential", MyStart=3600*frac)
        self.load( datapath = "data/", filename = "K40-sst_DayaBayAD2.list",  rate = 311.0  , MyMode = "sequential", MyStart=1400*frac)

        self.load( datapath = "data/", filename = "Co60-PMT_DayaBayAD1.list", rate = 1.46   , MyMode = "sequential", MyStart=10*frac)
        self.load( datapath = "data/", filename = "Co60-sst_DayaBayAD1.list", rate = 48.0   , MyMode = "sequential", MyStart=600*frac)
        self.load( datapath = "data/", filename = "Co60-PMT_DayaBayAD2.list", rate = 1.46   , MyMode = "sequential", MyStart=10*frac)
        self.load( datapath = "data/", filename = "Co60-sst_DayaBayAD2.list", rate = 48.0   , MyMode = "sequential", MyStart=600*frac)

        self.load( datapath = "data/", filename = "Muon.list",  rate = 523.0, MyMode = "sequential", MyStart=150000*frac)
        #self.load( datapath = "data/", filename = "MuonHitSim_DayaBay.list",  rate = 523.0 , MyMode = "sequential", MyStart=2400*frac )
        #self.load( datapath = "data/", filename = "Geant4Muon_DayaBay.list",  rate = 523.0 , MyMode = "sequential", MyStart=2400*frac )

    def configureElectronic(self):
        '''Configure the Electronics stage'''
        
        import ElecSim
        es = ElecSim.Configure(use_push_algs = False)
        #from ElecSim.ElecSimConf import EsFrontEndAlg
        #frontEndAlg = EsFrontEndAlg()
        #frontEndAlg.Detectors = ["DayaBayAD1","DayaBayAD2","DayaBayIWS","DayaBayOWS"]
        
        from ElecSimProc.ElecSimProcConf import ElecSimProc
        esp = ElecSimProc()
        esp.PmtTool = "EsPmtEffectPulseTool"
        esp.RpcTool = "EsIdealPulseTool"
        esp.FeeTool = "EsIdealFeeTool"
        esp.FecTool = "EsIdealFecTool"
        esp.Detectors = ["DayaBayAD1","DayaBayAD2","DayaBayIWS","DayaBayOWS"]

        from ElecSim.ElecSimConf import EsIdealFeeTool
        feeTool = EsIdealFeeTool()
        feeTool.LinearityThreshold = 20000
        feeTool.EnableESumH = False
        feeTool.EnableESumL = False
        feeTool.EnableESumTotal = False
        
        esp.ThisStageName = "Electronic"
        esp.LowerStageName = "Detector"
        #esp.OutputLevel = 2
        self.stage_cfg.ElectronicSequence.Members.append(esp)
        return
    
    def configureTrigRead(self):
        '''Configure the Trigger and Readout stage'''
        from TrigReadProc.TrigReadProcConf import TrigReadProc
        tsp = TrigReadProc()
        #tsp.TrigTools = [...]
        #tsp.RoTools = [...]
        #tsp.OutputLevel = 2
        from TrigSim.TrigSimConf import TsMultTriggerTool
        nhit = TsMultTriggerTool()
        nhit.NHitTriggerThreshold={"DayaBayAD1":30,
                                   "DayaBayAD2":30,
                                   "DayaBayIWS":10,
                                   "DayaBayOWS":10,
                                   "LingAoAD1":30,
                                   "LingAoAD2":30,
                                   "LingAoIWS":10,
                                   "LingAoOWS":10,
                                   "FarAD1":30,
                                   "FarAD2":30,
                                   "FarAD3":30,
                                   "FarAD4":30,
                                   "FarIWS":10,
                                   "FarOWS":10}

        tsp.ThisStageName = "TrigRead"
        tsp.LowerStageName = "Electronic"
        #tsp.OutputLevel = 2
        self.stage_cfg.TrigReadSequence.Members.append(tsp)
        return
    
    def configureSingleLoader(self):
        '''Configure the SingleLoader stage'''
        from SingleLoader.SingleLoaderConf import SingleLoader
        sll = SingleLoader()
        sll.ThisStageName = "SingleLoader"
        sll.LowerStageName = "TrigRead"
        #sll.OutputLevel = 2
        self.stage_cfg.SingleLoaderSequence.Members.append(sll)
        
    def configureSim15(self):
        from Stage.StageConf import Sim15
        sim15=Sim15()
        sim15.TopStage = self.opts.top_stage
        #sim15.TimeRange = 60*60*units.s  # 1 hour

        from DybEventMgr.DybEventMgrConf import ArchiveTrimSvc
        trimSvc = ArchiveTrimSvc()
        trimSvc.DefaultWindowSeconds = 0 #in second
        #trimSvc.OutputLevel=0
        
        from Gaudi.Configuration import ApplicationMgr
        theApp = ApplicationMgr()
        theApp.TopAlg.append(sim15)

    def configure(self):
        from Stage import Configure as StageConfigure
        self.stage_cfg = StageConfigure()

        stagedic={'Detector':2,'Electronic':3,'TrigRead':4,'SingleLoader':5}

        if not self.opts.top_stage in stagedic:
            print 'Error, wrong top stage parameter', self.opts.top_stage
            print 'Valid stage is Detector, Electronic, TrigRead or SingleLoader'
            
        for stg,idx in stagedic.iteritems():
            if idx <= stagedic[self.opts.top_stage]:
                self.stage_cfg.addStages([stg])           ## stage tools are configured here.
                    
        for stg in self.stage_cfg.stages:
            #self.stage_cfg.__dict__[stg].OutputLevel = 2
            pass

        if stagedic[self.opts.top_stage]>=2:
            self.configureLoadingProc()
        if stagedic[self.opts.top_stage]>=3:
            self.configureElectronic()
        if stagedic[self.opts.top_stage]>=4:
            self.configureTrigRead()
        if stagedic[self.opts.top_stage]>=5:
            self.configureSingleLoader()
            
        self.configureSim15()
        
        return
        
def configure(argv=[]):
    cfc = ConfigureSandglass(argv)
    cfc.configure()
    return

if __name__ == "__main__":
    configure()
    pass

---