/*
 * Copyright 2017 Google Inc. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef COBALT_LOADER_IMAGE_ANIMATED_WEBP_IMAGE_H_
#define COBALT_LOADER_IMAGE_ANIMATED_WEBP_IMAGE_H_

#include <vector>

#include "base/basictypes.h"
#include "base/callback.h"
#include "base/debug/trace_event.h"
#include "base/memory/scoped_ptr.h"
#include "base/synchronization/lock.h"
#include "base/threading/thread.h"
#include "base/time.h"
#include "cobalt/loader/image/image.h"
#include "cobalt/render_tree/resource_provider.h"
#include "third_party/libwebp/webp/demux.h"

namespace cobalt {
namespace loader {
namespace image {

class AnimatedWebPImage : public AnimatedImage {
 public:
  AnimatedWebPImage(const math::Size& size, bool is_opaque,
                    render_tree::ResourceProvider* resource_provider);

  const math::Size& GetSize() const OVERRIDE { return size_; }

  uint32 GetEstimatedSizeInBytes() const OVERRIDE {
    return static_cast<uint32>(data_buffer_.size());
  }

  bool IsOpaque() const OVERRIDE { return is_opaque_; }

  scoped_refptr<render_tree::Image> GetFrame() OVERRIDE;

  void AppendChunk(const uint8* data, size_t input_byte);

  void SetDecodedFramesCallback(const base::Closure& callback) {
    decoded_callback_ = callback;
  }

 private:
  ~AnimatedWebPImage() OVERRIDE;

  void DecodeFrames();

  // Updates the index and time info of the current and next frames.
  void UpdateTimelineInfo();

  void FillImageBufferWithBackgroundColor();

  scoped_ptr<render_tree::ImageData> AllocateImageData();

  base::Thread thread_;
  const math::Size size_;
  const bool is_opaque_;
  WebPDemuxer* demux_;
  WebPDemuxState demux_state_;
  int frame_count_;
  // The remaining number of times to loop the animation. kLoopInfinite means
  // looping infinitely.
  int loop_count_;
  uint32_t background_color_;
  bool should_dispose_previous_frame_;
  int current_frame_index_;
  int next_frame_index_;
  render_tree::ResourceProvider* resource_provider_;

  base::TimeTicks current_frame_time_;
  base::TimeTicks next_frame_time_;
  // The original encoded data.
  std::vector<uint8> data_buffer_;
  // The alpha-blended frame.
  std::vector<uint8> image_buffer_;
  scoped_refptr<render_tree::Image> current_frame_image_;
  base::Closure decoded_callback_;
  base::Lock lock_;
};

}  // namespace image
}  // namespace loader
}  // namespace cobalt

#endif  // COBALT_LOADER_IMAGE_ANIMATED_WEBP_IMAGE_H_