import pandas as pd

import math
import itertools
import pprint
import sys

import matplotlib.pyplot as plt
plt.style.use('ggplot')

def printMinMax(key, df):
	print 'the (MIN, MAX) value of %10s is (%8s, %8s)' % (key, df[key].min(axis = 0), df[key].max(axis = 0))


modelparam = ['tbeta', 'lambda', 'kappa',  'mueff',  's_A_lambda', 's_A_kappa',  's_A_d3',  's_A_e3',  's_A_u3',  's_M_1',
            's_M_2',   's_M_3',   's_m_L3',  's_m_Q3',  's_m_d3',  's_m_e3',  's_m_u3',  'm_H1',    'w_H1',    'x_H1_ggF_bbH',    'b_H1_gg',
                 'b_H1_ee', 'b_H1_mumu',   'b_H1_tautau', 'b_H1_ss', 'b_H1_cc', 'b_H1_bb', 'b_H1_tt', 'b_H1_WW', 'b_H1_ZZ', 'b_H1_gamgam', 'b_H1_Zgam',
                   'b_H1_A1A1',   'b_H1_A1Z',    'm_H2',    'w_H2',    'x_H2_ggF_bbH',    'b_H2_gg', 'b_H2_ee', 'b_H2_mumu',   'b_H2_tautau', 'b_H2_ss',
                    'b_H2_cc', 'b_H2_bb', 'b_H2_tt', 'b_H2_WW', 'b_H2_ZZ', 'b_H2_gamgam', 'b_H2_Zgam',   'b_H2_A1A1',   'b_H2_A1Z',    'b_H2_H1H1',   'm_H3',
                        'w_H3',    'x_H3_ggF_bbH',    'b_H3_gg', 'b_H3_ee', 'b_H3_mumu',   'b_H3_tautau', 'b_H3_ss', 'b_H3_cc', 'b_H3_bb', 'b_H3_tt', 'b_H3_WW',
                         'b_H3_ZZ', 'b_H3_gamgam', 'b_H3_Zgam',   'b_H3_H1H1',   'b_H3_H1H2',   'b_H3_H2H2',   'b_H3_A1A1',   'b_H3_A1Z',    'b_H3_A2Z',    'b_H3_A2A2',
                           'm_A1',    'w_A1',    'x_A1_ggF_bbH',    'b_A1_gg', 'b_A1_ee', 'b_A1_mumu',   'b_A1_tautau', 'b_A1_ss', 'b_A1_cc', 'b_A1_bb', 'b_A1_tt', 'b_A1_gamgam',
                             'b_A1_Zgam',   'b_A1_ZH1',    'b_A1_ZH2',    'm_A2',    'w_A2',    'x_A2_ggF_bbH',    'b_A2_gg', 'b_A2_ee', 'b_A2_mumu',   'b_A2_tautau', 'b_A2_ss',
                              'b_A2_cc', 'b_A2_bb', 'b_A2_tt', 'b_A2_gamgam' 'b_A2_Zgam',   'b_A2_A1H1',   'b_A2_A1H2',   'b_A2_ZH1',    'b_A2_ZH2']

param_table = pd.read_table('./NMSSM_sample_for_CMS.out', sep='\t')

#print param_table


df = pd.DataFrame(param_table)

#fig1 = plt.figure()
ax = plt.gca()

#df.plot( x = 'm_A2', y = 'm_H1', kind = 'scatter')
#df.plot.scatter( x = 'm_A2', y = 'm_H1', ax=ax)

#fig1.savefig("H1A2.png")

#plt.show()


#fig2 = plt.figure()

#df.plot.scatter( x = 'm_A2', y = 'm_H2', ax=ax)

#fig2.savefig("H2A2.png")



#plt.show()

#fig3 = plt.figure()

#df.plot.scatter( x = 'm_A2', y = 'm_H3', ax=ax)

#fig3.savefig("H1A3.png")
#plt.show()

fig = plt.figure()


df.plot.scatter( x = 'm_H1', y = 'x_H1_ggF_bbH', ax=ax)



fig.savefig("XsH1.png")

plt.show()


#fig5 = plt.figure()

#df.plot.scatter( x = 'm_H2', y = 'x_H2_ggF_bbH', ax=ax)

#fig5.savefig("XsH2.png")

#plt.show()

#fig6 = plt.figure()


#df.plot.scatter( x = 'm_H3', y = 'x_H3_ggF_bbH', ax=ax)

#fig6.savefig("WqH3.png")
#plt.show()