src/ui/testing/util/flatland_test_view.cc

// Copyright 2022 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 "src/ui/testing/util/flatland_test_view.h"

#include <lib/syslog/cpp/macros.h>
#include <lib/ui/scenic/cpp/view_creation_tokens.h>
#include <lib/ui/scenic/cpp/view_identity.h>
#include <zircon/status.h>

namespace ui_testing {

void FlatlandTestView::CreateView2(fuchsia::ui::app::CreateView2Args args) {
  flatland_ = mock_handles_->svc().Connect<fuchsia::ui::composition::Flatland>();
  flatland_.set_error_handler([](zx_status_t status) {
    FX_LOGS(ERROR) << "Error from fuchsia::ui::composition::Flatland: "
                   << zx_status_get_string(status);
  });

  // Set up parent watcher to retrieve layout info.
  parent_watcher_.set_error_handler([](zx_status_t status) {
    FX_LOGS(ERROR) << "Error from fuchsia::ui::composition::ParentViewportWatcher: "
                   << zx_status_get_string(status);
  });
  // Create a11y view's ViewRef.
  auto view_identity = scenic::NewViewIdentityOnCreation();
  view_ref_ = fidl::Clone(view_identity.view_ref);

  flatland_->CreateView2(std::move(*args.mutable_view_creation_token()), std::move(view_identity),
                         /* view_bound_protocols = */ {}, parent_watcher_.NewRequest());

  flatland_->CreateTransform(fuchsia::ui::composition::TransformId({.value = kRootTransformId}));
  flatland_->SetRootTransform(fuchsia::ui::composition::TransformId({.value = kRootTransformId}));

  parent_watcher_->GetLayout([this](fuchsia::ui::composition::LayoutInfo layout_info) {
    layout_info_ = std::move(layout_info);

    DrawContent();
  });
}

uint32_t FlatlandTestView::width() {
  FX_CHECK(layout_info_);
  return layout_info_->logical_size().width;
}

uint32_t FlatlandTestView::height() {
  FX_CHECK(layout_info_);
  return layout_info_->logical_size().height;
}

void FlatlandTestView::DrawRectangle(int32_t x, int32_t y, int32_t z, uint32_t width,
                                     uint32_t height, uint8_t red, uint8_t green, uint8_t blue,
                                     uint8_t alpha) {
  const fuchsia::ui::composition::ContentId kFilledRectId = {next_resource_id_++};
  const fuchsia::ui::composition::TransformId kTransformId = {next_resource_id_++};

  float red_f = static_cast<float>(red) / 255.f;
  float green_f = static_cast<float>(green) / 255.f;
  float blue_f = static_cast<float>(blue) / 255.f;
  float alpha_f = static_cast<float>(alpha) / 255.f;

  flatland_->CreateFilledRect(kFilledRectId);
  flatland_->SetSolidFill(kFilledRectId, {red_f, green_f, blue_f, alpha_f}, {width, height});

  // Associate the rect with a transform.
  flatland_->CreateTransform(kTransformId);
  flatland_->SetContent(kTransformId, kFilledRectId);
  flatland_->SetTranslation(kTransformId, {x, y});

  // Attach the transform to the view.
  flatland_->AddChild(fuchsia::ui::composition::TransformId{kRootTransformId}, kTransformId);
}

void FlatlandTestView::PresentChanges() {
  flatland_->Present(fuchsia::ui::composition::PresentArgs{});
}

}  // namespace ui_testing