image_manip_node_test.cpp

Go to the documentation of this file.

#include <chrono>
 
#include "depthai/depthai.hpp"
 
// TEST
// ImageManipNode test
int main() {
    using namespace std;
    using namespace std::chrono;
    using namespace std::chrono_literals;
 
    // Create pipeline
    dai::Pipeline p;
    auto xin = p.create<dai::node::XLinkIn>();
    auto imageManip = p.create<dai::node::ImageManip>();
    auto xout = p.create<dai::node::XLinkOut>();
 
    // resize
    const int originalWidth = 640, originalHeight = 360;
    const int width = 337, height = 225;
 
    // Properties
    xin->setStreamName("in");
    xout->setStreamName("out");
    imageManip->initialConfig.setResize(width, height);
 
    // Link plugins CAM -> XLINK
    xin->out.link(imageManip->inputImage);
    imageManip->out.link(xout->input);
 
    // Try connecting to an available device
    dai::Device d(p);
 
    auto in = d.getInputQueue("in");
    auto out = d.getOutputQueue("out");
 
    // Send 10 messages
    for(int i = 0; i < 10; i++) {
        // Create 'rgb interleaved' frame
        dai::ImgFrame inFrame;
        inFrame.getData().resize(originalWidth * originalHeight * 3);
        inFrame.setWidth(originalWidth);
        inFrame.setHeight(originalHeight);
        inFrame.setType(dai::ImgFrame::Type::RGB888p);
 
        // Send the frame
        in->send(inFrame);
 
        // Retrieve the resized frame
        auto outFrame = out->get<dai::ImgFrame>();
 
        // Check that out frame is resized
        if(outFrame->getWidth() != width || outFrame->getHeight() != height) {
            return -1;
        }
    }
 
    // Exit with success error code
    return 0;
}