O2MCApplication.h

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.
 
#ifndef O2_MC_APPLICATION_H
#define O2_MC_APPLICATION_H
 
#include "FairMCApplication.h"
#include "Steer/O2MCApplicationBase.h"
#include "Rtypes.h" // for Int_t, Bool_t, Double_t, etc
#include <iostream>
#include <TParticle.h>
#include <vector>
#include <DetectorsBase/Stack.h>
#include <SimulationDataFormat/PrimaryChunk.h>
#include <FairRootManager.h>
#include <FairDetector.h>
 
#include <fairmq/FwdDecls.h>
 
namespace o2
{
namespace steer
{
 
// O2 specific changes/overrides to FairMCApplication
// (like for device based processing in which we
//  forward the data instead of using FairRootManager::Fill())
class O2MCApplication : public O2MCApplicationBase
{
 public:
  using O2MCApplicationBase::O2MCApplicationBase;
  ~O2MCApplication() override = default;
 
  // triggers data sending/io
  void SendData();
 
  void initLate();
 
  void FinishEvent() override
  {
    // update the stack
    fStack->FillTrackArray();
    fStack->UpdateTrackIndex(fActiveDetectors);
 
    finishEventCommon();
 
    // detectors finalize their hits (e.g. sorting, summing duplicates) before these are sent
    for (auto det : listActiveDetectors) {
      det->FinishEvent();
    }
 
    // This special finish event version does not fill the output tree of FairRootManager
    // but forwards the data to the HitMerger
    SendData();
 
    // call end of event on active detectors
    for (auto det : listActiveDetectors) {
      det->EndOfEvent();
    }
    fStack->Reset();
  }
 
  void FinishRun();
 
  void attachSubEventInfo(fair::mq::Parts&, o2::data::SubEventInfo const& info) const;
 
  void GeneratePrimaries() override
  {
    // ordinarily we would call the event generator ...
    LOG(debug) << "O2MCApplication: Init primaries from external buffer " << mPrimaries.size();
    GetStack()->Reset();
    // but here we init the stack from
    // a vector of particles that someone sets externally
    static_cast<o2::data::Stack*>(GetStack())->initFromPrimaries(mPrimaries);
  }
 
  void setPrimaries(std::vector<TParticle> const& p)
  {
    mPrimaries = p;
  }
 
  void setSimDataChannel(fair::mq::Channel* channel) { mSimDataChannel = channel; }
  void setSubEventInfo(o2::data::SubEventInfo* i);
 
  std::vector<TParticle> mPrimaries; 
 
  fair::mq::Channel* mSimDataChannel;                  
  o2::data::SubEventInfo* mSubEventInfo = nullptr;     
  std::vector<o2::base::Detector*> mActiveO2Detectors; 
 
  ClassDefOverride(O2MCApplication, 1); //Interface to MonteCarlo application
};
 
} // end namespace steer
} // end namespace o2
 
#endif