img_rcc 0.1.0

A Rust library for image processing with CUDA, C++.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77

#define STB_IMAGE_IMPLEMENTATION

#include "stb_image.h"

#define STB_IMAGE_WRITE_IMPLEMENTATION

#include "stb_image_write.h"

#include "image.hpp"
#include <stdexcept>
#include <iostream>

Image *load_image_cpu(const char *file_path)
{
    Image *image = new Image();
    image->data = stbi_load(file_path, &image->width, &image->height, &image->channels, 0);
    if (!image->data)
    {
        throw std::runtime_error("failed to load image: " + std::string(file_path));
    }
    return image;
}

void save_image_cpu(const char *file_path, const Image *image)
{
    int success = stbi_write_png(file_path, image->width, image->height, image->channels, image->data, image->width * image->channels);
    if (!success)
    {
        throw std::runtime_error("failed to save image");
    }
}

void free_image_cpu(Image *image)
{
    if (image->data)
    {
        stbi_image_free(image->data);
        image->data = nullptr; // Set pointer to null to avoid double free
    }
}

// Image load_image(const char *file_path)
// {
//     Image image;
//     image.data = stbi_load(file_path, &image.width, &image.height, &image.channels, 0);
//     if (!image.data)
//     {
//         throw std::runtime_error("failed to load image: " + std::string(file_path));
//     }
//     // unsigned char *data = stbi_load(file_path.c_str(), &image.width, &image.height, &image.channels, 0);
//     // if (!data)
//     // {
//     //     throw std::runtime_error("Failed to load image: " + file_path);
//     // }

//     // image.data = std::vector<unsigned char>(data, data + (image.width * image.height * image.channels));
//     // stbi_image_free(data);

//     return image;
// }

// void save_image(const char *file_path, const Image &image)
// {
//     // int success = stbi_write_png(file_path.c_str(), image.width, image.height, image.channels, image.data.data(), image.width * image.channels);
//     // if (!success)
//     // {
//     //     throw std::runtime_error("Failed to save image: " + file_path);
//     // }
//     int success = stbi_write_png(file_path, image.width, image.height, image.channels, image.data, image.width * image.channels);
//     if (!success)
//     {
//         throw std::runtime_error("failed to save image");
//     }
// }

// void free_image(Image image)
// {
//     stbi_image_free(image.data);
// }