ADDING ANALYSIS FOR REMOVAL OF BAD CHUNKS (D.Dobrikeit & AK)

[u/mrichter/AliRoot.git] / ANALYSIS / BadChunkFilter / ProcessBadChunks02.C

#include <iostream>\r
#include <fstream>\r
\r
#include "TFile.h"\r
#include "TTree.h"\r
#include "TString.h"\r
\r
using namespace std;\r
\r
int ProcessBadChunks02(){\r
  // Open the file\r
  TFile* file = TFile::Open("BadChunkTree.root", "READ");\r
  if (!file || !file->IsOpen()) {\r
    cout<<"File not found!"<<endl;\r
    return 1;\r
  }\r
  // Get the tree\r
  TTree* ftree = (TTree*)file->FindObjectAny("fTree");\r
  if (!ftree) {\r
    cout<<"File doesn't contain fTree!"<<endl;\r
    return 2;\r
  }\r
  //Print out "I'm alive"\r
  cout<<"----------------------------------------------------"<<endl;\r
  cout<<" Bad Chunk Analysis Macro"<<endl;\r
  cout<<"----------------------------------------------------"<<endl;  \r
  cout<<" Entries in Event Tree......"<<ftree->GetEntries()<<endl;\r
  cout<<"----------------------------------------------------"<<endl;\r
  Int_t llRunNumber = 0;\r
  TString * llFileName = 0x0;\r
  Int_t llNTracks = 0;\r
  Int_t llNGlobalTracks = 0;\r
  \r
  ftree->SetBranchAddress("fRunNumber"      ,&llRunNumber     );\r
  ftree->SetBranchAddress("fFileName"       , &llFileName   );\r
  ftree->SetBranchAddress("fNTracks"        ,&llNTracks       );\r
  ftree->SetBranchAddress("fNGlobalTracks"  ,&llNGlobalTracks );\r
\r
  //Output ostreams: text files with chunk location, good and bad\r
  \r
  filebuf fbgood;\r
  fbgood.open ("goodguys.txt",ios::out);\r
  ostream osgood(&fbgood);\r
\r
  filebuf fbbad;\r
  fbbad.open ("badguys.txt",ios::out);\r
  ostream osbad(&fbbad);\r
  \r
  //Main Tree Loop\r
  Long_t lProcessedChunks = 0;\r
  Long_t lProcessedChunksGood = 0;\r
  Long_t lProcessedChunksBad = 0;\r
  Long_t lTracks = 0;\r
  Long_t lGlobalTracks = 0;\r
  TString *lChunkName = new TString();\r
  \r
  //First Chunk entry\r
  ftree->GetEntry(0);\r
  *lChunkName = *llFileName;\r
  \r
  cout<<"TEST  "<<lChunkName->Data() <<", track = "<<lTracks<<", globals = "<<lGlobalTracks<<" " << llFileName->Data()  << endl;\r
  \r
  for(Long_t iEvent = 1; iEvent<ftree->GetEntries(); iEvent++){\r
    ftree->GetEntry(iEvent);\r
    if( !llFileName->EqualTo(*lChunkName) ){\r
      lProcessedChunks++;\r
      //Change Chunk\r
      if( lTracks > 0 && lGlobalTracks ==0){\r
        //Candidate bad chunk found!\r
        cout<<"BAD CHUNK at "<<lChunkName->Data() <<", track = "<<lTracks<<", globals = "<<lGlobalTracks<< endl;\r
        osbad<<lChunkName->Data()<<endl;\r
        lProcessedChunksBad++;\r
      }else{\r
        //This looks OK...\r
        //cout<<"GOOD CHUNK at "<<lChunkName<<endl;\r
        osgood<<lChunkName->Data()<<endl;\r
        lProcessedChunksGood++;\r
      }\r
      //Get ready to loop over new chunk\r
      *lChunkName = *llFileName;\r
      lTracks = llNTracks;\r
      lGlobalTracks = llNGlobalTracks;\r
      if(lProcessedChunks%5000==0) cout<<"---> Processed "<<lProcessedChunks<<"..."<<endl;\r
    }else{\r
      lTracks        += llNTracks;\r
      lGlobalTracks  += llNGlobalTracks;\r
    }\r
  }\r
  //CLOSE the processing: one extra pass...\r
  lProcessedChunks++;\r
  if( lTracks > 0 && lGlobalTracks ==0){\r
    //Candidate bad chunk found!\r
    cout<<"BAD CHUNK at "<<lChunkName->Data()<<endl;\r
    osbad<<lChunkName->Data() <<endl;\r
    lProcessedChunksBad++;\r
  }else{\r
    //This looks OK...\r
    //cout<<"GOOD CHUNK at "<<lChunkName<<endl;\r
    osgood<<lChunkName->Data()<<endl;\r
    lProcessedChunksGood++;\r
  }\r
  fbgood.close();\r
  fbbad.close();\r
  cout<<"----------------------------------------------------"<<endl;\r
  cout<<"Processed chunks, total..: "<<lProcessedChunks<<endl;\r
  cout<<"Processed chunks, good...: "<<lProcessedChunksGood<<endl;\r
  cout<<"Processed chunks, bad....: "<<lProcessedChunksBad<<endl;\r
  cout<<"----------------------------------------------------"<<endl;\r
  cout<<endl;\r
  cout<<"---> Good chunks saved to \"goodguys.txt\""<<endl;\r
  cout<<"---> Bad chunks saved to \"badguys.txt\""<<endl;\r
  cout<<endl;\r
  cout<<"DONE!"<<endl;\r
  return 0;\r
   \r
}\r