NtpMCRecord.cxx

Go to the documentation of this file.

// 
// NtpMCRecord
//
// NtpMCRecord is an ntuple record serving the monte carlo data
//

#include <iostream>
using namespace std;

#include "TClonesArray.h"
#include "Record/RecArrayAllocator.h"
#include "MCNtuple/NtpMCRecord.h"
#include "MCNtuple/NtpMCTruth.h"
#include "MCNtuple/NtpMCStdHep.h"
#include "PhotonTransport/PhotonEventResult.h"
#include "DetSim/SimEventResult.h"
#include "MessageService/MsgService.h"

ClassImp(NtpMCRecord)

CVSID("$Id: NtpMCRecord.cxx,v 1.15 2007/03/27 18:53:01 schubert Exp $");

NtpMCRecord::NtpMCRecord() : RecRecordImp<RecCandHeader>(),mc(0),stdhep(0),
                             flsdigit(0),digihit(0) {
  // Purpose: Default constructor
  MSG("NtpMC",Msg::kVerbose) << "NtpMCRecord def ctor @ " << this << endl;
  this -> Init();  
}

NtpMCRecord::NtpMCRecord(const RecCandHeader& hdr) : 
  RecRecordImp<RecCandHeader>(hdr),mc(0),stdhep(0),flsdigit(0),digihit(0) {
  // Purpose: Normal constructor

  MSG("NtpMC",Msg::kVerbose) << "NtpMCRecord normal ctor @ " << this << endl;
  this -> Init(); 
}

NtpMCRecord::~NtpMCRecord() {
  // Purpose: Destructor

  MSG("NtpMC",Msg::kVerbose) << "NtpMCRecord dtor @ " << this << endl;
  
  // Release arrays back to TClonesArray pool for reuse
  // Allocated memory of stored objects is retrieved via object Clear call
  RecArrayAllocator& allocator = RecArrayAllocator::Instance();
  if ( mc ) { allocator.ReleaseArray(mc); mc = 0; }
  if ( stdhep ) { allocator.ReleaseArray(stdhep); stdhep = 0; }
  if ( flsdigit ) { allocator.ReleaseArray(flsdigit); flsdigit = 0; }
  if ( digihit ) { allocator.ReleaseArray(digihit); digihit = 0; }
  
}

void NtpMCRecord::Clear(Option_t* /* option */) {
  // Purpose: Clear memory allocated to arrays so that record can
  // be reused.  

  detsim.Clear();
  if ( mc ) { mc -> Clear("C"); }
  if ( stdhep ) { stdhep -> Clear("C"); }
  if ( flsdigit ) { flsdigit -> Clear("C"); }
  if ( digihit ) { digihit -> Clear("C"); }
}

void NtpMCRecord::Init() {
  // 
  // Purpose: Initialize ntuple TClonesArrays
  //

  // Set variable in record base class to indicate that it is possible
  // to recover dynamic memory using Clear() method for this record type
  SetClearable(true);

  RecArrayAllocator& allocator = RecArrayAllocator::Instance();
  if ( !mc ) mc = allocator.GetArray("NtpMCTruth");
  if ( !stdhep ) stdhep = allocator.GetArray("NtpMCStdHep");
  if ( !flsdigit ) flsdigit = allocator.GetArray("NtpMCFLSDigit");
  if ( !digihit ) digihit = allocator.GetArray("NtpMCDigiScintHit");
}

std::ostream& NtpMCRecord::Print(std::ostream& os) const {
  //
  // Purpose: Print status of ntuple record on ostream
  //

  os << "NtpMCRecord::Print" << endl;
  RecRecordImp<RecCandHeader>::Print(os);
  mchdr.Print(os);
  if ( mc ) {
    for ( int imc = 0; imc < mc->GetLast()+1; imc++ ) {
      (dynamic_cast<NtpMCTruth*>(mc -> At(imc))) -> Print(os);
    }
  }
  if ( stdhep ) {
    Int_t nstdhep = stdhep->GetLast()+1;
    std::string entrystr = " entries:";
    if ( nstdhep == 1 ) entrystr = " entry:";

    os << "\nPrint stdhep array of NtpMCStdHep w/"
       << nstdhep << entrystr.c_str() << endl;
    for ( int istd = 0; istd < nstdhep; istd++ ) {
      const NtpMCStdHep* ntpstdhep
        = dynamic_cast<const NtpMCStdHep*>(stdhep->At(istd));
      if ( ntpstdhep->parent[0] == -1 ) 
        ntpstdhep->Print(std::cout,"","",stdhep);
    }
  }
  else {
    os << "Null stdhep array." << endl;
  }
  
  return os;

}

void NtpMCRecord::Print(const Option_t* /* option */) const {
  //
  // Purpose: Print record in form supported by TObject::Print
  //

  Print(std::cout);
  return;

}

---