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

# <CENTER><img src="../../images/ATLASOD.gif" style="width:50%"></CENTER>

# <CENTER><h1>Get running the full Hyy analysis using the 13 TeV dataset in 5 minutes!

# **Introduction**
# The analysis is based on the 13 TeV Open Data.
# The ATLAS note [ATL-OREACH-PUB-2020-001](https://atlas.web.cern.ch/Atlas/GROUPS/PHYSICS/PUBNOTES/ATL-OREACH-PUB-2020-001) can be used as a guide on the content, properties, capabilities and limitations of the released datasets.
# 
# In the following, in about **5 minutes** we are going to re-produce the H->yy analysis plots from the note.
# <CENTER>
#     <img src="../../images/fig_12b.png" alt="mass" width="350" height="200">
# </CENTER>

# In[ ]:


import os
import ROOT
from ROOT import TMath
import time


# In[ ]:


start = time.time()


# ##  First, we will download from github the 13 TeV C++ analysis software
# 

# In[ ]:


myCmd = os.popen('ls -lhrt').read()
print(myCmd)


# in case it already exists, remove it

# In[ ]:


folder_demo = 'atlas-demo-Hyy'
command1 = 'rm -rf '+folder_demo
os.system(command1)


# In[ ]:


myCmd = os.popen('ls -lhrt').read()
print(myCmd)


# In[ ]:


folder_demo = 'atlas-demo-Hyy'
command1 = 'mkdir '+folder_demo
os.system(command1)


# Show the contents in the current folder, and checck your new folder is there

# In[ ]:


myCmd = os.popen('ls -lhrt').read()
print(myCmd)


# Get into the folder which we just created

# In[ ]:


os.chdir(folder_demo+"/")


# Print the current path and check that you are really there

# In[ ]:


myCmd = os.popen('pwd').read()
print(myCmd)


# Clone the analysis code **(do this only once!)**

# In[ ]:


myCmd = os.popen('git clone https://github.com/atlas-outreach-data-tools/atlas-outreach-cpp-framework-13tev.git').read()
print(myCmd)


# Get into the folder which contains the analysis code

# In[ ]:


os.chdir("atlas-outreach-cpp-framework-13tev/")


# In[ ]:


myCmd = os.popen('ls -lhrt').read()
print(myCmd)


# let's create the output folders **do it once**

# In[ ]:


command3 = "./welcome_web.sh"
os.system(command3)


# ### Let's run the Hyy physics analysis
# 
# #### From the list below, we will selection the proper HyyAnalysis option:

# In[ ]:


os.chdir("Analysis/HyyAnalysis")


# In[ ]:


myCmd = os.popen('pwd').read()
print(myCmd)


# In[ ]:


myCmd = os.popen('ls -lhrt').read()
print(myCmd)


# Check that the directory **Output_HyyAnalysis** is there! If not, **create it**

# In[ ]:


#command1 = 'mkdir Output_HyyAnalysis'
#os.system(command1)


# ### Now, we will execute the C++ analysis code. It will take some time (less than 5mins)

# First, we will run over MC samples

# In[ ]:


command4 = "./run_web.sh"
os.system(command4)


# #### Now the analysis is done

# Inside that directory **Output_HyyAnalysis** you will find the output root file: data.root

# In[ ]:


myCmd = os.popen('ls -lhrt Output_HyyAnalysis/*').read()
print(myCmd)


# Now we will run over MC samples

# In[ ]:


command4 = "./run_web2.sh"
os.system(command4)


# **only** if there are 5 root files, you can proceed with plotting

# In[ ]:


myCmd = os.popen('ls -lhrt Output_HyyAnalysis/*').read()
print(myCmd)


# ### Let's run the Plotting code

# In[ ]:


os.chdir("../../Plotting/")


# In[ ]:


myCmd = os.popen('pwd').read()
print(myCmd)


# In[ ]:


myCmd = os.popen('ls -lhrt').read()
print(myCmd)


# Check that the directory **histograms** is there! If not, **create it**

# In[ ]:


#command5 = 'mkdir histograms/'
#os.system(command5)


# In[ ]:


command5 = "./plotme_Hyyweb.sh"
os.system(command5)


# In[ ]:


myCmd = os.popen('ls -lhrt histograms/*png').read()
print(myCmd)


# <CENTER>
#     <h2> The two photons invariant mass </h2>
#     <img src="atlas-demo-Hyy/atlas-outreach-cpp-framework-13tev/Plotting/histograms/hist_mYY_bin1.png"  alt="mass" width="700" height="400">
# </CENTER>

# In[ ]:


end = time.time()
duration = end-start
print("Finished in {} min {} s".format(int(duration//60),int(duration%60))) # Python3


# Finally, you can compare it to the ATLAS plot from a published [paper](https://arxiv.org/abs/1802.04146) that uses **3 times more data!**
# 
# <CENTER>
#     <img src="../../images/fig_09.png"  alt="mass" width="350" height="200">
# </CENTER>

# ### Done!