Detector.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_FV0_DETECTOR_H_
#define ALICEO2_FV0_DETECTOR_H_
 
#include <TLorentzVector.h>
 
#include "DataFormatsFV0/Hit.h"
#include "DetectorsBase/Detector.h"
#include "FV0Base/Geometry.h"
#include "SimulationDataFormat/BaseHits.h"
 
class FairModule;
class FairVolume;
class TGeoVolume;
class TGraph;
 
namespace o2
{
namespace fv0
{
class Detector : public o2::base::DetImpl<Detector>
{
 public:
  Detector();
 
  ~Detector() override;
 
  explicit Detector(Bool_t isActive);
 
  void InitializeO2Detector() override;
  bool isMTSafe() const override { return true; }
 
  Bool_t ProcessHits(FairVolume* v = nullptr) override;
 
  // ------------------------------------------------------------------
 
  void Register() override;
 
  std::vector<o2::fv0::Hit>* getHits(Int_t iColl)
  {
    if (iColl == 0) {
      return mHits;
    }
    return nullptr;
  }
 
  void Reset() override;
 
  void EndOfEvent() override;
 
  void createMaterials();
 
  void ConstructGeometry() override; // inherited from FairModule
 
  enum EMedia {
    Zero,
    Air,
    Scintillator,
    Plastic,
    FiberRing1,
    FiberRing2,
    FiberRing3,
    FiberRing4,
    FiberRing5,
    FiberPMT1,
    FiberPMT2,
    FiberPMT3,
    FiberPMT4,
    FiberPMT5,
    Aluminium,
    Steel,
    Titanium,
    PMT,
  }; // media IDs used in createMaterials()
 
  void addAlignableVolumes() const override;
 
 private:
  std::vector<o2::fv0::Hit>* mHits = nullptr;
  int mTrackRefLastTrackID = -1; 
 
  Geometry* mGeometry = nullptr; 
 
  struct TrackData {               // this is transient
    bool mHitStarted;              
    TLorentzVector mPositionStart; 
    TLorentzVector mMomentumStart; 
    double mEnergyLoss;            
  } mTrackData;                    
 
  o2::fv0::Hit* addHit(Int_t trackId, Int_t cellId,
                       const math_utils::Point3D<float>& startPos, const math_utils::Point3D<float>& endPos,
                       const math_utils::Vector3D<float>& startMom, double startE,
                       double endTime, double eLoss, Int_t particlePdg);
 
  template <typename Det>
  friend class o2::base::DetImpl;
  ClassDefOverride(Detector, 1);
};
 
// Input and output function for standard C++ input/output.
std::ostream& operator<<(std::ostream& os, Detector& source);
std::istream& operator>>(std::istream& os, Detector& source);
} // namespace fv0
} // namespace o2
 
#ifdef USESHM
namespace o2
{
namespace base
{
template <>
struct UseShm<o2::fv0::Detector> {
  static constexpr bool value = true;
};
} // namespace base
} // namespace o2
#endif
 
#endif