SVDAbsoluteClusterTimeShift.py

#!/usr/bin/env python3
 

 

 
 
from ROOT import Belle2
import basf2 as b2
import datetime
import json
import argparse
import math
 
now = datetime.datetime.now()
 
 
def arg_parser():
    """ argument parser """
 
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument('-j', '--json',
                        type=str,
                        help='input json file',
                        metavar='JSON')
    return parser
 
 
if __name__ == "__main__":
 
    # Argument parsing
    args = arg_parser().parse_args()
 
    # open json file
    f = open(args.json)
    param = json.load(f)
 
    print("")
    print("--> json INFO:")
    print("              "+str(param["_COMMENT"]))
    print("")
 
    iov = Belle2.IntervalOfValidity.always()
 
    uniqueID = "SVDAbsoluteClusterTimeShift_" + str(now.isoformat())
    uniqueID += "_in-" + param["uniqueID"]["INFO"]["source"]
    #    uniqueID += "_useFor-" + param["uniqueID"]["INFO"]["useFor"]
    #    uniqueID += "_" + param["uniqueID"]["INFO"]["config"]
    uniqueID += "_" + param["uniqueID"]["INFO"]["tag"]
    print("uniqueID ->", uniqueID)
    print("")
 
    payload = Belle2.SVDAbsoluteClusterTimeShift(uniqueID)
 
    for alg in param["shiftValue"]:
        for sType in param["shiftValue"][alg]:
            shiftVal = param["shiftValue"][alg][sType]
            if math.fabs(shiftVal) > 10:
                b2.B2WARNING(f"cluster time shift ({shiftVal}) is very large for ({alg} : {sType})")
            payload.setClusterTimeShift(alg, sType, shiftVal)
 
    # write out the payload to localdb directory
    Belle2.Database.Instance().storeData(Belle2.SVDAbsoluteClusterTimeShift.name, payload, iov)