ImageManipConfig.hpp

Go to the documentation of this file.

#pragma once
 
#include <unordered_map>
#include <vector>
 
#include "depthai-shared/datatype/RawImageManipConfig.hpp"
#include "depthai/pipeline/datatype/Buffer.hpp"
#include "depthai/pipeline/datatype/ImgFrame.hpp"
 
namespace dai {
 
class ImageManipConfig : public Buffer {
    std::shared_ptr<RawBuffer> serialize() const override;
    RawImageManipConfig& cfg;
 
   public:
    // Alias
    using CropConfig = RawImageManipConfig::CropConfig;
    using ResizeConfig = RawImageManipConfig::ResizeConfig;
    using FormatConfig = RawImageManipConfig::FormatConfig;
 
    ImageManipConfig();
    explicit ImageManipConfig(std::shared_ptr<RawImageManipConfig> ptr);
    virtual ~ImageManipConfig() = default;
 
    // Functions to set properties
    ImageManipConfig& setCropRect(float xmin, float ymin, float xmax, float ymax);
 
    ImageManipConfig& setCropRect(std::tuple<float, float, float, float> coordinates);
 
    ImageManipConfig& setCropRotatedRect(RotatedRect rr, bool normalizedCoords = true);
 
    ImageManipConfig& setCenterCrop(float ratio, float whRatio = 1.0f);
 
    ImageManipConfig& setWarpTransformFourPoints(std::vector<Point2f> pt, bool normalizedCoords);
 
    ImageManipConfig& setWarpTransformMatrix3x3(std::vector<float> mat);
 
    ImageManipConfig& setWarpBorderReplicatePixels();
 
    ImageManipConfig& setWarpBorderFillColor(int red, int green, int blue);
 
    ImageManipConfig& setRotationDegrees(float deg);
 
    ImageManipConfig& setRotationRadians(float rad);
 
    ImageManipConfig& setResize(int w, int h);
 
    ImageManipConfig& setResize(std::tuple<int, int> size);
 
    ImageManipConfig& setResizeThumbnail(int w, int h, int bgRed = 0, int bgGreen = 0, int bgBlue = 0);
 
    ImageManipConfig& setResizeThumbnail(std::tuple<int, int> size, int bgRed = 0, int bgGreen = 0, int bgBlue = 0);
 
    ImageManipConfig& setFrameType(ImgFrame::Type name);
 
    ImageManipConfig& setColormap(Colormap colormap, int min, int max);
    ImageManipConfig& setColormap(Colormap colormap, float maxf);
    ImageManipConfig& setColormap(Colormap colormap, int max = 255);
 
    ImageManipConfig& setHorizontalFlip(bool flip);
 
    void setVerticalFlip(bool flip);
 
    ImageManipConfig& setReusePreviousImage(bool reuse);
 
    ImageManipConfig& setSkipCurrentImage(bool skip);
 
    ImageManipConfig& setKeepAspectRatio(bool keep);
 
    ImageManipConfig& setInterpolation(dai::Interpolation interpolation);
 
    // Functions to retrieve properties
    float getCropXMin() const;
 
    float getCropYMin() const;
 
    float getCropXMax() const;
 
    float getCropYMax() const;
 
    int getResizeWidth() const;
 
    int getResizeHeight() const;
 
    CropConfig getCropConfig() const;
 
    ResizeConfig getResizeConfig() const;
 
    FormatConfig getFormatConfig() const;
 
    bool isResizeThumbnail() const;
 
    Colormap getColormap() const;
 
    ImageManipConfig& set(dai::RawImageManipConfig config);
 
    dai::RawImageManipConfig get() const;
 
    dai::Interpolation getInterpolation() const;
};
 
}  // namespace dai