readTrajectory.cc

// Include files
#include "TROOT.h"
#include "TFile.h"
#include "TSystem.h"
#include "TKey.h"
//
#include <iostream>
#include <map>
#include <string>
#include "include/SimStep.hh"
#include "include/SimTrajectory.hh"

int main(int argc, char** argv)
{
  // initialize ROOT
  TSystem ts;
  gSystem->Load("liblArTestClassesDict");
  if(argc < 2)
    std::cout << "Missing name of the file to read!" << std::endl;

  TFile fo(argv[1]);
  std::cout << " reading: " << argv[1] << std::endl;
  std::map<int, SimTrajectory*>* tmap;
  std::vector<SimStep*>* trajectory;
  // fo.GetListOfKeys()->Print();

  TIter next(fo.GetListOfKeys());
  TKey* key;
  // double tot_en;
  while((key = (TKey*) next()))
  {
    //        fo.GetObject(key->GetName(), hits);

    // tot_en = 0;
    //       std::std::cout << "Collection: " << key->GetName() << std::endl;
    //  std::std::cout << "Number of hits: " << hits->size() << std::endl;
    //  for (size_t i=0;i!=hits->size();i++)
    //  {
    //    (*hits)[i]->Print();
    //  }

    fo.GetObject(key->GetName(), tmap);

    std::cout << "Collection: " << key->GetName() << std::endl;
    key->Print();

    const char* Collectionname = key->GetName();
    std::string s(Collectionname);
    std::string s2("Simtrajectory");
    std::size_t found = s.find(s2);
    // s.compare(0, 13, s)
    if(found == 0)
    {
      std::cout << "Number of trajectories: " << tmap->size() << std::endl;
      for(auto itr = tmap->begin(); itr != tmap->end(); itr++)
      {
        std::cout << "ID:  " << itr->first << "  steps:  " << (itr->second)->GetTrajectory()->size()
                  << '\n';
        trajectory = (itr->second)->GetTrajectory();
        for(auto vitr = trajectory->begin(); vitr != trajectory->end(); vitr++)
        {
          SimStep* st = *vitr;
          std::cout << " Edep:  " << st->GetEdep() << "  Len:  " << st->GetLen()
                    << "  Time: " << st->GetT() << "  X:    " << st->GetX()
                    << "  Y:    " << st->GetY() << "  Z:    " << st->GetZ() << std::endl;
        }
      }
    }
  }
}