Functions
	mergeEBWeightsFiles (input_files)
	createWeightsNode (xmlDoc, xmlRoot, weights_list)
	createEventsNode (xmlDoc, xmlRoot, events_list)
	checkRunNumber (runNumber, input_files)

Variables
	log = logging.getLogger('MergeEBWeigtsFiles.py')
	parser = ArgumentParser()
	nargs
	help
	default
	type
	int
	args = parser.parse_args()
	input_files = args.inputfiles
	runNumber = re.compile(r'\d+')
dict	parsed_files = {}
	weights_list
	events_list
	doc = minidom.parseString("<run/>")
	root = doc.documentElement
	output_filename = args.outputfile
	xml_str = root.toprettyxml(indent = " ")

Function Documentation

◆ checkRunNumber()

MergeEBWeightsFiles.checkRunNumber	(		runNumber,
			input_files )

Definition at line 73 of file MergeEBWeightsFiles.py.

def checkRunNumber(runNumber, input_files):
    for fin in input_files[1:]:
        rn = re.compile(r'\d+')
        rn = rn.findall(fin)
        if rn[-1] != runNumber[-1]:
            sys.exit("ERROR: run number should be the same for all input files. Exiting.")   
 

◆ createEventsNode()

MergeEBWeightsFiles.createEventsNode	(	xmlDoc,
		xmlRoot,
		events_list )

Definition at line 63 of file MergeEBWeightsFiles.py.

def createEventsNode(xmlDoc, xmlRoot, events_list):
    events_element = xmlDoc.createElement("events")
    xmlRoot.appendChild(events_element)
    for event in events_list:
        e=doc.createElement("e")
        e.setAttribute("n", event["n"])
        e.setAttribute("w", event["w"])
        events_element.appendChild(e)
 
# Check that all the input files have the same run number    

◆ createWeightsNode()

MergeEBWeightsFiles.createWeightsNode	(	xmlDoc,
		xmlRoot,
		weights_list )

Definition at line 52 of file MergeEBWeightsFiles.py.

def createWeightsNode(xmlDoc, xmlRoot, weights_list):
    weights_element = xmlDoc.createElement("weights")
    xmlRoot.appendChild(weights_element)
    for weight in weights_list:
        w=doc.createElement("weight")
        w.setAttribute("id", weight["id"])
        w.setAttribute("value", weight["value"])
        w.setAttribute("unbiased", weight["unbiased"])
        weights_element.appendChild(w)
 
# Create events node

◆ mergeEBWeightsFiles()

MergeEBWeightsFiles.mergeEBWeightsFiles ( input_files )

Definition at line 13 of file MergeEBWeightsFiles.py.

def mergeEBWeightsFiles(input_files):
    check_weight = []
    weightsList = []
    eventsList = []
 
    new_id=0
    log.debug("Processing file {0}".format(list(input_files.keys())[0]))
    weights = list(input_files.values())[0].getElementsByTagName("weight")
    for weight in weights:
        weightsList.append({"value":weight.getAttribute("value"), "unbiased":weight.getAttribute("unbiased"), "id":str(new_id)})
        check_weight.append({"value": weight.getAttribute("value"), "unbiased":weight.getAttribute("unbiased")})
        new_id+=1
    events = list(input_files.values())[0].getElementsByTagName("e")
    for event in events:
        eventsList.append({"n":event.getAttribute("n"), "w":event.getAttribute("w")})
 
    for fin_name, fin in list(input_files.items())[1:]:
        log.debug("Processing file {0}".format(fin_name))
        id_map = {} # mapping between old and new ids
        weights = fin.getElementsByTagName("weight")
        for weight in weights:
            element = {"value": weight.getAttribute("value"), "unbiased":weight.getAttribute("unbiased")}
            # If the weight is not already in the list, add it to the list, update the id and associate the new id to the old one through id_map
            # If the weight is already in the list, only associate the new id to the old one through id_map 
            if element not in check_weight:
                check_weight.append(element.copy())
                element["id"] = str(new_id)
                weightsList.append(element)
                id_map[weight.getAttribute("id")]=element["id"]
                new_id+=1
            else:
                id_map[weight.getAttribute("id")]=weightsList[check_weight.index(element)]["id"]
        events = fin.getElementsByTagName("e")
        # Loop on events to update the weights ids
        for event in events:
            eventsList.append({"n":event.getAttribute("n"), "w":id_map[event.getAttribute("w")]})
    return weightsList, eventsList
 
# Create weights node

Variable Documentation

◆ args

MergeEBWeightsFiles.args = parser.parse_args()

Definition at line 87 of file MergeEBWeightsFiles.py.

◆ default

MergeEBWeightsFiles.default

Definition at line 85 of file MergeEBWeightsFiles.py.

◆ doc

MergeEBWeightsFiles.doc = minidom.parseString("<run/>")

Definition at line 111 of file MergeEBWeightsFiles.py.

◆ events_list

MergeEBWeightsFiles.events_list

Definition at line 108 of file MergeEBWeightsFiles.py.

◆ help

MergeEBWeightsFiles.help

Definition at line 84 of file MergeEBWeightsFiles.py.

◆ input_files

MergeEBWeightsFiles.input_files = args.inputfiles

Definition at line 91 of file MergeEBWeightsFiles.py.

◆ int

MergeEBWeightsFiles.int

Definition at line 86 of file MergeEBWeightsFiles.py.

◆ log

MergeEBWeightsFiles.log = logging.getLogger('MergeEBWeigtsFiles.py')

Definition at line 10 of file MergeEBWeightsFiles.py.

◆ nargs

MergeEBWeightsFiles.nargs

Definition at line 84 of file MergeEBWeightsFiles.py.

◆ output_filename

MergeEBWeightsFiles.output_filename = args.outputfile

Definition at line 119 of file MergeEBWeightsFiles.py.

◆ parsed_files

dict MergeEBWeightsFiles.parsed_files = {}

Definition at line 103 of file MergeEBWeightsFiles.py.

◆ parser

MergeEBWeightsFiles.parser = ArgumentParser()

Definition at line 83 of file MergeEBWeightsFiles.py.

◆ root

MergeEBWeightsFiles.root = doc.documentElement

Definition at line 112 of file MergeEBWeightsFiles.py.

◆ runNumber

MergeEBWeightsFiles.runNumber = re.compile(r'\d+')

Definition at line 99 of file MergeEBWeightsFiles.py.

◆ type

MergeEBWeightsFiles.type