moduleOccupancyTrend.cc

Go to the documentation of this file.

#include <iostream>
#include <fstream>
#include <sstream>
#include <string>
 
#include <TROOT.h>
#include <TStyle.h>
#include "TFile.h"
#include "THStack.h"
#include "TH1D.h"
#include "TDirectoryFile.h"
#include "TCanvas.h"
#include "TLine.h"
#include "TLegend.h"
#include "TRint.h"
 
int main(int argc, char *argv[]) {
 
  gROOT->SetStyle("Plain");
 
  char* filelist;
  char* modulelist;
 
  filelist= argv[1];
  modulelist= argv[2];
 
  int stripmin = 999;
  int stripmax = 999;
 
  if (argv[3] && argv[4]){
    stripmin=atoi(argv[3]);
    stripmax=atoi(argv[4]);
  }
 
  std::string detid;
  std::string hn;
 
  TLegend leg(0.1,0.7,0.2,0.9);
  leg.SetFillStyle(0);
 
  std::ifstream inmodules(modulelist);
 
  while(true){
    
    inmodules >> detid;
    if (!inmodules.good()) break;
 
    hn="ClusterDigiPosition__det__" + detid;//std::to_string(detid);
    
    TCanvas c1("c1","c1",1600,900);
  
    c1.SetBatch(kTRUE);
    c1.SetLogy(1);
    c1.SetGridy(1);
 
    //cout << detid << endl;
    std::ifstream fileToCountLines(filelist);
    std::size_t lines_count =0;
    std::string line;
 
    while (std::getline(fileToCountLines , line))
        ++lines_count;
 
    const float dim=lines_count;
 
    std::ifstream filesin(filelist);
 
    std::string filename;
    int k=0;
 
    TH1D* trend=new TH1D("trend","trend",int(dim),0.5,dim+0.5);
    trend->SetMarkerSize(3);
    trend->SetMarkerStyle(8);
    trend->SetMarkerColor(4);
 
    std::string ttitle=hn+"_trend";
    trend->SetTitle(ttitle.c_str());
 
    double max=-1;
    double min=1000000;
 
    leg.Clear();
    TFile* fin;
 
    THStack *hs = new THStack("hs","");
 
 
    while(true){
 
      filesin >> filename;
      if (!filesin.good()) break;      
 
      std::string runNum= filename.substr(filename.find("run_")+4, 6);
      //std::cout << runNum << std::endl;
     
      fin=TFile::Open(filename.c_str());
 
      if (!fin) {std::cout << "Cannot open file " << filename.c_str() << std::endl;
    return 0;
      }
      
      TH1D* Events=(TH1D*)fin->Get("TotEvents");
 
      double EvtNum=Events->GetBinContent(1);
 
      TH1D* histo=(TH1D*) fin->Get(hn.c_str());
      
      if (!histo) {std::cout << "Cannot open histo " << hn.c_str() << std::endl;
    return 0;
      }
 
      histo->SetDirectory(nullptr);
      histo->SetStats(kFALSE);
 
      if (hn.find("Summary")==std::string::npos) histo->Scale(1/EvtNum);
 
      double numberPerEvent;
      if (stripmin==999 && stripmax==999) numberPerEvent=histo->Integral();
      else numberPerEvent=histo->Integral(stripmin,stripmax);
 
      
      if (max<=histo->GetBinContent(histo->GetMaximumBin())) max=histo->GetBinContent(histo->GetMaximumBin());
      if (min>histo->GetBinContent(histo->GetMinimumBin())) min=histo->GetBinContent(histo->GetMinimumBin());
 
      histo->SetLineColor(k+1);
      histo->SetMarkerStyle(9);
      histo->SetMarkerColor(k+1);
           
      trend->SetBinContent(k+1,numberPerEvent);
      trend->GetXaxis()->SetBinLabel(k+1,runNum.c_str());
 
      leg.AddEntry(histo, runNum.c_str(), "L");
      hs->Add(histo);
      k++;
 
      fin->Close();
    }
 
    if (min==0) min=1.e-6;
 
    max=max*10;// in a way one can read the legend
 
    hs->SetMaximum(max);
    hs->SetMinimum(min);
    hs->SetTitle(hn.c_str());
 
    hs->Draw("nostack");
 
    TLine l;
    
    l.SetLineColor(4);
    l.SetLineStyle(2);
    l.SetLineWidth(3);
 
    l.DrawLine(128,min,128,max);
    l.DrawLine(256,min,256,max);
    l.DrawLine(384,min,384,max);
    l.DrawLine(384,min,384,max);
    l.DrawLine(512,min,512,max);
    l.DrawLine(640,min,640,max);
 
    leg.Draw();
    std::string outname= hn+"_Super.png";
    c1.SaveAs(outname.c_str());
    
    c1.SetGridx(1);
    
    double mintrend=0;
    if (trend->GetMinimum()==0) mintrend=1.e-4;
    else mintrend=trend->GetMinimum();
 
    trend->SetStats(kFALSE);
 
    trend->GetYaxis()->SetRangeUser(mintrend*0.5,trend->GetMaximum()*2);
    trend->Draw("P");
    outname=hn+"_Trend.png";
    c1.SaveAs(outname.c_str());
    c1.Clear();
 
    delete trend;      
    
  }  
  return 0;
}