blob: 65cd9468a3f1656c56045f912e5638aeb6ae4765 [file] [log] [blame]
// Copyright 2021 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <fuchsia/sysmem/cpp/fidl.h>
#include <lib/async-loop/cpp/loop.h>
#include <lib/async-loop/default.h>
#include <lib/async/cpp/task.h>
#include <mutex>
#include <queue>
namespace camera {
class CapturedFrame {
CapturedFrame(std::unique_ptr<uint8_t[]> buffer, uint32_t buffer_size, std::string path_name)
: buffer_(std::move(buffer)), buffer_size_(buffer_size), path_name_(std::move(path_name)) {}
~CapturedFrame() = default;
std::unique_ptr<uint8_t[]> take_buffer() { return std::move(buffer_); }
uint32_t buffer_size() const { return buffer_size_; }
std::string take_path_name() { return std::move(path_name_); }
std::unique_ptr<uint8_t[]> buffer_;
uint32_t buffer_size_;
std::string path_name_;
class FrameCapture {
~FrameCapture() = default;
// Must initialize before attempting to capture.
zx_status_t Initialize();
// Caller must call each time with the VMO of interest, along with the image width, height, stride
// and image size. The image format is assumed to be NV12.
void Capture(const zx::vmo& vmo, uint32_t coded_width, uint32_t coded_height,
uint32_t coded_stride, uint32_t coded_image_size);
async::Loop archive_loop_;
// Do the actual work of archiving the captured frame by writing to the file system.
void Archive(std::unique_ptr<uint8_t[]> buffer, uint32_t buffer_size, std::string path_name);
// Dispatch work whenever work is available.
void ArchiveTask();
// Set of captured frames.
// "captured_frame_count" goes from 0 to N-1 (whenever a captured frame is passed in).
// "archived_frame_count" goes from 0 to N-1 (whenever a frame is about to be written to a file).
// This is really just an explicit queue with a handshake to ensure previous archive operation
// is done.
uint32_t captured_frame_count_ = 0;
uint32_t archived_frame_count_ = 0;
std::mutex captured_frames_lock_;
std::queue<std::unique_ptr<CapturedFrame>> captured_frames_;
} // namespace camera