CaloGeoParams.h

Go to the documentation of this file.

/*
 * CaloGeoParams.h
 *
 *  Created on: 19 Set 2019
 *      Author: Lorenzo Pacini
 */

#ifndef HERD_CALOGEOPARAMS_H_
#define HERD_CALOGEOPARAMS_H_

#include "common/AxesArray.h"
#include "common/DefaultValues.h"
#include "dataobjects/Line.h"
#include "dataobjects/Point.h"

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

#include <vector>

namespace Herd {

class CaloGeoParams {
public:
  CaloGeoParams(unsigned int nCubes)
      : DEFAULT_INIT(_cubeSize), DEFAULT_INIT(_cubeSlotSize), DEFAULT_INIT(_lysoDensity) {
    _caloCubes.resize(nCubes);
  }

  CaloGeoParams() : DEFAULT_INIT(_cubeSize), DEFAULT_INIT(_cubeSlotSize), DEFAULT_INIT(_lysoDensity) {}

  virtual ~CaloGeoParams() = default;

  const Point &Position(unsigned int volID) const { return _caloCubes[volID]._position; };

  const CooArray<unsigned char> &CooIndexes(unsigned int volID) const { return _caloCubes[volID]._indexes; };

  struct CooIndexRange {
    unsigned char begin = defaultValue<unsigned char>;
    unsigned char end = defaultValue<unsigned char>;

    bool operator==(const CooIndexRange &other) const { return begin == other.begin && end == other.end; }
    unsigned char Size() const { return end - begin + 1; }
  };

  CooIndexRange CooIndexesRange(RefFrame::Axis axis, unsigned char index1, unsigned char index2);

  bool IsEmpty(unsigned int volID) const { return _caloCubes[volID]._isEmpty; };

  const float &CubeSize() const { return _cubeSize; };

  float LYSODensity() const { return _lysoDensity; };

  float LYSORadiationLength() const { return 8.6728 / _lysoDensity; }; // 8.6728 = 1.172 * 7.4

  float LYSONuclearInteractionLength() const { return 152.3586 / _lysoDensity; }; // 152.3586 = 20.589 * 7.4

  const float &CubeSlotSize() const { return _cubeSlotSize; };

  unsigned int NCubes() const { return _caloCubes.size(); };

  unsigned int VolumeID(const CooArray<unsigned char> &cooIndex);

  unsigned int VolumeID(const Point &position);

  std::pair<Point, Point> GetCubeIntersections(const Line &track, const unsigned int volID,
                                               const Herd::RefFrame::Coo sort_by_coo = Herd::RefFrame::Coo::NONE) const;

  void SetNCubes(unsigned int nCubes) { _caloCubes.resize(nCubes); };

  void SetPosition(unsigned int volID, Point position) { _caloCubes[volID]._position = position; };

  void SetCooIndexes(unsigned int volID, CooArray<unsigned char> indexes) { _caloCubes[volID]._indexes = indexes; };

  void SetIsEmpty(unsigned int volID, bool isEmpty) { _caloCubes[volID]._isEmpty = isEmpty; };

  void SetCubeSize(float cubeSize) { _cubeSize = cubeSize; };

  void SetCubeSlotSize(float cubeSlotSize) { _cubeSlotSize = cubeSlotSize; };

  void SetLYSODensity(float density) { _lysoDensity = density; };

  struct CaloCube {

    Point _position;                  
    CooArray<unsigned char> _indexes; 
    bool _isEmpty;                    

    CaloCube() : DEFAULT_INIT(_position), DEFAULT_INIT(_indexes), _isEmpty(false) {}
  };

private:
  void FillCooId2VolIDMatrix();

  std::vector<CaloCube> _caloCubes; 
  float _cubeSize;                  
  float _cubeSlotSize;              
  float _lysoDensity;               
  std::vector<std::vector<std::vector<unsigned short>>>
      _cooId2VolIDMatrix; 

#ifdef HS_USE_ROOT
  ClassDef(CaloGeoParams, 4)
#endif
};

} // namespace Herd

#endif /* HERD_PSDGEOPARAMS_H_ */