f0254c2dcf
This is the first step into merging `DRW_gpu_wrapper.hh` into the GPU module. This is very similar to #119825. Pull Request: https://projects.blender.org/blender/blender/pulls/142732
87 lines
2.5 KiB
C++
87 lines
2.5 KiB
C++
/* SPDX-FileCopyrightText: 2023 Blender Authors
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-or-later */
|
|
|
|
#pragma once
|
|
|
|
#include <cstdint>
|
|
#include <memory>
|
|
#include <string>
|
|
|
|
#include "BLI_map.hh"
|
|
|
|
#include "GPU_texture.hh"
|
|
|
|
#include "DNA_image_types.h"
|
|
|
|
#include "RE_pipeline.h"
|
|
|
|
#include "COM_cached_resource.hh"
|
|
#include "COM_result.hh"
|
|
|
|
namespace blender::compositor {
|
|
|
|
class Context;
|
|
|
|
/* ------------------------------------------------------------------------------------------------
|
|
* Cached Image Key.
|
|
*/
|
|
class CachedImageKey {
|
|
public:
|
|
ImageUser image_user;
|
|
std::string pass_name;
|
|
|
|
CachedImageKey(ImageUser image_user, std::string pass_name);
|
|
|
|
uint64_t hash() const;
|
|
};
|
|
|
|
bool operator==(const CachedImageKey &a, const CachedImageKey &b);
|
|
|
|
/* -------------------------------------------------------------------------------------------------
|
|
* Cached Image.
|
|
*
|
|
* A cached resource that computes and caches a result containing the contents of the image with
|
|
* the given image user. */
|
|
class CachedImage : public CachedResource {
|
|
public:
|
|
Result result;
|
|
|
|
private:
|
|
/* For GPU, the result wraps an external GPU texture that is generated by the IMB module and
|
|
* stored in this member to be freed when the cached resource is deleted. */
|
|
blender::gpu::Texture *texture_ = nullptr;
|
|
|
|
public:
|
|
CachedImage(Context &context, Image *image, ImageUser *image_user, const char *pass_name);
|
|
|
|
~CachedImage();
|
|
|
|
private:
|
|
/* Populates the meta data of the image. */
|
|
void populate_meta_data(const RenderResult *render_result, const ImageUser &image_user);
|
|
};
|
|
|
|
/* ------------------------------------------------------------------------------------------------
|
|
* Cached Image Container.
|
|
*/
|
|
class CachedImageContainer : CachedResourceContainer {
|
|
private:
|
|
Map<std::string, Map<CachedImageKey, std::unique_ptr<CachedImage>>> map_;
|
|
|
|
/* A map that stores the update counts of the images at the moment they were cached. */
|
|
Map<std::string, uint64_t> update_counts_;
|
|
|
|
public:
|
|
void reset() override;
|
|
|
|
/* Check if the given image has changed since it was cached, and if so, invalidate its cache
|
|
* entry. Then, check if there is an available CachedImage cached resource with the given image
|
|
* user and pass_name in the container, if one exists, return it, otherwise, return a newly
|
|
* created one and add it to the container. In both cases, tag the cached resource as needed to
|
|
* keep it cached for the next evaluation. */
|
|
Result get(Context &context, Image *image, const ImageUser *image_user, const char *pass_name);
|
|
};
|
|
|
|
} // namespace blender::compositor
|