zircon/system/dev/virtual_camera/virtual_camera_device.h

// Copyright 2019 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.

#pragma once

#include <map>

#include "virtual_camera_stream.h"

#include <ddktl/device-internal.h>
#include <ddktl/device.h>
#include <ddktl/protocol/empty-protocol.h>
#include <fuchsia/hardware/camera/c/fidl.h>
#include <fuchsia/camera/common/c/fidl.h>
#include <lib/async-loop/cpp/loop.h>
#include <lib/async/default.h>
#include <lib/async/wait.h>
#include <lib/fidl-utils/bind.h>
#include <lib/fzl/vmo-pool.h>

namespace virtual_camera {

class VirtualCameraDevice;
using VirtualCameraDeviceType =
    ddk::Device<VirtualCameraDevice, ddk::Unbindable, ddk::Messageable>;

class VirtualCameraDevice : public VirtualCameraDeviceType,
                            public ddk::EmptyProtocol<ZX_PROTOCOL_CAMERA> {
public:
    DISALLOW_COPY_ASSIGN_AND_MOVE(VirtualCameraDevice);

    static zx_status_t Create(void* ctx, zx_device_t* parent);

    VirtualCameraDevice(zx_device_t* parent)
        : VirtualCameraDeviceType(parent), count_(0) {
    }

    ~VirtualCameraDevice() {}

    // DDK device implementation
    void DdkRelease();
    void DdkUnbind();
    zx_status_t DdkMessage(fidl_msg_t* msg, fidl_txn_t* txn);

    void RemoveStream(uint64_t);


private:
    // DDK Helper Functions.
    zx_status_t GetFormats(uint32_t index, fidl_txn_t* txn);
    zx_status_t CreateStream(
        const fuchsia_sysmem_BufferCollectionInfo* buffer_collection_info,
        const fuchsia_camera_common_FrameRate* rate, zx_handle_t stream,
        zx_handle_t stream_token);
    zx_status_t GetDeviceInfo(fidl_txn_t* txn);

    static constexpr fuchsia_hardware_camera_ControlV2_ops_t control_ops = {
        .GetFormats = fidl::Binder<VirtualCameraDevice>::BindMember<
            &VirtualCameraDevice::GetFormats>,
        .CreateStream = fidl::Binder<VirtualCameraDevice>::BindMember<
            &VirtualCameraDevice::CreateStream>,
        .GetDeviceInfo = fidl::Binder<VirtualCameraDevice>::BindMember<
            &VirtualCameraDevice::GetDeviceInfo>,
    };

    std::map<uint64_t, fbl::unique_ptr<VirtualCameraStream>> streams_;

    // Variable to keep track of the number of streams created.
    uint64_t count_;

};

} // namespace virtual_camera