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

#ifndef MEDIA_FORMATS_WEBM_WEBM_VIDEO_CLIENT_H_
#define MEDIA_FORMATS_WEBM_WEBM_VIDEO_CLIENT_H_

#include <string>
#include <vector>

#include "packager/base/memory/scoped_ptr.h"
#include "packager/media/base/video_stream_info.h"
#include "packager/media/formats/webm/webm_parser.h"

namespace shaka {
namespace media {
class VideoDecoderConfig;

/// Helper class used to parse a Video element inside a TrackEntry element.
class WebMVideoClient : public WebMParserClient {
 public:
  WebMVideoClient();
  ~WebMVideoClient() override;

  /// Reset this object's state so it can process a new video track element.
  void Reset();

  /// Create a VideoStreamInfo with the data in |track_num|, |codec_id|,
  /// |codec_private|, |is_encrypted| and the fields parsed from the last video
  /// track element this object was used to parse.
  /// @return A VideoStreamInfo scoped_refptr if successful.
  /// @return An empty scoped_refptr if there was unexpected values in the
  ///         provided parameters or video track element fields.
  scoped_refptr<VideoStreamInfo> GetVideoStreamInfo(
      int64_t track_num,
      const std::string& codec_id,
      const std::vector<uint8_t>& codec_private,
      bool is_encrypted);

 private:
  // WebMParserClient implementation.
  bool OnUInt(int id, int64_t val) override;
  bool OnBinary(int id, const uint8_t* data, int size) override;
  bool OnFloat(int id, double val) override;

  int64_t pixel_width_;
  int64_t pixel_height_;
  int64_t crop_bottom_;
  int64_t crop_top_;
  int64_t crop_left_;
  int64_t crop_right_;
  int64_t display_width_;
  int64_t display_height_;
  int64_t display_unit_;
  int64_t alpha_mode_;

  DISALLOW_COPY_AND_ASSIGN(WebMVideoClient);
};

}  // namespace media
}  // namespace shaka

#endif  // MEDIA_FORMATS_WEBM_WEBM_VIDEO_CLIENT_H_