AssocModeTask.h

Go to the documentation of this file.

 
#ifndef _SEIMPLEMENTATION_PLUGIN_ASSOCMODE_ASSOCMODETASK_H_
#define _SEIMPLEMENTATION_PLUGIN_ASSOCMODE_ASSOCMODETASK_H_
 
 
#include "AssocModeConfig.h"
#include "SEUtils/Types.h"
#include "SEUtils/KdTree.h"
 
#include "SEFramework/Task/SourceTask.h"
 
 
namespace SourceXtractor {
 
class AssocModeTask : public SourceTask {
public:
  using CatalogEntry = AssocModeConfig::CatalogEntry;
  using GetAssocResult = std::function<AssocModeConfig::CatalogEntry(const std::vector<AssocModeConfig::CatalogEntry>&)>;
 
  virtual ~AssocModeTask() = default;
 
  AssocModeTask(const std::vector<std::vector<AssocModeConfig::CatalogEntry>>& catalogs, AssocModeConfig::AssocMode assoc_type, double radius);
 
  void computeProperties(SourceInterface& source) const override;
 
private:
  std::vector<KdTree<AssocModeConfig::CatalogEntry>> m_catalogs;
  AssocModeConfig::AssocMode m_assoc_mode;
  double m_radius;
};
 
}
 
 
 
#endif /* _SEIMPLEMENTATION_PLUGIN_ASSOCMODE_ASSOCMODETASK_H_ */