/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#ifndef CODEC_STATISTICS_H_
#define CODEC_STATISTICS_H_
#include <math.h>

#include "nsTArray.h"
#include "nsISupportsImpl.h"
#include "mozilla/TimeStamp.h"
#include "webrtc/common_types.h"
#include "webrtc/video_engine/include/vie_codec.h"
#include "MediaEngineWrapper.h"
#include "RunningStat.h"

namespace mozilla {

// Statistics-gathering observer for Video Encoder and Decoder

class VideoCodecStatistics : public webrtc::ViEEncoderObserver
                           , public webrtc::ViEDecoderObserver
{
public:
  VideoCodecStatistics(int channel, webrtc::ViECodec* vieCodec);
  ~VideoCodecStatistics();
  void Register(bool encoder);

  void SentFrame();
  virtual void OutgoingRate(const int video_channel,
    const unsigned int framerate, const unsigned int bitrate) override;

  virtual void IncomingCodecChanged(const int video_channel,
    const webrtc::VideoCodec& video_codec) override;

  virtual void IncomingRate(const int video_channel,
                            const unsigned int framerate,
                            const unsigned int bitrate) override;

  void ReceiveStateChange(const int video_channel, webrtc::VideoReceiveState state) override;

  void EndOfCallStats();

  virtual void RequestNewKeyFrame(const int video_channel) override {};

  virtual void SuspendChange(int video_channel, bool is_suspended) override {};
  virtual void DecoderTiming(int decode_ms,
                             int max_decode_ms,
                             int current_delay_ms,
                             int target_delay_ms,
                             int jitter_buffer_ms,
                             int min_playout_delay_ms,
                             int render_delay_ms) override {}

  bool GetEncoderStats(double* framerateMean,
                       double* framerateStdDev,
                       double* bitrateMean,
                       double* bitrateStdDev,
                       uint32_t* droppedFrames)
  {
    *framerateMean   = mEncoderFps.Mean();
    *framerateStdDev = mEncoderFps.StandardDeviation();
    *bitrateMean     = mEncoderBitRate.Mean();
    *bitrateStdDev   = mEncoderBitRate.StandardDeviation();
    *droppedFrames   = mEncoderDroppedFrames;
    return true;
  }

  bool GetDecoderStats(double* framerateMean,
                       double* framerateStdDev,
                       double* bitrateMean,
                       double* bitrateStdDev,
                       uint32_t* discardedPackets)
  {
    *framerateMean    = mDecoderFps.Mean();
    *framerateStdDev  = mDecoderFps.StandardDeviation();
    *bitrateMean      = mDecoderBitRate.Mean();
    *bitrateStdDev    = mDecoderBitRate.StandardDeviation();
    *discardedPackets = mDecoderDiscardedPackets;
    return true;
  }

  void Dump();
private:
  void Dump(RunningStat& s, const char *name);

  int mChannel;
  uint32_t mSentRawFrames;
  ScopedCustomReleasePtr<webrtc::ViECodec> mPtrViECodec; // back-pointer

  RunningStat mEncoderBitRate;
  RunningStat mEncoderFps;
  uint32_t mEncoderDroppedFrames;
  RunningStat mDecoderBitRate;
  RunningStat mDecoderFps;
  uint32_t mDecoderDiscardedPackets;
  bool mRegisteredEncode;
  bool mRegisteredDecode;

  webrtc::VideoReceiveState mReceiveState;
#ifdef MOZILLA_INTERNAL_API
  TimeStamp mFirstDecodeTime;
  TimeStamp mReceiveFailureTime;
  TimeDuration mTotalLossTime;
  uint32_t mRecoveredBeforeLoss;
  uint32_t mRecoveredLosses;
#endif
};

}

#endif //CODEC_STATISTICS_H_