Generator.cxx

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
 
#include "Generators/Generator.h"
#include "Generators/Trigger.h"
#include "Generators/PrimaryGenerator.h"
#include "SimulationDataFormat/MCEventHeader.h"
#include "SimulationDataFormat/ParticleStatus.h"
#include "SimulationDataFormat/MCGenProperties.h"
#include "FairPrimaryGenerator.h"
#include <fairlogger/Logger.h>
#include <cmath>
#include "TClonesArray.h"
#include "TParticle.h"
 
namespace o2
{
namespace eventgen
{
 
std::atomic<int> Generator::InstanceCounter{0};
unsigned int Generator::gTotalNEvents = 0;
/*****************************************************************/
/*****************************************************************/
 
Generator::Generator() : FairGenerator("ALICEo2", "ALICEo2 Generator"),
                         mBoost(0.)
{
  mThisInstanceID = Generator::InstanceCounter;
  Generator::InstanceCounter++;
}
 
/*****************************************************************/
 
Generator::Generator(const Char_t* name, const Char_t* title) : FairGenerator(name, title),
                                                                mBoost(0.)
{
  mThisInstanceID = Generator::InstanceCounter;
  Generator::InstanceCounter++;
}
 
/*****************************************************************/
 
Bool_t
  Generator::Init()
{
  return kTRUE;
}
 
/*****************************************************************/
 
Bool_t
  Generator::ReadEvent(FairPrimaryGenerator* primGen)
{
  while (true) {
    mReadEventCounter++;
 
    mParticles.clear();
 
    mSubGeneratorId = -1;
 
    if (!generateEvent()) {
      LOG(error) << "ReadEvent failed in generateEvent";
      return kFALSE;
    }
 
    if (!importParticles()) {
      LOG(error) << "ReadEvent failed in importParticles";
      return kFALSE;
    }
 
    if (mSubGeneratorsIdToDesc.empty() && mSubGeneratorId > -1) {
      LOG(fatal) << "ReadEvent failed because no SubGenerator description given";
    }
 
    if (!mSubGeneratorsIdToDesc.empty() && mSubGeneratorId < 0) {
      LOG(fatal) << "ReadEvent failed because SubGenerator description given but sub-generator not set";
    }
 
    if (triggerEvent()) {
      mTriggerOkHook(mParticles, mReadEventCounter);
      break;
    } else {
      mTriggerFalseHook(mParticles, mReadEventCounter);
    }
  }
 
  if (!addTracks(primGen)) {
    LOG(error) << "ReadEvent failed in addTracks";
    return kFALSE;
  }
 
  auto header = primGen->GetEvent();
  auto o2header = dynamic_cast<o2::dataformats::MCEventHeader*>(header);
  if (!header) {
    LOG(fatal) << "MC event header is not a 'o2::dataformats::MCEventHeader' object";
    return kFALSE;
  }
  updateHeader(o2header);
  updateSubGeneratorInformation(o2header);
 
  return kTRUE;
}
 
/*****************************************************************/
 
Bool_t
  Generator::addTracks(FairPrimaryGenerator* primGen)
{
  auto o2primGen = dynamic_cast<PrimaryGenerator*>(primGen);
  if (!o2primGen) {
    LOG(fatal) << "PrimaryGenerator is not a o2::eventgen::PrimaryGenerator";
    return kFALSE;
  }
 
  for (const auto& particle : mParticles) {
    o2primGen->AddTrack(particle.GetPdgCode(),
                        particle.Px() * mMomentumUnit,
                        particle.Py() * mMomentumUnit,
                        particle.Pz() * mMomentumUnit,
                        particle.Vx() * mPositionUnit,
                        particle.Vy() * mPositionUnit,
                        particle.Vz() * mPositionUnit,
                        particle.GetMother(0),
                        particle.GetMother(1),
                        particle.GetDaughter(0),
                        particle.GetDaughter(1),
                        particle.TestBit(ParticleStatus::kToBeDone),
                        particle.Energy() * mEnergyUnit,
                        particle.T() * mTimeUnit,
                        particle.GetWeight(),
                        (TMCProcess)particle.GetUniqueID(),
                        particle.GetStatusCode()); // generator status information passed as status code field
  }
 
  return kTRUE;
}
 
/*****************************************************************/
 
Bool_t
  Generator::boostEvent()
{
  return kTRUE;
}
 
/*****************************************************************/
 
Bool_t
  Generator::triggerEvent()
{
  if (mTriggers.size() == 0 && mDeepTriggers.size() == 0) {
    return kTRUE;
  }
 
  Bool_t triggered;
  if (mTriggerMode == kTriggerOFF) {
    return kTRUE;
  } else if (mTriggerMode == kTriggerOR) {
    triggered = kFALSE;
  } else if (mTriggerMode == kTriggerAND) {
    triggered = kTRUE;
  } else {
    return kTRUE;
  }
 
  for (const auto& trigger : mTriggers) {
    auto retval = trigger(mParticles);
    if (mTriggerMode == kTriggerOR) {
      triggered |= retval;
    }
    if (mTriggerMode == kTriggerAND) {
      triggered &= retval;
    }
  }
 
  for (const auto& trigger : mDeepTriggers) {
    auto retval = trigger(mInterface, mInterfaceName);
    if (mTriggerMode == kTriggerOR) {
      triggered |= retval;
    }
    if (mTriggerMode == kTriggerAND) {
      triggered &= retval;
    }
  }
 
  return triggered;
}
 
/*****************************************************************/
 
void Generator::addSubGenerator(int subGeneratorId, std::string const& subGeneratorDescription)
{
  if (subGeneratorId < 0) {
    LOG(fatal) << "Sub-generator IDs must be >= 0, instead, passed value is " << subGeneratorId;
  }
  mSubGeneratorsIdToDesc.insert({subGeneratorId, subGeneratorDescription});
}
 
/*****************************************************************/
 
void Generator::updateSubGeneratorInformation(o2::dataformats::MCEventHeader* header) const
{
  if (mSubGeneratorId < 0) {
    return;
  }
  header->putInfo<int>(o2::mcgenid::GeneratorProperty::SUBGENERATORID, mSubGeneratorId);
  header->putInfo<std::unordered_map<int, std::string>>(o2::mcgenid::GeneratorProperty::SUBGENERATORDESCRIPTIONMAP, mSubGeneratorsIdToDesc);
}
 
/*****************************************************************/
/*****************************************************************/
 
} /* namespace eventgen */
} /* namespace o2 */
 
ClassImp(o2::eventgen::Generator);