python.MadGraphUtilsHelpers Namespace Reference

Functions
def	getDictFromCard (card_loc, lowercase=False)
def	settingIsTrue (setting)
def	totallyStripped (x)
def	checkSetting (key_, value_, mydict_)
def	checkSettingIsTrue (key_, mydict_)
def	checkSettingExists (key_, mydict_)
def	is_version_or_newer (args)
def	isNLO_from_run_card (run_card)
def	get_runArgs_info (runArgs)
def	get_physics_short ()

Function Documentation

checkSetting()

def python.MadGraphUtilsHelpers.checkSetting	(		key_,
			value_,
			mydict_
	)

Definition at line 33 of file MadGraphUtilsHelpers.py.

def checkSetting(key_,value_,mydict_):
    key=totallyStripped(key_)
    value=totallyStripped(value_)
    mydict={}
    for k in mydict_:
        mydict[totallyStripped(k)]=totallyStripped(mydict_[k])
    return key in mydict and mydict[key]==value
 

checkSettingExists()

def python.MadGraphUtilsHelpers.checkSettingExists	(		key_,
			mydict_
	)

Definition at line 48 of file MadGraphUtilsHelpers.py.

def checkSettingExists(key_,mydict_):
    key=totallyStripped(key_)
    keys=[]
    for k in mydict_:
        keys+=[totallyStripped(k)]
    return key in keys
 

checkSettingIsTrue()

def python.MadGraphUtilsHelpers.checkSettingIsTrue	(		key_,
			mydict_
	)

Definition at line 41 of file MadGraphUtilsHelpers.py.

def checkSettingIsTrue(key_,mydict_):
    key=totallyStripped(key_)
    mydict={}
    for k in mydict_:
        mydict[totallyStripped(k)]=totallyStripped(mydict_[k])
    return key in mydict and mydict[key] in ['t','true']
 

get_physics_short()

def python.MadGraphUtilsHelpers.get_physics_short

(

)

Definition at line 107 of file MadGraphUtilsHelpers.py.

def get_physics_short():
    import os
    FIRST_DIR = (os.environ['JOBOPTSEARCHPATH']).split(":")[0]
    jofiles = [f for f in os.listdir(FIRST_DIR) if (f.startswith('mc') and f.endswith('.py'))]
    if len(jofiles)==0:
        raise RuntimeError('No job options found in '+FIRST_DIR)
    joparts = os.path.basename(jofiles[0]).split('.')
    if len(joparts)<2:
        raise RuntimeError('Malformed job options file name: '+jofiles[0])
    return joparts[1]

get_runArgs_info()

def python.MadGraphUtilsHelpers.get_runArgs_info

(

runArgs

)

Definition at line 94 of file MadGraphUtilsHelpers.py.

def get_runArgs_info(runArgs):
    if runArgs is None:
        raise RuntimeError('runArgs must be provided!')
    if hasattr(runArgs,'ecmEnergy'):
        beamEnergy = runArgs.ecmEnergy / 2.
    else:
        raise RuntimeError("No center of mass energy found in runArgs.")
    if hasattr(runArgs,'randomSeed'):
        random_seed = runArgs.randomSeed
    else:
        raise RuntimeError("No random seed found in runArgs.")
    return beamEnergy,random_seed
 

getDictFromCard()

def python.MadGraphUtilsHelpers.getDictFromCard	(		card_loc,
			lowercase = False
	)

Definition at line 3 of file MadGraphUtilsHelpers.py.

def getDictFromCard(card_loc,lowercase=False):
    card=open(card_loc)
    mydict={}
    for line in iter(card):
        if not line.strip().startswith('#'): # line commented out
            command = line.split('!', 1)[0]
            if '=' in command:
                setting = command.split('=')[-1].strip()
                value = '='.join(command.split('=')[:-1]).strip()
                if lowercase:
                    value=value.lower()
                    setting=setting.lower()
                mydict[setting]=value
    card.close()
    return mydict
 

is_version_or_newer()

def python.MadGraphUtilsHelpers.is_version_or_newer

(

args

)

Definition at line 55 of file MadGraphUtilsHelpers.py.

def is_version_or_newer(args):
    # also need to find out the version (copied from generate)
    import os
    version=None
    version_file = open(os.environ['MADPATH']+'/VERSION','r')
 
    for line in version_file:
        if 'version' in line:
            version=line.split('=')[1].strip()
    version_file.close()
 
    if not version:
        raise RuntimeError('Failed to find MadGraph/MadGraph5_aMC@NLO version in '+version_file)
 
    vs=[int(v) for v in version.split('.')]
 
    # this is lazy, let's hope there wont be a subversion > 100...
    y=int(100**max(len(vs),len(args)))
    testnumber=0
    for x in args:
        testnumber+=x*y
        y/=100
 
    y=int(100**max(len(vs),len(args)))
    versionnumber=0
    for x in vs:
        versionnumber+=x*y
        y/=100
    return versionnumber>=testnumber
 

isNLO_from_run_card()

def python.MadGraphUtilsHelpers.isNLO_from_run_card

(

run_card

)

Definition at line 85 of file MadGraphUtilsHelpers.py.

def isNLO_from_run_card(run_card):
    f = open(run_card,'r')
    if "parton_shower" in f.read().lower():
        f.close()
        return True
    else:
        f.close()
        return False
 

settingIsTrue()

def python.MadGraphUtilsHelpers.settingIsTrue

(

setting

)

Definition at line 19 of file MadGraphUtilsHelpers.py.

def settingIsTrue(setting):
    if setting.replace("'",'').replace('"','').replace('.','').lower() in ['t','true']:
        return True
    return False
 

totallyStripped()

def python.MadGraphUtilsHelpers.totallyStripped

(

x

)

Definition at line 24 of file MadGraphUtilsHelpers.py.

def totallyStripped(x):
    y=str(x).lower().strip()
    # remove leading and trailing "/'
    while len(y)>0 and (y[0]=='"' or y[0]=="'"):
        y=y[1:]
    while len(y)>0 and (y[-1]=='"' or y[-1]=="'"):
        y=y[:-1]
    return y