vk_cv.cpp

Go to the documentation of this file.

/**
 * @file vk_cv.cpp
 * @brief Implementation of mx::MXCapture (Vulkan + OpenCV variant).
 */
#include"vk_cv.hpp"
#include<filesystem>
 
namespace mx {
 
    bool MXCapture::open(const std::string &filename) {
        return cap.open(filename);
    }
 
    bool MXCapture::open(int id, int mode) {
        if(mode == 0)
            mode = cv::CAP_V4L2;
        if(cap.open(id, mode)) {
            cap.set(cv::CAP_PROP_FOURCC, cv::VideoWriter::fourcc('M','J','P','G'));
            return true;
        }
        return false;
    }
    
    void MXCapture::close() {
        cap.release();
    }
 
    bool MXCapture::createImage(VKWindow *window, size_t width, size_t height, const std::string &vert, const std::string &frag) {
        sprite = window->createSprite(width, height, vert, frag);
        if(sprite) {
            sprite->enableExtendedUBO();
            sprite->rebuildPipeline();
            std::cout << ">> MXCapture - CreateSprite for - [OK]\n";
            return true;
        }
        std::cout << ">> MXCapture = CreateSprite for MXCapture failed.\n";
        return false;
    }
 
    bool MXCapture::reload(size_t width, size_t height, const std::string &vert, const std::string &frag) {
        if(sprite) {
            sprite->createEmptySprite(width, height,vert, frag);
            sprite->enableExtendedUBO();
            std::cout << ">> MXCapture shader reloaded \n   Vertex: [" << std::filesystem::path(vert).filename().string() << "] - Fragment: [" << std::filesystem::path(frag).filename().string() << "] \n";
            return true;
        }
        std::cout << ">> MXCapture failure to reload shader\n";
        return false;
    }
 
    bool MXCapture::read(cv::Mat &frame) {
        return cap.read(frame);
    }
 
    bool MXCapture::read() {
        if(!cap.read(frame)) {
            return false;
        }
        cv::Mat rgba;
        cv::cvtColor(frame, rgba, cv::COLOR_BGR2RGBA);
        sprite->updateTexture(rgba.ptr(), rgba.cols, rgba.rows, rgba.step);
        return true;
    }
 
    void MXCapture::draw(int x, int y, int width, int height) {
        if(sprite)
            sprite->drawSpriteRect(x, y, width, height);
    }
 
    void MXCapture::draw(int x, int y) {
        if(sprite)
            sprite->drawSprite(x, y);
    }
        
    void MXCapture::set(unsigned int option, double value) {
        cap.set(option, value);
    }
        
    double MXCapture::get(unsigned int option) {
        return cap.get(option);
    }
 
}