PyNTupleProcessor/run_EDM.py at master · lucien1011/PyNTupleProcessor · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77

# alphatwirl
from Core.EventReader import EventLoopRunner, MPEventLoopRunner, EventLoop
from Core.ProgressBar import ProgressBar,ProgressReport,ProgressMonitor,BProgressMonitor
from Core.Concurrently import CommunicationChannel,CommunicationChannel0

from Core.EDMResult.BEventBuilder import BEventBuilder

from Core.HeppyResult import ComponentLoop
from Core.HeppyResult.UFComponentReader import UFComponentReader

from Core.Utils.git import getGitDescribe,getGitDiff

# Standard package
import imp,sys,os,time

cfgFileName             = sys.argv[1]
file                    = open( cfgFileName,'r')
cfg                     = imp.load_source( 'UFEDM.__cfg_to_run__', cfgFileName, file)

#rootTree                = "Events"

nCores                  = cfg.nCores
componentList           = cfg.componentList
nEvents                 = cfg.nEvents
disableProgressBar      = cfg.disableProgressBar
sequence                = cfg.sequence
outputInfo              = cfg.outputInfo
endSequence             = cfg.endSequence
justEndSequence         = cfg.justEndSequence if hasattr(cfg,"justEndSequence") else False

start_time = time.time()

if not justEndSequence:
    progressBar = ProgressBar()

    if nCores != 1:
        progressMonitor      = BProgressMonitor(progressBar)
        communicationChannel = CommunicationChannel(nCores,progressMonitor)
    else:
        progressMonitor      = ProgressMonitor(progressBar)
        communicationChannel = CommunicationChannel0(progressMonitor)
        pass

    if not disableProgressBar: progressMonitor.begin()
    communicationChannel.begin()

    print "\nLoading samples:\n"

    eventLoopRunner = MPEventLoopRunner(communicationChannel)
    eventBuilder    = BEventBuilder()
    componentReader = UFComponentReader(eventBuilder, eventLoopRunner, sequence, outputInfo)
    componentLoop   = ComponentLoop(componentReader)

    print "\nBegin Running\n"
    componentLoop(componentList)

    print "\nEnd Running\n"
    print "\nOutput saved in "+outputInfo.outputDir+"\n"

    gitFile        = os.path.join(outputInfo.outputDir,"gitDetails.txt")
    gitVerboseFile = os.path.join(outputInfo.outputDir,"gitVerboseDetails.txt")
    with open(gitFile,'w') as f:
        f.write(getGitDescribe())

    with open(gitVerboseFile,'w') as f:
        f.write(getGitDiff())

    if not disableProgressBar: progressMonitor.end()
    communicationChannel.end()
if endSequence:
    print "\nBegin Summarising\n"
    print "\nInput used: "+outputInfo.outputDir+"\n"
    endSequence.run(outputInfo)

elapsed_time = time.time() - start_time
print "Time used: "+str(elapsed_time)+"s"