Modularize and cleanHEADmain

1 files changed, 52 insertions, 0 deletions

diff --git a/src/pipeline.h b/src/pipeline.h
new file mode 100644
index 0000000..93a7ea9
--- /dev/null
+++ b/src/pipeline.h
@@ -0,0 +1,52 @@
+#pragma once
+
+#include <hip/hip_runtime.h>
+#include <stdexcept>
+#include <string>
+#include <vector>
+#include <source_location>
+
+
+inline void hip_check(
+    hipError_t err,
+    std::source_location loc = std::source_location::current())
+{
+    if (err != hipSuccess) {
+        throw std::runtime_error(
+            std::string(hipGetErrorString(err)) +
+            " at " + loc.file_name() + ":" + std::to_string(loc.line()));
+    }
+}
+
+
+struct PyramidLevel {
+    int width = 0, height = 0;
+    float4* color_smax = nullptr; // rgb = color, a = s_max
+    float4* tensor     = nullptr; // x=E, y=F, z=G, w=unused
+};
+
+struct Pyramid {
+    std::vector<PyramidLevel> levels; // 0 = finest, back() = coarsest
+};
+
+struct Params {
+    float radius      = 5.0f;
+    float alpha       = 1.0f; // eccentricity tuning
+    int   num_sectors = 8;
+    float q           = 8.0f;
+    float tau_w       = 0.02f;
+    float p_s         = 0.5f;
+    float p_d         = 1.25f;
+    float tau_v       = 0.1f;
+};
+
+
+PyramidLevel make_level(int width, int height);
+void         free_level(PyramidLevel& lvl);
+Pyramid make_pyramid(int base_width, int base_height, int num_levels);
+void            destroy_pyramid(Pyramid& pyr);
+
+void build_pyramid (Pyramid& pyr);
+void compute_tensor(Pyramid& pyr, const Params& p);
+void smooth_tensor (Pyramid& pyr, const Params& p);
+void run_pipeline  (Pyramid& pyr, const Params& p);