src/ui/lib/escher/paper/paper_render_funcs.cc

// Copyright 2018 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/lib/escher/paper/paper_render_funcs.h"

#include <lib/syslog/cpp/macros.h>

#include "src/ui/lib/escher/paper/paper_render_queue_context.h"
#include "src/ui/lib/escher/paper/paper_shader_structs.h"
#include "src/ui/lib/escher/renderer/frame.h"
#include "src/ui/lib/escher/renderer/render_queue_item.h"
#include "src/ui/lib/escher/shape/mesh.h"
#include "src/ui/lib/escher/util/trace_macros.h"
#include "src/ui/lib/escher/vk/texture.h"

namespace escher {

void PaperRenderFuncs::MeshData::Bind(CommandBuffer* cb) const {
  TRACE_DURATION("gfx", "PaperRenderFuncs::MeshData::Bind");
  index_binding.Bind(cb);
  for (uint32_t i = 0; i < vertex_binding_count; ++i) {
    vertex_bindings[i].Bind(cb);
  }
  for (uint32_t i = 0; i < vertex_attribute_count; ++i) {
    vertex_attributes[i].Bind(cb);
  }
  for (uint32_t i = 0; i < uniform_binding_count; ++i) {
    uniform_bindings[i].Bind(cb);
  }
  cb->BindTexture(1, 1, texture);
}

void PaperRenderFuncs::RenderMesh(CommandBuffer* cb, const RenderQueueContext* context_in,
                                  const RenderQueueItem* items, uint32_t instance_count) {
  TRACE_DURATION("gfx", "PaperRenderFuncs::RenderMesh");
  FX_DCHECK(cb && items && instance_count > 0);
  FX_DCHECK(context_in);
  auto* context = static_cast<const PaperRenderQueueContext*>(context_in);
  auto* mesh_data = static_cast<const MeshData*>(items[0].object_data);

  // Set up per-object state.
  mesh_data->Bind(cb);

  // TODO(fxbug.dev/7249): this assumes that all meshes in this render-queue pass are
  // drawn exactly the same way.  We will need something better soon.
  const SamplerPtr& sampler =
      mesh_data->texture->sampler()->is_immutable() ? mesh_data->texture->sampler() : nullptr;
  const PaperShaderListSelector shader_selector = context->shader_selector();

  // For each instance, set up per-instance state and draw.
  for (uint32_t i = 0; i < instance_count; ++i) {
    FX_DCHECK(items[i].object_data == mesh_data);

    const MeshDrawData* instance_data = static_cast<const MeshDrawData*>(items[i].instance_data);

    cb->SetShaderProgram(instance_data->shader_list.get_shader(shader_selector), sampler);

    auto& b = instance_data->object_properties;
    cb->BindUniformBuffer(b.descriptor_set_index, b.binding_index, b.buffer, b.offset, b.size);
    cb->DrawIndexed(instance_data->num_indices);
  }
}

PaperRenderFuncs::MeshData* PaperRenderFuncs::NewMeshData(const FramePtr& frame, Mesh* mesh,
                                                          const TexturePtr& texture) {
  TRACE_DURATION("gfx", "PaperRenderFuncs::NewMeshData");
  FX_DCHECK(mesh);
  FX_DCHECK(texture);
  auto& mesh_spec = mesh->spec();

  // TODO(fxbug.dev/7194): avoid reaching in to impl::CommandBuffer for keep-alive.
  frame->cmds()->KeepAlive(mesh);
  frame->cmds()->KeepAlive(texture.get());

  BlockAllocator* allocator = frame->host_allocator();
  auto* obj = allocator->Allocate<MeshData>();

  obj->index_binding.index_buffer = mesh->vk_index_buffer();
  obj->index_binding.index_type = MeshSpec::IndexTypeEnum;
  obj->index_binding.index_buffer_offset = mesh->index_buffer_offset();

  // Set up vertex buffer bindings.
  obj->vertex_binding_count = mesh_spec.vertex_buffer_count();
  obj->vertex_bindings = allocator->AllocateMany<VertexBinding>(obj->vertex_binding_count);

  {
    uint32_t binding_count = 0;
    for (uint32_t i = 0; i < VulkanLimits::kNumVertexBuffers; ++i) {
      if (auto& attribute_buffer = mesh->attribute_buffer(i)) {
        // TODO(fxbug.dev/7194): avoid reaching in to impl::CommandBuffer for keep-alive.
        frame->cmds()->KeepAlive(attribute_buffer.buffer);

        obj->vertex_bindings[binding_count++] =
            VertexBinding{.binding_index = i,
                          .buffer = attribute_buffer.buffer->vk(),
                          .offset = attribute_buffer.offset,
                          .stride = attribute_buffer.stride};
      }
    }
    FX_DCHECK(binding_count == obj->vertex_binding_count);
  }

  // Set up vertex attribute bindings.
  const uint32_t total_attribute_count = mesh_spec.total_attribute_count();
  obj->vertex_attribute_count = total_attribute_count;
  obj->vertex_attributes = NewVertexAttributeBindings(kMeshAttributeBindingLocations, allocator,
                                                      mesh_spec, total_attribute_count);
  obj->uniform_binding_count = 0;
  obj->texture = texture.get();

  return obj;
}

PaperRenderFuncs::MeshDrawData* PaperRenderFuncs::NewMeshDrawData(
    const FramePtr& frame, const mat4& transform, const vec4& color, float gamma_power,
    const PaperShaderList& shader_list, uint32_t num_indices) {
  MeshDrawData* draw_data = frame->Allocate<MeshDrawData>();

  auto writable_binding = NewPaperShaderUniformBinding<PaperShaderMeshInstance>(frame);
  writable_binding.first->model_transform = transform;
  writable_binding.first->color = color;
  writable_binding.first->gamma_power = gamma_power;
  // TODO(fxbug.dev/7243): populate field for vertex-shader clip-planes.

  draw_data->object_properties = writable_binding.second;
  draw_data->shader_list = shader_list;
  draw_data->num_indices = num_indices;

  return draw_data;
}

}  // namespace escher