AssocMode.h

Go to the documentation of this file.

 
#ifndef _SEIMPLEMENTATION_PLUGIN_ASSOCMODE_ASSOCMODE_H_
#define _SEIMPLEMENTATION_PLUGIN_ASSOCMODE_ASSOCMODE_H_
 
#include "NdArray/NdArray.h"
 
#include "SEUtils/Types.h"
#include "SEFramework/Property/Property.h"
#include "SEFramework/Source/SourceFlags.h"
 
namespace SourceXtractor {
 
class AssocMode : public Property {
public:
  template<typename T>
  using NdArray = Euclid::NdArray::NdArray<T>;
 
  virtual ~AssocMode() = default;
 
  AssocMode() : m_has_assoc(false), m_assoc_data(std::vector<size_t>({0})), m_ref_frame_pixel_radius(1), m_group_id(0) {
  }
 
  AssocMode(bool has_assoc, const std::vector<double>& assoc_data,
      double ref_frame_pixel_radius=0.0, unsigned int group_id=0) :
      m_has_assoc(has_assoc), m_assoc_data(std::vector<size_t>({assoc_data.size()})),
      m_ref_frame_pixel_radius(ref_frame_pixel_radius), m_group_id(group_id) {
    for (size_t i=0; i<assoc_data.size(); i++) {
      m_assoc_data.at(i) = assoc_data[i];
    }
  }
 
  bool getMatch() const {
    return m_has_assoc;
  }
 
  const NdArray<SeFloat>& getAssocValues() const {
    return m_assoc_data;
  }
 
  double getRefFramePixelRadius() const {
    return m_ref_frame_pixel_radius;
  }
 
  unsigned int getGroupId() const {
    return m_group_id;
  }
 
private:
  bool m_has_assoc;
 
  NdArray<SeFloat> m_assoc_data;
  double m_ref_frame_pixel_radius;
  unsigned int m_group_id;
};
 
} /* namespace SourceXtractor */
 
#endif /* _SEIMPLEMENTATION_PLUGIN_ASSOCMODE_ASSOCMODE_H_ */