TkAlCaSkimTreeMerger Class Reference

Inheritance diagram for TkAlCaSkimTreeMerger:

Public Member Functions
void	analyze (const edm::Event &, const edm::EventSetup &) override
void	beginJob () override
void	endJob () override
	TkAlCaSkimTreeMerger (const edm::ParameterSet &iConfig)
	~TkAlCaSkimTreeMerger ()
Public Member Functions inherited from edm::EDAnalyzer
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
	EDAnalyzer ()
ModuleDescription const &	moduleDescription () const
std::string	workerType () const
virtual	~EDAnalyzer ()
Public Member Functions inherited from edm::EDConsumerBase
	EDConsumerBase ()
ProductHolderIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const
void	itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const
void	itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const
std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const
void	labelsForToken (EDGetToken iToken, Labels &oLabels) const
bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const
bool	registeredToConsumeMany (TypeID const &, BranchType) const
void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)
virtual	~EDConsumerBase ()

Private Types
typedef std::map< uint32_t, uint32_t >	DetHitMap

Private Attributes
TChain *	ch_
std::string	filelist_
std::string	firstfilename_
TTree *	firsttree_
DetHitMap	hitmap_
int	maxhits_
edm::ParameterSet	maxhitsSet_
int	maxPXBhits_
int	maxPXFhits_
int	maxTEChits_
int	maxTIBhits_
int	maxTIDhits_
int	maxTOBhits_
TStopwatch	myclock
TTree *	out_
std::string	outfilename_
DetHitMap	overlapmap_
std::string	treename_

Additional Inherited Members
Public Types inherited from edm::EDAnalyzer
typedef EDAnalyzer	ModuleType
Static Public Member Functions inherited from edm::EDAnalyzer
static const std::string &	baseType ()
static void	fillDescriptions (ConfigurationDescriptions &descriptions)
static void	prevalidate (ConfigurationDescriptions &)
Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)
EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)
template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)
ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...
template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()
void	consumesMany (const TypeToGet &id)
template<BranchType B>
void	consumesMany (const TypeToGet &id)
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)
template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

Definition at line 25 of file TkAlCaSkimTreeMerger.cc.

Member Typedef Documentation

typedef std::map<uint32_t,uint32_t> TkAlCaSkimTreeMerger::DetHitMap

private

Definition at line 44 of file TkAlCaSkimTreeMerger.cc.

Constructor & Destructor Documentation

TkAlCaSkimTreeMerger::TkAlCaSkimTreeMerger

(

const edm::ParameterSet &

iConfig

)

Definition at line 62 of file TkAlCaSkimTreeMerger.cc.

References ch_, gather_cfg::cout, firsttree_, edm::ParameterSet::getParameter(), maxhits_, maxhitsSet_, maxPXBhits_, maxPXFhits_, maxTEChits_, maxTIBhits_, maxTIDhits_, maxTOBhits_, and out_.

                                                                         :
    filelist_(iConfig.getParameter<std::string>("FileList")), 
    treename_(iConfig.getParameter<std::string>("TreeName")),
    outfilename_(iConfig.getParameter<std::string>("OutputFile")),
    // maxhits_(iConfig.getParameter<vint>("NhitsMaxLimit"))
    maxhits_(iConfig.getParameter<int32_t>("NhitsMaxLimit")),
    maxhitsSet_(iConfig.getParameter<edm::ParameterSet>("NhitsMaxSet"))
{
  maxPXBhits_=maxhitsSet_.getParameter<int32_t>("PXBmaxhits");
  maxPXFhits_=maxhitsSet_.getParameter<int32_t>("PXFmaxhits");
  maxTIBhits_=maxhitsSet_.getParameter<int32_t>("TIBmaxhits");
  maxTIDhits_=maxhitsSet_.getParameter<int32_t>("TIDmaxhits");
  maxTOBhits_=maxhitsSet_.getParameter<int32_t>("TOBmaxhits");
  maxTEChits_=maxhitsSet_.getParameter<int32_t>("TECmaxhits");
  //anything you want to do for initializing
  std::cout<<"\n\n*** MAX N HITS = "<<maxhits_<<std::endl<<std::endl;
  out_=0;
  firsttree_=0;
  ch_=0;
}

TkAlCaSkimTreeMerger::~TkAlCaSkimTreeMerger

(

)

Definition at line 84 of file TkAlCaSkimTreeMerger.cc.

References ch_, and gather_cfg::cout.

                                           {
  //default destructor
  // delete out_;
  // delete firsttree_;

  delete ch_;
  std::cout<<"finished."<<std::endl;
}

Member Function Documentation

void TkAlCaSkimTreeMerger::analyze ( const edm::Event &  , const edm::EventSetup &    )

overridevirtual

