#include "TROOT.h"
#include "TStyle.h"
#include "TLatex.h"
#include "TCanvas.h"
#include "TColor.h"
#include "TAxis.h"
#include "TH2.h"
#include <fstream>

void setstyle(){
  gROOT->SetStyle("Plain");
  gStyle->SetTitleAlign(23);
  gStyle->SetTitleX(0.54);
  gStyle->SetTitleY(0.97);
  gStyle->SetTitleBorderSize(0);
  gStyle->SetTitleFillColor(0);
  gStyle->SetTitleColor(1);
  gStyle->SetLabelFont(132,"xyz");
  gStyle->SetTitleFont(132,"xyz");
  gStyle->SetTitleFont(132,"");
  gStyle->SetTextFont(132);
  gStyle->SetStatFont(132);
  gStyle->SetOptStat(0);
  gStyle->SetNdivisions(510,"xyz");
  gStyle->SetNdivisions(1015,"z");
  gStyle->SetTickLength(0.02,"xyz");
  gStyle->SetLabelSize(0.075,"xy");
  gStyle->SetLabelSize(0.05,"z");
  gStyle->SetTitleSize(0.075,"xyz");
  gStyle->SetTitleSize(0.075,"");
  gStyle->SetPadTopMargin(0.15);
  gStyle->SetPadRightMargin(0.10);
  gStyle->SetPadLeftMargin(0.15);
  gStyle->SetPadBottomMargin(0.18);
  gStyle->SetOptLogz(0);

  Int_t MyPalette[15];
  Double_t r[]    = {0.00, 0.00, 1.00, 1.00, 0.50};
  Double_t g[]    = {0.00, 0.00, 1.00, 0.00, 0.00};
  Double_t b[]    = {0.50, 1.00, 1.00, 0.00, 0.00};
  Double_t stop[] = {0.00, 0.30, 0.50, 0.70, 1.00};
  Int_t FI = TColor::CreateGradientColorTable(5, stop, r, g, b, 15);
  for (int i=0;i<15;i++) MyPalette[i] = FI+i;
  gStyle->SetPalette(15, MyPalette);
  gStyle->SetNumberContours(15);
}

void plot_evenz_oddn_jpi(){
  setstyle();

  TCanvas *c1 = new TCanvas("c1","c1",500,400); 
  TH2D *h = new TH2D("h","#it{J^{#pi}} of ground states",81,0.,162.,56,-1.,111.);
  std::ifstream ifs("evenz_oddn_jpi.dat");
  Int_t n,z;
  Double_t jpi2;
  while (ifs >> z >> n >> jpi2) {
    h->Fill(n,z,jpi2/2.);
  }
  h->SetMaximum(7.3);
  h->SetMinimum(-7.3);
  h->Draw("colz");
  h->GetXaxis()->SetTitle("Neutron number #it{N}");
  h->GetYaxis()->SetTitle("Proton number #it{Z}");
  h->GetXaxis()->CenterTitle();
  h->GetYaxis()->CenterTitle();

  TLatex Tl;
  Tl.SetTextAlign(12);
  Tl.SetNDC();
  Tl.SetTextSize(0.075);
  Tl.DrawLatex(0.20,0.80,"#it{Z}: even");
  Tl.DrawLatex(0.20,0.73,"#it{N}: odd");
}