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

#ifndef CAMERA_INCLUDE_CROS_CAMERA_CAMERA_METRICS_H_
#define CAMERA_INCLUDE_CROS_CAMERA_CAMERA_METRICS_H_

#include <memory>

#include <base/containers/flat_map.h>
#include <base/time/time.h>

#include "cros-camera/export.h"

namespace cros {

enum class JpegProcessType { kDecode, kEncode };

enum class JpegProcessMethod { kHardware, kSoftware };

// These values are persisted to logs. Entries should not be renumbered and
// numeric values should never be reused.
enum class FaceAeFunction {
  // Doesn't support ROI control.
  kUnsupported = 0,
  // Supports ROI control, but doesn't enable face AE.
  kNotEnabled = 1,
  // Supports ROI control and enabled face AE.
  kEnabled = 2,
  // Supports ROI control and enabled face AE from app, but forcedly disabled by
  // user.
  kForceDisabled = 3,
  // For SendEnumToUMA() usage.
  kMaxValue = kForceDisabled,
};

// These values are persisted to logs. Entries should not be renumbered and
// numeric values should never be reused.
enum class HdrnetStreamConfiguration {
  kSingleYuvStream = 0,
  kSingleYuvStreamWithBlob = 1,
  kMultipleYuvStreams = 2,
  kMultipleYuvStreamsWithBlob = 3,
  kMultipleYuvStreamsOfDifferentAspectRatio = 4,
  kMultipleYuvStreamsOfDifferentAspectRatioWithBlob = 5,
  kMaxValue = kMultipleYuvStreamsOfDifferentAspectRatioWithBlob,
};

// These values are persisted to logs. Entries should not be renumbered and
// numeric values should never be reused.
enum class HdrnetStreamType {
  // HDRnet stream for YUV output.
  kYuv = 0,
  // HDRnet stream for BLOB output.
  kBlob = 1,
  kMaxValue = kBlob,
};

// These values are persisted to logs. Entries should not be renumbered and
// numeric values should never be reused.
enum class HdrnetProcessingType {
  // Pre-processing of input YUV into linear RGB domain.
  kPreprocessing = 0,
  // Main HDRnet inferencing and rendering.
  kRgbPipeline = 1,
  // Post-processing of HDRnet RGB output to final YUV output(s).
  kPostprocessing = 2,
  kMaxValue = kPostprocessing,
};

// These values are persisted to logs. Entries should not be renumbered and
// numeric values should never be reused.
enum class HdrnetError {
  kNoError = 0,
  // Error during HDRnet stream manipulator initialization.
  kInitializationError = 1,
  // Error when waiting for buffer acquire fence.
  kSyncWaitError = 2,
  // Error when running HDRnet processor.
  kHdrnetProcessorError = 3,
  // Error in pre-processing input buffer to the HDRnet pipeline.
  kPreprocessingError = 4,
  // Error when running linear RGB pipeline.
  kRgbPipelineError = 5,
  // Error in post-processing the RGB buffer to produce the output buffers.
  kPostprocessingError = 6,
  // Error triggered by camera HAL.
  kCameraHal3Error = 7,
  kMaxValue = kCameraHal3Error,
};

// These values are persisted to logs. Entries should not be renumbered and
// numeric values should never be reused.
enum class AutoFramingError {
  kNoError = 0,
  // Error in auto-framing stream manipulator initialization.
  kInitializationError = 1,
  // Error in auto-framing stream manipulator configuration.
  kConfigurationError = 2,
  // Error in auto-framing stream manipulator processing requests.
  kProcessRequestError = 3,
  // Error in auto-framing stream manipulator processing results.
  kProcessResultError = 4,
  // Error when initializing auto-framing pipeline.
  kPipelineInitializationError = 5,
  // Error when adding inputs to auto-framing pipeline.
  kPipelineInputError = 6,
  // Error when obtaining outputs auto-framing pipeline.
  kPipelineOutputError = 7,
  kMaxValue = kPipelineOutputError,
};

// These values are persisted to logs. Entries should not be renumbered and
// numeric values should never be reused.
enum class CameraEffect {
  kNone = 0,  // Not used, but kept for consistency
  kBlur = 1,
  kRelight = 2,
  kBlurAndRelight = 3,
  kMaxValue = kBlurAndRelight,
};

// Could use HdrnetStreamType here but that wouldn't read very well.
//
// These values are persisted to logs. Entries should not be renumbered and
// numeric values should never be reused.
enum class CameraEffectStreamType {
  kYuv = 0,
  kBlob = 1,  // Also JPEG
  kMaxValue = kBlob,
};

// These values are persisted to logs. Entries should not be renumbered and
// numeric values should never be reused.
enum class CameraEffectError {
  kNoError = 0,
  // Error during EffectsStreamManipulator GPU initialization.
  kGPUInitializationError = 1,
  // Error while allocating buffers.
  kBufferAllocationError = 2,
  // A failed buffer was sent into the stream manipulator.
  kReceivedFailedBuffer = 3,
  // A synchronous buffer wait failed.
  kSyncWaitTimeout = 4,
  // Unable to register a buffer.
  kBufferRegistrationFailed = 5,
  // Unable to unregister a buffer.
  kBufferUnregistrationFailed = 6,
  // Initializing GPU images failed.
  kGPUImageInitializationFailed = 7,
  // YUV to RGB or YUV to YUV conversion failed.
  kYUVConversionFailed = 8,
  // Effects pipeline rendering failed.
  kPipelineFailed = 9,
  kMaxValue = kPipelineFailed,
};

// These values are persisted to logs. Entries should not be renumbered and
// numeric values should never be reused.
enum class PortraitModeError {
  kNoError = 0,
  // Error in Portrait Mode stream manipulator initialization.
  kInitializationError = 1,
  // Error in Portrait Mode stream manipulator configuration.
  kConfigurationError = 2,
  // Error in Portrait Mode stream manipulator processing requests.
  kProcessRequestError = 3,
  // Error in Portrait Mode stream manipulator processing results.
  kProcessResultError = 4,
  kMaxValue = kProcessResultError,
};

class CROS_CAMERA_EXPORT CameraMetrics {
 public:
  static std::unique_ptr<CameraMetrics> New();

