working_image_gpu_halide.h

Go to the documentation of this file.

 
#pragma once
 
#include "image_processing/gpu/working_image_gpu.h"
#include "image_processing/halide/working_image_halide.h"
#include "common/error_handling/core_error.h"
#include "image_processing/common/image_view.h"
 
#include <memory>
#include <expected>
 
namespace CaptureMoment::Core {
 
namespace ImageProcessing {
 
class WorkingImageGPU_Halide final : public WorkingImageGPU, public WorkingImageHalide {
public:
    WorkingImageGPU_Halide() = default;
 
    ~WorkingImageGPU_Halide() override = default;
 
    [[nodiscard]] bool bindView(const ImageView& view) override;
 
    [[nodiscard]] std::expected<void, ErrorHandling::CoreError>
    transferToVRAM() override;
 
    [[nodiscard]] bool isValid() const override;
 
    void resetExecutionBuffer();
 
    [[nodiscard]] Halide::Buffer<float>& getExecutionBuffer();
 
    [[nodiscard]] bool isOriginalHalideBufferValid() const { return m_original_halide_buffer.defined(); };
 
    [[nodiscard]] Halide::Buffer<float>& getOriginalHalideBuffer() { return m_original_halide_buffer; };
 
    void syncToHostRAM();
 
    [[nodiscard]] std::expected<std::unique_ptr<Common::ImageRegion>, ErrorHandling::CoreError>
    downsample(Common::ImageDim target_width, Common::ImageDim target_height) override;
 
protected:
    [[nodiscard]] bool downloadDeviceToHost() override;
 
private:
    void initDataForHalide();
 
    void buildDownsamplePipeline();
 
    // --- GPU Memory Management ---
    Halide::Buffer<float> m_original_halide_buffer; 
 
    // --- Pipelines Halide ---
    Halide::Pipeline m_reset_pipeline; 
    Halide::Pipeline m_downsample_pipeline; 
 
    // --- Downsample Pipelines ---
    Halide::ImageParam m_downsample_input{Halide::Float(32), 3, "downsample_src"}; 
    Halide::Param<float> m_downsample_scale_x{"downsample_scale_x"}; 
    Halide::Param<float> m_downsample_scale_y{"downsample_scale_y"}; 
    Halide::Buffer<float> m_downsample_src_buffer; 
 
    bool m_downsample_built{false}; 
    bool m_pipelines_initialized{false}; 
 
};
 
} // namespace ImageProcessing
 
} // namespace CaptureMoment::Core