runReconstruction.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
'''
An example script to reconstruct osmics events with standalone CDC.
Usage :
basf2 runReconstruction.py <input> <output>
input: Input root file (Raw data)
       GCR1 (RAW) /hsm/belle2/bdata/Data/Raw/e0001/
       GCR2 (RAW) /hsm/belle2/bdata/Data/Raw/e0002/
 
output : Output root file, which contains helix parameters.
         N.B. this is not the basf2 root file!
         To see the helix parameters.
         Please use compare2Tracks.C for example.
'''
 
import basf2
from basf2 import *
import ROOT
import os
import os.path
import argparse
from reconstruction import add_cosmics_reconstruction
from ROOT import Belle2
from cdc.cr import *
 
# Set your suitable DB
reset_database()
use_database_chain()
use_central_database("332_COPY-OF_GT_gen_prod_004.11_Master-20171213-230000", LogLevel.INFO)
use_central_database("MagneticFieldPhase2QCSoff")
# use_local_database("/home/belle/muchida/basf2/work/caf/gcr2/test6/localDB/database.txt")
 
 
def rec(input, output, topInCounter=False, magneticField=True,
        unpacking=True, fieldMapper=False):
    main_path = basf2.create_path()
    logging.log_level = LogLevel.INFO
 
    # Get experiment and runnumber.
    exp_number, run_number = getExpRunNumber(input)
 
    # Set the peiod of data taking.
    data_period = getDataPeriod(exp=exp_number,
                                run=run_number)
 
    mapperAngle = getMapperAngle(exp=exp_number,
                                 run=run_number)
 
    # print(data_period)
    if os.path.exists('output') is False:
        os.mkdir('output')
 
    # RootInput
    main_path.add_module('RootInput',
                         #                          entrySequences=['0:1000'],
                         inputFileNames=input)
    if unpacking is True:
        main_path.add_module('CDCUnpacker')
 
    if data_period == 'gcr2017':
        gearbox = register_module('Gearbox',
                                  fileName="/geometry/GCR_Summer2017.xml",
                                  override=[("/Global/length", "8.", "m"),
                                            ("/Global/width", "8.", "m"),
                                            ("/Global/height", "8.", "m"),
                                            ])
        main_path.add_module(gearbox)
    else:
        main_path.add_module('Gearbox')
 
    if fieldMapper is True:
        main_path.add_module('CDCJobCntlParModifier',
                             MapperGeometry=True,
                             MapperPhiAngle=mapperAngle)
 
    if magneticField is True:
        main_path.add_module('Geometry', useDB=True)
    else:
        main_path.add_module('Geometry',
                             components=['CDC', 'ECL'])
 
    main_path.add_module('Progress')
 
    # Add CDC CR reconstruction.
    set_cdc_cr_parameters(data_period)
    add_cdc_cr_reconstruction(main_path)
    # add_cosmics_reconstruction(main_path,
    #                               data_taking_period=data_period,
    #                           merge_tracks=False)
 
    # Simple analysis module.
    output = "/".join(['output', output])
    main_path.add_module('CDCCosmicAnalysis',
                         noBFit=not magneticField,
                         Output=output)
 
    #    main_path.add_module("RootOutput", outputFileName='full.root')
    basf2.print_path(main_path)
    basf2.process(main_path)
    print(basf2.statistics)
 
 
if __name__ == "__main__":
    # Make the parameters accessible form the outside.
    parser = argparse.ArgumentParser()
    parser.add_argument('input', help='Input file to be processed (unpacked CDC data).')
    parser.add_argument('output', help='Output file you want to store the results.')
    args = parser.parse_args()
    rec(args.input, args.output, topInCounter=False, magneticField=True,
        unpacking=True, fieldMapper=False)