TMCTTreeWriter.cxx

Go to the documentation of this file.

//------------------------------------------------
// The Geant4 Virtual Monte Carlo package
// Copyright (C) 2013 - 2018 Ivana Hrivnacova
// All rights reserved.
//
// For the licensing terms see geant4_vmc/LICENSE.
// Contact: root-vmc@cern.ch
//-------------------------------------------------

 
#include "TMCTTreeWriter.h"
 
#include <TError.h>
 
#include <iostream>
 
//
// ctors/dtors
//
 
//_____________________________________________________________________________
TMCTTreeWriter::TMCTTreeWriter(const char *storageName, TFile* file, Bool_t workerMode)
 : TMCVNtupleWriter(storageName, file, workerMode)
{
   if (!fFile) {
      Error("TMCTTreeWriter", "File not found, cannot create tree.");
      return;
   }
 
   if (fFile->IsWritable()) {
      // we are in writing mode 
      fTree = new TTree(storageName, fStorageName.c_str());
   }
   else {
      // we are in reading mode 
      fTree = (TTree *)fFile->Get(storageName);
      if (!fTree) {
        Error("TMCTTreeWriter", "Failed to get tree %s from the file.", storageName);
      }
   }
}
 
//
// public methods
//
 
//_____________________________________________________________________________
void TMCTTreeWriter::Register(const char *name, const char *className, void *objAddress)
{
   if (!fTree) {
      Error("Register", "Tree not found, cannot register data.");
      return;
   }
 
   // Check if we are in READ mode or WRITE mode
   TBranch* branch = fTree->GetBranch(name);
 
   if (branch) {
      // --- READING MODE ---
      // The branch exists in the file; attach memory address to read into
      // std::cout << "Going to set branch address: " << name  << " " << objAddress << std::endl;
      branch->SetAddress(objAddress);
   }
   else if (fFile && fFile->IsWritable()) {
      // --- WRITING MODE ---
      // The branch doesn't exist AND file is open for writing; create new branch
      // std::cout << "Going to create branch: " << name << " " << objAddress << std::endl;
      fFile->cd();
      fTree->Branch(name, className, objAddress, 32000, 99);
   }
   else {
      Error("Register", "Branch %s not found in tree for reading!", name);
   }
}
//_____________________________________________________________________________
void TMCTTreeWriter::Register(const char *name, const char *className, const void *objAddress)
{
 
   Register(name, className, const_cast<void *&>(objAddress));
}
 
//_____________________________________________________________________________
void TMCTTreeWriter::Fill()
{
   fFile->cd();
   fTree->Fill();
}
 
//_____________________________________________________________________________
void TMCTTreeWriter::WriteAll()
{
   fFile->cd();
   fFile->Write();
}
 
//_____________________________________________________________________________
void TMCTTreeWriter::ReadEvent(Int_t i)
{
 
   fTree->GetEntry(i);
}