PndSttGeometryMap.h

Go to the documentation of this file.

#ifndef PNDSTTGEOMETRYMAP_H
#define PNDSTTGEOMETRYMAP_H

#include <iostream>
#include <map>
#include <vector>
#include "TVector3.h"
#include "TObject.h"
#include "TArrayI.h"

class PndSttTubeParameters;
class PndSttTube;
class PndGeoSttPar;
class PndSttHit;
class FairGeoNode;

class FairHit;

class TClonesArray;

using std::map;
using std::vector;

class PndSttGeometryMap : public TObject {
 public:
  PndSttGeometryMap();
  PndSttGeometryMap(TClonesArray *tubearray, Int_t geoType);
  ~PndSttGeometryMap();
  PndSttGeometryMap(const PndSttGeometryMap &) = delete;
  PndSttGeometryMap &operator=(const PndSttGeometryMap &) = delete;
  // ----------------------------
  void SetGeneralParameters();
  void GenerateStrawMap(Int_t map);
  Bool_t FillGeometryParameters();
  // ----------------------------

  // *** GEO TYPE 1 ***
  void SetGeneralParametersGeoType1();
  void GenerateStrawMapGeoType1(Int_t map);
  Bool_t FillGeometryParametersGeoType1();

  void GenerateStrawMapTubeIDGeoType1();
  void GenerateStrawMapAngleGeoType1();

  int GetRow(int strawindex) const { return fLayerOfStraw.at(strawindex); }
  int GetSector(int strawindex) const { return fSectorOfStraw.at(strawindex); }
  const vector<int> &GetStrawRow(int sector, int row) const { return (fStrawIndex.find(sector))->second.at(row); }
  const vector<vector<int>> &GetStrawSector(int sector) const { return (fStrawIndex.find(sector))->second; }

  bool IsEdgeStraw(int strawindex) const;
  int IsSectorBorderStraw(int strawindex) const;

  bool IsAxialStraw(int strawindex) const { return fAxialStraw.at(strawindex); }
  bool IsSkewedStraw(int strawindex) const { return !(fAxialStraw.at(strawindex)); }

  bool IsAxialRow(int rowindex) const;
  bool IsSkewedRow(int rowindex) const;

  TArrayI FindNeighborings(PndSttTube *tube);
  TArrayI FindNeighborings(int tubeId);

  TArrayI GetNeighboringsByMap(int tubeId);

  void FillStrawNeighborsMap();

  Double_t
  CalculateStrawPoca(PndSttHit *hit1, PndSttHit *hit2, TVector3 &poca); // returns smallest distance between two Stt Straws. poca is set to the point between the two straws.

  double GetAngleBetweenTubes(int tubeID1, int tubeID2) const;

  bool InStraightLine(int tube1, int tube2, int tube3) const;

  // *********************

  void SetVerbose(int ver) { fVerbose = ver; }

 private:
  Int_t fGeoType, fVerbose;

  void GenerateAngles();
  map<int, vector<vector<int>>> fStrawIndex;
  map<int, TArrayI> fStrawNeighbors;
  vector<int> fSectorOfStraw;
  vector<int> fLayerOfStraw;
  vector<bool> fAxialStraw;
  vector<double> fSectorStart;
  vector<double> fSectorEnd;
  bool fStrawMapInitialized;

  // ------------------ SECTORS/LAYERS
  Int_t fNLayers, fNSectors, fNTubes, fNTubes_inner_parallel, fNTubes_outer_parallel, fNTubes_fillup_parallel, fNTubes_skewed, fNLayers_inner_parallel, fNLayers_skew,
    fNLayers_outer_parallel, fNLayers_fillup_parallel;
  Int_t **fStartTube, **fEndTube, **fShift, **fShiftSkew;

  TClonesArray *fTubeArray;

 protected:
  ClassDef(PndSttGeometryMap, 2)
};

#endif