NicaGeneratorWriteUnigen.cxx

Go to the documentation of this file.

/*
 * NicaGeneratorWriteUnigen.cxx
 *
 *  Created on: 13 sie 2015
 *      Author: Daniel Wielanek
 *              E-mail: daniel.wielanek@gmail.com
 *              Warsaw University of Technology, Faculty of Physics
 */
#include "NicaGeneratorWriteUnigen.h"
#include "NicaRunSim.h"
 
NicaGeneratorWriteUnigen::NicaGeneratorWriteUnigen()
  : NicaGeneratorWrite(4, 3, 1, 0, 9, 9, 0, 0) {
  fEvent = new UEvent();
  // parameter 0 is reserverd for random stuff !
  //register event parameters int
  Register(4, "Nes", "Step_nr", "Npa", "Nr");
  //register event parameters double
  Register(3, "B", "Phi", "Step_t");
  //register track int parameters
  Register(9,
           "index",
           "pdg",
           "status",
           "parent",
           "parent_decay",
           "mate",
           "decay",
           "first_child",
           "last_child");
  //register track double parameters
  Register(9, "px", "py", "pz", "e", "frx", "fry", "frz", "frt", "weight");
}
 
InitStatus NicaGeneratorWriteUnigen::Init() {
  InitStatus stat     = NicaGeneratorWrite::Init();
  NicaRunSim* manager = NicaRunSim::Instance();
  manager->GetTree()->Branch("event", &fEvent);
  LOG(info) << "Data will be set to Unigen format";
  return stat;
}
 
void NicaGeneratorWriteUnigen::ClearEvent() { fEvent->Clear(); }
 
void NicaGeneratorWriteUnigen::AddEvent() {
  fEvent->SetNes(fEventParameters->GetInt(1));
  fEvent->SetStepNr(fEventParameters->GetInt(2));
  fEvent->SetEventNr(fEventParameters->GetInt(4));
  fEvent->SetB(fEventParameters->GetDouble(1));
  fEvent->SetPhi(fEventParameters->GetDouble(2));
  fEvent->SetStepT(fEventParameters->GetDouble(3));
}
 
void NicaGeneratorWriteUnigen::AddParticle() {
  Int_t child[2] = {fTrackParameters->GetInt(8), fTrackParameters->GetInt(9)};
  fEvent->AddParticle(fTrackParameters->GetInt(1),  //index
                      fTrackParameters->GetInt(2),  //pdg
                      fTrackParameters->GetInt(3),  
                      fTrackParameters->GetInt(4),  //parent
                      fTrackParameters->GetInt(5),  //parent_decay
                      fTrackParameters->GetInt(6),  //mate
                      fTrackParameters->GetInt(7),  //decay
                      child,
                      fTrackParameters->GetDouble(1),
                      fTrackParameters->GetDouble(2),
                      fTrackParameters->GetDouble(3),
                      fTrackParameters->GetDouble(4),
                      fTrackParameters->GetDouble(5),
                      fTrackParameters->GetDouble(6),
                      fTrackParameters->GetDouble(7),
                      fTrackParameters->GetDouble(8),
                      fTrackParameters->GetDouble(9));
}
 
NicaGeneratorWriteUnigen::~NicaGeneratorWriteUnigen() {
  // TODO Auto-generated destructor stub
}