RayTracingBackend.h

Go to the documentation of this file.

 
#ifndef RAYTRACING_BACKEND_H
#define RAYTRACING_BACKEND_H
 
#include "RayTracingTypes.h"
#include <memory>
#include <string>
 
namespace helios {
 
    class RayTracingBackend {
    public:
        virtual ~RayTracingBackend() = default;
 
        // ========== Lifecycle Management ==========
 
        virtual void initialize() = 0;
 
        virtual void shutdown() = 0;
 
        // ========== Geometry Management ==========
 
        virtual void updateGeometry(const RayTracingGeometry &geometry) = 0;
 
        virtual void buildAccelerationStructure() = 0;
 
        // ========== Material/Optical Properties ==========
 
        virtual void updateMaterials(const RayTracingMaterial &materials) = 0;
 
        // ========== Radiation Sources ==========
 
        virtual void updateSources(const std::vector<RayTracingSource> &sources) = 0;
 
        // ========== Diffuse/Sky Radiation ==========
 
        virtual void updateDiffuseRadiation(const std::vector<float> &flux, const std::vector<float> &extinction, const std::vector<helios::vec3> &peak_dir, const std::vector<float> &dist_norm, const std::vector<float> &sky_energy) = 0;
 
        virtual void updateSkyModel(const std::vector<helios::vec4> &sky_radiance_params, const std::vector<float> &camera_sky_radiance, const helios::vec3 &sun_direction, const std::vector<float> &solar_disk_radiance,
                                    float solar_disk_cos_angle, const std::vector<float> &camera_diffuse_flux, const std::vector<uint32_t> &band_emission_flag) = 0;
 
        // ========== Ray Launching ==========
 
        virtual void launchDirectRays(const RayTracingLaunchParams &params) = 0;
 
        virtual void launchDiffuseRays(const RayTracingLaunchParams &params) = 0;
 
        virtual void launchCameraRays(const RayTracingLaunchParams &params) = 0;
 
        virtual void launchPixelLabelRays(const RayTracingLaunchParams &params) = 0;
 
        // ========== Results Retrieval ==========
 
        virtual void getRadiationResults(RayTracingResults &results) = 0;
 
        virtual void getCameraResults(std::vector<float> &pixel_data, std::vector<uint> &pixel_labels, std::vector<float> &pixel_depths, uint camera_id, const helios::int2 &resolution) = 0;
 
        // ========== Buffer Management Utilities ==========
 
        virtual void zeroRadiationBuffers(size_t launch_band_count) = 0;
 
        virtual void zeroScatterBuffers() = 0;
 
        virtual void zeroCameraPixelBuffers(const helios::int2 &resolution) = 0;
 
        virtual void copyScatterToRadiation() = 0;
 
        virtual void uploadRadiationOut(const std::vector<float> &radiation_out_top, const std::vector<float> &radiation_out_bottom) = 0;
 
        virtual void uploadCameraScatterBuffers(const std::vector<float> &scatter_top_cam, const std::vector<float> &scatter_bottom_cam) = 0;
 
        virtual void zeroCameraScatterBuffers(size_t launch_band_count) = 0;
 
        virtual void uploadSourceFluxes(const std::vector<float> &fluxes) = 0;
 
        virtual void uploadSourceFluxesCam(const std::vector<float> &fluxes_cam) = 0;
 
        // ========== Diagnostics ==========
 
        virtual void queryGPUMemory() const = 0;
 
        virtual std::string getBackendName() const = 0;
 
    protected:
        void validateGeometryBeforeUpload(const RayTracingGeometry &geometry) const {
#ifndef NDEBUG
            geometry.validate();
#endif
        }
 
    public:
        // ========== Factory Method ==========
 
        static std::unique_ptr<RayTracingBackend> create(const std::string &backend_type);
    };
 
    bool probeAnyGPUBackend() noexcept;
 
} // namespace helios
 
#endif // RAYTRACING_BACKEND_H