def _exists(analysis, topology, path=None): """ Check if the dictionary 'limit_topo' in <path>/sms.py exists. For topologies with intermediate masses, check if all dictionaries listed in the axes-information exist. If topology == None, check if <path>/sms.py exists. """ path2 = smsHelpers.getPath(analysis, path) if not topology: import os base = smsHelpers.base pydict = "%s/%s/%s/sms.py" % (base, path2, analysis) if os.path.exists(pydict): return True else: return False axes = getaxes(analysis, topology) if not axes: return False hasDict = smsHelpers.hasDictionary(analysis, path2) if not hasDict: return False return smsHelpers.hasDictionary(analysis, path2, topology)
def getUpperLimit(analysis, topology, mx=None, my=None, path=None, interpolate=False, expected=False): """ Get the upper limit for path/analysis/topology. :returns: None, if it does not exist; entire dictionary, if mx and my are None; upper limit at mx/my, if mx and my are floats; """ path = smsHelpers.getPath(analysis, path) if smsHelpers.hasDictionary(analysis, path): return getUpperLimitFromDictionary(analysis, topology, mx, my, path, interpolate=interpolate, expected=expected) logger.warning("No upper limits found for %s", analysis) return None
def _exists(analysis, topology, path=None): """ Check if the dictionary 'limit_topo' in <path>/sms.py exists. For topologies with intermediate masses, check if all dictionaries listed in the axes-information exist. If topology == None, check if <path>/sms.py exists. """ path2 = smsHelpers.getPath(analysis, path) if not topology: import os base = smsHelpers.base pydict = "%s/%s/%s/sms.py" % (base, path2, analysis) if os.path.exists(pydict): return True else: return False axes = getaxes(analysis, topology) if not axes: return False hasDict = smsHelpers.hasDictionary(analysis, path2) if not hasDict: return False return smsHelpers.hasDictionary(analysis, path2, topology)
def getConditions(analysis, topology="all", fuzzy=True, path=None): """ Get conditions of an analysis. :returns: dictionary of conditions, if topology == "all"; single condition for the passed topology, if only one topology is passed; None if non-existent. """ key = analysis + topology + str(fuzzy) + str(path) if key in conditions: return conditions[key] path = smsHelpers.getPath(analysis, path) if fuzzy: ret = smsHelpers.getLines(analysis, path, "fuzzycondition") else: ret = smsHelpers.getLines(analysis, path, "condition") if topology == "all": conditions[key] = ret return ret if not topology in ret: conditions[key] = None return None conditions[key] = ret[topology] return ret[topology]
def getConditions(analysis, topology="all", fuzzy=True, path=None): """ Get conditions of an analysis. :returns: dictionary of conditions, if topology == "all"; single condition for the passed topology, if only one topology is passed; None if non-existent. """ key = analysis + topology + str(fuzzy) + str(path) if key in conditions: return conditions[key] path = smsHelpers.getPath(analysis, path) if fuzzy: ret = smsHelpers.getLines(analysis, path, "fuzzycondition") else: ret = smsHelpers.getLines(analysis, path, "condition") if topology == "all": conditions[key] = ret return ret if not topology in ret: conditions[key] = None return None conditions[key] = ret[topology] return ret[topology]
def getTopologies(analysis): """ Get all topologies of an analysis with constraints. """ path = smsHelpers.getPath(analysis) x = getConstraints(analysis, path=path) return x.keys()
def getTopologies(analysis): """ Get all topologies of an analysis with constraints. """ path = smsHelpers.getPath(analysis) x = getConstraints(analysis, path=path) return x.keys()
def getExperiment(analysis, path=None): """ Check if path is to ATLAS directory, else return CMS. """ logger.warning("getExperiment is deprecated") path1 = getPath(analysis, path) if path1.find("ATLAS") > -1: return "ATLAS" return "CMS"
def getExperiment(analysis, path=None): """ Check if path is to ATLAS directory, else return CMS. """ logger.warning("getExperiment is deprecated") path1 = getPath(analysis, path) if path1.find("ATLAS") > -1: return "ATLAS" return "CMS"
def upperLimit(analysis, topology, masses, path=None): """ Return upper limit for analysis-topology for given masses. :param masses: list of masses, with (mother, intermediate(s), LSP). For intermediate masses: if possible do interpolation over upper limits for different x-values. If interpolation is not possible: check if masses are comparable to the assumptions in the histogram. """ d = smsResults.getaxes(analysis, topology) if not path: path = smsHelpers.getPath(analysis) if not d: logger.error("%s/%s not found.", analysis, topology) return None for entry in masses: if not type(entry) == type(1 * GeV): logger.error("Please give input masses with unum units.") return None if len(masses) < 2: logger.error("No results for single mass entry") return None if len(masses) == 2 and not d[0]['mz']: return smsResults.getUpperLimit(analysis, topology, masses[_getAxis('x', d[0]['axes'])], masses[_getAxis('y', d[0]['axes'])], interpolate=True) if len(masses) == 2 and d[0]['mz']: logger.error("Need intermediate mass input for %s/%s.", analysis, topology) return None if len(masses) > 2 and not d[0]['mz']: logger.error("No intermediate mass in %s/%s.", analysis, topology) return None if len(masses) > 3 or len(d[0]['mz']) > 1: logger.error( "More than one intermediate mass in %s/%s. Cannot find " "upper limit for topologies with more than one " "intermediate mass.", analysis, topology) return None if len(masses) > 2 and len(d) == 1: if _compareMasses(masses, d[0]): return smsResults.getUpperLimit( analysis, _getHistName(topology, d[0]['mz'][0]), masses[_getAxis('x', d[0]['axes'])], masses[_getAxis('y', d[0]['axes'])], interpolate=True) logger.warning( "Only one histogram available for %s/%s, cannot " "interpolate for intermediate mass.", analysis, topology) return None return _doGridData(analysis, topology, masses, d, path)
def getBranchCondition(anaName, txName, path=None): """ Get the branch condition for an analysis. :returns: string containing the branch condition (e.g. equal branch masses) """ path = smsHelpers.getPath(anaName, path) ret = smsHelpers.getLines(anaName, path, "branchcondition") if not txName in ret: return None else: return ret[txName]
def getBranchCondition(anaName, txName, path=None): """ Get the branch condition for an analysis. :returns: string containing the branch condition (e.g. equal branch masses) """ path = smsHelpers.getPath(anaName, path) ret = smsHelpers.getLines(anaName, path, "branchcondition") if not txName in ret: return None else: return ret[txName]
def upperLimit(analysis, topology, masses, path=None): """ Return upper limit for analysis-topology for given masses. :param masses: list of masses, with (mother, intermediate(s), LSP). For intermediate masses: if possible do interpolation over upper limits for different x-values. If interpolation is not possible: check if masses are comparable to the assumptions in the histogram. """ d = smsResults.getaxes(analysis, topology) if not path: path = smsHelpers.getPath(analysis) if not d: logger.error("%s/%s not found.", analysis, topology) return None for entry in masses: if not type(entry)==type(1*GeV): logger.error("Please give input masses with unum units.") return None if len(masses) < 2: logger.error("No results for single mass entry") return None if len(masses) == 2 and not d[0]['mz']: return smsResults.getUpperLimit(analysis, topology, masses[_getAxis('x', d[0]['axes'])], masses[_getAxis('y', d[0]['axes'])], interpolate=True) if len(masses) == 2 and d[0]['mz']: logger.error("Need intermediate mass input for %s/%s.", analysis, topology) return None if len(masses) > 2 and not d[0]['mz']: logger.error("No intermediate mass in %s/%s.", analysis, topology) return None if len(masses) > 3 or len(d[0]['mz']) > 1: logger.error("More than one intermediate mass in %s/%s. Cannot find " "upper limit for topologies with more than one " "intermediate mass.", analysis, topology) return None if len(masses) > 2 and len(d) == 1: if _compareMasses(masses, d[0]): return smsResults.getUpperLimit(analysis, _getHistName(topology, d[0]['mz'][0]), masses[_getAxis('x', d[0]['axes'])], masses[_getAxis('y', d[0]['axes'])], interpolate=True) logger.warning("Only one histogram available for %s/%s, cannot " "interpolate for intermediate mass.", analysis, topology) return None return _doGridData(analysis, topology, masses, d, path)
def getUpperLimit(analysis, topology, mx=None, my=None, path=None, interpolate=False, expected=False): """ Get the upper limit for path/analysis/topology. :returns: None, if it does not exist; entire dictionary, if mx and my are None; upper limit at mx/my, if mx and my are floats; """ path = smsHelpers.getPath(analysis, path) if smsHelpers.hasDictionary(analysis, path): return getUpperLimitFromDictionary(analysis, topology, mx, my, path, interpolate=interpolate, expected=expected) logger.warning("No upper limits found for %s", analysis) return None
def getConstraints(analysis, topology="all", path=None): """ Get constraints of an analysis. :returns: dictionary of constraints, if topology == "all"; single constraint for the passed topology, if only one topology is passed; None if non-existent; """ key = analysis + topology + str(path) if key in constraints: return constraints[key] path = smsHelpers.getPath(analysis, path) ret = smsHelpers.getLines(analysis, path, "constraint") if topology == "all": constraints[key] = ret return ret if not topology in ret: constraints[key] = None return None constraints[key] = ret[topology] return ret[topology]
def getConstraints(analysis, topology="all", path=None): """ Get constraints of an analysis. :returns: dictionary of constraints, if topology == "all"; single constraint for the passed topology, if only one topology is passed; None if non-existent; """ key = analysis + topology + str(path) if key in constraints: return constraints[key] path = smsHelpers.getPath(analysis, path) ret = smsHelpers.getLines(analysis, path, "constraint") if topology == "all": constraints[key] = ret return ret if not topology in ret: constraints[key] = None return None constraints[key] = ret[topology] return ret[topology]
def load(analyses=None, topologies=None, sqrts=[7, 8], usePrivate=False, useSuperseded=False): """ Create an analysis objects from the info given in the SMS results database. :param analyses: If a list of analyses is passed, create only objects for these analyses (the database naming convention is used). :param topologies: If a list of topologies is passed, only these topologies are considered. :param sqrts: Array of center-of-mass energies of the analyses that are to be considered. :param usePrivate: Include analyses flagged as private. :returns: list of analyses """ #info message if superseded results are not used if not useSuperseded: logger.info("useSuperseded is not set, skipping superseded results") #to have readable input, we can give all instead of None if topologies == "all": topologies = None if analyses == "all": analyses = None # Enable supplying a single analysis/topology if isinstance(topologies, str): topologies = [topologies] if isinstance(analyses, str): analyses = [analyses] if isinstance(sqrts, int): sqrts = [sqrts] if isinstance(sqrts, float): sqrts = [int(sqrts)] listOfAnalyses = [] if analyses == None: analyses = smsResults.getAllResults().keys() for ana in analyses: if not smsHelpers.getPath(ana): logger.warning("Analysis %s not found" % ana) continue if smsResults.isPrivate(ana) and not usePrivate: logger.info("Skipping private analysis %s.", str(ana)) continue if smsResults.isSuperseded(ana) and not useSuperseded: continue logger.debug("Building analysis %s.", str(ana)) ss = smsResults.getSqrts(ana) / TeV if ss == None: logger.debug("SS: " + str(ss) + str(ana)) continue ss = int(ss) if not ss in sqrts: continue for tx in smsResults.getTopologies(ana): if topologies != None and tx not in topologies: continue logger.debug(str(tx)) newAnalysis = analysis.ULanalysis() newAnalysis.sqrts = smsResults.getSqrts(ana) newAnalysis.lum = smsResults.getLumi(ana) stopo = _getRealTopo(tx) newAnalysis.label = ana + ":" + tx # "2012" constraint = smsResults.getConstraints(ana, topology=stopo) cond = smsResults.getConditions(ana, topology=stopo) if not constraint or constraint == "Not yet assigned": logger.debug("Constraint for %s %s (%s) does not exist.", str(ana), str(tx), str(stopo)) continue if not cond: logger.warning( "Skipping %s %s (%s), Condition does not exist.", str(ana), str(tx), str(stopo)) continue cond = cond.replace("'", "").replace(" ", "").split(';') newAnalysis.constraint = constraint newAnalysis.conditions = cond newAnalysis.elementsEff = _getElementsEffs(constraint, cond) #Skip analyses with invalid elements if not newAnalysis.elementsEff: logger.warning( "Skipping analysis %s due to invalid elements." % (newAnalysis.label)) continue # Add analysis to list of analyses: listOfAnalyses.append(newAnalysis) if not listOfAnalyses: logger.warning("Zero analyses loaded.") return listOfAnalyses
def load(analyses=None, topologies=None, sqrts=[7, 8], usePrivate=False, useSuperseded=False ): """ Create an analysis objects from the info given in the SMS results database. :param analyses: If a list of analyses is passed, create only objects for these analyses (the database naming convention is used). :param topologies: If a list of topologies is passed, only these topologies are considered. :param sqrts: Array of center-of-mass energies of the analyses that are to be considered. :param usePrivate: Include analyses flagged as private. :returns: list of analyses """ #info message if superseded results are not used if not useSuperseded: logger.info("useSuperseded is not set, skipping superseded results") #to have readable input, we can give all instead of None if topologies == "all": topologies = None if analyses == "all": analyses = None # Enable supplying a single analysis/topology if isinstance(topologies, str): topologies = [topologies] if isinstance(analyses, str): analyses = [analyses] if isinstance(sqrts, int): sqrts = [sqrts] if isinstance(sqrts, float): sqrts = [int(sqrts)] listOfAnalyses = [] if analyses == None: analyses = smsResults.getAllResults().keys() for ana in analyses: if not smsHelpers.getPath(ana): logger.warning("Analysis %s not found" % ana) continue if smsResults.isPrivate(ana) and not usePrivate: logger.info("Skipping private analysis %s.",str(ana)) continue if smsResults.isSuperseded(ana) and not useSuperseded: continue logger.debug("Building analysis %s.", str(ana)) ss = smsResults.getSqrts(ana) / TeV if ss == None: logger.debug("SS: " + str(ss) + str(ana)) continue ss = int(ss) if not ss in sqrts: continue for tx in smsResults.getTopologies(ana): if topologies != None and tx not in topologies: continue logger.debug(str(tx)) newAnalysis = analysis.ULanalysis() newAnalysis.sqrts = smsResults.getSqrts(ana) newAnalysis.lum = smsResults.getLumi(ana) stopo = _getRealTopo(tx) newAnalysis.label = ana + ":" + tx # "2012" constraint = smsResults.getConstraints(ana, topology=stopo) cond = smsResults.getConditions(ana, topology=stopo) if not constraint or constraint == "Not yet assigned": logger.debug("Constraint for %s %s (%s) does not exist.", str(ana), str(tx), str(stopo)) continue if not cond: logger.warning("Skipping %s %s (%s), Condition does not exist.", str(ana), str(tx), str(stopo)) continue cond = cond.replace("'", "").replace(" ", "").split(';') newAnalysis.constraint = constraint newAnalysis.conditions = cond newAnalysis.elementsEff = _getElementsEffs(constraint,cond) #Skip analyses with invalid elements if not newAnalysis.elementsEff: logger.warning("Skipping analysis %s due to invalid elements." % (newAnalysis.label)) continue # Add analysis to list of analyses: listOfAnalyses.append(newAnalysis) if not listOfAnalyses: logger.warning("Zero analyses loaded.") return listOfAnalyses