Digitizer.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_MID_DIGITIZER_H
#define O2_MID_DIGITIZER_H
 
#include <random>
#include <vector>
#include <array>
#include "MathUtils/Cartesian.h"
#include "SimulationDataFormat/MCTruthContainer.h"
#include "DataFormatsMID/ColumnData.h"
#include "DataFormatsMID/ChEffCounter.h"
#include "MIDBase/Mapping.h"
#include "MIDBase/GeometryTransformer.h"
#include "MIDSimulation/ChamberResponse.h"
#include "MIDSimulation/ChamberEfficiencyResponse.h"
#include "MIDSimulation/Hit.h"
#include "DataFormatsMID/MCLabel.h"
 
namespace o2
{
namespace mid
{
class Digitizer
{
 public:
  Digitizer(const ChamberResponse& chamberResponse, const ChamberEfficiencyResponse& efficiencyResponse, const GeometryTransformer& transformer);
  virtual ~Digitizer() = default;
 
  void process(const std::vector<Hit>& hits, std::vector<ColumnData>& digitStore, o2::dataformats::MCTruthContainer<MCLabel>& mcContainer);
 
  void setEventID(int entryID) { mEventID = entryID; }
 
  void setSrcID(int sourceID) { mSrcID = sourceID; }
 
  void setGeometryTransformer(const GeometryTransformer& transformer) { mTransformer = transformer; }
 
  void setChamberResponse(const ChamberResponse& chamberResponse) { mResponse = chamberResponse; }
 
  ChamberResponse& getChamberResponse() { return mResponse; }
 
  void setChamberEfficiency(const std::vector<ChEffCounter>& counters) { mEfficiencyResponse.setFromCounters(counters); }
 
  void setSeed(unsigned int seed) { mGenerator.seed(seed); }
 
 private:
  void addStrip(const Mapping::MpStripIndex& stripIndex, int cathode, int deId);
  bool addBPStrips(double xPos, double yPos, int deId, double prob, double xOffset);
  bool addNeighbours(const Mapping::MpStripIndex& stripIndex, int cathode, int deId, double prob,
                     const std::array<double, 2>& initialDist, double xOffset = 0.);
  bool hitToDigits(const Hit& hit);
  bool getLabelLimits(int cathode, const ColumnData& col, int& firstStrip, int& lastStrip) const;
 
  int mEventID{0};
  int mSrcID{0};
 
  std::default_random_engine mGenerator;          
  std::uniform_real_distribution<double> mRandom; 
  ChamberResponse mResponse;                      
  ChamberEfficiencyResponse mEfficiencyResponse;  
  Mapping mMapping;                               
  GeometryTransformer mTransformer;               
  std::vector<ColumnData> mDigits;                
};
 
Digitizer createDefaultDigitizer();
 
} // namespace mid
} // namespace o2
 
#endif /* O2_MID_DIGITIZER_H */