PsdGeoParams.h

Go to the documentation of this file.

/*
 * PsdGeoParams.h
 *
 *  Created on: 7 Dec 2018
 *      Author: Nicola Mori
 */
 
#ifndef HERD_PSDGEOPARAMS_H_
#define HERD_PSDGEOPARAMS_H_
 
#include "herd/analysis/common/AxesArray.h"
#include "herd/analysis/common/CooArray.h"
#include "herd/analysis/common/DefaultValues.h"
#include "herd/analysis/dataobjects/Parallelepiped.h"
#include "herd/analysis/dataobjects/Point.h"
#include "herd/analysis/dataobjects/TrapezoidalPrism.h"
 
#ifdef HS_USE_ROOT
#include "Rtypes.h"
#endif
 
#include <vector>
 
namespace Herd {
 
struct PsdGeoParams {
 
  PsdGeoParams()
      : DEFAULT_INIT(_nElements), DEFAULT_INIT(_sizes), DEFAULT_INIT(_normal), DEFAULT_INIT(_mainSegmDir),
        DEFAULT_INIT(_secSegmDir) {}
 
  PsdGeoParams(unsigned int nTotElements)
      : DEFAULT_INIT(_nElements), DEFAULT_INIT(_sizes), DEFAULT_INIT(_normal), DEFAULT_INIT(_mainSegmDir),
        DEFAULT_INIT(_secSegmDir), _psdElements{nTotElements} {}
 
  virtual ~PsdGeoParams() = default;
 
  const Parallelepiped &BoundingBox() const;
 
  [[nodiscard]] const AxesArray<unsigned int> &NElements() const { return _nElements; };
 
  [[nodiscard]] unsigned int NTotElements() const { return _psdElements.size(); };
 
  [[nodiscard]] RefFrame::Direction NormalDirection() const { return _normal; };
 
  [[nodiscard]] const AxesArray<float> &Sizes() const { return _sizes; };
 
  [[nodiscard]] const Point &Position(unsigned int volID) const { return _psdElements[volID]._position; };
 
  [[nodiscard]] const CooArray<unsigned char> &CooIndexes(unsigned int volID) const {
    return _psdElements[volID]._indexes;
  };
 
  [[nodiscard]] TrapezoidalPrism ElementShape(unsigned int volID) const;
 
  [[nodiscard]] float BaseAngle() const { return _baseAngle; };
 
  bool IsSmallerFaceOutward(unsigned int volID) { return (bool)(volID % 2); };
 
  [[nodiscard]] std::pair<RefFrame::Direction, RefFrame::Direction> SegmentationDirections() const {
    return std::make_pair(_mainSegmDir, _secSegmDir);
  };
 
  void SetNElements(AxesArray<unsigned int> nElements);
 
  void SetNormal(RefFrame::Direction normal) { _normal = normal; };
 
  void SetSizes(AxesArray<float> sizes) { _sizes = sizes; };
 
  void SetPosition(unsigned int volID, Point position) { _psdElements[volID]._position = position; };
 
  void SetBaseAngle(float baseAngle) { _baseAngle = baseAngle; };
 
  void SetSegmentationDirections(RefFrame::Direction mainSegmDir, RefFrame::Direction secSegmDir) {
    _mainSegmDir = mainSegmDir;
    _secSegmDir = secSegmDir;
  };
 
  struct PsdElement {
    Point _position;                  
    CooArray<unsigned char> _indexes; 
 
    PsdElement() : _position() {
      _indexes[RefFrame::Coo::X] = std::numeric_limits<unsigned char>::max();
      _indexes[RefFrame::Coo::Y] = std::numeric_limits<unsigned char>::max();
      _indexes[RefFrame::Coo::Z] = std::numeric_limits<unsigned char>::max();
    };
  };
 
private:
  AxesArray<unsigned int> _nElements;   
  AxesArray<float> _sizes;              
  RefFrame::Direction _normal;          
  RefFrame::Direction _mainSegmDir;     
  RefFrame::Direction _secSegmDir;      
  float _baseAngle;                     
  std::vector<PsdElement> _psdElements; 
  bool mutable _bbIsValid{false};       
  Parallelepiped mutable _boundingBox;  
 
  void ComputeBoundingBox() const;
 
  void SetDefaultSegmDirections(); // backward-compatibility fallback setting of default directions
 
#ifdef HS_USE_ROOT
  ClassDef(PsdGeoParams, 3)
#endif
};
 
} // namespace Herd
 
#endif /* HERD_PSDGEOPARAMS_H_ */