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

#include "services/image_annotation/image_annotation_metrics.h"

#include <map>
#include <string>

#include "base/metrics/histogram_functions.h"
#include "base/metrics/histogram_macros.h"
#include "base/no_destructor.h"
#include "base/strings/strcat.h"
#include "base/strings/stringprintf.h"

namespace image_annotation {

using metrics_internal::kAnnotationConfidence;
using metrics_internal::kAnnotationEmpty;

void ReportCacheHit(const bool cache_hit) {
  UMA_HISTOGRAM_BOOLEAN(metrics_internal::kCacheHit, cache_hit);
}

void ReportJsonParseSuccess(const bool success) {
  UMA_HISTOGRAM_BOOLEAN(metrics_internal::kJsonParseSuccess, success);
}

void ReportPixelFetchSuccess(const bool success) {
  UMA_HISTOGRAM_BOOLEAN(metrics_internal::kPixelFetchSuccess, success);
}

void ReportOcrAnnotation(const double confidence, const bool empty) {
  const int confidence_percent = static_cast<int>(std::round(confidence * 100));
  UMA_HISTOGRAM_PERCENTAGE(base::StringPrintf(kAnnotationConfidence, "Ocr"),
                           confidence_percent);
  UMA_HISTOGRAM_BOOLEAN(base::StringPrintf(kAnnotationEmpty, "Ocr"), empty);
}

void ReportDescAnnotation(const mojom::AnnotationType type,
                          const double confidence,
                          const bool empty) {
  static const base::NoDestructor<std::map<mojom::AnnotationType, std::string>>
      kTypeNames({{mojom::AnnotationType::kOcr, "Ocr"},
                  {mojom::AnnotationType::kLabel, "Label"},
                  {mojom::AnnotationType::kCaption, "Caption"}});

  const auto lookup = kTypeNames->find(type);
  const std::string type_name = base::StrCat(
      {"Desc", lookup == kTypeNames->end() ? "Unknown" : lookup->second});

  const int confidence_percent = static_cast<int>(std::round(confidence * 100));

  // We use function variants here since our histogram name is not a "runtime
  // constant".
  base::UmaHistogramPercentageObsoleteDoNotUse(
      base::StringPrintf(kAnnotationConfidence, type_name.c_str()),
      confidence_percent);
  base::UmaHistogramBoolean(
      base::StringPrintf(kAnnotationEmpty, type_name.c_str()), empty);

  UMA_HISTOGRAM_ENUMERATION(metrics_internal::kDescType, type);
}

void ReportDescFailure(const DescFailureReason reason) {
  UMA_HISTOGRAM_ENUMERATION(metrics_internal::kDescFailure, reason);
}

void ReportServerNetError(const int code) {
  base::UmaHistogramSparse(metrics_internal::kServerNetError, code);
}

void ReportServerResponseCode(const int code) {
  base::UmaHistogramSparse(metrics_internal::kServerHttpResponseCode, code);
}

void ReportServerLatency(const base::TimeDelta latency) {
  // Use a custom time histogram with ~10 buckets per order of magnitude between
  // 1ms and 30sec.
  UMA_HISTOGRAM_CUSTOM_TIMES(metrics_internal::kServerLatency, latency,
                             base::Milliseconds(1), base::Seconds(30), 50);
}

void ReportImageRequestIncludesDesc(const bool includes_desc) {
  UMA_HISTOGRAM_BOOLEAN(metrics_internal::kImageRequestIncludesDesc,
                        includes_desc);
}

void ReportImageRequestIncludesIcon(const bool includes_icon) {
  UMA_HISTOGRAM_BOOLEAN(metrics_internal::kImageRequestIncludesIcon,
                        includes_icon);
}

void ReportServerRequestSizeKB(const size_t size_kb) {
  // Use a custom memory histogram with ~10 buckets per order of magnitude
  // between 1KB and 30MB.
  UMA_HISTOGRAM_CUSTOM_COUNTS(metrics_internal::kServerRequestSize, size_kb, 1,
                              30000, 50);
}

void ReportServerResponseSizeBytes(const size_t size_bytes) {
  // Use a custom memory histogram with ~10 buckets per order of magnitude
  // between 1byte and 1MB (at which point we stop downloading).
  UMA_HISTOGRAM_CUSTOM_COUNTS(metrics_internal::kServerResponseSize, size_bytes,
                              1, 1000000, 70);
}

void ReportOcrStatus(const int status) {
  base::UmaHistogramSparse(
      base::StringPrintf(metrics_internal::kAnnotationStatus, "Ocr"), status);
}

void ReportDescStatus(const int status) {
  base::UmaHistogramSparse(
      base::StringPrintf(metrics_internal::kAnnotationStatus, "Desc"), status);
}

void ReportEngineKnown(const bool known) {
  UMA_HISTOGRAM_BOOLEAN(metrics_internal::kEngineKnown, known);
}

void ReportSourcePixelCount(const size_t pixel_count) {
  // Use a custom memory histogram with ~10 buckets per order of magnitude
  // 1x1 and 8K image resolution.
  UMA_HISTOGRAM_CUSTOM_COUNTS(metrics_internal::kSourcePixelCount, pixel_count,
                              1, 7680 * 4320, 80);
}

void ReportEncodedJpegSize(const size_t size_bytes) {
  // Use a custom memory histogram with ~10 buckets per order of magnitude
  // between 1byte and 1MB.
  UMA_HISTOGRAM_CUSTOM_COUNTS(metrics_internal::kEncodedJpegSize, size_bytes, 1,
                              1000000, 70);
}

void ReportClientResult(const ClientResult result) {
  UMA_HISTOGRAM_ENUMERATION(metrics_internal::kClientResult, result);
}

}  // namespace image_annotation