stats.C

Go to the documentation of this file.

#include <iostream>
#include <fstream>
#include <sstream>
#include <string>

#include "TH1F.h"
#include "TH2F.h"
#include "TCanvas.h"
#include "TPad.h"
#include "TStyle.h"

//int stats(std::string inputFilePath, int firstRun, int lastRun, Float_t thresh){
int stats(TString inputFilePath, Int_t firstRun, Int_t lastRun, Float_t thresh, TString myDet){
  
  //std::string myfile = inputFilePath+"/bad_chan_stats.txt";
  //TString myfile = inputFilePath+"/bad_chan_stats.txt";
  //std::cout<<"my file: "<< myfile <<std::endl;
  //std::cout<<std::endl;
  
  // Open and load file
  std::ifstream openfile;
  // NOTE: PATH TO TXT FILE WOULD BE A PARAMETER!!
  //openfile.open("bad_chan_stats.txt");  // original
  //openfile.open(myfile.c_str()); //It works with string
  openfile.open(inputFilePath+"/bad_chan_stats.txt");
  
  if ( !(openfile.is_open()) ){
    std::cout<<std::endl;
    std::cout << "  ===  Unable to open bad_chan_stats.txt file ===" << std::endl;
    std::cout<<std::endl;
  }
  //else{
  //  std::cout<<"SUCCESS!! "<<std::endl;
  //}
  
  TH1F *hmismatch_distro = new TH1F("mismatch_distro",
                                    myDet+" Distribution of Mismatches;Mismatch(%);frequency",
                                    101,-0.2,100.8);
                                    //28,-0.2,1.2);
  
  TH1F *hfracdb_distro = new TH1F("fracdb_distro",
                                  myDet+" Fraction channels DB sees good, OnMon sees bad;FracDB(%);frequency",
                                  101,-0.2,100.8);
                                  //28,-0.2,1.2);
  
  TH1F *hfracom_distro = new TH1F("fracom_distro",
                                  myDet+" Fraction of Channels OnMon sees good, DB sees bad;FracOM(%);frequency",
                                  101,-0.2,100.8);
  //28,-0.2,1.8);
  
  // For binning purposes
  int nBins = (lastRun-firstRun)+3;
  int lowR  = firstRun-1;
  int highR = lastRun+2;
  
  TH2F *hmismatchVsRun = new TH2F("mismatchVsRun",
                                  myDet+" Mismatch vs Run number; Run; Mismatch (%)",
                                  //70,10930,11000,100,0.0,100.0);
                                  //275,22901,23176,100,0.0,100.0); // NOTE: NUMBER OF BINS AND START/END RUN WOULD BE A PARAMETER!!
                                  //3,23977,23979,100,0.0,100.0); 
                                  nBins,lowR,highR,100,0.0,100.0); 
  
  std::string Line;
  int counter = 0;
  int counterTotal = 0;
  
  std::cout<<std::endl;
  while (getline(openfile,Line)){
    
    std::istringstream is(Line);
    float run;
    //float srun;
    string srun;
    float mismatch;
    float fracdb;
    float fracom;
    
    // Reads each column separately from the file
    is >> run >> srun >> mismatch >> fracdb >> fracom; 
    counterTotal += 1;
    
    hmismatch_distro->Fill(mismatch);
    hfracdb_distro->Fill(fracdb);
    hfracom_distro->Fill(fracom);
    hmismatchVsRun->Fill(run,mismatch);
    
    //if( (mismatch == 0.0) ){
    //if( (mismatch > 0.000) && (mismatch <= 0.2)){
    //if( (mismatch > 0.4) && (mismatch <= 1.0)){
    //if( (mismatch > 1.0) && (mismatch <= 20.0)){
    //if( (mismatch > 20.0) && (mismatch <= 30.0)){
    
    // NOTE: WARKING MESSAGE WOULD BE A PARAMETER!! (?)
    if( (mismatch > thresh) ){
      std::cout<< run <<", "<< srun <<", "<< mismatch 
               <<", "<< fracdb <<", "<< fracom <<std::endl;  
      counter += 1;
    }
  } // end while
  std::cout<<std::endl;
  
  // TO DO: IMPLEMENT THIS
  // ls -l badchan/*.root | wc -l output
  //const int Nproc = 2795; // No. files processed
  /*
  // The number of files processed comes from % ls -l badchan_temp/*root | wc -l
  std::cout<<"Counter for current condition: "<< counter 
           <<" (out of "<<counterTotal<<" in txt file, "<<Nproc <<" processed) as percentage: "
           <<(float)counter*100/Nproc<<" % from proceesed ("<< Nproc <<")"<<std::endl;
  */
  
  TCanvas *c1 = new TCanvas("c1","c1",880,590);
  hmismatch_distro->SetLineColor(kBlue);
  hmismatch_distro->SetLineWidth(2);
  hmismatch_distro->SetStats(kTRUE);
  gPad->SetLogy();
  hmismatch_distro->Draw();  
  c1->SaveAs("mismatch_distro_"+myDet+".C");
  //c1->SaveAs("mismatch_distro.png");
  c1->SaveAs("/nusoft/app/web/htdoc/nova/users/novadq/BadChanValidationPlots/mismatch_distro_"+myDet+".png");
  
  //---
  /*
  // Draw an arrow on the canvas 
  TArrow arrow2(0.16,90.0,0.35,90.0,0.02,"----|>"); 
  arrow2.SetLineWidth(3);
  arrow2.SetLineColor(kRed);
  arrow2.SetFillColor(kRed);
  arrow2.DrawClone();
  
  // Draw a line on the canvas
  //TLine line(0.16,0.0,0.16,350.0);
  TLine line(0.16,0.0,0.16,1450.0);
  line.SetLineWidth(3);
  line.SetLineColor(kRed);
  line.DrawClone();
  
  // Add some text to the plot
  TLatex text(0.2,120.0,"Cut: 0.16");
  text.SetTextColor(kRed);
  text.DrawClone();
  */
  //--
  
  
  TCanvas *c2 = new TCanvas("c2","c2",880,590);
  hfracdb_distro->SetLineColor(kBlue);
  hfracdb_distro->SetLineWidth(2);
  hfracdb_distro->SetStats(kTRUE);
  gPad->SetLogy();
  hfracdb_distro->Draw();
  c2->SaveAs("fracdb_distro_"+myDet+".C"); 
  //c2->SaveAs("fracdb_distro.png"); 
  c2->SaveAs("/nusoft/app/web/htdoc/nova/users/novadq/BadChanValidationPlots/fracdb_distro_"+myDet+".png");
  
  TCanvas *c3 = new TCanvas("c3","c3",880,590);
  hfracom_distro->SetLineColor(kBlue);
  hfracom_distro->SetLineWidth(2);
  hfracom_distro->SetStats(kTRUE);
  gPad->SetLogy();
  hfracom_distro->Draw();
  c3->SaveAs("fracom_distro_"+myDet+".C"); 
  //c3->SaveAs("fracom_distro.png"); 
  c3->SaveAs("/nusoft/app/web/htdoc/nova/users/novadq/BadChanValidationPlots/fracom_distro_"+myDet+".png");
  
  // Set colors  
  const Int_t NRGBs2 = 3;        
  const Int_t NCont2 = 255;
  Double_t stops2[NRGBs2] = { 0.00, 0.3, 1.00};
  Double_t red2[NRGBs2]   = { 0.85, 0.25, 0.00};
  Double_t green2[NRGBs2] = { 0.96, 0.61, 0.00};
  Double_t blue2[NRGBs2]  = { 1.00, 0.98, 0.59};
  TColor::CreateGradientColorTable(NRGBs2, stops2, red2, green2, blue2, NCont2);
  gStyle->SetNumberContours(NCont2);
  
  TCanvas *c4 = new TCanvas("c4","c4",880,590);
  hmismatchVsRun->SetStats(kTRUE);
  hmismatchVsRun->Draw("colz");
  c4->SaveAs("mismatchVsRun_"+myDet+".C");
  //c4->SaveAs("mismatchVsRun.png");
  c4->SaveAs("/nusoft/app/web/htdoc/nova/users/novadq/BadChanValidationPlots/mismatchVsRun_"+myDet+".png");
  
  return 0;
}