Implements edm::EDAnalyzer.

Definition at line 183 of file TkAlCaSkimTreeMerger.cc.

                                                                       {
    // std::cout<<firsttree_->GetEntries()<<std::endl;
}//end analyze

void TkAlCaSkimTreeMerger::beginJob ( void  )

overridevirtual

Reimplemented from edm::EDAnalyzer.

Definition at line 94 of file TkAlCaSkimTreeMerger.cc.

References ch_, gather_cfg::cout, filelist_, lut2db_cfg::filename, first, firstfilename_, hitmap_, recoMuon::in, myclock, overlapmap_, AlCaHLTBitMon_QueryRunRegistry::string, and treename_.

                                   {
 
  myclock.Start();

  //prepare the chain
  ch_=new TChain(treename_.c_str());
  std::cout<<"The chain contains "<<ch_->GetNtrees()<<" trees"<<std::endl;
  
  //load the trees into the chain
  std::ifstream flist(filelist_.c_str(),std::ios::in);
  std::string filename;
  std::string firstfilename;
  bool first=true;
  while(!flist.eof()){
    filename="";
    flist>>filename;
    if(filename.empty())continue;
    //std::cout<<"Adding "<<filename<<std::endl;
    ch_->Add(filename.c_str());
    if(first){
      firstfilename_=filename;
      first=false;
    }
   
  }
  std::cout<<"Now the chain contains "<<ch_->GetNtrees()<<" trees ("<<ch_->GetEntries()<<" entries)"<<std::endl;

 
  unsigned int id_ch=0;
  uint32_t nhits_ch=0,noverlaps_ch=0;
  ch_->SetBranchAddress("DetId",    &id_ch);
  ch_->SetBranchAddress("Nhits",    &nhits_ch);
  ch_->SetBranchAddress("Noverlaps",&noverlaps_ch);

  ch_->SetBranchStatus("SubDet",0);
  ch_->SetBranchStatus("Layer",0);
  ch_->SetBranchStatus("is2D",0);
  ch_->SetBranchStatus("isStereo",0);
  ch_->SetBranchStatus("posX",0);
  ch_->SetBranchStatus("posY",0);
  ch_->SetBranchStatus("posZ",0);
  ch_->SetBranchStatus("posR",0);
  ch_->SetBranchStatus("posEta",0);
  ch_->SetBranchStatus("posPhi",0);//now only id, nhits and noverlaps are on...


  int totnhits(0),totnoverlaps(0);


  //look if you find this detid in the map
  DetHitMap::iterator mapiter;
  DetHitMap::iterator overlapiter;
  
  for(int ent=0;ent<ch_->GetEntries();++ent){
  //  for(int ent=0;ent<100;++ent){
    ch_->GetEntry(ent);
    totnhits+=nhits_ch;
    totnoverlaps+=noverlaps_ch;

    mapiter= hitmap_.find(id_ch);
    if(mapiter!=hitmap_.end() ){//present, increase its value
      hitmap_[id_ch]=hitmap_[id_ch]+nhits_ch;
    }
    else{//not present, let's add this key to the map with value=1
      hitmap_.insert(std::pair<uint32_t, uint32_t>(id_ch, nhits_ch));
    }
    //do the same for overlaps
    overlapiter= overlapmap_.find(id_ch);
    if(overlapiter!=overlapmap_.end() ){
      overlapmap_[id_ch]=overlapmap_[id_ch]+noverlaps_ch;
    }
    else{
      overlapmap_.insert(std::pair<uint32_t, uint32_t>(id_ch, noverlaps_ch));
    }
    
  }//end loop on ent - entries in the chain


  std::cout<<"Nhits in the chain: "<<totnhits<<std::endl;
  std::cout<<"NOverlaps in the chain: "<<totnoverlaps<<std::endl;


  myclock.Stop();
  std::cout<<"Finished beginJob after "<<myclock.RealTime()<<" s (real time) / "<<myclock.CpuTime()<<" s (cpu time)"<<std::endl;
  myclock.Continue();
}//end beginJob

void TkAlCaSkimTreeMerger::endJob ( void  )

overridevirtual

Reimplemented from edm::EDAnalyzer.

Definition at line 188 of file TkAlCaSkimTreeMerger.cc.

