AODMcProducerWorkflowSpec.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_AODMCPRODUCER_WORKFLOW_SPEC
#define O2_AODMCPRODUCER_WORKFLOW_SPEC
 
#include "AODProducerHelpers.h"
#include "AODMcProducerHelpers.h"
#include "CommonDataFormat/InteractionRecord.h"
#include "Framework/AnalysisDataModel.h"
#include "Framework/AnalysisHelpers.h"
#include "Framework/Task.h"
#include "Framework/DataProcessorSpec.h"
#include "Steer/MCKinematicsReader.h"
#include <TStopwatch.h>
 
#include <string>
#include <vector>
 
using namespace o2::framework;
 
namespace o2::aodmcproducer
{
 
class AODMcProducerWorkflowDPL : public Task
{
 public:
  AODMcProducerWorkflowDPL() = default;
  ~AODMcProducerWorkflowDPL() override = default;
  void init(InitContext& ic) final;
  void run(ProcessingContext& pc) final;
  void endOfStream(EndOfStreamContext& ec) final;
 
  using MCEventHeader = dataformats::MCEventHeader;
  using McCollisions = aod::McCollisions;
  using McParticles = aod::StoredMcParticles;
  using Origins = aod::Origins;
  using XSections = aod::HepMCXSections;
  using PdfInfos = aod::HepMCPdfInfos;
  using HeavyIons = aod::HepMCHeavyIons;
  using MCKinematicsReader = steer::MCKinematicsReader;
 private:
  using CollisionCursor = aodmchelpers::CollisionCursor;
  using XSectionCursor = aodmchelpers::XSectionCursor;
  using PdfInfoCursor = aodmchelpers::PdfInfoCursor;
  using HeavyIonCursor = aodmchelpers::HeavyIonCursor;
  using HepMCUpdate = aodmchelpers::HepMCUpdate;
  void updateHeader(CollisionCursor& collisionCursor,
                    XSectionCursor& xSectionCursor,
                    PdfInfoCursor& pdfInfoCursor,
                    HeavyIonCursor& heavyIonCursor,
                    const MCEventHeader& header,
                    int collisionID, // Index
                    int bcID,
                    float time,
                    short generatorID,
                    int sourceID);
  int64_t mTFNumber{1};
  int mFilterMC{0};
  bool mEnableEmbed{false};
  TString mLPMProdTag{""};
  TString mAnchorPass{""};
  TString mAnchorProd{""};
  TString mRecoPass{""};
  TStopwatch mTimer;
  HepMCUpdate mXSectionUpdate = HepMCUpdate::anyKey;
  HepMCUpdate mPdfInfoUpdate = HepMCUpdate::anyKey;
  HepMCUpdate mHeavyIonUpdate = HepMCUpdate::anyKey;
 
  std::string mSimPrefix;
 
  // MC production metadata holder
  bool mIsMDSent{false};
 
  // truncation is enabled by default
  uint32_t mCollisionPosition = 0xFFFFFFF0; // 19 bits mantissa
  uint32_t mMcParticleW = 0xFFFFFFF0;       // 19 bits
  uint32_t mMcParticlePos = 0xFFFFFFF0;     // 19 bits
  uint32_t mMcParticleMom = 0xFFFFFFF0;     // 19 bits
};
 
framework::DataProcessorSpec getAODMcProducerWorkflowSpec();
 
} // namespace o2::aodmcproducer
 
#endif /* O2_AODMCPRODUCER_WORKFLOW_SPEC */