modules/gapi/src/streaming/onevpl/engine/preproc/preproc_session.cpp - 4.x - Alx89/OpenCV

modules/gapi/src/streaming/onevpl/engine/preproc/preproc_session.cpp
85 строк3 KB
Sergey Ivanov
Merge pull request #21658 from sivanov-work:vpp_core_add_roi
05 мар 2022, 13:50
05 мар 2022, 13:5044c2c77
// This file is part of OpenCV project.
// It is subject to the license terms in the LICENSE file found in the top-level directory
// of this distribution and at http://opencv.org/license.html.
//
// Copyright (C) 2022 Intel Corporation

#ifdef HAVE_ONEVPL

#include <chrono>
#include <exception>

#include "streaming/onevpl/engine/preproc/preproc_session.hpp"
#include "streaming/onevpl/engine/preproc/preproc_engine.hpp"
#include "streaming/onevpl/accelerators/surface/surface.hpp"
#include "streaming/onevpl/utils.hpp"
#include "logger.hpp"

namespace cv {
namespace gapi {
namespace wip {
namespace onevpl {
VPPPreprocSession::VPPPreprocSession(mfxSession sess, const mfxVideoParam& vpp_out_param) :
    EngineSession(sess),
    mfx_vpp_out_param(vpp_out_param),
    processing_surface_ptr(),
    sync_in_queue(),
    vpp_out_queue(),
    preprocessed_frames_count()
{
}

VPPPreprocSession::~VPPPreprocSession() {
    GAPI_LOG_INFO(nullptr, "Close VPP for session: " << session);
    MFXVideoVPP_Close(session);
}

Data::Meta VPPPreprocSession::generate_frame_meta() {
    const auto now = std::chrono::system_clock::now();
    const auto dur = std::chrono::duration_cast<std::chrono::microseconds>
                (now.time_since_epoch());
    Data::Meta meta {
                        {cv::gapi::streaming::meta_tag::timestamp, int64_t{dur.count()} },
                        {cv::gapi::streaming::meta_tag::seq_id, int64_t{preprocessed_frames_count++}}
                    };
    return meta;
}

void VPPPreprocSession::swap_surface(VPPPreprocEngine& engine) {
    VPLAccelerationPolicy* acceleration_policy = engine.get_accel();
    GAPI_Assert(acceleration_policy && "Empty acceleration_policy");
    request_free_surface(session, vpp_pool_id, *acceleration_policy,
                         processing_surface_ptr, true);
}

void VPPPreprocSession::init_surface_pool(VPLAccelerationPolicy::pool_key_t key) {
    GAPI_Assert(key && "Init preproc pull with empty key");
    vpp_pool_id = key;
}

const mfxFrameInfo& VPPPreprocSession::get_video_param() const {
    return mfx_vpp_out_param.vpp.Out;
}

VPPPreprocSession::outgoing_task::outgoing_task(mfxSyncPoint acquired_sync_handle,
                                                mfxFrameSurface1* acquired_surface_ptr,
                                                VPPPreprocSession::incoming_task &&in) :
    sync_handle(acquired_sync_handle),
    vpp_surface_ptr(acquired_surface_ptr),
    original_surface_ptr(in.decoded_surface_ptr),
    original_frame_info(std::move(in.decoded_frame_info)),
    original_frame(in.decoded_frame_copy) {
}

void VPPPreprocSession::outgoing_task::release_frame() {
    // restore initial surface params
    memcpy(&(original_surface_ptr->Info),
           &original_frame_info, sizeof(Surface::info_t));
    // release references on frame adapter
    original_frame = cv::MediaFrame();
}
} // namespace onevpl
} // namespace wip
} // namespace gapi
} // namespace cv
#endif // HAVE_ONEVPL