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;
}