tools: update iris.c

2026-06-17 21:44:24 +02:00
parent 5cacde7711
commit 20be299c4e
29 changed files with 23392 additions and 20990 deletions
@@ -0,0 +1,43 @@
+/*
+ * iris_upscale.h - RealESRGAN_x4plus image upscaler (RRDBNet)
+ *
+ * Self-contained, CPU-only inference for the RealESRGAN_x4plus model
+ * (RRDBNet: scale=4, num_feat=64, num_block=23, num_grow_ch=32).
+ * Loads weights from a single safetensors file and upscales an RGB image
+ * by 4x.
+ */
+
+#ifndef IRIS_UPSCALE_H
+#define IRIS_UPSCALE_H
+
+#include "iris.h"
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+typedef struct iris_upscale iris_upscale_t;
+
+/*
+ * Load the RealESRGAN_x4plus model from a safetensors file.
+ * Returns NULL on error (use iris_get_error() for details).
+ */
+iris_upscale_t *iris_upscale_load(const char *path);
+
+/*
+ * Free the model and its resources.
+ */
+void iris_upscale_free(iris_upscale_t *model);
+
+/*
+ * Upscale an RGB(A) image by 4x. Alpha is ignored; the result is always
+ * a 3-channel RGB image with dimensions (4*width, 4*height).
+ * Returns a newly allocated image (free with iris_image_free), NULL on error.
+ */
+iris_image *iris_upscale_run(iris_upscale_t *model, const iris_image *input);
+
+#ifdef __cplusplus
+}
+#endif
+
+#endif /* IRIS_UPSCALE_H */