src/camera/examples/camera_display/demo_view.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.

#ifndef SRC_CAMERA_EXAMPLES_DEMO_DEMO_VIEW_H_
#define SRC_CAMERA_EXAMPLES_DEMO_DEMO_VIEW_H_

#include <fuchsia/camera2/cpp/fidl.h>
#include <lib/async-loop/cpp/loop.h>
#include <lib/async/cpp/wait.h>
#include <stream_provider.h>
#include <text_node.h>

#include <map>
#include <random>

#include <src/lib/ui/base_view/base_view.h>
#include <trace-provider/provider.h>
#include <trace/event.h>

#include "lib/ui/scenic/cpp/session.h"
#include "src/camera/lib/stream_utils/image_io_util.h"

namespace camera {

// Draws a scenic scene containing a single rectangle with an image pipe material,
// constructed with buffers populated by a stream provider.
class DemoView : public scenic::BaseView {
 public:
  explicit DemoView(scenic::ViewContext context, async::Loop* loop, bool chaos, bool image_io);
  ~DemoView() override;
  static std::unique_ptr<DemoView> Create(scenic::ViewContext context, async::Loop* loop,
                                          bool chaos, bool image_io);

 private:
  // |scenic::BaseView|
  void OnSceneInvalidated(fuchsia::images::PresentationInfo presentation_info) override;
  void OnInputEvent(fuchsia::ui::input::InputEvent event) override;

  // |scenic::SessionListener|
  void OnScenicError(std::string error) override;

  void OnFrameAvailable(uint32_t index, fuchsia::camera2::FrameAvailableInfo info);

  void SleepIfChaos();

  struct ImagePipeProperties {
    explicit ImagePipeProperties(scenic::Session* session) : node(session) {}
    fuchsia::camera2::StreamPtr stream;
    scenic::ShapeNode node;
    fuchsia::images::ImagePipePtr image_pipe;
    std::map<uint32_t, uint32_t> image_ids;
    float shape_width;
    float shape_height;
    bool should_rotate;
    std::map<uint32_t, std::pair<std::unique_ptr<async::Wait>, zx::event>> waiters;
    std::unique_ptr<camera::ImageIOUtil> image_io_util;
  };

  async::Loop* loop_;
  bool chaos_;
  std::mt19937 chaos_gen_;
  std::binomial_distribution<uint32_t> chaos_dist_;
  std::unique_ptr<StreamProvider> stream_provider_;
  std::vector<ImagePipeProperties> image_pipe_properties_;
  float total_width_;
  float max_height_;
  TextNode text_node_;
  bool image_io_;
  trace::TraceProviderWithFdio trace_provider_;
};

}  // namespace camera

#endif  // SRC_CAMERA_EXAMPLES_DEMO_DEMO_VIEW_H_