LHETools Namespace Reference

Classes
class	Event
class	EventInfo
class	Particle
class	Record

Functions
def	FindDuplicates (inFileName=None, crashMode=False, printDuplicated=False)

Variables
	LHEToolsLog = Logging.logging.getLogger('MCJobOptionUtils/LHETools.py')

Function Documentation

FindDuplicates()

def LHETools.FindDuplicates	(		inFileName = None,
			crashMode = False,
			printDuplicated = False
	)

Definition at line 220 of file LHETools.py.

def FindDuplicates(inFileName=None, crashMode=False, printDuplicated=False):
    if inFileName == None:
        LHEToolsLog.error("No input file for Duplicate finder")
        raise RuntimeError
 
    if printDuplicated:
        LHEToolsLog.info("Possible duplicated events in lhe file will be printed out if any")
 
    if printDuplicated:
        LHEToolsLog.info("Possible duplicated events in lhe file will be printed out if any")
    
    # writing awk script to be executed on the input lhe file
    fScript = open("finder.awk", 'w')
    fScript.write(
"""BEGIN{RS="<event>"; ORS=""; FS="</event>"; OFS=""; nDuplicates=0}
{
if (NR==1) {
    print $1 > \""""+inFileName+"-new"+"""\"
    ORS=\"\\n\"
}
else {
    c[$1]++;
    if (c[$1]==1) {print \"<event>\"$1\"</event>\" > \""""+inFileName+"-new"+"""\" }
    else {print \"<event>\"$1\"</event>\" > \""""+inFileName+"-duplicates"+"""\"; nDuplicates++}
}
}
END{print \"</LesHouchesEvents>\\n\" > \""""+inFileName+"-new"+"""\"; print nDuplicates}""")
    fScript.close()
 
    command = 'awk -f finder.awk '+inFileName+" > nDuplicates.txt"
    try:
        os.system(command)
    except:
        LHEToolsLog.error("Impossible to execute awk script")
        raise RuntimeError
 
    # read the number of duplicate events from temp file
    fnDuplicates = open("nDuplicates.txt", 'r')
    s = fnDuplicates.read().rstrip()
    fnDuplicates.close()
 
    if int(s) >0:
        fDuplicates = open(inFileName+"-duplicates", 'r')
        # some usefull prints
        LHEToolsLog.info(s+" duplicate events were found in "+inFileName)
        if printDuplicated:
            LHEToolsLog.info("Possible duplicated events in lhe file will be printed out if any")
            for line in fDuplicates:
                LHEToolsLog.info(line)
        fDuplicates.close() # don't need it anymore
        # either crash of replace file
        if(crashMode):
            LHEToolsLog.error("Duplicate events found. To avoid this crash, use crashmode=False. Exiting...")
            raise RuntimeError
        else:
            LHEToolsLog.info("Duplicate events found. Replacing input file "+inFileName+" by fixed file "+inFileName+"-new")
            LHEToolsLog.info("Old file saved in "+inFileName+"-old")
            os.rename(inFileName,inFileName+"-old")
            os.rename(inFileName+"-new",inFileName)
    else:
        LHEToolsLog.info("No duplicate events was found in "+inFileName)
        os.remove(inFileName+"-new") # removing the useless "-new" file

Variable Documentation

LHEToolsLog

LHETools.LHEToolsLog = Logging.logging.getLogger('MCJobOptionUtils/LHETools.py')

Definition at line 7 of file LHETools.py.