writeNtuple.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/

// This class has been automatically generated on
// Wed Jul 20 23:04:22 2011 by ROOT version 5.28/00e
// from TTree NNinput/NNtraining
// found on file: root://castoratlas//castor/cern.ch/grid/atlas/atlasscratchdisk/user.lorenzi/user.lorenzi.T1.TRKVAL.NNtrain.clustersOnTrack.110701122007_sub022943841/user.lorenzi.000160.TRKVAL._00027.root
 
#ifndef writeNtuple_h
#define writeNtuple_h
 
#include "RIOs.h"
 
#include <TROOT.h>
#include <TSystem.h>
#include <TChain.h>
#include <TFile.h>
#include <TString.h>
#include <TChainElement.h>
#include <vector>
#include <stdio.h>
#include <stdlib.h>
 
#pragma link C++ class std::vector<int> +;
#pragma link C++ class std::vector<double> +;
#pragma link C++ class std::vector<float> +; Loader
 
 
 
 
class writeNtuple {
public :
   TTree          *fChain;   
   TChain         *tree2;   
   Int_t           fCurrent; 
   RIOs       *fOther;
   // Declaration of leaf types
   std::vector<int>     *NN_sizeX;
   std::vector<int>     *NN_sizeY;
   std::vector<std::vector<float> > *NN_matrixOfToT;
   std::vector<std::vector<float> > *NN_vectorOfPitchesY;
   std::vector<int>     *NN_ClusterPixLayer;
   std::vector<int>     *NN_ClusterPixBarrelEC;
   std::vector<float>   *NN_phiBS;
   std::vector<float>   *NN_thetaBS;
   std::vector<float>   *NN_etaModule;
   std::vector<bool>    *NN_useTrackInfo;
   std::vector<int>     *NN_columnWeightedPosition;
   std::vector<int>     *NN_rowWeightedPosition;
   std::vector<double>  *NN_localColumnWeightedPosition;
   std::vector<double>  *NN_localRowWeightedPosition;
   std::vector<std::vector<float> > *NN_positionX;
   std::vector<std::vector<float> > *NN_positionY;
   std::vector<std::vector<float> > *NN_position_idX;
   std::vector<std::vector<float> > *NN_position_idY;
   std::vector<std::vector<float> > *NN_theta;
   std::vector<std::vector<float> > *NN_phi;
 
   // List of branches
   TBranch        *b_NN_sizeX;   
   TBranch        *b_NN_sizeY;   
   TBranch        *b_NN_matrixOfToT;   
   TBranch        *b_NN_vectorOfPitchesY;   
   TBranch        *b_NN_ClusterPixLayer;   
   TBranch        *b_NN_ClusterPixBarrelEC;   
   TBranch        *b_NN_phiBS;   
   TBranch        *b_NN_thetaBS;   
   TBranch        *b_NN_etaModule;   
   TBranch        *b_NN_useTrackInfo;   
   TBranch        *b_NN_columnWeightedPosition;   
   TBranch        *b_NN_rowWeightedPosition;   
   TBranch        *b_NN_localColumnWeightedPosition;   
   TBranch        *b_NN_localRowWeightedPosition;   
   TBranch        *b_NN_positionX;   
   TBranch        *b_NN_positionY;   
   TBranch        *b_NN_position_idX;   
   TBranch        *b_NN_position_idY;   
   TBranch        *b_NN_theta;   
   TBranch        *b_NN_phi;   
 
