frame_with_timestamp_recorder_impl.hpp

// frame_with_timestamp_recorder_impl.hpp
// Created by greg on 2/19/25.
// Copyright (C) 2025 Presage Security, Inc.
//
// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 3 of the License, or (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with this program; if not, write to the Free Software Foundation,
// Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 
// === standard library includes (if any) ===
// === third-party includes (if any) ===
// === local includes (if any) ===
#include "frame_with_timestamp_recorder.hpp"
 
#pragma once
namespace presage::smartspectra::grpc_bindings {
 
 
template<container::settings::IntegrationMode TIntegrationMode>
FrameWithTimestampRecorder<TIntegrationMode>::FrameWithTimestampRecorder(
    std::shared_ptr<BackgroundContainer> container,
    video_source::InputTransformMode input_transform_mode
) :
    container(std::move(container)),
    input_transformer{input_transform_mode} {
    StartRead(&this->frame_with_timestamp);
};
 
template<container::settings::IntegrationMode TIntegrationMode>
void FrameWithTimestampRecorder<TIntegrationMode>::SwapContainer(std::shared_ptr<BackgroundContainer> new_container) {
    this->container.reset();
    this->container = std::move(new_container);
}
 
inline cv::Mat BuildOpenCVMatFromProtoBuf(const smartspectra::Mat& mat_protobuf) {
    cv::Mat mat_cv(mat_protobuf.rows(), mat_protobuf.cols(), mat_protobuf.element_type());
    size_t data_bytesize = mat_cv.rows * mat_cv.cols * mat_cv.elemSize();
    auto data_protobuf = const_cast<char*>(mat_protobuf.data().data());
    std::copy(reinterpret_cast<unsigned char*>(data_protobuf),
              reinterpret_cast<unsigned char*>(data_protobuf + data_bytesize),
              mat_cv.data);
    return mat_cv;
}
 
template<container::settings::IntegrationMode TIntegrationMode>
void FrameWithTimestampRecorder<TIntegrationMode>::OnReadDone(bool ok) {
    if (ok) {
        if (this->container == nullptr || !this->container->GraphIsRunning()) {
            Finish(grpc::Status(grpc::StatusCode::FAILED_PRECONDITION,
                                "Physiology Preprocessing is not running, unable to add frame."));
        } else {
            cv::Mat frame = BuildOpenCVMatFromProtoBuf(frame_with_timestamp.frame());
            frame = this->input_transformer.apply(frame);
            rpc::StatusAdapter status_adapter = container
                ->AddFrameWithTimestamp(frame, frame_with_timestamp.timestamp());
            if (!status_adapter.ok()) {
                Finish(status_adapter);
            }
            StartRead(&this->frame_with_timestamp);
        }
    } else {
        Finish(grpc::Status(grpc::StatusCode::INTERNAL, "Unexpected failure reading frame with timestamp."));
    }
}
 
} // namespace presage::smartspectra::grpc_bindings