// Copyright 2016 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/vr/ui_element_renderer.h"

#include <math.h>
#include <algorithm>
#include <string>

#include "base/trace_event/trace_event.h"
#include "chrome/browser/vr/renderers/base_quad_renderer.h"
#include "chrome/browser/vr/renderers/base_renderer.h"
#include "chrome/browser/vr/renderers/external_textured_quad_renderer.h"
#include "chrome/browser/vr/renderers/radial_gradient_quad_renderer.h"
#include "chrome/browser/vr/renderers/texture_copy_renderer.h"
#include "chrome/browser/vr/renderers/textured_quad_renderer.h"
#include "chrome/browser/vr/renderers/transparent_quad_renderer.h"
#include "ui/gfx/geometry/point3_f.h"
#include "ui/gfx/geometry/vector3d_f.h"

namespace vr {

UiElementRenderer::UiElementRenderer() : UiElementRenderer(true) {}

UiElementRenderer::UiElementRenderer(bool use_gl) {
  if (!use_gl)
    return;
  Init();
  BaseQuadRenderer::CreateBuffers();
  TexturedQuadRenderer::CreateBuffers();
  Stars::Renderer::CreateBuffers();
}

UiElementRenderer::~UiElementRenderer() = default;

void UiElementRenderer::Init() {
  external_textured_quad_renderer_ =
      std::make_unique<ExternalTexturedQuadRenderer>();
  transparent_quad_renderer_ = std::make_unique<TransparentQuadRenderer>();
  textured_quad_renderer_ = std::make_unique<TexturedQuadRenderer>();
  radial_gradient_quad_renderer_ =
      std::make_unique<RadialGradientQuadRenderer>();
  texture_copy_renderer_ = std::make_unique<TextureCopyRenderer>();
  reticle_renderer_ = std::make_unique<Reticle::Renderer>();
  laser_renderer_ = std::make_unique<Laser::Renderer>();
  controller_renderer_ = std::make_unique<Controller::Renderer>();
  gradient_grid_renderer_ = std::make_unique<Grid::Renderer>();
  shadow_renderer_ = std::make_unique<Shadow::Renderer>();
  stars_renderer_ = std::make_unique<Stars::Renderer>();
  background_renderer_ = std::make_unique<Background::Renderer>();
}

void UiElementRenderer::DrawTexturedQuad(
    int texture_data_handle,
    int overlay_texture_data_handle,
    GlTextureLocation texture_location,
    const gfx::Transform& model_view_proj_matrix,
    const gfx::RectF& clip_rect,
    float opacity,
    const gfx::SizeF& element_size,
    float corner_radius,
    bool blend) {
  TRACE_EVENT0("gpu", "UiElementRenderer::DrawTexturedQuad");
  // TODO(vollick): handle drawing this degenerate situation crbug.com/768922
  if (corner_radius * 2.0 > element_size.width() ||
      corner_radius * 2.0 > element_size.height()) {
    return;
  }
  TexturedQuadRenderer* renderer =
      texture_location == kGlTextureLocationExternal
          ? external_textured_quad_renderer_.get()
          : textured_quad_renderer_.get();
  if (!texture_data_handle && !overlay_texture_data_handle) {
    // If we're blending, why are we even drawing a transparent quad?
    DCHECK(!blend);
    renderer = transparent_quad_renderer_.get();
  }
  FlushIfNecessary(renderer);
  renderer->AddQuad(texture_data_handle, overlay_texture_data_handle,
                    model_view_proj_matrix, clip_rect, opacity, element_size,
                    corner_radius, blend);
}

void UiElementRenderer::DrawRadialGradientQuad(
    const gfx::Transform& model_view_proj_matrix,
    const SkColor edge_color,
    const SkColor center_color,
    const gfx::RectF& clip_rect,
    float opacity,
    const gfx::SizeF& element_size,
    const CornerRadii& radii) {
  TRACE_EVENT0("gpu", "UiElementRenderer::DrawRadialGradientQuad");
  FlushIfNecessary(radial_gradient_quad_renderer_.get());
  radial_gradient_quad_renderer_->Draw(model_view_proj_matrix, edge_color,
                                       center_color, clip_rect, opacity,
                                       element_size, radii);
}

void UiElementRenderer::DrawGradientGridQuad(
    const gfx::Transform& model_view_proj_matrix,
    const SkColor grid_color,
    int gridline_count,
    float opacity) {
  TRACE_EVENT0("gpu", "UiElementRenderer::DrawGradientGridQuad");
  FlushIfNecessary(gradient_grid_renderer_.get());
  gradient_grid_renderer_->Draw(model_view_proj_matrix, grid_color,
                                gridline_count, opacity);
}

void UiElementRenderer::DrawController(
    float opacity,
    const gfx::Transform& model_view_proj_matrix) {
  TRACE_EVENT0("gpu", "UiElementRenderer::DrawController");
  FlushIfNecessary(controller_renderer_.get());
  controller_renderer_->Draw(opacity, model_view_proj_matrix);
}

void UiElementRenderer::DrawLaser(
    float opacity,
    const gfx::Transform& model_view_proj_matrix) {
  TRACE_EVENT0("gpu", "UiElementRenderer::DrawLaser");
  FlushIfNecessary(laser_renderer_.get());
  laser_renderer_->Draw(opacity, model_view_proj_matrix);
}

void UiElementRenderer::DrawReticle(
    float opacity,
    const gfx::Transform& model_view_proj_matrix) {
  TRACE_EVENT0("gpu", "UiElementRenderer::DrawReticle");
  FlushIfNecessary(reticle_renderer_.get());
  reticle_renderer_->Draw(opacity, model_view_proj_matrix);
}

void UiElementRenderer::DrawTextureCopy(int texture_data_handle,
                                        const float (&uv_transform)[16],
                                        float xborder,
                                        float yborder) {
  FlushIfNecessary(texture_copy_renderer_.get());
  texture_copy_renderer_->Draw(texture_data_handle, uv_transform, xborder,
                               yborder);
}

void UiElementRenderer::DrawShadow(const gfx::Transform& model_view_proj_matrix,
                                   const gfx::SizeF& element_size,
                                   float x_padding,
                                   float y_padding,
                                   float y_offset,
                                   SkColor color,
                                   float opacity,
                                   float corner_radius) {
  TRACE_EVENT0("gpu", "UiElementRenderer::DrawShadow");
  FlushIfNecessary(shadow_renderer_.get());
  shadow_renderer_->Draw(model_view_proj_matrix, element_size, x_padding,
                         y_padding, y_offset, color, opacity, corner_radius);
}

void UiElementRenderer::DrawStars(
    float t,
    const gfx::Transform& model_view_proj_matrix) {
  TRACE_EVENT0("gpu", "UiElementRenderer::DrawStars");
  FlushIfNecessary(stars_renderer_.get());
  stars_renderer_->Draw(t, model_view_proj_matrix);
}

void UiElementRenderer::DrawBackground(
    const gfx::Transform& model_view_proj_matrix,
    int texture_data_handle,
    int normal_gradient_texture_data_handle,
    int incognito_gradient_texture_data_handle,
    int fullscreen_gradient_texture_data_handle,
    float normal_factor,
    float incognito_factor,
    float fullscreen_factor) {
  TRACE_EVENT0("gpu", "UiElementRenderer::DrawBackground");
  FlushIfNecessary(background_renderer_.get());
  background_renderer_->Draw(model_view_proj_matrix, texture_data_handle,
                             normal_gradient_texture_data_handle,
                             incognito_gradient_texture_data_handle,
                             fullscreen_gradient_texture_data_handle,
                             normal_factor, incognito_factor,
                             fullscreen_factor);
}

void UiElementRenderer::Flush() {
  textured_quad_renderer_->Flush();
  external_textured_quad_renderer_->Flush();
  last_renderer_ = nullptr;
}

void UiElementRenderer::FlushIfNecessary(BaseRenderer* renderer) {
  if (last_renderer_ && renderer != last_renderer_) {
    last_renderer_->Flush();
  }
  last_renderer_ = renderer;
}

}  // namespace vr