SiliconDetectorGeoParams.h

Go to the documentation of this file.

/*
 * SiliconDetectorGeoParams.h
 *
 *  Created on: 14 Nov 2018
 *      Author: Nicola Mori
 */

#ifndef HERD_SILICONDETECTORGEOPARAMS_H_
#define HERD_SILICONDETECTORGEOPARAMS_H_

// HERD headers
#include "common/AxesArray.h"
#include "common/DefaultValues.h"
#include "common/SidesArray.h"
#include "common/TypeUtils.h"
#include "dataobjects/Point.h"
#include "dataobjects/Rectangle.h"
#include "dataobjects/TrackingDetectorGeoParams.h"

#ifdef HS_USE_ROOT
#include "Rtypes.h"
#endif

// C++ headers
#include <vector>

namespace Herd {

class SiliconDetectorGeoParams : public TrackingDetectorGeoParams {
public:
  SiliconDetectorGeoParams()
      : TrackingDetectorGeoParams(), DEFAULT_INIT(_firstStripPos), DEFAULT_INIT(_nStripsPerLadder),
        DEFAULT_INIT(_waferSize), DEFAULT_INIT(_nWafersPerLadder), DEFAULT_INIT(_nLadders), _waferPos{} {};
  SiliconDetectorGeoParams(RefFrame::Side dside, size_t layerNumber, RefFrame::Direction stripMainSegm);

  ~SiliconDetectorGeoParams() override = default;

  size_t NWafers() const { return _waferPos.size(); }

  float StripPitch() const { return ElementPitch(); }

  int NStripsPerLadder() const { return _nStripsPerLadder; }

  float FirstStripPosition() const { return _firstStripPos; }

  const AxesArray<int> &NWafersPerLadder() const { return _nWafersPerLadder; }

  const AxesArray<int> &NLadders() const { return _nLadders; }

  const AxesArray<float> &WaferSize() const { return _waferSize; }

  Point ElementPosition(unsigned int volID) const override;

  Point StripPosition(unsigned int volID) const { return ElementPosition(volID); };

  const Point &WaferPosition(unsigned int volID) const { return _waferPos.at(volID); }

  Point LadderPosition(unsigned int ladderID) const;

  std::pair<RefFrame::Direction, RefFrame::Direction> StripSegmentationDirections() const {
    return SegmentationDirections();
  };

  std::pair<RefFrame::Direction, RefFrame::Direction> WaferSegmentationDirections() const;

  std::pair<RefFrame::Direction, RefFrame::Direction> LadderSegmentationDirections() const;

  unsigned int LadderIDFromStripID(unsigned int stripID) const;

  unsigned int LadderIDFromWaferID(unsigned int waferID) const;

  std::pair<unsigned int, unsigned int> WaferIDsFromLadderID(unsigned int ladderID) const;

  std::pair<unsigned int, unsigned int> StripIDsFromLadderID(unsigned int ladderID) const;

  unsigned int StripIDOnLadder(unsigned int stripID) const;

  unsigned int WaferID(const Point &position) const;

  unsigned int ElementID(const Point &position) const override;

  unsigned int StripID(const Point &position) const { return ElementID(position); };

  void SetSideAndStripMainSegmDir(RefFrame::Side dside, RefFrame::Direction segmDir) {
    SetSideAndMainSegmDir(dside, segmDir);
  };

  void SetSideAndMainSegmDir(RefFrame::Side dside, RefFrame::Direction segmDir) override;

  void SetStripPitch(float pitch);

  void SetNStripsPerLadder(int nStrips);

  void SetNWafersPerLadder(AxesArray<int> nWafers);

  void SetNLadders(AxesArray<int> nLadders);

  void SetWaferSize(const AxesArray<float> &sizes) {
    _waferSize = sizes;
    m_waferLUT.isValid = false;
  }

  void SetWaferPosition(size_t waferID, Point position) {
    _waferPos.at(waferID) = position;
    m_waferLUT.isValid = false;
  }

private:
  float _firstStripPos;             
  int _nStripsPerLadder;            
  AxesArray<float> _waferSize;      
  AxesArray<int> _nWafersPerLadder; 
  AxesArray<int> _nLadders;         
  std::vector<Point> _waferPos;     

protected:
  void UpdateLUT() const;

  struct WaferLookupTable {
    bool isValid = false;
    std::array<size_t, 2> ncells; 
    std::array<float, 2> sizes;   
    std::array<float, 2>
        cellDimension; 

    Rectangle surface;

    std::vector<std::vector<unsigned int>> waferIDs; 

    WaferLookupTable() : DEFAULT_INIT(ncells), DEFAULT_INIT(cellDimension) {}

    const std::vector<unsigned int> &WaferIDsAtPos(std::array<float, 2> position);

    size_t PosToCellID(std::array<float, 2> position);

    void UpdateNCells(std::array<size_t, 2> newNcells);
    void UpdateSurface(std::array<float, 2> newOrigin, std::array<float, 2> newSizes);
    void UpdateInternals(std::vector<Point> waferPositions, std::array<float, 2> waferSizes,
                         std::pair<RefFrame::Direction, RefFrame::Direction> segmDirections);
  } mutable m_waferLUT; 

#ifdef HS_USE_ROOT
  ClassDefOverride(SiliconDetectorGeoParams, 1)
#endif
};

} // namespace Herd

#endif /* HERD_SILICONDETECTORGEOPARAMS_H_ */