teca_connected_components.h

#ifndef teca_connected_components_h
#define teca_connected_components_h
 
#include "teca_shared_object.h"
#include "teca_algorithm.h"
#include "teca_metadata.h"
 
#include <string>
#include <vector>
 
TECA_SHARED_OBJECT_FORWARD_DECL(teca_connected_components)
 
 
class teca_connected_components : public teca_algorithm
{
public:
    TECA_ALGORITHM_STATIC_NEW(teca_connected_components)
    TECA_ALGORITHM_DELETE_COPY_ASSIGN(teca_connected_components)
    TECA_ALGORITHM_CLASS_NAME(teca_connected_components)
    ~teca_connected_components();
 
    // set the input array containing a binary segmentation
    // see teca_binary_segmentation
    TECA_ALGORITHM_PROPERTY(std::string, segmentation_variable)
 
    // set the name of the output array to store the component labels in
    TECA_ALGORITHM_PROPERTY(std::string, component_variable)
 
 
protected:
    teca_connected_components();
 
    std::string get_component_variable(const teca_metadata &request);
    std::string get_segmentation_variable(const teca_metadata &request);
 
private:
    teca_metadata get_output_metadata(unsigned int port,
        const std::vector<teca_metadata> &input_md) override;
 
    std::vector<teca_metadata> get_upstream_request(
        unsigned int port, const std::vector<teca_metadata> &input_md,
        const teca_metadata &request) override;
 
    const_p_teca_dataset execute( unsigned int port,
        const std::vector<const_p_teca_dataset> &input_data,
        const teca_metadata &request) override;
 
private:
    std::string component_variable;
    std::string segmentation_variable;
};
 
#endif