  virtual ~CameraMetrics() = default;

  // Records the process time of JDA/JEA in microseconds.
  virtual void SendJpegProcessLatency(JpegProcessType process_type,
                                      JpegProcessMethod process_layer,
                                      base::TimeDelta latency) = 0;

  // Records the resolution of image that JDA/JEA process in pixels.
  virtual void SendJpegResolution(JpegProcessType process_type,
                                  JpegProcessMethod process_layer,
                                  int width,
                                  int height) = 0;

  // Records the process time of ConfigureStreams().
  virtual void SendConfigureStreamsLatency(base::TimeDelta latency) = 0;

  // Records the resolution of streams that configured.
  virtual void SendConfigureStreamResolution(int width,
                                             int height,
                                             int format) = 0;

  // Records the type of the client that called OpenDevice().
  virtual void SendOpenDeviceClientType(int client_type) = 0;

  // Records the process time of OpenDevice().
  virtual void SendOpenDeviceLatency(base::TimeDelta latency) = 0;

  // Records the error type which triggers Notify().
  virtual void SendError(int error_code) = 0;

  // Records the camera facing of current session.
  virtual void SendCameraFacing(int facing) = 0;

  // Records the duration of the closing session.
  virtual void SendSessionDuration(base::TimeDelta duration) = 0;

  // Records the face AE function.
  virtual void SendFaceAeFunction(FaceAeFunction function) = 0;

  // Records the max number of detected faces in a camera session
  virtual void SendFaceAeMaxDetectedFaces(int number) = 0;

  // *** HDRnet metrics ***

  // Records the stream configuration including the number of streams, the type
  // of streams, and if the streams are of the same aspect ratio.
  virtual void SendHdrnetStreamConfiguration(
      HdrnetStreamConfiguration config) = 0;

  // Records the maximum size (in width * height) of the HDRnet stream
  // configured for |stream_type| output.
  virtual void SendHdrnetMaxStreamSize(HdrnetStreamType stream_type,
                                       int size) = 0;

  // Records the number of concurrent HDRnet streams in a session.
  virtual void SendHdrnetNumConcurrentStreams(int num_streams) = 0;

  // Records the maximum number of output buffers a HDRnet stream produces (> 1
  // means there are multiple streams with the same aspect ratio) in a session.
  virtual void SendHdrnetMaxOutputBuffersRendered(int num_buffers) = 0;

  // Records whether there's an error that can compromise the HDRnet feature,
  // either causing frame drops or stops the pipeline from running completely,
  // in a session.
  virtual void SendHdrnetError(HdrnetError error) = 0;

