479 lines
14 KiB
C++
479 lines
14 KiB
C++
/* SPDX-FileCopyrightText: 2023 Blender Authors
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-or-later */
|
|
|
|
/** \file
|
|
* \ingroup bke
|
|
*/
|
|
|
|
#include <cmath>
|
|
#include <cstdlib>
|
|
|
|
#include "BLI_math_base.h"
|
|
#include "BLI_math_color.h"
|
|
#include "BLI_math_vector.h"
|
|
|
|
#include "BKE_image.h"
|
|
|
|
#include "IMB_imbuf.h"
|
|
#include "IMB_imbuf_types.h"
|
|
|
|
#include "BLF_api.h"
|
|
|
|
struct FillColorThreadData {
|
|
uchar *rect;
|
|
float *rect_float;
|
|
int width;
|
|
float color[4];
|
|
};
|
|
|
|
static void image_buf_fill_color_slice(
|
|
uchar *rect, float *rect_float, int width, int height, const float color[4])
|
|
{
|
|
int x, y;
|
|
|
|
/* blank image */
|
|
if (rect_float) {
|
|
for (y = 0; y < height; y++) {
|
|
for (x = 0; x < width; x++) {
|
|
copy_v4_v4(rect_float, color);
|
|
rect_float += 4;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (rect) {
|
|
uchar ccol[4];
|
|
rgba_float_to_uchar(ccol, color);
|
|
for (y = 0; y < height; y++) {
|
|
for (x = 0; x < width; x++) {
|
|
rect[0] = ccol[0];
|
|
rect[1] = ccol[1];
|
|
rect[2] = ccol[2];
|
|
rect[3] = ccol[3];
|
|
rect += 4;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
static void image_buf_fill_color_thread_do(void *data_v, int scanline)
|
|
{
|
|
FillColorThreadData *data = (FillColorThreadData *)data_v;
|
|
const int num_scanlines = 1;
|
|
size_t offset = size_t(scanline) * data->width * 4;
|
|
uchar *rect = (data->rect != nullptr) ? (data->rect + offset) : nullptr;
|
|
float *rect_float = (data->rect_float != nullptr) ? (data->rect_float + offset) : nullptr;
|
|
image_buf_fill_color_slice(rect, rect_float, data->width, num_scanlines, data->color);
|
|
}
|
|
|
|
void BKE_image_buf_fill_color(
|
|
uchar *rect, float *rect_float, int width, int height, const float color[4])
|
|
{
|
|
if (size_t(width) * height < 64 * 64) {
|
|
image_buf_fill_color_slice(rect, rect_float, width, height, color);
|
|
}
|
|
else {
|
|
FillColorThreadData data;
|
|
data.rect = rect;
|
|
data.rect_float = rect_float;
|
|
data.width = width;
|
|
copy_v4_v4(data.color, color);
|
|
IMB_processor_apply_threaded_scanlines(height, image_buf_fill_color_thread_do, &data);
|
|
}
|
|
}
|
|
|
|
static void image_buf_fill_checker_slice(
|
|
uchar *rect, float *rect_float, int width, int height, int offset)
|
|
{
|
|
/* these two passes could be combined into one, but it's more readable and
|
|
* easy to tweak like this, speed isn't really that much of an issue in this situation... */
|
|
|
|
int checkerwidth = 32;
|
|
int x, y;
|
|
|
|
uchar *rect_orig = rect;
|
|
float *rect_float_orig = rect_float;
|
|
|
|
float hsv[3] = {0.0f, 0.9f, 0.9f};
|
|
float rgb[3];
|
|
|
|
float dark_linear_color = 0.0f, bright_linear_color = 0.0f;
|
|
if (rect_float != nullptr) {
|
|
dark_linear_color = srgb_to_linearrgb(0.25f);
|
|
bright_linear_color = srgb_to_linearrgb(0.58f);
|
|
}
|
|
|
|
/* checkers */
|
|
for (y = offset; y < height + offset; y++) {
|
|
int dark = powf(-1.0f, floorf(y / checkerwidth));
|
|
|
|
for (x = 0; x < width; x++) {
|
|
if (x % checkerwidth == 0) {
|
|
dark = -dark;
|
|
}
|
|
|
|
if (rect_float) {
|
|
if (dark > 0) {
|
|
rect_float[0] = rect_float[1] = rect_float[2] = dark_linear_color;
|
|
rect_float[3] = 1.0f;
|
|
}
|
|
else {
|
|
rect_float[0] = rect_float[1] = rect_float[2] = bright_linear_color;
|
|
rect_float[3] = 1.0f;
|
|
}
|
|
rect_float += 4;
|
|
}
|
|
else {
|
|
if (dark > 0) {
|
|
rect[0] = rect[1] = rect[2] = 64;
|
|
rect[3] = 255;
|
|
}
|
|
else {
|
|
rect[0] = rect[1] = rect[2] = 150;
|
|
rect[3] = 255;
|
|
}
|
|
rect += 4;
|
|
}
|
|
}
|
|
}
|
|
|
|
rect = rect_orig;
|
|
rect_float = rect_float_orig;
|
|
|
|
/* 2nd pass, colored `+`. */
|
|
for (y = offset; y < height + offset; y++) {
|
|
float hoffs = 0.125f * floorf(y / checkerwidth);
|
|
|
|
for (x = 0; x < width; x++) {
|
|
float h = 0.125f * floorf(x / checkerwidth);
|
|
|
|
if ((abs((x % checkerwidth) - (checkerwidth / 2)) < 4) &&
|
|
(abs((y % checkerwidth) - (checkerwidth / 2)) < 4))
|
|
{
|
|
if ((abs((x % checkerwidth) - (checkerwidth / 2)) < 1) ||
|
|
(abs((y % checkerwidth) - (checkerwidth / 2)) < 1))
|
|
{
|
|
hsv[0] = fmodf(fabsf(h - hoffs), 1.0f);
|
|
hsv_to_rgb_v(hsv, rgb);
|
|
|
|
if (rect) {
|
|
rect[0] = char(rgb[0] * 255.0f);
|
|
rect[1] = char(rgb[1] * 255.0f);
|
|
rect[2] = char(rgb[2] * 255.0f);
|
|
rect[3] = 255;
|
|
}
|
|
|
|
if (rect_float) {
|
|
srgb_to_linearrgb_v3_v3(rect_float, rgb);
|
|
rect_float[3] = 1.0f;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (rect_float) {
|
|
rect_float += 4;
|
|
}
|
|
if (rect) {
|
|
rect += 4;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
struct FillCheckerThreadData {
|
|
uchar *rect;
|
|
float *rect_float;
|
|
int width;
|
|
};
|
|
|
|
static void image_buf_fill_checker_thread_do(void *data_v, int scanline)
|
|
{
|
|
FillCheckerThreadData *data = (FillCheckerThreadData *)data_v;
|
|
size_t offset = size_t(scanline) * data->width * 4;
|
|
const int num_scanlines = 1;
|
|
uchar *rect = (data->rect != nullptr) ? (data->rect + offset) : nullptr;
|
|
float *rect_float = (data->rect_float != nullptr) ? (data->rect_float + offset) : nullptr;
|
|
image_buf_fill_checker_slice(rect, rect_float, data->width, num_scanlines, scanline);
|
|
}
|
|
|
|
void BKE_image_buf_fill_checker(uchar *rect, float *rect_float, int width, int height)
|
|
{
|
|
if (size_t(width) * height < 64 * 64) {
|
|
image_buf_fill_checker_slice(rect, rect_float, width, height, 0);
|
|
}
|
|
else {
|
|
FillCheckerThreadData data;
|
|
data.rect = rect;
|
|
data.rect_float = rect_float;
|
|
data.width = width;
|
|
IMB_processor_apply_threaded_scanlines(height, image_buf_fill_checker_thread_do, &data);
|
|
}
|
|
}
|
|
|
|
/* Utility functions for BKE_image_buf_fill_checker_color */
|
|
|
|
#define BLEND_FLOAT(real, add) (real + add <= 1.0f) ? (real + add) : 1.0f
|
|
#define BLEND_CHAR(real, add) \
|
|
((real + char(add * 255.0f)) <= 255) ? (real + char(add * 255.0f)) : 255
|
|
|
|
static void checker_board_color_fill(
|
|
uchar *rect, float *rect_float, int width, int height, int offset, int total_height)
|
|
{
|
|
int hue_step, y, x;
|
|
float hsv[3], rgb[3];
|
|
|
|
hsv[1] = 1.0;
|
|
|
|
hue_step = power_of_2_max_i(width / 8);
|
|
if (hue_step < 8) {
|
|
hue_step = 8;
|
|
}
|
|
|
|
for (y = offset; y < height + offset; y++) {
|
|
/* Use a number lower than 1.0 else its too bright. */
|
|
hsv[2] = 0.1 + (y * (0.4 / total_height));
|
|
|
|
for (x = 0; x < width; x++) {
|
|
hsv[0] = float(double(x / hue_step) * 1.0 / width * hue_step);
|
|
hsv_to_rgb_v(hsv, rgb);
|
|
|
|
if (rect) {
|
|
rect[0] = char(rgb[0] * 255.0f);
|
|
rect[1] = char(rgb[1] * 255.0f);
|
|
rect[2] = char(rgb[2] * 255.0f);
|
|
rect[3] = 255;
|
|
|
|
rect += 4;
|
|
}
|
|
|
|
if (rect_float) {
|
|
rect_float[0] = rgb[0];
|
|
rect_float[1] = rgb[1];
|
|
rect_float[2] = rgb[2];
|
|
rect_float[3] = 1.0f;
|
|
|
|
rect_float += 4;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
static void checker_board_color_tint(
|
|
uchar *rect, float *rect_float, int width, int height, int size, float blend, int offset)
|
|
{
|
|
int x, y;
|
|
float blend_half = blend * 0.5f;
|
|
|
|
for (y = offset; y < height + offset; y++) {
|
|
for (x = 0; x < width; x++) {
|
|
if (((y / size) % 2 == 1 && (x / size) % 2 == 1) ||
|
|
((y / size) % 2 == 0 && (x / size) % 2 == 0)) {
|
|
if (rect) {
|
|
rect[0] = char(BLEND_CHAR(rect[0], blend));
|
|
rect[1] = char(BLEND_CHAR(rect[1], blend));
|
|
rect[2] = char(BLEND_CHAR(rect[2], blend));
|
|
rect[3] = 255;
|
|
|
|
rect += 4;
|
|
}
|
|
if (rect_float) {
|
|
rect_float[0] = BLEND_FLOAT(rect_float[0], blend);
|
|
rect_float[1] = BLEND_FLOAT(rect_float[1], blend);
|
|
rect_float[2] = BLEND_FLOAT(rect_float[2], blend);
|
|
rect_float[3] = 1.0f;
|
|
|
|
rect_float += 4;
|
|
}
|
|
}
|
|
else {
|
|
if (rect) {
|
|
rect[0] = char(BLEND_CHAR(rect[0], blend_half));
|
|
rect[1] = char(BLEND_CHAR(rect[1], blend_half));
|
|
rect[2] = char(BLEND_CHAR(rect[2], blend_half));
|
|
rect[3] = 255;
|
|
|
|
rect += 4;
|
|
}
|
|
if (rect_float) {
|
|
rect_float[0] = BLEND_FLOAT(rect_float[0], blend_half);
|
|
rect_float[1] = BLEND_FLOAT(rect_float[1], blend_half);
|
|
rect_float[2] = BLEND_FLOAT(rect_float[2], blend_half);
|
|
rect_float[3] = 1.0f;
|
|
|
|
rect_float += 4;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
static void checker_board_grid_fill(
|
|
uchar *rect, float *rect_float, int width, int height, float blend, int offset)
|
|
{
|
|
int x, y;
|
|
for (y = offset; y < height + offset; y++) {
|
|
for (x = 0; x < width; x++) {
|
|
if (((y % 32) == 0) || ((x % 32) == 0) || x == 0) {
|
|
if (rect) {
|
|
rect[0] = BLEND_CHAR(rect[0], blend);
|
|
rect[1] = BLEND_CHAR(rect[1], blend);
|
|
rect[2] = BLEND_CHAR(rect[2], blend);
|
|
rect[3] = 255;
|
|
|
|
rect += 4;
|
|
}
|
|
if (rect_float) {
|
|
rect_float[0] = BLEND_FLOAT(rect_float[0], blend);
|
|
rect_float[1] = BLEND_FLOAT(rect_float[1], blend);
|
|
rect_float[2] = BLEND_FLOAT(rect_float[2], blend);
|
|
rect_float[3] = 1.0f;
|
|
|
|
rect_float += 4;
|
|
}
|
|
}
|
|
else {
|
|
if (rect_float) {
|
|
rect_float += 4;
|
|
}
|
|
if (rect) {
|
|
rect += 4;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/* Defined in `image.cc`. */
|
|
|
|
static void checker_board_text(
|
|
uchar *rect, float *rect_float, int width, int height, int step, int outline)
|
|
{
|
|
int x, y;
|
|
int pen_x, pen_y;
|
|
char text[3] = {'A', '1', '\0'};
|
|
const int mono = blf_mono_font_render;
|
|
|
|
BLF_size(mono, 54.0f); /* hard coded size! */
|
|
|
|
/* OCIO_TODO: using nullptr as display will assume using sRGB display
|
|
* this is correct since currently generated images are assumed to be in sRGB space,
|
|
* but this would probably needed to be fixed in some way
|
|
*/
|
|
BLF_buffer(mono, rect_float, rect, width, height, 4, nullptr);
|
|
|
|
const float text_color[4] = {0.0, 0.0, 0.0, 1.0};
|
|
const float text_outline[4] = {1.0, 1.0, 1.0, 1.0};
|
|
|
|
const char char_array[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
|
|
/* Subtract one because of null termination. */
|
|
const int char_num = sizeof(char_array) - 1;
|
|
|
|
int first_char_index = 0;
|
|
for (y = 0; y < height; y += step) {
|
|
text[0] = char_array[first_char_index];
|
|
|
|
int second_char_index = 27;
|
|
for (x = 0; x < width; x += step) {
|
|
text[1] = char_array[second_char_index];
|
|
|
|
/* hard coded offset */
|
|
pen_x = x + 33;
|
|
pen_y = y + 44;
|
|
|
|
/* terribly crappy outline font! */
|
|
BLF_buffer_col(mono, text_outline);
|
|
|
|
BLF_position(mono, pen_x - outline, pen_y, 0.0);
|
|
BLF_draw_buffer(mono, text, 2);
|
|
BLF_position(mono, pen_x + outline, pen_y, 0.0);
|
|
BLF_draw_buffer(mono, text, 2);
|
|
BLF_position(mono, pen_x, pen_y - outline, 0.0);
|
|
BLF_draw_buffer(mono, text, 2);
|
|
BLF_position(mono, pen_x, pen_y + outline, 0.0);
|
|
BLF_draw_buffer(mono, text, 2);
|
|
|
|
BLF_position(mono, pen_x - outline, pen_y - outline, 0.0);
|
|
BLF_draw_buffer(mono, text, 2);
|
|
BLF_position(mono, pen_x + outline, pen_y + outline, 0.0);
|
|
BLF_draw_buffer(mono, text, 2);
|
|
BLF_position(mono, pen_x - outline, pen_y + outline, 0.0);
|
|
BLF_draw_buffer(mono, text, 2);
|
|
BLF_position(mono, pen_x + outline, pen_y - outline, 0.0);
|
|
BLF_draw_buffer(mono, text, 2);
|
|
|
|
BLF_buffer_col(mono, text_color);
|
|
BLF_position(mono, pen_x, pen_y, 0.0);
|
|
BLF_draw_buffer(mono, text, 2);
|
|
|
|
second_char_index = (second_char_index + 1) % char_num;
|
|
}
|
|
first_char_index = (first_char_index + 1) % char_num;
|
|
}
|
|
|
|
/* cleanup the buffer. */
|
|
BLF_buffer(mono, nullptr, nullptr, 0, 0, 0, nullptr);
|
|
}
|
|
|
|
static void checker_board_color_prepare_slice(
|
|
uchar *rect, float *rect_float, int width, int height, int offset, int total_height)
|
|
{
|
|
checker_board_color_fill(rect, rect_float, width, height, offset, total_height);
|
|
checker_board_color_tint(rect, rect_float, width, height, 1, 0.03f, offset);
|
|
checker_board_color_tint(rect, rect_float, width, height, 4, 0.05f, offset);
|
|
checker_board_color_tint(rect, rect_float, width, height, 32, 0.07f, offset);
|
|
checker_board_color_tint(rect, rect_float, width, height, 128, 0.15f, offset);
|
|
checker_board_grid_fill(rect, rect_float, width, height, 1.0f / 4.0f, offset);
|
|
}
|
|
|
|
struct FillCheckerColorThreadData {
|
|
uchar *rect;
|
|
float *rect_float;
|
|
int width, height;
|
|
};
|
|
|
|
static void checker_board_color_prepare_thread_do(void *data_v, int scanline)
|
|
{
|
|
FillCheckerColorThreadData *data = (FillCheckerColorThreadData *)data_v;
|
|
const int num_scanlines = 1;
|
|
size_t offset = size_t(data->width) * scanline * 4;
|
|
uchar *rect = (data->rect != nullptr) ? (data->rect + offset) : nullptr;
|
|
float *rect_float = (data->rect_float != nullptr) ? (data->rect_float + offset) : nullptr;
|
|
checker_board_color_prepare_slice(
|
|
rect, rect_float, data->width, num_scanlines, scanline, data->height);
|
|
}
|
|
|
|
void BKE_image_buf_fill_checker_color(uchar *rect, float *rect_float, int width, int height)
|
|
{
|
|
if (size_t(width) * height < 64 * 64) {
|
|
checker_board_color_prepare_slice(rect, rect_float, width, height, 0, height);
|
|
}
|
|
else {
|
|
FillCheckerColorThreadData data;
|
|
data.rect = rect;
|
|
data.rect_float = rect_float;
|
|
data.width = width;
|
|
data.height = height;
|
|
IMB_processor_apply_threaded_scanlines(height, checker_board_color_prepare_thread_do, &data);
|
|
}
|
|
|
|
checker_board_text(rect, rect_float, width, height, 128, 2);
|
|
|
|
if (rect_float != nullptr) {
|
|
/* TODO(sergey): Currently it's easier to fill in form buffer and
|
|
* linearize it afterwards. This could be optimized with some smart
|
|
* trickery around blending factors and such.
|
|
*/
|
|
IMB_buffer_float_from_float_threaded(rect_float,
|
|
rect_float,
|
|
4,
|
|
IB_PROFILE_LINEAR_RGB,
|
|
IB_PROFILE_SRGB,
|
|
true,
|
|
width,
|
|
height,
|
|
width,
|
|
width);
|
|
}
|
|
}
|