        // --- ntuple items. Please document here so they can be automatically parsed for doxygen/wiki tables ---
   mutable std::vector<int>*   m_PixDetType; 
   mutable std::vector<int>*   m_PixHasGanged; 
   mutable std::vector<int>*   m_PixIsFake;   
   mutable std::vector<int>*   m_PixLVL1A;    
   mutable std::vector<int>*   m_PixToT;      
   mutable std::vector<float>* m_PixCharge;   
   mutable std::vector<float>* m_PixTLorPhi;  
   mutable std::vector<float>* m_PixTLorEta;  
   mutable std::vector<float>* m_PixBiasVolt; 
   mutable std::vector<float>* m_PixTemp;     
   mutable std::vector<float>* m_PixDepVolt;  
   mutable std::vector<float>* m_PixDigResPhi; 
   mutable std::vector<float>* m_PixDigResEta; 
   mutable std::vector<float>* m_PixResPhi; 
   mutable std::vector<float>* m_PixResEta; 
   mutable std::vector<float>* m_PixErrPhi;    
   mutable std::vector<float>* m_PixErrEta;    
   mutable std::vector<float>* m_PixEta_t;       
   mutable std::vector<float>* m_PixGloX;      
   mutable std::vector<float>* m_PixGloY;      
   mutable std::vector<float>* m_PixGloZ;      
   mutable std::vector<float>* m_PixLocX;      
   mutable std::vector<float>* m_PixLocY;      
   mutable std::vector<int>*   m_PixEtaModule_t; 
   mutable std::vector<int>*   m_PixPhiModule_t; 
   mutable std::vector<float>* m_PixFirstRow;  
   mutable std::vector<float>* m_PixFirstCol;  
   mutable std::vector<float>* m_PixDeltaPhi_t;  
   mutable std::vector<float>* m_PixDeltaEta_t;  
   mutable std::vector<int>*   m_PixDeltaRow_t;  
   mutable std::vector<int>*   m_PixDeltaCol_t;  
   mutable std::vector<float>* m_PixOmegaPhi_t;  
   mutable std::vector<float>* m_PixOmegaEta_t;  
   mutable std::vector<float>* m_PixTrkAngle;  
   mutable std::vector<float>* m_PixTrkThetaI; 
   mutable std::vector<float>* m_PixTrkEta;    
   mutable std::vector<float>* m_PixTrkPt;     
   mutable std::vector<float>* m_PixTrkQ;      
   mutable std::vector<float>* m_PixTrkClusGroupSize;  
   mutable std::vector<std::vector<float> >* m_PixChargeList;
   
   mutable int m_nPixelHits;
   mutable int m_nTRTHits;
   mutable int m_nSCTHits;
   mutable std::vector<float>*   m_PixPullPhi;
   mutable std::vector<float>*   m_PixPullEta;
 
   TString  m_pathWeightsWithTracks;
   TString  m_pathWeightsWithoutTracks;
 
   writeNtuple(TTree *tree=0);
   writeNtuple (const writeNtuple&) = delete;
   writeNtuple& operator= (const writeNtuple&) = delete;
   virtual ~writeNtuple();
   virtual Int_t    Cut(Long64_t entry);
   virtual Int_t    GetEntry(Long64_t entry);
   virtual Long64_t LoadTree(Long64_t entry);
           void     Init(TTree *tree);
   virtual void     Loop(int nParticlesTraining, bool useTrackEstimate, TString outFileName);
           Bool_t   Notify();
   virtual void     Show(Long64_t entry = -1);
   virtual void     FillClusterInfo(int clus);
   virtual void     InitializeVectors();
   virtual void     ResetVectors();
 
}; 
 
#endif
 
#ifdef writeNtuple_cxx
 
 
writeNtuple::writeNtuple(TTree *tree)
  : m_PixChargeList(nullptr),
    m_nPixelHits(0),
    m_nTRTHits(0),
    m_nSCTHits(0)
{
// if parameter tree is not specified (or zero), connect the file
// used to generate this class and read the Tree.
   if (tree == 0) {
      TFile *f = (TFile*)gROOT->GetListOfFiles()->FindObject("AnalysisSkeleton.aan.root");
      if (!f) {
         f = new TFile("AnalysisSkeleton.aan.root");
      }
      tree = (TTree*)f->Get("Validation/NNinput");
   }
   Init(tree);
 
 
   tree2 = new TChain("Validation/PixelRIOs");
   TIter next(static_cast<TChain*>(tree)->GetListOfFiles());
   TObject *obj;
   while( ( obj = static_cast<TChainElement*> (next()) ) ) {
     tree2->Add(obj->GetTitle());
   }
   fOther = new RIOs( tree2 );
}
   
 
 
 
 
