SegmentationImage.h

Go to the documentation of this file.

/*
 * Copyright (C) 2006-2020 Istituto Italiano di Tecnologia (IIT)
 * All rights reserved.
 *
 * This software may be modified and distributed under the terms of the
 * BSD-3-Clause license. See the accompanying LICENSE file for details.
 */
 
#ifndef YARP_CARRIER_SEGMENTATION_CONVERTER_H
#define YARP_CARRIER_SEGMENTATION_CONVERTER_H
 
#include <yarp/os/Bottle.h>
#include <yarp/os/Things.h>
#include <yarp/os/MonitorObject.h>
#include <yarp/sig/Image.h>
 
//example usage:
//yarp connect /segmentationimage:o /yarpview/img:i tcp+recv.portmonitor+type.dll+file.segmentationimage
 
struct rgbColor
{
    char r;
    char g;
    char b;
};
 
class SegmentationImageConverter : public yarp::os::MonitorObject
{
public:
    bool create(const yarp::os::Property& options) override;
    void destroy() override;
 
    bool setparam(const yarp::os::Property& params) override;
    bool getparam(yarp::os::Property& params) override;
 
    bool accept(yarp::os::Things& thing) override;
    yarp::os::Things& update(yarp::os::Things& thing) override;
 
private:
    int max_colors;
    yarp::os::Bottle bt;
    yarp::os::Things th;
    yarp::sig::FlexImage outImg;
    std::map<int, rgbColor> colormap;
};
 
#endif  // YARP_CARRIER_SEGMENTATION_CONVERTER_H