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
|
// Copyright 2009-2021 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
#include "image.h"
#include "../../../common/sys/estring.h"
#include <iostream>
#include <vector>
#define STB_IMAGE_IMPLEMENTATION
#include "stb_image.h"
#define STB_IMAGE_WRITE_IMPLEMENTATION
#include "stb_image_write.h"
namespace embree
{
Ref<Image> loadSTB(const FileName& fileName)
{
int width, height, channels;
unsigned char *pixels = stbi_load(fileName.str().c_str(), &width, &height, &channels, 0);
if(pixels == nullptr) {
THROW_RUNTIME_ERROR("Could not load " + fileName.str());
}
Image* out = new Image4uc(width, height, fileName);
const float rcpMaxColor = 1.f/255.f;
for (size_t y = 0; y < height; y++)
{
for (size_t x = 0; x < width; x++)
{
float r = (channels > 0) ? (float(pixels[(y*width+x)*channels+0]) * rcpMaxColor) : 0.f;
float g = (channels > 1) ? (float(pixels[(y*width+x)*channels+1]) * rcpMaxColor) : r;
float b = (channels > 2) ? (float(pixels[(y*width+x)*channels+2]) * rcpMaxColor) : ((channels == 1) ? r : 0.f);
float a = (channels > 3) ? (float(pixels[(y*width+x)*channels+3]) * rcpMaxColor) : 1.f;
out->set(x, y, Color4(r,g,b,a));
}
}
stbi_image_free(pixels);
return out;
}
void storeSTB(const Ref<Image>& img, const FileName& fileName)
{
std::string ext = toLowerCase(fileName.ext());
if (ext != "bmp" && ext != "png" && ext != "jpg") {
THROW_RUNTIME_ERROR("Could not store image")
return;
}
int width = int(img->width);
int height = int(img->height);
int channels = 3;
std::vector<unsigned char> pixels(width*height*channels);
const float maxColor = 255.f;
for (int y = 0; y < height; y++)
{
for (int x = 0; x < width; x++)
{
Color4 c = img->get(x, y);
pixels[(y*img->width+x)*channels+0] = (unsigned char)(clamp(c.r) * maxColor);
pixels[(y*img->width+x)*channels+1] = (unsigned char)(clamp(c.g) * maxColor);
pixels[(y*img->width+x)*channels+2] = (unsigned char)(clamp(c.b) * maxColor);
}
}
if (ext == "bmp" ) { stbi_write_bmp(fileName.str().c_str(), width, height, channels, pixels.data()); return; }
if (ext == "png" ) { stbi_write_png(fileName.str().c_str(), width, height, channels, pixels.data(), width * channels); return; }
if (ext == "jpg" ) { stbi_write_jpg(fileName.str().c_str(), width, height, channels, pixels.data(), 100); return; }
}
}
|
