mirror of
https://github.com/opencv/opencv.git
synced 2024-12-14 00:31:07 +08:00
a92cba8484
[G-API] Implement G-API pipeline modeling tool * Implement G-API pipeline builder tool * Remove whitespaces from config * Remove unused unittest stuff * Fix comments to review * Fix MAC warning * Rename to pipeline_modeling_tool * Move to opencv apps * Move config to gapi/samples/data * gapi: samples sources are installed automatically
67 lines
2.1 KiB
C++
67 lines
2.1 KiB
C++
#ifndef OPENCV_GAPI_PIPELINE_MODELING_TOOL_DUMMY_SOURCE_HPP
|
|
#define OPENCV_GAPI_PIPELINE_MODELING_TOOL_DUMMY_SOURCE_HPP
|
|
|
|
#include <thread>
|
|
#include <memory>
|
|
#include <chrono>
|
|
|
|
#include <opencv2/gapi.hpp>
|
|
#include <opencv2/gapi/streaming/cap.hpp> // cv::gapi::wip::IStreamSource
|
|
|
|
#include "utils.hpp"
|
|
|
|
class DummySource final: public cv::gapi::wip::IStreamSource {
|
|
public:
|
|
using Ptr = std::shared_ptr<DummySource>;
|
|
DummySource(const double latency,
|
|
const OutputDescr& output);
|
|
bool pull(cv::gapi::wip::Data& data) override;
|
|
cv::GMetaArg descr_of() const override;
|
|
|
|
private:
|
|
double m_latency;
|
|
cv::Mat m_mat;
|
|
using TimePoint =
|
|
std::chrono::time_point<std::chrono::high_resolution_clock>;
|
|
cv::optional<TimePoint> m_prev_pull_tp;
|
|
};
|
|
|
|
DummySource::DummySource(const double latency,
|
|
const OutputDescr& output)
|
|
: m_latency(latency), m_mat(output.dims, output.precision) {
|
|
if (output.dims.size() == 1) {
|
|
//FIXME: Well-known 1D mat WA
|
|
m_mat.dims = 1;
|
|
}
|
|
utils::generateRandom(m_mat);
|
|
}
|
|
|
|
bool DummySource::pull(cv::gapi::wip::Data& data) {
|
|
using namespace std::chrono;
|
|
using namespace cv::gapi::streaming;
|
|
// NB: In case it's the first pull.
|
|
if (!m_prev_pull_tp) {
|
|
m_prev_pull_tp = cv::util::make_optional(high_resolution_clock::now());
|
|
}
|
|
// NB: Just increase reference counter not to release mat memory
|
|
// after assigning it to the data.
|
|
cv::Mat mat = m_mat;
|
|
auto end = high_resolution_clock::now();
|
|
auto elapsed =
|
|
duration_cast<duration<double, std::milli>>(end - *m_prev_pull_tp).count();
|
|
auto delta = m_latency - elapsed;
|
|
if (delta > 0) {
|
|
utils::sleep(delta);
|
|
}
|
|
data.meta[meta_tag::timestamp] = int64_t{utils::timestamp<milliseconds>()};
|
|
data = mat;
|
|
m_prev_pull_tp = cv::util::make_optional(high_resolution_clock::now());
|
|
return true;
|
|
}
|
|
|
|
cv::GMetaArg DummySource::descr_of() const {
|
|
return cv::GMetaArg{cv::descr_of(m_mat)};
|
|
}
|
|
|
|
#endif // OPENCV_GAPI_PIPELINE_MODELING_TOOL_DUMMY_SOURCE_HPP
|