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 "common/AxesArray.h"
#include "common/CooArray.h"
#include "common/DefaultValues.h"
#include "dataobjects/Point.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) {}

  PsdGeoParams(unsigned int nTotElements)
      : DEFAULT_INIT(_nElements), DEFAULT_INIT(_sizes), DEFAULT_INIT(_normal), _psdElements{nTotElements} {}

  virtual ~PsdGeoParams() = default;

  const AxesArray<unsigned int> &NElements() const { return _nElements; };

  unsigned int NTotElements() const { return _psdElements.size(); };

  RefFrame::Direction Normal() const { return _normal; };

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

  const AxesArray<float> &Sizes() const { return _sizes; };

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

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

  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; };

  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;          
  std::vector<PsdElement> _psdElements; 

#ifdef HS_USE_ROOT
  ClassDef(PsdGeoParams, 1)
#endif
};

} // namespace Herd

#endif /* HERD_PSDGEOPARAMS_H_ */