You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and dots ('.'), can be up to 35 characters long. Letters must be lowercase.
89 lines
2.2 KiB
89 lines
2.2 KiB
#ifdef _WIN32 |
|
#define STBR_ASSERT(x) \ |
|
if (!(x)) \ |
|
__debugbreak(); |
|
#endif |
|
|
|
#define STB_RESAMPLE_IMPLEMENTATION |
|
#define STB_RESAMPLE_STATIC |
|
#include "stb_resample.h" |
|
|
|
#define STB_IMAGE_WRITE_IMPLEMENTATION |
|
#include "stb_image_write.h" |
|
|
|
#define STB_IMAGE_IMPLEMENTATION |
|
#include "stb_image.h" |
|
|
|
#ifdef _WIN32 |
|
#include <sys/timeb.h> |
|
#endif |
|
|
|
int main(int argc, char** argv) |
|
{ |
|
unsigned char* input_data; |
|
unsigned char* output_data; |
|
int w, h; |
|
int n; |
|
int out_w, out_h, out_stride; |
|
|
|
if (argc <= 1) |
|
{ |
|
printf("No input image\n"); |
|
return 1; |
|
} |
|
|
|
input_data = stbi_load(argv[1], &w, &h, &n, 0); |
|
if (!input_data) |
|
{ |
|
printf("Input image could not be loaded"); |
|
return 1; |
|
} |
|
|
|
out_w = 256; |
|
out_h = 256; |
|
out_stride = (out_w + 10) * n; |
|
|
|
output_data = (unsigned char*)malloc(out_stride * out_h); |
|
|
|
int in_w = 512; |
|
int in_h = 512; |
|
|
|
size_t memory_required = stbr_calculate_memory(in_w, in_h, out_w, out_h, n, STBR_FILTER_CATMULLROM); |
|
void* extra_memory = malloc(memory_required); |
|
|
|
// Cut out the outside 64 pixels all around to test the stride. |
|
int border = 64; |
|
STBR_ASSERT(in_w + border <= w); |
|
STBR_ASSERT(in_h + border <= h); |
|
|
|
#ifdef PERF_TEST |
|
struct timeb initial_time_millis, final_time_millis; |
|
|
|
long average = 0; |
|
for (int j = 0; j < 10; j++) |
|
{ |
|
ftime(&initial_time_millis); |
|
for (int i = 0; i < 100; i++) |
|
stbr_resize_arbitrary(input_data + w * border * n + border * n, in_w, in_h, w*n, output_data, out_w, out_h, out_stride, n, STBR_TYPE_UINT8, STBR_FILTER_CATMULLROM, STBR_EDGE_CLAMP, STBR_COLORSPACE_SRGB, extra_memory, memory_required); |
|
ftime(&final_time_millis); |
|
long lapsed_ms = (long)(final_time_millis.time - initial_time_millis.time) * 1000 + (final_time_millis.millitm - initial_time_millis.millitm); |
|
printf("Resample: %dms\n", lapsed_ms); |
|
|
|
average += lapsed_ms; |
|
} |
|
|
|
average /= 10; |
|
|
|
printf("Average: %dms\n", average); |
|
#else |
|
stbr_resize_arbitrary(input_data + w * border * n + border * n, in_w, in_h, w*n, output_data, out_w, out_h, out_stride, n, STBR_TYPE_UINT8, STBR_FILTER_CATMULLROM, STBR_EDGE_CLAMP, STBR_COLORSPACE_SRGB, extra_memory, memory_required); |
|
#endif |
|
|
|
free(extra_memory); |
|
|
|
stbi_write_png("output.png", out_w, out_h, n, output_data, out_stride); |
|
|
|
free(output_data); |
|
|
|
return 0; |
|
}
|
|
|