FileLumper.cxx

Go to the documentation of this file.

//==========================================================
//
// FileLumper.cxx
//
// Authors: Pedro Ochoa
//
// Written: December 21 2007
//
//===========================================================

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

#include "TSystem.h"
#include "MCNNAnalysis/FileLumper.h"
#include "MCNNAnalysis/NNReader.h"

using namespace std;

FileLumper::FileLumper(){
  
  OutputFile = new TFile();
  OutputTree = new TTree("nn","Minos NuE Analysis Results");
  nread = new NNReader("nn",true);

  cce = new MinosCompactEvent();

  mar = new MinosAnalysisResult();
  
}

FileLumper::FileLumper(const char* path_to_unmerged, const char* filename, Int_t pieces_in){

  cce = new MinosCompactEvent();
  mar = new MinosAnalysisResult();
  
  Bool_t catastrophe=false;
  nread = new NNReader("nn",true);

  //-->Putting input files in chain
  for(int ii=0;ii<pieces_in;++ii){
    
    //make file name
    ostringstream line;
    line << path_to_unmerged << "/mc*nn_" << ii << "_" << filename;
    string linestr = line.str();
    //NOTE: did mc*nn instead of mcnn just to add a wildcard in there. For some reason in 
    // root 05-16-00 TChain:Add always returns 1 if file is completely specified without
    // wildcards, even if file name is bogus
    
    //Adding to chain
    Int_t filenum = nread->Add(linestr.c_str());
    cout << "Adding " << linestr.c_str() << endl;
    if(filenum!=1){
      cout << "WARNING: Could not add " << linestr.c_str() << endl;
      cout << "-->Aborting; Will not write lumped file." << endl;
      catastrophe=true;
      break;
    }   
  }

  tot_entries=nread->GetEntries();
  cout << "Have " << tot_entries << " in big nread chain" << endl;
  
  if(catastrophe==false){

    ostringstream fileoutput;
    // hardcoding $MCNNTMP as a hack
    fileoutput << "$MCNNTMP/mcnn_" << filename;
    string fileoutputstr = gSystem->ExpandPathName((fileoutput.str()).c_str());

    OutputFile = new TFile(fileoutputstr.c_str(),"recreate");
    OutputTree = new TTree("nn","Minos NuE Analysis Results");
    OutputTree->SetDirectory(OutputFile);   
    
    SetBranches();
    
    //Saving to output tree
    Int_t srCtr=0;
    while(nread->GetEntry(srCtr)){
      ++srCtr;
            
      cce = nread->GetInputCCE();
      mar = nread->GetMinosAnalysisResult();
      
      OutputTree->Fill();
      
    }
    
    OutputTree->Write();
    OutputFile->Close();

  }//only writing file if all input files were present
  
}

FileLumper::~FileLumper(){

  delete OutputFile;
  delete nread;
  delete cce;
  delete mar;

}

//===================================================================
void FileLumper::SetBranches(){
  
  OutputTree->Branch("cce","MinosCompactEvent",&cce);
  OutputTree->Branch("comparisonResults", "MinosAnalysisResult", &mar);
  OutputTree->Branch("run", &nread->run, "run/I");
  OutputTree->Branch("snarl", &nread->snarl, "snarl/I");
  OutputTree->Branch("evt", &nread->evt, "evt/I");
  OutputTree->Branch("nuid", &nread->nuid, "nuid/I");
  OutputTree->Branch("nuidnoosc",&nread->nuidnoosc,"nuidnoosc/I");
  OutputTree->Branch("action", &nread->action, "action/I");
  OutputTree->Branch("nue", &nread->nue, "nue/F");
  OutputTree->Branch("nuenoosc",&nread->nuenoosc,"nuenoosc/F");
  OutputTree->Branch("reco_nue",&nread->reco_nue,"reco_nue/F");
  OutputTree->Branch("y", &nread->y, "y/F");
  OutputTree->Branch("emfrac", &nread->emfrac, "emfrac/F");
  OutputTree->Branch("ires",&nread->ires,"ires/I");
  OutputTree->Branch("qtot", &nread->qtot, "qtot/F");
  OutputTree->Branch("vtxx", &nread->vtxx, "vtxx/F");
  OutputTree->Branch("vtxy", &nread->vtxy, "vtxy/F");
  OutputTree->Branch("vtxz", &nread->vtxz, "vtxz/F");
  OutputTree->Branch("vtxu", &nread->vtxu, "vtxu/F");
  OutputTree->Branch("vtxv", &nread->vtxv, "vtxv/F");
  OutputTree->Branch("tvtxx", &nread->tvtxx, "tvtxx/F");
  OutputTree->Branch("tvtxy", &nread->tvtxy, "tvtxy/F");
  OutputTree->Branch("tvtxz", &nread->tvtxz, "tvtxz/F");
  OutputTree->Branch("tptype",&nread->tptype,"tptype/I");
  OutputTree->Branch("tpx",&nread->tpx,"tpx/F");
  OutputTree->Branch("tpy",&nread->tpy,"tpy/F");
  OutputTree->Branch("tpz",&nread->tpz,"tpz/F");
  OutputTree->Branch("npln", &nread->npln, "npln/I");
  OutputTree->Branch("nstp", &nread->nstp, "nstp/I");
  OutputTree->Branch("nplnu", &nread->nplnu, "nplnu/I");
  OutputTree->Branch("nplnv", &nread->nplnv, "nplnv/I");          
  OutputTree->Branch("mcpresel",&nread->mcpresel,"mcpresel/I");
  OutputTree->Branch("datapresel",&nread->datapresel,"datapresel/I");
  OutputTree->Branch("nxtalk",&nread->nxtalk,"nxtalk/I");
  OutputTree->Branch("nxtalk_tagged",&nread->nxtalk_tagged,"nxtalk_tagged/I");
  OutputTree->Branch("nxtalk_mistagged",&nread->nxtalk_mistagged,"nxtalk_mistagged/I"); 
  OutputTree->Branch("nphysics",&nread->nphysics,"nphysics/I");
  OutputTree->Branch("w2",&nread->w2,"w2/F");
  OutputTree->Branch("totpt",&nread->totpt,"totpt/F");
  OutputTree->Branch("snarl_pot",&nread->snarl_pot,"snarl_pot/F");
  OutputTree->Branch("good_snarl",&nread->good_snarl,"good_snarl/I");
  OutputTree->SetDirectory(OutputFile);

}

---