Многочастная граница у бустового зверя - PullRequest
Новая
       39

Многочастная граница у бустового зверя

1 голос
/ 24 октября 2019

Я пытаюсь передавать кадры из видео в виде mjpegs на локально размещенный сервер. Вот мой код: 

#include <iostream>
#include <string>
#include <memory>
#include <thread>
#include <cstdlib>

#include <opencv2/opencv.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include <opencv2/core/core.hpp>

#include <boost/beast/core.hpp>
#include <boost/beast/http.hpp>
#include <boost/beast/version.hpp>
#include <boost/asio/ip/tcp.hpp>
#include <boost/config.hpp>

using boost::asio::ip::tcp;
namespace http = boost::beast::http;

cv::Mat frame;
std::vector<uchar> buffer;

int main(){
    try{
        boost::asio::io_context io_service;
        tcp::acceptor acceptor(io_service, tcp::endpoint(tcp::v4(), 1112));
        boost::system::error_code err;
        tcp::socket socket(io_service);
        acceptor.accept(socket);

        boost::beast::flat_buffer request_buffer;

        cv::VideoCapture cap("zz.mp4");

        // http::request<http::string_body> req;
        // http::read(socket, request_buffer, req, err);
        // if(err){
        //     std::cerr << "read: " << err.message() << "\n";
        // } 

        while(cap.isOpened()){
            cap >> frame;
            cv::imencode(".jpg", frame, buffer, std::vector<int> {cv::IMWRITE_JPEG_QUALITY, 95});

            auto const size = buffer.size();

            http::request<http::string_body> req;
            http::read(socket, request_buffer, req, err);
            if(err){
                std::cerr << "read: " << err.message() << "\n";
                break;
            } 

            http::response<http::empty_body> res{http::status::ok, req.version()};
            res.set(http::field::server, BOOST_BEAST_VERSION_STRING);
            // res.set(http::field::content_type, "multipart/x-mixed-replace; boundary=frame\r\n\r\n--frame\n");
            res.set(http::field::content_type, "multipart/x-mixed-replace; boundary=frame");
            res.content_length(size);
            http::response_serializer<http::empty_body> sr{res};
            http::write_header(socket, sr);
            while(true){
                http::response<http::vector_body<unsigned char>> res{std::piecewise_construct,
                            std::make_tuple(std::move(buffer)),
                            std::make_tuple(http::status::ok, req.version())};
                res.set(http::field::server, BOOST_BEAST_VERSION_STRING);
                res.set(http::field::content_type, "image/jpeg");
                res.content_length(size);
                res.keep_alive(req.keep_alive());
                http::write(socket, res, err);
            }
        }
    }
    catch(std::exception& e)
    {
        std::cerr << e.what() << std::endl;
        return EXIT_FAILURE;
    }
    return 0;
}

Я хочу, чтобы каждый кадр отображался без обновления страницы, чтобы каждый раз отправлять новый запрос. Я пришел к выводу, что мне нужно отправить заголовок с типом содержимого multipart/x-mixed-replace перед отправкой каждого буфера изображения с mime-типом image / jpeg. Но я не могу понять, как установить конечную границу (очевидно, для multipart) в моем ответе в boost-beast. Или, может быть, мои проблемы лежат в другом месте. Любая помощь / совет будет оценен. Ура! * * 1005

1 Ответ

1 голос
/ 24 октября 2019

Вот ответ. Нам нужно установить границу, используя boost::beast::http::field::body

#include <iostream>
#include <string>
#include <memory>
#include <thread>
#include <cstdlib>

#include <opencv2/opencv.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include <opencv2/core/core.hpp>

#include <boost/beast/core.hpp>
#include <boost/beast/http.hpp>
#include <boost/beast/version.hpp>
#include <boost/asio/ip/tcp.hpp>
#include <boost/config.hpp>

using boost::asio::ip::tcp;
namespace http = boost::beast::http;

cv::Mat frame;
std::vector<uchar> buffer;
cv::Mat grayframe;

int main(){
    try{
        boost::asio::io_context io_service;
        tcp::acceptor acceptor(io_service, tcp::endpoint(tcp::v4(), 1112));
        boost::system::error_code err;
        tcp::socket socket(io_service);
        acceptor.accept(socket);

        boost::beast::flat_buffer request_buffer;

        cv::VideoCapture cap("zz.mp4");

        http::request<http::string_body> req;
        http::read(socket, request_buffer, req, err);
        if(err){
            std::cerr << "read: " << err.message() << "\n";
        } 

        http::response<http::empty_body> res{http::status::ok, req.version()};
        res.set(http::field::server, BOOST_BEAST_VERSION_STRING);
        res.set(http::field::content_type, "multipart/x-mixed-replace; boundary=frame");
        res.keep_alive();
        http::response_serializer<http::empty_body> sr{res};
        http::write_header(socket, sr);

        while(cap.isOpened()){
            cap >> frame;
            cv::cvtColor(frame, grayframe, CV_BGR2GRAY);
            cv::imencode(".jpg", grayframe, buffer, std::vector<int> {cv::IMWRITE_JPEG_QUALITY, 95});

            auto const size = buffer.size();

            http::response<http::vector_body<unsigned char>> res{std::piecewise_construct,
                        std::make_tuple(std::move(buffer)),
                        std::make_tuple(http::status::ok, req.version())};
            res.set(http::field::body, "--frame");
            res.set(http::field::server, BOOST_BEAST_VERSION_STRING);
            res.set(http::field::content_type, "image/jpeg");
            res.content_length(size);
            res.keep_alive(req.keep_alive());
            http::write(socket, res, err);
        }
    }
    catch(std::exception& e)
    {
        std::cerr << e.what() << std::endl;
        return EXIT_FAILURE;
    }
    return 0;
}
...