writeNtuple::~writeNtuple()
{
   if (!fChain) return;
   delete fChain->GetCurrentFile();
}
 
Int_t writeNtuple::GetEntry(Long64_t entry)
{
// Read contents of entry.
   if (!fChain) return 0;
   return fChain->GetEntry(entry);
}
Long64_t writeNtuple::LoadTree(Long64_t entry)
{
// Set the environment to read one entry
   if (!fChain) return -5;
   Long64_t centry = fChain->LoadTree(entry);
   if (centry < 0) return centry;
   if (!fChain->InheritsFrom(TChain::Class()))  return centry;
   TChain *chain = static_cast<TChain*>(fChain);
   if (chain->GetTreeNumber() != fCurrent) {
      fCurrent = chain->GetTreeNumber();
      Notify();
   }
   return centry;
}
 
void writeNtuple::Init(TTree *tree)
{
   // The Init() function is called when the selector needs to initialize
   // a new tree or chain. Typically here the branch addresses and branch
   // pointers of the tree will be set.
   // It is normally not necessary to make changes to the generated
   // code, but the routine can be extended by the user if needed.
   // Init() will be called many times when running on PROOF
   // (once per file to be processed).
 
   // Set object pointer
   NN_sizeX = 0;
   NN_sizeY = 0;
   NN_matrixOfToT = 0;
   NN_vectorOfPitchesY = 0;
   NN_ClusterPixLayer = 0;
   NN_ClusterPixBarrelEC = 0;
   NN_phiBS = 0;
   NN_thetaBS = 0;
   NN_etaModule = 0;
   NN_useTrackInfo = 0;
   NN_columnWeightedPosition = 0;
   NN_rowWeightedPosition = 0;
   NN_localColumnWeightedPosition = 0;
   NN_localRowWeightedPosition = 0;
   NN_positionX = 0;
   NN_positionY = 0;
   NN_position_idX = 0;
   NN_position_idY = 0;
   NN_theta = 0;
   NN_phi = 0;
   // Set branch addresses and branch pointers
   if (!tree) return;
   fChain = tree;
   fCurrent = -1;
 
   fChain->SetMakeClass(1);
 
   fChain->SetBranchAddress("NN_sizeX", &NN_sizeX, &b_NN_sizeX);
   fChain->SetBranchAddress("NN_sizeY", &NN_sizeY, &b_NN_sizeY);
   fChain->SetBranchAddress("NN_matrixOfToT", &NN_matrixOfToT, &b_NN_matrixOfToT);
   fChain->SetBranchAddress("NN_vectorOfPitchesY", &NN_vectorOfPitchesY, &b_NN_vectorOfPitchesY);
   fChain->SetBranchAddress("NN_ClusterPixLayer", &NN_ClusterPixLayer, &b_NN_ClusterPixLayer);
   fChain->SetBranchAddress("NN_ClusterPixBarrelEC", &NN_ClusterPixBarrelEC, &b_NN_ClusterPixBarrelEC);
   fChain->SetBranchAddress("NN_phiBS", &NN_phiBS, &b_NN_phiBS);
   fChain->SetBranchAddress("NN_thetaBS", &NN_thetaBS, &b_NN_thetaBS);
   fChain->SetBranchAddress("NN_etaModule", &NN_etaModule, &b_NN_etaModule);
   fChain->SetBranchAddress("NN_useTrackInfo", &NN_useTrackInfo, &b_NN_useTrackInfo);
   fChain->SetBranchAddress("NN_columnWeightedPosition", &NN_columnWeightedPosition, &b_NN_columnWeightedPosition);
   fChain->SetBranchAddress("NN_rowWeightedPosition", &NN_rowWeightedPosition, &b_NN_rowWeightedPosition);
   fChain->SetBranchAddress("NN_localColumnWeightedPosition", &NN_localColumnWeightedPosition, &b_NN_localColumnWeightedPosition);
   fChain->SetBranchAddress("NN_localRowWeightedPosition", &NN_localRowWeightedPosition, &b_NN_localRowWeightedPosition);
   fChain->SetBranchAddress("NN_positionX", &NN_positionX, &b_NN_positionX);
   fChain->SetBranchAddress("NN_positionY", &NN_positionY, &b_NN_positionY);
   fChain->SetBranchAddress("NN_position_idX", &NN_position_idX, &b_NN_position_idX);
   fChain->SetBranchAddress("NN_position_idY", &NN_position_idY, &b_NN_position_idY);
   fChain->SetBranchAddress("NN_theta", &NN_theta, &b_NN_theta);
   fChain->SetBranchAddress("NN_phi", &NN_phi, &b_NN_phi);
     
 
   //InitializeVectors();
   
  m_PixDetType    = new std::vector<int>();
  m_PixHasGanged  = new std::vector<int>();
  m_PixIsFake     = new std::vector<int>();
  m_PixLVL1A      = new std::vector<int>();
  m_PixToT        = new std::vector<int>();
  m_PixCharge     = new std::vector<float>();
  m_PixTLorPhi    = new std::vector<float>();
  m_PixTLorEta    = new std::vector<float>();
  m_PixBiasVolt   = new std::vector<float>();
  m_PixTemp       = new std::vector<float>();
  m_PixDepVolt    = new std::vector<float>();
  m_PixLocX       = new std::vector<float>();
  m_PixLocY       = new std::vector<float>();
  m_PixEtaModule_t= new std::vector<int>();
  m_PixPhiModule_t= new std::vector<int>();
  m_PixGloX       = new std::vector<float>();
  m_PixGloY       = new std::vector<float>();
  m_PixGloZ       = new std::vector<float>();
  m_PixEta_t      = new std::vector<float>();
  m_PixFirstRow   = new std::vector<float>();
  m_PixFirstCol   = new std::vector<float>();
  m_PixDigResPhi  = new std::vector<float>();
  m_PixDigResEta  = new std::vector<float>();
  m_PixResPhi  = new std::vector<float>();
  m_PixResEta  = new std::vector<float>();
  m_PixErrPhi     = new std::vector<float>();
  m_PixErrEta     = new std::vector<float>();
  m_PixDeltaRow_t   = new std::vector<int>();
  m_PixDeltaCol_t   = new std::vector<int>();
  m_PixDeltaPhi_t   = new std::vector<float>();
  m_PixDeltaEta_t   = new std::vector<float>();
  m_PixOmegaPhi_t   = new std::vector<float>();
  m_PixOmegaEta_t   = new std::vector<float>();
  m_PixTrkAngle   = new std::vector<float>();
  m_PixTrkThetaI  = new std::vector<float>();
  m_PixTrkEta     = new std::vector<float>();
  m_PixTrkPt      = new std::vector<float>();
  m_PixTrkQ       = new std::vector<float>();
  m_PixTrkClusGroupSize = new std::vector<float>();
 
  m_PixPullPhi = new std::vector<float>();
  m_PixPullEta = new std::vector<float>();
 
   Notify();
}
 
