DigitizerTask.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 ALICEO2_MFT_DIGITIZERTASK_H
#define ALICEO2_MFT_DIGITIZERTASK_H
 
#include <cstdio>
#include <memory>
#include "FairTask.h"
#include "Rtypes.h"
 
#include "ITSMFTSimulation/DigiParams.h"
#include "ITSMFTSimulation/Digitizer.h"
#include "ITSMFTSimulation/Hit.h"
#include "SimulationDataFormat/MCTruthContainer.h"
#include "SimulationDataFormat/MCCompLabel.h"
 
class TBranch;
 
namespace o2
{
namespace mft
{
class DigitizerTask : public FairTask
{
  using Digitizer = o2::itsmft::Digitizer;
 
 public:
  DigitizerTask();
 
  ~DigitizerTask() override;
 
  InitStatus Init() override;
 
  void Exec(Option_t* option) override;
  void FinishTask() override;
 
  Digitizer& getDigitizer() { return mDigitizer; }
  o2::itsmft::DigiParams& getDigiParams() { return mDigitizer.getParams(); }
  const o2::itsmft::DigiParams& getDigiParams() const { return mDigitizer.getParams(); }
 
  void setContinuous(bool v) { mDigitizer.getParams().setContinuous(v); }
  bool isContinuous() const { return mDigitizer.getParams().isContinuous(); }
  void setFairTimeUnitInNS(double tinNS) { mFairTimeUnitInNS = tinNS < 1. ? 1. : tinNS; }
  double getFairTimeUnitInNS() const { return mFairTimeUnitInNS; }
  void setAlpideROFramLength(float l) { mDigitizer.getParams().setROFrameLength(l); }
  float getAlpideROFramLength() const { return mDigitizer.getParams().getROFrameLength(); }
 
  void setQEDInput(TBranch* qed, float timebin, UChar_t srcID);
 
 private:
  void processQEDBackground(double tMax);
 
  double mFairTimeUnitInNS = 1; 
  Int_t mSourceID = 0;          
  Int_t mEventID = 0;           
  Digitizer mDigitizer;         
 
  const std::vector<o2::itsmft::Hit>* mHitsArray = nullptr; 
 
  TBranch* mQEDBranch = nullptr;                               
  const std::vector<o2::itsmft::Hit>* mHitsArrayQED = nullptr; 
  float mQEDEntryTimeBinNS = 0.f;                              
  double mLastQEDTimeNS = 0.;                                  
  int mLastQEDEntry = -1;                                      
  UChar_t mQEDSourceID = 0;                                    
 
  std::vector<o2::itsmft::Digit> mDigitsArray;                     
  std::vector<o2::itsmft::Digit>* mDigitsArrayPtr = &mDigitsArray; 
 
  o2::dataformats::MCTruthContainer<o2::MCCompLabel> mMCTruthArray;                      
  o2::dataformats::MCTruthContainer<o2::MCCompLabel>* mMCTruthArrayPtr = &mMCTruthArray; 
 
  ClassDefOverride(DigitizerTask, 1);
};
} // namespace mft
} // namespace o2
 
#endif