Functions
def	mergeEBWeightsFiles (input_files)

def	createWeightsNode (xmlDoc, xmlRoot, weights_list)

def	createEventsNode (xmlDoc, xmlRoot, events_list)

def	checkRunNumber (runNumber, input_files)

Variables
	log

	parser

	nargs

	help

	default

	type

	int

	args

	input_files

	runNumber

	parsed_files

	weights_list

	events_list

	doc

	root

	output_filename

	xml_str

	indent

Function Documentation

◆ checkRunNumber()

def 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()

def 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()

def 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()

def 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

Definition at line 87 of file MergeEBWeightsFiles.py.

◆ default

MergeEBWeightsFiles.default

Definition at line 85 of file MergeEBWeightsFiles.py.

◆ doc

MergeEBWeightsFiles.doc

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.

◆ indent

MergeEBWeightsFiles.indent

Definition at line 124 of file MergeEBWeightsFiles.py.

◆ input_files

MergeEBWeightsFiles.input_files

Definition at line 91 of file MergeEBWeightsFiles.py.

◆ int

MergeEBWeightsFiles.int

Definition at line 86 of file MergeEBWeightsFiles.py.

◆ log

MergeEBWeightsFiles.log

Definition at line 10 of file MergeEBWeightsFiles.py.

◆ nargs

MergeEBWeightsFiles.nargs

Definition at line 84 of file MergeEBWeightsFiles.py.

◆ output_filename

MergeEBWeightsFiles.output_filename

Definition at line 119 of file MergeEBWeightsFiles.py.

◆ parsed_files

MergeEBWeightsFiles.parsed_files

Definition at line 103 of file MergeEBWeightsFiles.py.

◆ parser

MergeEBWeightsFiles.parser

Definition at line 83 of file MergeEBWeightsFiles.py.

◆ root

MergeEBWeightsFiles.root

Definition at line 112 of file MergeEBWeightsFiles.py.

◆ runNumber

MergeEBWeightsFiles.runNumber

Definition at line 99 of file MergeEBWeightsFiles.py.

◆ type

MergeEBWeightsFiles.type