  // Records the number of HDRnet-rendered still capture shots taken in a
  // session.
  virtual void SendHdrnetNumStillShotsTaken(int num_shots) = 0;

  // Records the average CPU latency in processing |processing_type| in a
  // session.
  virtual void SendHdrnetAvgLatency(HdrnetProcessingType processing_type,
                                    int latency_us) = 0;

  // *** Gcam AE metrics ***

  // Records the average AE convergence latency in frame count per session.
  virtual void SendGcamAeAvgConvergenceLatency(int latency_frames) = 0;

  // Records the average HDR ratio per session.
  virtual void SendGcamAeAvgHdrRatio(int hdr_ratio) = 0;

  // Records the average total exposure time (TET) per session.
  virtual void SendGcamAeAvgTet(int tet) = 0;

  // *** Auto-framing metrics ***

  // Records auto-framing enabled time in ratio per session.
  virtual void SendAutoFramingEnabledTimePercentage(int percentage) = 0;

  // Records auto-framing enabled count per session.
  virtual void SendAutoFramingEnabledCount(int count) = 0;

  // Records auto-framing detection hit rate per session.
  virtual void SendAutoFramingDetectionHitPercentage(int percentage) = 0;

  // Records auto-framing average detection latency per session.
  virtual void SendAutoFramingAvgDetectionLatency(base::TimeDelta latency) = 0;

  // Records auto-framing median zoom ratio per session.
  virtual void SendAutoFramingMedianZoomRatio(int zoom_ratio_tenths) = 0;

  // Records whether there's an error that can compromise the auto-framing
  // feature.
  virtual void SendAutoFramingError(AutoFramingError error) = 0;

  // *** Effects metrics ***

  // Records the user selecting an effect during the session.
  virtual void SendEffectsSelectedEffect(CameraEffect effect) = 0;

  // Records the average processing latency of the EffectsStreamManipulator
  // ProcessCaptureResult method per session.
  virtual void SendEffectsAvgProcessingLatency(
      CameraEffect effect,
      CameraEffectStreamType stream_type,
      base::TimeDelta latency) = 0;

  // Records the average interval between successfully processed frames in the
  // EffectsStreamManipulator ProcessCaptureResult method per session.
  virtual void SendEffectsAvgProcessedFrameInterval(
      CameraEffect effect,
      CameraEffectStreamType stream_type,
      base::TimeDelta interval) = 0;

  // Records the requested frame rate for an EffectsStreamManipulator session.
  virtual void SendEffectsRequestedFrameRate(int fps) = 0;

  // Records the minimum size (in width * height) of the
  // EffectsStreamManipulator stream configured for |stream_type| output.
  virtual void SendEffectsMinStreamSize(CameraEffectStreamType stream_type,
                                        int size) = 0;

  // Records the maximum size (in width * height) of the
  // EffectsStreamManipulator stream configured for |stream_type| output.
  virtual void SendEffectsMaxStreamSize(CameraEffectStreamType stream_type,
                                        int size) = 0;

  // Records the number of concurrent EffectsStreamManipulator streams in a
  // session.
  virtual void SendEffectsNumConcurrentStreams(int num_streams) = 0;

  // Records the number of concurrent EffectsStreamManipulator streams which
  // involved rendering effects in a session.
  virtual void SendEffectsNumConcurrentProcessedStreams(int num_streams) = 0;

  // Records whether there's an error that can compromise the
  // EffectsStreamManipulator feature.
  virtual void SendEffectsError(CameraEffectError error) = 0;

  // Records the number of EffectsStreamManipulator-rendered still capture shots
  // taken in a session.
  virtual void SendEffectsNumStillShotsTaken(int num_shots) = 0;

  // *** Portrait Mode metrics ***

  // Records the number of still capture shots taken in Portrait Mode per
  // session.
  virtual void SendPortraitModeNumStillShotsTaken(int num_shots) = 0;

  // Records the average processing latency of the
  // PortraitModeStreamManipulator::ProcessCaptureResult() method per session.
  // This method only takes into account the processing time for successful
  // portrait photos (human face detected).
  virtual void SendPortraitModeProcessAvgLatency(base::TimeDelta latency) = 0;

  // Records whether there's an error that can compromise the
  // Portrait Mode feature.
  virtual void SendPortraitModeError(PortraitModeError error) = 0;
};

}  // namespace cros

#endif  // CAMERA_INCLUDE_CROS_CAMERA_CAMERA_METRICS_H_