DigitsVectorStream.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_DIGITSVECTORSTREAM_H_
#define ALICEO2_EMCAL_DIGITSVECTORSTREAM_H_
 
#include <memory>
#include <unordered_map>
#include <vector>
#include <deque>
#include <list>
#include <optional>
#include <gsl/span>
#include "TRandom3.h"
#include "DataFormatsEMCAL/Digit.h"
#include "EMCALSimulation/LabeledDigit.h"
#include "DataFormatsEMCAL/TriggerRecord.h"
#include "SimulationDataFormat/MCTruthContainer.h"
#include "EMCALSimulation/SimParam.h"
#include "EMCALSimulation/DigitTimebin.h"
 
namespace o2
{
namespace emcal
{
 
 
class DigitsVectorStream
{
 
 public:
  DigitsVectorStream() = default;
 
  ~DigitsVectorStream() = default;
 
  void clear();
 
  void init();
 
  void fill(std::deque<o2::emcal::DigitTimebin>& digitlist, o2::InteractionRecord record);
 
  const std::vector<o2::emcal::Digit>& getDigits() const { return mDigits; }
  const std::vector<o2::emcal::TriggerRecord>& getTriggerRecords() const { return mTriggerRecords; }
  const o2::dataformats::MCTruthContainer<o2::emcal::MCLabel>& getMCLabels() const { return mLabels; }
 
  void doSimulateNoiseDigits(bool doNoise = true) { mSimulateNoiseDigits = doNoise; }
  void addNoiseDigits(LabeledDigit& d1);
 
  void doRemoveDigitsBelowThreshold(bool doThreshold = true) { mRemoveDigitsBelowThreshold = doThreshold; }
 
 private:
  unsigned int mStartIndex = 0; 
 
  std::vector<o2::emcal::Digit> mDigits;                         
  o2::dataformats::MCTruthContainer<o2::emcal::MCLabel> mLabels; 
  std::vector<o2::emcal::TriggerRecord> mTriggerRecords;         
 
  bool mSimulateNoiseDigits = true;        
  bool mRemoveDigitsBelowThreshold = true; 
 
  const SimParam* mSimParam = nullptr;  
  TRandom3* mRandomGenerator = nullptr; 
 
  ClassDefNV(DigitsVectorStream, 1);
};
 
} // namespace emcal
 
} // namespace o2
 
#endif /* ALICEO2_EMCAL_DIGITSVECTORSTREAM_H_ */