histgen/draw_check_curve.cxx

Go to the documentation of this file.

#include "TCanvas.h"
#include "TFile.h"
//#include "TKey.h"
#include "TTree.h"
#include "TH1F.h"
//#include <TROOT.h>
#include <sstream>
#include <iostream>
#include <string>
#include <cstring>
#include <vector>
#include <cmath>
#include "TGraph.h"
 
#include "curve.h"
using namespace std;
const float  masses[5] = { 0.00051100, 0.10566, 0.13957, 0.4937, 0.93827 };
 
const bool m_debug = false;
void draw_check_curve(){
  TFile *f = new TFile("DedxSampleHists.root");
  TTree *t = (TTree*)f->Get("bin");
  int num, total;
  t->SetBranchAddress("betagammaBounds",&num);
  t->SetBranchAddress("totalNum",&total);
  t->GetEntry(0);
  double *pre_x = new double[num];
  t->SetBranchAddress("betagamma", pre_x);
  t->GetEntry(0);
  
  double *x = new double[total];
  double *y = new double[total];
  ReadCurPara(0);
  double mean_exp[5];
  double sigma_exp[5];
  const double theta(1), Nohit(15), t0(1200);
  
  double *curve_y = new double[total];
  for(int i=0; i<total; i++){
    TH1F *h = (TH1F*)f->Get(Form("h%d",i));
    y[i] = h->GetMean();
    x[i] = pre_x[i/10];
    dedx_pid_exp(x[i]*masses[2], theta, Nohit, t0, mean_exp, sigma_exp);    
    curve_y[i] = mean_exp[2];
    if(m_debug) cout << "i= " << i << "  x= " << x[i] << "  y= " << y[i] << endl;
  }  
   
  TCanvas *c1 = new TCanvas("c1","curve",10,10,800,642);
  gPad->SetLogx(1);
  TGraph *g = new TGraph(total, x, y);
  g->Draw("a*");
  TGraph *curve_g = new TGraph(total, x, curve_y);
  curve_g->SetLineColor(2);
  curve_g->Draw("c,same");
 
  c1->SaveAs("figs/curve.eps");
  TFile *save_file = new TFile("figs/curve.root","recreate");
  c1->Write();
  save_file->Close();
}