read.cc

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file 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";
        }
      }

    }

  }


}