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_ITS_DETECTOR_H_
#define ALICEO2_ITS_DETECTOR_H_
 
#include <vector>                                    // for vector
#include "DetectorsBase/GeometryManager.h"           // for getSensID
#include "DetectorsBase/Detector.h"                  // for Detector
#include "DetectorsCommonDataFormats/DetID.h"        // for Detector
#include "ITSMFTSimulation/Hit.h"                    // for Hit
#include "ITSSimulation/DescriptorInnerBarrelITS2.h" // for Description of Inner Barrel (ITS2)
#include "Rtypes.h"                                  // for Int_t, Double_t, Float_t, Bool_t, etc
#include "TArrayD.h"                                 // for TArrayD
#include "TGeoManager.h"                             // for gGeoManager, TGeoManager (ptr only)
#include "TLorentzVector.h"                          // for TLorentzVector
#include "TVector3.h"                                // for TVector3
 
#ifdef ENABLE_UPGRADES
#include "ITS3Simulation/DescriptorInnerBarrelITS3.h" // for Description of Inner Barrel (ITS3)
#endif
 
class FairVolume;
class TGeoVolume;
 
class TParticle;
 
class TString;
 
namespace o2
{
namespace itsmft
{
class Hit;
}
} // namespace o2
 
namespace o2
{
namespace its
{
class GeometryTGeo;
}
} // namespace o2
namespace o2
{
namespace its
{
class V3Layer;
}
} // namespace o2
 
namespace o2
{
namespace its
{
class V3Layer;
class V3Services;
 
class Detector : public o2::base::DetImpl<Detector>
{
 public:
  static constexpr Int_t sNumberOuterLayers = 4;      
  static constexpr Int_t sNumberOfWrapperVolumes = 3; 
 
  Detector(Bool_t active, TString name = "ITS");
 
  // Factory method
  static o2::base::Detector* create(const char* name, bool active)
  {
    return new Detector(active, name);
  }
 
  Detector();
 
  ~Detector() override;
 
  void InitializeO2Detector() override;
 
  Bool_t ProcessHits(FairVolume* v = nullptr) override;
 
  void Register() override;
 
  void configOuterBarrelITS(int nInnerBarrelLayers, int buildLevel = 0);
 
  std::vector<o2::itsmft::Hit>* getHits(Int_t iColl) const
  {
    if (iColl == 0) {
      return mHits;
    }
    return nullptr;
  }
 
 public:
  void Reset() override;
 
  void ConstructGeometry() override;
 
  void defineLayer(Int_t nlay, Double_t phi0, Double_t r, Int_t nladd, Int_t nmod, Double_t lthick = 0.,
                   Double_t dthick = 0., UInt_t detType = 0, Int_t buildFlag = 0) override;
 
  virtual void getLayerParameters(Int_t nlay, Double_t& phi0, Double_t& r, Int_t& nladd, Int_t& nmod, Double_t& width,
                                  Double_t& tilt, Double_t& lthick, Double_t& mthick, UInt_t& dettype) const;
 
  o2::itsmft::Hit* addHit(int trackID, int detID, const TVector3& startPos, const TVector3& endPos,
                          const TVector3& startMom, double startE, double endTime, double eLoss,
                          unsigned char startStatus, unsigned char endStatus);
 
  void defineWrapperVolume(Int_t id, Double_t rmin, Double_t rmax, Double_t zspan) override;
 
  void addAlignableVolumes() const override;
 
  void fillParallelWorld() const override;
 
  void addAlignableVolumesLayer(Int_t lr, TString& parent, Int_t& lastUID) const;
 
  void addAlignableVolumesHalfBarrel(Int_t lr, Int_t hb, TString& parent, Int_t& lastUID) const;
 
  void addAlignableVolumesStave(Int_t lr, Int_t hb, Int_t st, TString& parent, Int_t& lastUID) const;
 
  void addAlignableVolumesHalfStave(Int_t lr, Int_t hb, Int_t st, Int_t hst, TString& parent, Int_t& lastUID) const;
 
  void addAlignableVolumesModule(Int_t lr, Int_t hb, Int_t st, Int_t hst, Int_t md, TString& parent, Int_t& lastUID) const;
 
  void addAlignableVolumesChip(Int_t lr, Int_t hb, Int_t st, Int_t hst, Int_t md, Int_t ch, TString& parent,
                               Int_t& lastUID) const;
 
  Int_t chipVolUID(Int_t id) const { return o2::base::GeometryManager::getSensID(o2::detectors::DetID::ITS, id); }
 
  void EndOfEvent() override;
 
  void FinishPrimary() override { ; }
  virtual void finishRun() { ; }
  void BeginPrimary() override { ; }
  void PostTrack() override { ; }
  void PreTrack() override { ; }
 
  Int_t getNumberOfLayers() const { return mNumberLayers; }
 
  GeometryTGeo* mGeometryTGeo; 
 
 protected:
  struct TrackData {               // this is transient
    bool mHitStarted;              
    unsigned char mTrkStatusStart; 
    TLorentzVector mPositionStart; 
    TLorentzVector mMomentumStart; 
    double mEnergyLoss;            
  } mTrackData;                    
 
  int mNumberInnerLayers; 
  int mNumberLayers;      
 
  std::vector<int> mLayerID;       
  std::vector<TString> mLayerName; 
 
  Int_t mNumberOfDetectors;
 
  Bool_t mModifyGeometry;
 
  Double_t mWrapperMinRadius[sNumberOfWrapperVolumes]; 
  Double_t mWrapperMaxRadius[sNumberOfWrapperVolumes]; 
  Double_t mWrapperZSpan[sNumberOfWrapperVolumes];     
  std::vector<int> mWrapperLayerId;                    
 
  std::vector<bool> mTurboLayer;          
  std::vector<double> mLayerPhi0;         
  std::vector<double> mLayerRadii;        
  std::vector<int> mStavePerLayer;        
  std::vector<int> mUnitPerStave;         
  std::vector<double> mChipThickness;     
  std::vector<double> mStaveWidth;        
  std::vector<double> mStaveTilt;         
  std::vector<double> mDetectorThickness; 
  std::vector<uint> mChipTypeID;          
  std::vector<int> mBuildLevel;           
 
  std::vector<o2::itsmft::Hit>* mHits;
 
  TGeoVolume* createWrapperVolume(const Int_t nLay);
 
  virtual void createMaterials();
 
  void constructDetectorGeometry();
 
  void defineSensitiveVolumes();
 
  void createMiddlBarrelServices(TGeoVolume* motherVolume);
 
  void createOuterBarrelServices(TGeoVolume* motherVolume);
 
  void createOuterBarrelSupports(TGeoVolume* motherVolume);
 
  void createITSServices(TGeoVolume* motherVolume);
 
  Detector(const Detector&);
 
  Detector& operator=(const Detector&);
 
  std::vector<V3Layer*> mGeometry; 
  V3Services* mServicesGeometry;   
 
  std::shared_ptr<DescriptorInnerBarrel> mDescriptorIB; 
 
  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 its
} // namespace o2
 
#ifdef USESHM
namespace o2
{
namespace base
{
template <>
struct UseShm<o2::its::Detector> {
  static constexpr bool value = true;
};
} // namespace base
} // namespace o2
#endif
 
#endif