read.cc

#include <Track.h>
#include <GFRaveVertex.h>
 
#include <TDatabasePDG.h>
#include <TEveManager.h>
#include <TGeoManager.h>
#include <TRandom.h>
#include <TVector3.h>
#include <vector>
 
#include <TROOT.h>
#include <TClonesArray.h>
#include <TFile.h>
#include <TTree.h>
 
#include <iostream>
 
 
int main() {
 
  genfit::Track tr; // pull in genfit libraries
 
  //genfit::Track* aTrackPtr(nullptr);
  genfit::GFRaveVertex* aVertexPtr(nullptr);
 
  TFile* trackFile = TFile::Open("tracks.root", "READ");
  if (!trackFile) {
    std::cerr << "Couldn't open 'tracks.root'." << std::endl;
    return -1;
  }
  TTree* tree = (TTree*)trackFile->Get("tree");
  if (!tree) {
    std::cerr << "Couldn't find tree 'tree' in file 'tracks.root'." << std::endl;
    return -1;
  }
  TClonesArray* trackArray = new TClonesArray("genfit::Track");
  tree->SetBranchAddress("trackBranch", &trackArray);
 
  tree->Print();
 
  TClonesArray* vertexArray = new TClonesArray("genfit::GFRaveVertex");
  tree->SetBranchAddress("vertexBranch", &vertexArray);
 
 
  for (Long_t i = 0; i < tree->GetEntries(); ++i) {
    tree->GetEntry(i);
 
    std::cout << "trackArray nr of entries: " << trackArray->GetEntries() << "\n";
 
    for (Long_t j = 0; j < trackArray->GetEntriesFast(); ++j) {
      std::cout << "track uniqueID: " << static_cast<genfit::Track*>(trackArray->At(j))->GetUniqueID() <<
          " (" << static_cast<genfit::Track*>(trackArray->At(j))->GetUniqueID() - 16777216 << ")\n";
    }
 
    for (Long_t j = 0; j < vertexArray->GetEntriesFast(); ++j) {
 
      aVertexPtr = (genfit::GFRaveVertex*)(vertexArray->At(j));
      //aVertexPtr->Print();
 
      for (unsigned int k=0; k<aVertexPtr->getNTracks(); ++k) {
        std::cout << "track parameters uniqueID: " << aVertexPtr->getParameters(k)->GetUniqueID() << "\n";
      }
 
      // when the track branch from the tracks.root file is loaded, the TRefs to the tracks
      // in the GFRaveTrackParameters are again pointing to them.
      for (unsigned int k = 0; k<aVertexPtr->getNTracks(); ++k) {
        if (aVertexPtr->getParameters(k)->hasTrack()) {
          std::cout << "track parameters have track \n";
        }
        else {
          std::cout << "track parameters have NO track <--------------------------------- \n";
        }
      }
 
    }
 
  }
 
 
}