References gather_cfg::cout, firstfilename_, firsttree_, hitmap_, maxhits_, maxPXBhits_, maxPXFhits_, maxTEChits_, maxTIBhits_, maxTIDhits_, maxTOBhits_, mod(), myclock, out_, EdgesToViz::outfile, outfilename_, overlapmap_, and treename_.

                                 {


  //address variables in the first tree and in the chain
  TFile *firstfile=new TFile(firstfilename_.c_str(),"READ");
  firsttree_=(TTree*)firstfile->Get(treename_.c_str());
  std::cout<<"the first tree has "<<firsttree_->GetEntries() <<" entries"<<std::endl; 
  unsigned int id=0;
  uint32_t nhits=0,noverlaps=0;
  float posX(-99999.0),posY(-77777.0),posZ(-88888.0);
  float posEta(-6666.0),posPhi(-5555.0),posR(-4444.0);
  int subdet=0;
  unsigned int layer=0; 
  // bool is2D=false,isStereo=false;
  firsttree_->SetBranchAddress("DetId",    &id);
  firsttree_->SetBranchAddress("Nhits",    &nhits);
  firsttree_->SetBranchAddress("Noverlaps",&noverlaps);
  firsttree_->SetBranchAddress("SubDet",   &subdet);
  firsttree_->SetBranchAddress("Layer",    &layer);
  //  firsttree_->SetBranchAddress("is2D" ,    &is2D);
  // firsttree_->SetBranchAddress("isStereo", &isStereo);
  firsttree_->SetBranchAddress("posX",     &posX);
  firsttree_->SetBranchAddress("posY",     &posY);
  firsttree_->SetBranchAddress("posZ",     &posZ);
  firsttree_->SetBranchAddress("posR",     &posR);
  firsttree_->SetBranchAddress("posEta",   &posEta);
  firsttree_->SetBranchAddress("posPhi",   &posPhi);


  //create and book the output
 
 
  TFile *outfile=new TFile(outfilename_.c_str(),"RECREATE");
  out_=new TTree(treename_.c_str(),"AlignmentHitMapsTOTAL");
  unsigned int id_out=0;
  uint32_t nhits_out=0,noverlaps_out=0;
  float posX_out(-99999.0),posY_out(-77777.0),posZ_out(-88888.0);
  float posEta_out(-6666.0),posPhi_out(-5555.0),posR_out(-4444.0);
  int subdet_out=0;
  unsigned int layer_out=0; 
  bool is2D_out=false,isStereo_out=false;
  float prescfact_out=1.0;
  float prescfact_overlap_out=1.0;

  out_->Branch("DetId",    &id_out ,      "DetId/i");
  out_->Branch("Nhits",    &nhits_out ,   "Nhits/i");
  out_->Branch("Noverlaps",&noverlaps_out,"Noverlaps/i");
  out_->Branch("SubDet",   &subdet_out,   "SubDet/I");
  out_->Branch("Layer",    &layer_out,    "Layer/i");
  out_->Branch("is2D" ,    &is2D_out,     "is2D/B");
  out_->Branch("isStereo", &isStereo_out, "isStereo/B");
  out_->Branch("posX",     &posX_out,     "posX/F");
  out_->Branch("posY",     &posY_out,     "posY/F");
  out_->Branch("posZ",     &posZ_out,     "posZ/F");
  out_->Branch("posR",     &posR_out,     "posR/F");
  out_->Branch("posEta",   &posEta_out,   "posEta/F");
  out_->Branch("posPhi",   &posPhi_out,   "posPhi/F");
  out_->Branch("PrescaleFactor",&prescfact_out,"PrescaleFact/F");  
  out_->Branch("PrescaleFactorOverlap",&prescfact_overlap_out,"PrescaleFactOverlap/F"); 


  //look if you find this detid in the map
  DetHitMap::iterator mapiter;
  
   for(int mod=0;mod<firsttree_->GetEntries();mod++){
     //for(int mod=0;mod<100;++mod){
    //   nhits_out=0;
    // noverlaps_out=0;
 
    firsttree_->GetEntry(mod);
    nhits_out=hitmap_[id];
    noverlaps_out=overlapmap_[id];
    // if(mod<25)std::cout<<"Nhits 1st tree: "<<nhits<<"\tTotal nhits chain: "<<nhits_out<<std::endl;
    id_out=id;
    subdet_out=subdet;
    layer_out=layer;
    posX_out=posX;
    posY_out=posY;
    posZ_out=posZ;
    posR_out=posR;
    posEta_out=posEta;
    posPhi_out=posPhi;

    //calculate prescaling factor
    int subdetmax=-1;
    if(subdet_out==1)subdetmax=maxPXBhits_;
    else if(subdet_out==2)subdetmax=maxPXFhits_;
    else if(subdet_out==3)subdetmax=maxTIBhits_;
    else if(subdet_out==4)subdetmax=maxTIDhits_;
    else if(subdet_out==5)subdetmax=maxTOBhits_;
    else if(subdet_out==6)subdetmax=maxTEChits_;
    else subdetmax=-9999;

    if(maxhits_>-1){
      if(int(nhits_out)>maxhits_){
    prescfact_out=float(maxhits_)/float(nhits_out);
      }
      if(int(noverlaps_out)>maxhits_){
    prescfact_overlap_out=float(maxhits_)/float(noverlaps_out);
      }
    }
    else if(subdetmax>0){//calculate different prescaling factors for each subdet
      if(int(nhits_out)>subdetmax){
    prescfact_out=float(subdetmax/nhits_out);
      }
     if(int(noverlaps_out)>subdetmax){
    prescfact_overlap_out=float(subdetmax)/float(noverlaps_out);
      }
    }
    else{
      prescfact_out=1.0;
      prescfact_overlap_out=1.0;
    }
    out_->Fill();

  }//end loop on mod - first tree modules


  myclock.Stop();
  std::cout<<"Finished endJob after "<<myclock.RealTime()<<" s (real time) / "<<myclock.CpuTime()<<" s (cpu time)"<<std::endl;
  std::cout<<"Ending the tree merging."<<std::endl;
  out_->Write();
  std::cout<<"Deleting..."<<std::flush;
  delete firstfile;
  delete outfile;
 
  
}//end endJob

