DigitizerTRU.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_EMCAL_TRIGGERDIGITIZER_H
#define ALICEO2_EMCAL_TRIGGERDIGITIZER_H
 
#include <memory>
#include <unordered_map>
#include <vector>
#include <list>
 
#include "Rtypes.h"  // for DigitizerTRU::Class, Double_t, ClassDef, etc
#include "TObject.h" // for TObject
#include "TRandom3.h"
 
#include "DataFormatsEMCAL/Digit.h"
#include "EMCALBase/Hit.h"
#include "EMCALBase/TriggerMappingV2.h"
#include "EMCALSimulation/SimParam.h"
#include "EMCALSimulation/LabeledDigit.h"
#include "EMCALSimulation/DigitsWriteoutBufferTRU.h"
#include "EMCALSimulation/LZEROElectronics.h"
#include "SimulationDataFormat/MCTruthContainer.h"
#include "DataFormatsEMCAL/TriggerRecord.h"
#include "CommonUtils/TreeStreamRedirector.h"
#include "EMCALCalib/FeeDCS.h"
 
namespace o2
{
namespace utils
{
class TreeStreamRedirector;
}
namespace emcal
{
class FeeDCS;
 
class DigitizerTRU
{
 public:
  DigitizerTRU() = default;
  ~DigitizerTRU() = default;
  DigitizerTRU(const DigitizerTRU&) = delete;
  DigitizerTRU& operator=(const DigitizerTRU&) = delete;
 
  void init();
  void clear();
 
  void setPatches();
 
  void flush() { mDigits.flush(); }
 
  void finish();
 
  void process(const gsl::span<const Digit> summableDigits);
 
  std::vector<std::tuple<int, Digit>> makeAnaloguesFastorSums(const gsl::span<const Digit> sdigits);
 
  void setEventTime(o2::InteractionTimeRecord record);
 
  void setGeometry(o2::emcal::Geometry* gm) { mGeometry = gm; }
 
  void setFEE(o2::emcal::FeeDCS* fees) { mFeeDCS = fees; }
 
  void setMaskedFastOrsInLZERO();
  void printMaskedFastOrsInLZERO();
 
  void setWindowStartTime(int time) { mTimeWindowStart = time; }
  void setDebugStreaming(bool doStreaming) { mEnableDebugStreaming = doStreaming; }
 
  void fillOutputContainer(std::vector<Digit>& digits, o2::dataformats::MCTruthContainer<o2::emcal::MCLabel>& labels);
 
  bool doSmearEnergy() const { return mSmearEnergy; }
  double smearEnergy(double energy);
  double smearTime(double time, double energy);
  bool doSimulateTimeResponse() const { return mSimulateTimeResponse; }
 
  void sampleSDigit(const Digit& sdigit);
 
  void endDebugStream()
  {
    mDebugStream->Close();
    // mDebugStreamPatch->Close();
  }
 
  bool isDebugMode() { return mEnableDebugStreaming; }
 
  const std::vector<EMCALTriggerInputs>& getTriggerInputs() { return LZERO.getTriggerInputs(); }
 
  std::vector<TRUElectronics> getPatchesVector() { return patchesFromAllTRUs; }
 
  static double rawResponseFunction(double* x, double* par);
 
  int GetTRUIndexFromSTUIndex(Int_t id, Int_t detector);
  int GetChannelForMaskRun2(int mask, int bitnumber, bool onethirdsm);
  std::vector<int> GetAbsFastORIndexFromMask();
 
 private:
  short mEventTimeOffset = 0;        
  bool mSmearEnergy = true;          
  bool mSimulateTimeResponse = true; 
  // const SimParam* mSimParam = nullptr; ///< SimParam object
 
  std::vector<Digit> mTempDigitVector; 
  // std::unordered_map<Int_t, std::list<LabeledDigit>> mDigits; ///< used to sort digits and labels by tower
  o2::emcal::DigitsWriteoutBufferTRU mDigits;     
  o2::emcal::LZEROElectronics LZERO;              
  std::vector<TRUElectronics> patchesFromAllTRUs; 
 
  // TRandom3* mRandomGenerator = nullptr; ///< random number generator
  std::array<double, constants::EMCAL_MAXTIMEBINS>
    mAmplitudeInTimeBins; 
 
  // TriggerMappingV2* mTriggerMap = nullptr; ///< Trigger map for tower to fastor ID
  Geometry* mGeometry = nullptr; 
 
  int mTimeWindowStart = 7;      
  int mDelay = 7;                
  bool mWasTriggerFound = false; 
  int mPreviousTriggerSize = 0;  
 
  std::unique_ptr<o2::utils::TreeStreamRedirector> mDebugStream = nullptr;
  // std::unique_ptr<o2::utils::TreeStreamRedirector> mDebugStreamPatch = nullptr;
  bool mEnableDebugStreaming = false;
  o2::emcal::FeeDCS* mFeeDCS; 
 
  ClassDefNV(DigitizerTRU, 1);
};
} // namespace emcal
} // namespace o2
 
#endif /* ALICEO2_EMCAL_TRIGGERDIGITIZER_H */