python.subruns Namespace Reference

Functions
def	get_subruns

Variables
	myargv = sys.argv
tuple	run_subruns = get_subruns(str(sys.argv[1]))

Function Documentation

def python.subruns.get_subruns

(

inputfile

)

Definition at line 46 of file subruns.py.

def get_subruns(inputfile):

    # Initialize return value to empty list.

    result = []

    # Check whether this file exists.
    if not os.path.exists(inputfile) or not inputfile.endswith('.root'):
        return result
            
    # Root checks.

    file = ROOT.TFile.Open(larbatch_posix.root_stream(inputfile))
    if file and file.IsOpen() and not file.IsZombie():

        # Root file opened successfully.
        # Get runs and subruns fro SubRuns tree.

        subrun_tree = file.Get('SubRuns')
        if subrun_tree and subrun_tree.InheritsFrom('TTree'):
            nsubruns = subrun_tree.GetEntriesFast()
            tfr = ROOT.TTreeFormula('runs',
                                    'SubRunAuxiliary.id_.run_.run_',
                                    subrun_tree)
            tfs = ROOT.TTreeFormula('subruns',
                                    'SubRunAuxiliary.id_.subRun_',
                                    subrun_tree)
            for entry in range(nsubruns):
                subrun_tree.GetEntry(entry)
                run = tfr.EvalInstance64()
                subrun = tfs.EvalInstance64()
                run_subrun = (run, subrun)
                result.append(run_subrun)

        # If previous section didn't find anything, try extracting 
        # from beam data trees.

        if len(result) == 0:
            tdir = file.Get('beamdata')
            if tdir and tdir.InheritsFrom('TDirectory'):

                # Look for bnb tree.

                bnb_tree = tdir.Get('bnb')
                if bnb_tree and bnb_tree.InheritsFrom('TTree'):
                    nsubruns = bnb_tree.GetEntriesFast()
                    tfr = ROOT.TTreeFormula('runs', 'run', bnb_tree)
                    tfs = ROOT.TTreeFormula('subruns', 'subrun', bnb_tree)
                    for entry in range(nsubruns):
                        bnb_tree.GetEntry(entry)
                        run = tfr.EvalInstance64()
                        subrun = tfs.EvalInstance64()
                        run_subrun = (run, subrun)
                        if run_subrun not in result:
                            result.append(run_subrun)

                # Look for numi tree.

                numi_tree = tdir.Get('numi')
                if numi_tree and numi_tree.InheritsFrom('TTree'):
                    nsubruns = numi_tree.GetEntriesFast()
                    tfr = ROOT.TTreeFormula('runs', 'run', numi_tree)
                    tfs = ROOT.TTreeFormula('subruns', 'subrun', numi_tree)
                    for entry in range(nsubruns):
                        numi_tree.GetEntry(entry)
                        run = tfr.EvalInstance64()
                        subrun = tfs.EvalInstance64()
                        run_subrun = (run, subrun)
                        if run_subrun not in result:
                            result.append(run_subrun)

    else:

        # Root file could not be opened.

        result = []

    # Sort in order of increasing run, then increasing subrun.

    result.sort()

    # Done.

    return result