Bool_t writeNtuple::Notify()
{
   // The Notify() function is called when a new file is opened. This
   // can be either for a new TTree in a TChain or when when a new TTree
   // is started when using PROOF. It is normally not necessary to make changes
   // to the generated code, but the routine can be extended by the
   // user if needed. The return value is currently not used.
 
   return kTRUE;
}
 
void writeNtuple::Show(Long64_t entry)
{
// Print contents of entry.
// If entry is not specified, print current entry
   if (!fChain) return;
   fChain->Show(entry);
}
Int_t writeNtuple::Cut(Long64_t entry)
{
// This function may be called from Loop.
// returns  1 if entry is accepted.
// returns -1 otherwise.
   return 1;
}
 
 
void writeNtuple::InitializeVectors(){
  return;
}
 
void writeNtuple::ResetVectors(){
  
  m_PixDetType->clear();
  m_PixHasGanged->clear();
  m_PixIsFake->clear();
  m_PixLVL1A->clear();
  m_PixToT->clear();
  m_PixCharge->clear();
  m_PixTLorPhi->clear();
  m_PixTLorEta->clear();
  m_PixBiasVolt->clear();
  m_PixTemp->clear();
  m_PixDepVolt->clear();
  m_PixDigResPhi->clear();
  m_PixDigResEta->clear();
  m_PixResPhi->clear();
  m_PixResEta->clear();
  m_PixErrPhi->clear();
  m_PixErrEta->clear();
  m_PixEta_t->clear();
  m_PixGloX->clear();
  m_PixGloY->clear();
  m_PixGloZ->clear();
  m_PixLocX->clear();
  m_PixLocY->clear();
  m_PixEtaModule_t->clear();
  m_PixPhiModule_t->clear();
  m_PixFirstRow->clear();
  m_PixFirstCol->clear();
  m_PixDeltaPhi_t->clear();
  m_PixDeltaEta_t->clear();
  m_PixDeltaRow_t->clear();
  m_PixDeltaCol_t->clear();
  m_PixOmegaPhi_t->clear();
  m_PixOmegaEta_t->clear();
 
 
  m_PixTrkAngle->clear();
  m_PixTrkThetaI->clear();
  m_PixTrkEta->clear();
  m_PixTrkPt->clear();
  m_PixTrkQ->clear();
  m_PixTrkClusGroupSize->clear();
 
  m_PixPullPhi->clear();
  m_PixPullEta->clear();
 
   m_nPixelHits=0;
   m_nSCTHits=0;
   m_nTRTHits=0;
 
  return;
 
}
 
 
void 
writeNtuple::FillClusterInfo(int clus){
  m_PixEta_t->push_back( fOther->PixClusEta->at(clus)   );
  m_PixGloX->push_back(  fOther->PixClusGloX->at(clus)  );
  m_PixGloY->push_back(  fOther->PixClusGloY->at(clus)  );
  m_PixGloZ->push_back(  fOther->PixClusGloZ->at(clus)  );
  m_PixEtaModule_t->push_back( fOther->PixEtaModule->at(clus));
  m_PixPhiModule_t->push_back( fOther->PixPhiModule->at(clus));
  m_PixDeltaRow_t->push_back(fOther->PixDeltaRow->at(clus));
  m_PixDeltaCol_t->push_back(fOther->PixDeltaCol->at(clus));
  m_PixOmegaPhi_t->push_back(fOther->PixOmegaPhi->at(clus));
  m_PixOmegaEta_t->push_back(fOther->PixOmegaEta->at(clus));
 
  float PixTLorPhi    = 0.;
  float PixTLorEta    = -1000.;
  float PixBiasVolt    = -1000.;
  float PixTemp    = -1000.;
  float PixDepVolt    = -1000.;
  
  m_PixTLorPhi->push_back(PixTLorPhi);
  m_PixTLorEta->push_back(PixTLorEta);
  m_PixBiasVolt->push_back(PixBiasVolt);
  m_PixTemp->push_back(PixTemp);
  m_PixDepVolt->push_back(PixDepVolt);
  
  m_PixHasGanged->push_back(fOther->PixClusGanged->at(clus));
  m_PixIsFake->push_back(fOther->PixClusFake->at(clus));
  m_PixLVL1A->push_back(fOther->PixClusLvl1->at(clus));
  m_PixToT->push_back(fOther->PixClusToT->at(clus));
  m_PixCharge->push_back(fOther->PixClusCharge->at(clus));
 
  int   PixDetType    = 0;

 
  int ECBarrel =  fOther->PixECBarrel->at(clus);
  int Layer    =  fOther->PixClusLayer->at(clus);
   
//PixDetType = 100*ECBarrel/2+Layer ;   
 
  if( ECBarrel == 0 ){
    PixDetType = Layer;
  } 
  
  if( ECBarrel == 2 ){
    PixDetType = 100+Layer;
  }
  
  if( ECBarrel == -2 ){
    PixDetType = -100-Layer;
  }
  
  
  m_PixDetType->push_back(PixDetType);
  
  
  return;
 
}
 
#endif // #ifdef writeNtuple_cxx