blob: bbc264932671fc1daa6ee7c5198c8fa7b07fc415 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
#include "playback.h"
#include "labeller.h"
#include <opencv2/opencv.hpp>
#include <vector>
#include <stdexcept>
#include <iostream>
#include <thread>
namespace frontend {
struct playback_impl {
std::vector<cv::Mat> images;
int fps;
double reportedDuration;
std::size_t frameNum = 0;
};
playback::playback(const std::filesystem::path& video, std::size_t frameCap) {
data = std::shared_ptr<playback_impl>(new playback_impl);
cv::VideoCapture cap(video);
if(!cap.isOpened()) {
throw std::runtime_error("Error loading video " + video.string());
}
data->fps = cap.get(cv::CAP_PROP_FPS);
while(frameCap == 0 or data->images.size() < frameCap) {
cv::Mat frame;
cap >> frame;
if(frame.empty()) break;
data->images.push_back(frame);
}
data->reportedDuration = cap.get(cv::CAP_PROP_POS_MSEC) / 1000.0;
std::cout << "Reported duration: " << data->reportedDuration << " seconds" << std::endl;
cap.release();
}
void playback::display(const std::string& windowName, const backend::rect &r) const {
cv::Mat curr(data->images[data->frameNum]);
if(r.nonzero()) {
// Update with the rectangle
cv::rectangle(curr, cv::Point(r.x1, r.y1), cv::Point(r.x2, r.y2), cv::Scalar(255, 0, 0), 2, cv::LINE_8);
}
cv::imshow(windowName, data->images[data->frameNum]);
}
bool playback::seekFrame(std::size_t frameNum) {
if(frameNum >= data->images.size()) {
return false;
}
data->frameNum = frameNum;
return true;
}
std::size_t playback::getFrame() const {
return data->frameNum;
}
bool playback::seekTime(double time) {
return seekFrame(std::size_t(time * data->fps));
}
double playback::getTime() const {
return getFrame() / (double) data->fps;
}
void playback::interFrameSleep() const {
std::this_thread::sleep_for(std::chrono::milliseconds(1000/data->fps));
}
std::size_t playback::getMaxFrame() const {
return data->images.size() - 1;
}
double playback::getMaxTime() const {
return getMaxFrame() / (double) data->fps;
}
}
|