Member Data Documentation

TChain* TkAlCaSkimTreeMerger::ch_

private

Definition at line 37 of file TkAlCaSkimTreeMerger.cc.

Referenced by beginJob(), TkAlCaSkimTreeMerger(), and ~TkAlCaSkimTreeMerger().

std::string TkAlCaSkimTreeMerger::filelist_

private

Definition at line 38 of file TkAlCaSkimTreeMerger.cc.

Referenced by beginJob().

std::string TkAlCaSkimTreeMerger::firstfilename_

private

Definition at line 39 of file TkAlCaSkimTreeMerger.cc.

Referenced by beginJob(), and endJob().

TTree* TkAlCaSkimTreeMerger::firsttree_

private

Definition at line 36 of file TkAlCaSkimTreeMerger.cc.

Referenced by endJob(), and TkAlCaSkimTreeMerger().

DetHitMap TkAlCaSkimTreeMerger::hitmap_

private

Definition at line 45 of file TkAlCaSkimTreeMerger.cc.

Referenced by beginJob(), and endJob().

int TkAlCaSkimTreeMerger::maxhits_

private

Definition at line 47 of file TkAlCaSkimTreeMerger.cc.

Referenced by endJob(), and TkAlCaSkimTreeMerger().

edm::ParameterSet TkAlCaSkimTreeMerger::maxhitsSet_

private

Definition at line 48 of file TkAlCaSkimTreeMerger.cc.

Referenced by TkAlCaSkimTreeMerger().

int TkAlCaSkimTreeMerger::maxPXBhits_

private

Definition at line 49 of file TkAlCaSkimTreeMerger.cc.

Referenced by endJob(), and TkAlCaSkimTreeMerger().

int TkAlCaSkimTreeMerger::maxPXFhits_

private

Definition at line 49 of file TkAlCaSkimTreeMerger.cc.

Referenced by endJob(), and TkAlCaSkimTreeMerger().

int TkAlCaSkimTreeMerger::maxTEChits_

private

Definition at line 49 of file TkAlCaSkimTreeMerger.cc.

Referenced by endJob(), and TkAlCaSkimTreeMerger().

int TkAlCaSkimTreeMerger::maxTIBhits_

private

Definition at line 49 of file TkAlCaSkimTreeMerger.cc.

Referenced by endJob(), and TkAlCaSkimTreeMerger().

int TkAlCaSkimTreeMerger::maxTIDhits_

private

Definition at line 49 of file TkAlCaSkimTreeMerger.cc.

Referenced by endJob(), and TkAlCaSkimTreeMerger().

int TkAlCaSkimTreeMerger::maxTOBhits_

private

Definition at line 49 of file TkAlCaSkimTreeMerger.cc.

Referenced by endJob(), and TkAlCaSkimTreeMerger().

TStopwatch TkAlCaSkimTreeMerger::myclock

private

Definition at line 52 of file TkAlCaSkimTreeMerger.cc.

Referenced by beginJob(), and endJob().

TTree* TkAlCaSkimTreeMerger::out_

private

Definition at line 35 of file TkAlCaSkimTreeMerger.cc.

Referenced by endJob(), and TkAlCaSkimTreeMerger().

std::string TkAlCaSkimTreeMerger::outfilename_

private

Definition at line 41 of file TkAlCaSkimTreeMerger.cc.

Referenced by endJob().

DetHitMap TkAlCaSkimTreeMerger::overlapmap_

private

Definition at line 46 of file TkAlCaSkimTreeMerger.cc.

Referenced by beginJob(), and endJob().

std::string TkAlCaSkimTreeMerger::treename_

private

Definition at line 40 of file TkAlCaSkimTreeMerger.cc.

Referenced by beginJob(), and endJob().