xref: /aosp_15_r20/external/armnn/samples/common/src/CVUtils/CvVideoFileWriter.cpp (revision 89c4ff92f2867872bb9e2354d150bf0c8c502810)

//
// 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