polepp/plotprob.C at master · kdund/polepp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
{
  gROOT->Reset();
  //
  TTree *tree = new TTree("probTree", "probability");
  tree->ReadFile("construct.dat");
  tree->Draw("shyp:N:RL:P","","goff");
  Long64_t nent = tree->GetEntries();
  if (nent==0) {
    std::cout << "Error: no entries in tree! Check input file (construct.dat)" << std::endl;
    return;
  }
  Double_t smin = TMath::MinElement(nent, tree->GetV1());
  Double_t smax = TMath::MaxElement(nent, tree->GetV1());
  Double_t nmin = TMath::MinElement(nent, tree->GetV2());
  Double_t nmax = TMath::MaxElement(nent, tree->GetV2());
  Double_t wmin = TMath::MinElement(nent, tree->GetV4());
  Double_t wmax = TMath::MaxElement(nent, tree->GetV4());
  Double_t s,n,w,sp;
  Double_t ds;
  //
  Int_t i=0;
  sp = *(tree->GetV1()+0);
  while ((ds==0) && (i<nent)) {
    s = *(tree->GetV1()+i);
    ds = s-sp;
    sp = s;
    i++;
  }
  Int_t    nsbins = int((smax-smin)/ds)+1;
  TH2F *hist = new TH2F("prob","P(N|s)",int(nmax-nmin)+1,nmin,nmax,nsbins,smin,smax);
  hist->SetMinimum(wmin);
  hist->SetMaximum(wmax);

  for (int i=0; i<nent; i++) {
    s = *(tree->GetV1()+i);
    n = *(tree->GetV2()+i);
    w = *(tree->GetV4()+i);
    hist->Fill(n,s,w);
  }

  hist->GetXaxis()->SetTitle("N");
  hist->GetYaxis()->SetTitle("s_{hyp}");
  hist->Draw("colz");

}