circuitpython/shared-module/displayio/ColorConverter.c

218 lines
8.0 KiB
C
Raw Normal View History

/*
* This file is part of the Micro Python project, http://micropython.org/
*
* The MIT License (MIT)
*
* Copyright (c) 2018 Scott Shawcroft for Adafruit Industries
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
#include "shared-bindings/displayio/ColorConverter.h"
2019-08-23 18:27:21 -04:00
#include "py/misc.h"
#include "py/runtime.h"
2019-08-23 18:27:21 -04:00
#define NO_TRANSPARENT_COLOR (0x1000000)
2021-03-15 09:57:36 -04:00
uint32_t displayio_colorconverter_dither_noise_1(uint32_t n) {
n = (n >> 13) ^ n;
int nn = (n * (n * n * 60493 + 19990303) + 1376312589) & 0x7fffffff;
return (uint32_t)(((float)nn / (1073741824.0f * 2)) * 255);
2019-08-31 22:07:09 -04:00
}
uint32_t displayio_colorconverter_dither_noise_2(uint32_t x, uint32_t y) {
return displayio_colorconverter_dither_noise_1(x + y * 0xFFFF);
}
2021-03-15 09:57:36 -04:00
void common_hal_displayio_colorconverter_construct(displayio_colorconverter_t *self, bool dither) {
self->dither = dither;
self->transparent_color = NO_TRANSPARENT_COLOR;
}
uint16_t displayio_colorconverter_compute_rgb565(uint32_t color_rgb888) {
uint32_t r5 = (color_rgb888 >> 19);
uint32_t g6 = (color_rgb888 >> 10) & 0x3f;
uint32_t b5 = (color_rgb888 >> 3) & 0x1f;
2020-03-10 14:12:01 -04:00
return r5 << 11 | g6 << 5 | b5;
}
uint8_t displayio_colorconverter_compute_luma(uint32_t color_rgb888) {
uint32_t r8 = (color_rgb888 >> 16);
uint32_t g8 = (color_rgb888 >> 8) & 0xff;
uint32_t b8 = color_rgb888 & 0xff;
return (r8 * 19 + g8 * 182 + b8 * 54) / 255;
}
2019-08-02 19:17:38 -04:00
uint8_t displayio_colorconverter_compute_chroma(uint32_t color_rgb888) {
uint32_t r8 = (color_rgb888 >> 16);
uint32_t g8 = (color_rgb888 >> 8) & 0xff;
uint32_t b8 = color_rgb888 & 0xff;
2019-08-23 18:27:21 -04:00
uint8_t max = MAX(r8, MAX(g8, b8));
uint8_t min = MIN(r8, MIN(g8, b8));
2019-08-02 19:17:38 -04:00
return max - min;
}
uint8_t displayio_colorconverter_compute_hue(uint32_t color_rgb888) {
uint32_t r8 = (color_rgb888 >> 16);
uint32_t g8 = (color_rgb888 >> 8) & 0xff;
uint32_t b8 = color_rgb888 & 0xff;
2019-08-23 18:27:21 -04:00
uint8_t max = MAX(r8, MAX(g8, b8));
uint8_t min = MIN(r8, MIN(g8, b8));
2019-08-02 19:17:38 -04:00
uint8_t c = max - min;
if (c == 0) {
return 0;
}
int32_t hue = 0;
if (max == r8) {
2021-03-15 09:57:36 -04:00
hue = (((int32_t)(g8 - b8) * 40) / c) % 240;
2019-08-02 19:17:38 -04:00
} else if (max == g8) {
2021-03-15 09:57:36 -04:00
hue = (((int32_t)(b8 - r8) + (2 * c)) * 40) / c;
2019-08-02 19:17:38 -04:00
} else if (max == b8) {
2021-03-15 09:57:36 -04:00
hue = (((int32_t)(r8 - g8) + (4 * c)) * 40) / c;
2019-08-02 19:17:38 -04:00
}
if (hue < 0) {
hue += 240;
}
return hue;
}
2021-03-15 09:57:36 -04:00
void displayio_colorconverter_compute_tricolor(const _displayio_colorspace_t *colorspace, uint8_t pixel_hue, uint8_t pixel_luma, uint32_t *color) {
2019-08-02 19:17:38 -04:00
int16_t hue_diff = colorspace->tricolor_hue - pixel_hue;
if ((-10 <= hue_diff && hue_diff <= 10) || hue_diff <= -220 || hue_diff >= 220) {
if (colorspace->grayscale) {
*color = 0;
} else {
*color = 1;
}
} else if (!colorspace->grayscale) {
*color = 0;
}
}
2021-03-15 09:57:36 -04:00
void common_hal_displayio_colorconverter_convert(displayio_colorconverter_t *self, const _displayio_colorspace_t *colorspace, uint32_t input_color, uint32_t *output_color) {
2019-08-31 22:07:09 -04:00
displayio_input_pixel_t input_pixel;
input_pixel.pixel = input_color;
input_pixel.x = input_pixel.y = input_pixel.tile = input_pixel.tile_x = input_pixel.tile_y = 0;
2019-08-31 22:07:09 -04:00
displayio_output_pixel_t output_pixel;
output_pixel.pixel = 0;
output_pixel.opaque = false;
displayio_colorconverter_convert(self, colorspace, &input_pixel, &output_pixel);
(*output_color) = output_pixel.pixel;
}
2021-03-15 09:57:36 -04:00
void common_hal_displayio_colorconverter_set_dither(displayio_colorconverter_t *self, bool dither) {
self->dither = dither;
}
2021-03-15 09:57:36 -04:00
bool common_hal_displayio_colorconverter_get_dither(displayio_colorconverter_t *self) {
return self->dither;
}
2021-03-15 09:57:36 -04:00
void common_hal_displayio_colorconverter_make_transparent(displayio_colorconverter_t *self, uint32_t transparent_color) {
if (self->transparent_color != NO_TRANSPARENT_COLOR) {
mp_raise_RuntimeError(translate("Only one color can be transparent at a time"));
}
self->transparent_color = transparent_color;
}
2021-03-15 09:57:36 -04:00
void common_hal_displayio_colorconverter_make_opaque(displayio_colorconverter_t *self, uint32_t transparent_color) {
(void)transparent_color;
// NO_TRANSPARENT_COLOR will never equal a valid color
self->transparent_color = NO_TRANSPARENT_COLOR;
}
2021-03-15 09:57:36 -04:00
void displayio_colorconverter_convert(displayio_colorconverter_t *self, const _displayio_colorspace_t *colorspace, const displayio_input_pixel_t *input_pixel, displayio_output_pixel_t *output_color) {
2019-08-31 22:07:09 -04:00
uint32_t pixel = input_pixel->pixel;
if (self->transparent_color == pixel) {
output_color->opaque = false;
return;
}
2021-03-15 09:57:36 -04:00
if (self->dither) {
2019-08-31 22:07:09 -04:00
uint8_t randr = (displayio_colorconverter_dither_noise_2(input_pixel->tile_x,input_pixel->tile_y));
2021-03-15 09:57:36 -04:00
uint8_t randg = (displayio_colorconverter_dither_noise_2(input_pixel->tile_x + 33,input_pixel->tile_y));
uint8_t randb = (displayio_colorconverter_dither_noise_2(input_pixel->tile_x,input_pixel->tile_y + 33));
2019-08-31 22:07:09 -04:00
uint32_t r8 = (pixel >> 16);
uint32_t g8 = (pixel >> 8) & 0xff;
uint32_t b8 = pixel & 0xff;
if (colorspace->depth == 16) {
2021-03-15 09:57:36 -04:00
b8 = MIN(255,b8 + (randb & 0x07));
r8 = MIN(255,r8 + (randr & 0x07));
g8 = MIN(255,g8 + (randg & 0x03));
2019-08-31 22:07:09 -04:00
} else {
int bitmask = 0xFF >> colorspace->depth;
2020-10-26 19:58:00 -04:00
b8 = MIN(255,b8 + (randb & bitmask));
r8 = MIN(255,r8 + (randr & bitmask));
g8 = MIN(255,g8 + (randg & bitmask));
2019-08-31 22:07:09 -04:00
}
pixel = r8 << 16 | g8 << 8 | b8;
}
if (colorspace->depth == 16) {
2020-03-10 14:12:01 -04:00
uint16_t packed = displayio_colorconverter_compute_rgb565(pixel);
if (colorspace->reverse_bytes_in_word) {
// swap bytes
packed = __builtin_bswap16(packed);
}
output_color->pixel = packed;
2019-08-31 22:07:09 -04:00
output_color->opaque = true;
return;
2019-08-02 19:17:38 -04:00
} else if (colorspace->tricolor) {
2019-08-31 22:07:09 -04:00
uint8_t luma = displayio_colorconverter_compute_luma(pixel);
output_color->pixel = luma >> (8 - colorspace->depth);
if (displayio_colorconverter_compute_chroma(pixel) <= 16) {
2019-08-02 19:17:38 -04:00
if (!colorspace->grayscale) {
2019-08-31 22:07:09 -04:00
output_color->pixel = 0;
2019-08-02 19:17:38 -04:00
}
2019-08-31 22:07:09 -04:00
output_color->opaque = true;
return;
2019-08-02 19:17:38 -04:00
}
2019-08-31 22:07:09 -04:00
uint8_t pixel_hue = displayio_colorconverter_compute_hue(pixel);
displayio_colorconverter_compute_tricolor(colorspace, pixel_hue, luma, &output_color->pixel);
return;
2021-03-15 09:57:36 -04:00
} else if (colorspace->grayscale && colorspace->depth <= 8) {
2019-08-31 22:07:09 -04:00
uint8_t luma = displayio_colorconverter_compute_luma(pixel);
2020-10-26 19:58:00 -04:00
size_t bitmask = (1 << colorspace->depth) - 1;
output_color->pixel = (luma >> colorspace->grayscale_bit) & bitmask;
2019-08-31 22:07:09 -04:00
output_color->opaque = true;
return;
}
2019-08-31 22:07:09 -04:00
output_color->opaque = false;
}
// Currently no refresh logic is needed for a ColorConverter.
bool displayio_colorconverter_needs_refresh(displayio_colorconverter_t *self) {
return false;
}
void displayio_colorconverter_finish_refresh(displayio_colorconverter_t *self) {
}