blob: b76630049a1d9f5d10b4c6fcf44206a830c36f42 (
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
|
//
// Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
#include "CvVideoFileWriter.hpp"
namespace common
{
void CvVideoFileWriter::Init(const std::string& outputVideo, int encoding, double fps, int width, int height)
{
m_ready = m_cvWriter.open(outputVideo, cv::CAP_FFMPEG,
encoding,
fps,
cv::Size(width, height), true);
}
void CvVideoFileWriter::WriteFrame(std::shared_ptr<cv::Mat>& frame)
{
if(m_cvWriter.isOpened())
{
cv::cvtColor(*frame, *frame, cv::COLOR_RGB2BGR);
m_cvWriter.write(*frame);
}
}
bool CvVideoFileWriter::IsReady() const
{
return m_ready;
}
void CvVideoFileWriter::Close()
{
m_cvWriter.release();
}
}// namespace common
|
