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

#include "topaz/tests/benchmarks/gfx_benchmarks.h"

#include "src/lib/fxl/logging.h"

void AddGraphicsBenchmarks(benchmarking::BenchmarksRunner* benchmarks_runner) {
  FXL_DCHECK(benchmarks_runner != nullptr);

  struct Param {
    std::string benchmark_name;
    std::string command;
    std::optional<std::string> flutter_app_name;
    std::string renderer_params;
  };

  constexpr char kImageGridFlutterX3Command[] =
      "present_view fuchsia-pkg://fuchsia.com/tile_view#meta/tile_view.cmx "
      "image_grid_flutter image_grid_flutter image_grid_flutter";
  constexpr char kChoreographyCommand[] =
      "basemgr --test --enable_presenter "
      "--account_provider=dev_token_manager --base_shell=dev_base_shell "
      "--base_shell_args=--test_timeout_ms=60000 "
      "--session_shell=dev_session_shell "
      "--session_shell_args=--root_module=choreography "
      "--story_shell=mondrian";

  // clang-format off
  std::vector<Param> params = {
    //
    // image_grid_flutter
    //
    {"fuchsia.scenic.image_grid_flutter_noclipping_noshadows", "present_view image_grid_flutter", "image_grid_flutter", "--unshadowed --clipping_disabled"},
    {"fuchsia.scenic.image_grid_flutter_noshadows", "present_view image_grid_flutter", "image_grid_flutter", "--unshadowed --clipping_enabled"},
    {"fuchsia.scenic.image_grid_flutter_ssdo", "present_view image_grid_flutter", "image_grid_flutter", "--screen_space_shadows --clipping_enabled"},
    {"fuchsia.scenic.image_grid_flutter_shadow_map", "present_view image_grid_flutter", "image_grid_flutter", "--shadow_map --clipping_enabled"},
    {"fuchsia.scenic.image_grid_flutter_moment_shadow_map", "present_view image_grid_flutter", "image_grid_flutter", "--moment_shadow_map --clipping_enabled"},

    //
    // image_grid_flutter x3
    //
    // TODO: Support tracking multiple flutter apps of the same name in
    // process_scenic_trace.
    {"fuchsia.scenic.image_grid_flutter_x3_noclipping_noshadows", kImageGridFlutterX3Command, {}, "--unshadowed --clipping_disabled",},
    {"fuchsia.scenic.image_grid_flutter_x3_noshadows", kImageGridFlutterX3Command, {}, "--unshadowed --clipping_enabled",},
    {"fuchsia.scenic.image_grid_flutter_x3_ssdo", kImageGridFlutterX3Command, {}, "--screen_space_shadows --clipping_enabled",},
    {"fuchsia.scenic.image_grid_flutter_x3_shadow_map", kImageGridFlutterX3Command, {}, "--shadow_map --clipping_enabled",},
    {"fuchsia.scenic.image_grid_flutter_x3_moment_shadow_map", kImageGridFlutterX3Command, {}, "--moment_shadow_map --clipping_enabled",},

    //
    // choreography
    //
    {"fuchsia.scenic.choreography_noclipping_noshadows", kChoreographyCommand, "dashboard", "--unshadowed --clipping_disabled",},
    {"fuchsia.scenic.choreography_noshadows", kChoreographyCommand, "dashboard", "--unshadowed --clipping_enabled",},
    {"fuchsia.scenic.choreography_ssdo", kChoreographyCommand, "dashboard", "--screen_space_shadows --clipping_enabled",},
    {"fuchsia.scenic.choreography_shadow_map", kChoreographyCommand, "dashboard", "--shadow_map --clipping_enabled",},
    {"fuchsia.scenic.choreography_moment_shadow_map", kChoreographyCommand, "dashboard", "--moment_shadow_map --clipping_enabled",},
  };
  // clang-format on

  for (const auto& param : params) {
    std::string out_file = benchmarks_runner->MakeTempFile();

    // clang-format off
    std::vector<std::string> full_command = {
        "/pkgfs/packages/scenic_benchmarks/0/bin/run_scenic_benchmark.sh",
        "--out_file", out_file,
        "--benchmark_label", param.benchmark_name,
        "--cmd", param.command,
    };
    // clang-format on

    if (param.flutter_app_name) {
      full_command.push_back("--flutter_app_name");
      full_command.push_back(*param.flutter_app_name);
      full_command.push_back("--sleep_before_trace");
      full_command.push_back("5");
    }

    full_command.push_back(param.renderer_params);

    benchmarks_runner->AddCustomBenchmark(param.benchmark_name, full_command,
                                          out_file);
  }
}