// Copyright 2016 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 FLUTTER_FLOW_RASTER_CACHE_H_
#define FLUTTER_FLOW_RASTER_CACHE_H_

#include <memory>
#include <unordered_map>

#include "flutter/flow/instrumentation.h"
#include "lib/ftl/macros.h"
#include "lib/ftl/memory/weak_ptr.h"
#include "third_party/skia/include/core/SkImage.h"
#include "third_party/skia/include/core/SkSize.h"

namespace flow {

class RasterCache {
 public:
  RasterCache();
  ~RasterCache();

  sk_sp<SkImage> GetPrerolledImage(GrContext* context,
                                   SkPicture* picture,
                                   const SkMatrix& ctm,
                                   bool is_complex,
                                   bool will_change);
  void SweepAfterFrame();

  void Clear();

  void SetCheckboardCacheImages(bool checkerboard);

 private:
  struct Entry {
    Entry();
    ~Entry();

    bool used_this_frame = false;
    int access_count = 0;
    SkISize physical_size;
    sk_sp<SkImage> image;
  };

  using Cache = std::unordered_map<uint32_t, Entry>;

  Cache cache_;
  bool checkerboard_images_;
  ftl::WeakPtrFactory<RasterCache> weak_factory_;

  FTL_DISALLOW_COPY_AND_ASSIGN(RasterCache);
};

}  // namespace flow

#endif  // FLUTTER_FLOW_RASTER_CACHE_H_