#!/usr/bin/env python
# coding: utf-8

# # How To: Look up the upper limit of a particular result, for a particular set of masses

# In[1]:


# Set up the path to SModelS installation folder
import sys; sys.path.append("."); import smodels_paths


# In[2]:


# Import those parts of smodels that are needed for this exercise
from smodels.base.physicsUnits import GeV
from smodels.experiment.databaseObj import Database


# In[3]:


## Load the official database
db = Database( "official" )


# ## Look up upper limit for an Upper Limit-type result:

# In[4]:


# Select desired result:
resultID = ["CMS-PAS-SUS-13-016"]
txname = ["T1tttt"]
expResult = db.getExpResults(analysisIDs=resultID,txnames=txname)[0]
print ('selected result:',expResult)


# In[5]:


# Define the desired mass vector (must be consistent with the txname/simplified model):
masses = [[500*GeV, 150*GeV],[500*GeV, 150*GeV]]
print ('UL for mass\n',masses,' is: ',expResult.getUpperLimitFor(mass=masses,txname="T1tttt"))


# ## Look up upper limit for an Efficiency Map-type result:

# In[6]:


# Select desired result:
resultID = ["CMS-PAS-SUS-13-016"]
txname = ["T1tttt"]
expResult = db.getExpResults(analysisIDs=resultID,txnames=txname,dataTypes='efficiencyMap')[0]
print ('selected result:',expResult)


# In[7]:


# For an efficiency map result one only needs to specify the desired signal region (dataset)
datasetID = 'sr0'
print ('UL for dataset',datasetID,' is: ',expResult.getUpperLimitFor(dataID=datasetID))