// Copyright (c) 2005, Niels Martin Hansen, Rodrigo Braz Monteiro // All rights reserved. // // Redistribution and use in source and binary forms, with or without // modification, are permitted provided that the following conditions are met: // // * Redistributions of source code must retain the above copyright notice, // this list of conditions and the following disclaimer. // * Redistributions in binary form must reproduce the above copyright notice, // this list of conditions and the following disclaimer in the documentation // and/or other materials provided with the distribution. // * Neither the name of the Aegisub Group nor the names of its contributors // may be used to endorse or promote products derived from this software // without specific prior written permission. // // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" // AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE // ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE // LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR // CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF // SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS // INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN // CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) // ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE // POSSIBILITY OF SUCH DAMAGE. // // Aegisub Project http://www.aegisub.org/ // // $Id$ /// @file colorspace.cpp /// @brief Functions for converting colours between different representations /// @ingroup utility /// #include "config.h" #include "colorspace.h" #include "utils.h" // Matrix from http://forum.doom9.org/showthread.php?p=684080#post684080 void yuv_to_rgb(int Y, int U, int V, unsigned char *R, unsigned char *G, unsigned char *B) { U = U - 128; V = V - 128; *R = clip_colorval((Y*65536 + int(1.140*65536) * V) / 65536); *G = clip_colorval((Y*65536 - int(0.395*65536) * U - int(0.581*65536) * V) / 65536); *B = clip_colorval((Y*65536 + int(2.032*65536) * U ) / 65536); } // Algorithm from http://130.113.54.154/~monger/hsl-rgb.html void hsl_to_rgb(int H, int S, int L, unsigned char *R, unsigned char *G, unsigned char *B) { if (S == 0) { *R = L; *G = L; *B = L; return; } if (L == 128 && S == 255) { switch (H) { case 0: case 255: // actually this is wrong, since this is more like 359 degrees... but it's what you'd expect (sadly :) *R = 255; *G = 0; *B = 0; return; case 43: *R = 255; *G = 255; *B = 0; return; case 85: *R = 0; *G = 255; *B = 0; return; case 128: *R = 0; *G = 255; *B = 255; return; case 171: *R = 0; *G = 0; *B = 255; return; case 213: *R = 255; *G = 0; *B = 255; return; } } float h, s, l, r, g, b; h = H / 255.f; s = S / 255.f; l = L / 255.f; float temp2; if (l < .5) { temp2 = l * (1. + s); } else { temp2 = l + s - l*s; } float temp1 = 2.f * l - temp2; // assume h is in range [0;1] float temp3[3]; temp3[0] = h + 1.f/3.f; if (temp3[0] > 1.f) temp3[0] -= 1.f; temp3[1] = h; temp3[2] = h - 1.f/3.f; if (temp3[2] < 0.f) temp3[2] += 1.f; if (6.f * temp3[0] < 1.f) r = temp1 + (temp2 - temp1) * 6.f * temp3[0]; else if (2.f * temp3[0] < 1.f) r = temp2; else if (3.f * temp3[0] < 2.f) r = temp1 + (temp2 - temp1) * ((2.f/3.f) - temp3[0]) * 6.f; else r = temp1; if (6.f * temp3[1] < 1.f) g = temp1 + (temp2 - temp1) * 6.f * temp3[1]; else if (2.f * temp3[1] < 1.f) g = temp2; else if (3.f * temp3[1] < 2.f) g = temp1 + (temp2 - temp1) * ((2.f/3.f) - temp3[1]) * 6.f; else g = temp1; if (6.f * temp3[2] < 1.f) b = temp1 + (temp2 - temp1) * 6.f * temp3[2]; else if (2.f * temp3[2] < 1.f) b = temp2; else if (3.f * temp3[2] < 2.f) b = temp1 + (temp2 - temp1) * ((2.f/3.f) - temp3[2]) * 6.f; else b = temp1; *R = clip_colorval((int)(r*255)); *G = clip_colorval((int)(g*255)); *B = clip_colorval((int)(b*255)); } // Formulas taken from wikipedia: http://en.wikipedia.org/wiki/HSV_color_space // The range for H is 0..255 instead of 0..359, so 60 degrees has been translated to 256/6 here void hsv_to_rgb(int H, int S, int V, unsigned char *R, unsigned char *G, unsigned char *B) { *R = *G = *B = 0; // some special cases... oh yeah baby! if (S == 255) { switch (H) { case 0: case 255: // actually this is wrong, since this is more like 359 degrees... but it's what you'd expect (sadly :) *R = V; *G = 0; *B = 0; return; case 43: *R = V; *G = V; *B = 0; return; case 85: *R = 0; *G = V; *B = 0; return; case 128: *R = 0; *G = V; *B = V; return; case 171: *R = 0; *G = 0; *B = V; return; case 213: *R = V; *G = 0; *B = V; return; } } // Cap values unsigned int h = H * 360; unsigned int s = clip_colorval(S)*256; unsigned int v = clip_colorval(V)*256; // Saturation is zero, make grey if (S == 0) { *R = V; *G = V; *B = V; return; } unsigned int r, g, b; // Else, calculate color unsigned int Hi = h / 60 / 256; unsigned int f = h / 60 - Hi * 256; unsigned int p = v * (65535 - s) / 65536; unsigned int q = v * (65535 - (f * s)/256) / 65536; unsigned int t = v * (65535 - ((255 - f) * s)/256) / 65536; switch (Hi) { case 0: r = v; g = t; b = p; break; case 1: r = q; g = v; b = p; break; case 2: r = p; g = v; b = t; break; case 3: r = p; g = q; b = v; break; case 4: r = t; g = p; b = v; break; case 5: default: r = v; g = p; b = q; break; } *R = clip_colorval(r/256); *G = clip_colorval(g/256); *B = clip_colorval(b/256); } // Matrix from http://forum.doom9.org/showthread.php?p=684080#post684080 void rgb_to_yuv(int R, int G, int B, unsigned char *Y, unsigned char *U, unsigned char *V) { *Y = clip_colorval(( int(0.299*65536) * R + int(0.587*65536) * G + int(0.114*65536) * B) / 65536); *U = clip_colorval((-int(0.147*65536) * R - int(0.289*65536) * G + int(0.436*65536) * B) / 65536 + 128); *V = clip_colorval(( int(0.615*65536) * R - int(0.515*65536) * G - int(0.100*65536) * B) / 65536 + 128); } /// Algorithm from http://130.113.54.154/~monger/hsl-rgb.html void rgb_to_hsl(int R, int G, int B, unsigned char *H, unsigned char *S, unsigned char *L) { float r = R/255.f, g = G/255.f, b = B/255.f; float h, s, l; float maxrgb = std::max(r, std::max(g, b)), minrgb = std::min(r, std::min(g, b)); l = (minrgb + maxrgb) / 2; if (minrgb == maxrgb) { h = 0; s = 0; } else { if (l < 0.5) { s = (maxrgb - minrgb) / (maxrgb + minrgb); } else { s = (maxrgb - minrgb) / (2.f - maxrgb - minrgb); } if (r == maxrgb) { h = (g-b) / (maxrgb-minrgb) + 0; } else if (g == maxrgb) { h = (b-r) / (maxrgb-minrgb) + 2; } else { // if b == maxrgb h = (r-g) / (maxrgb-minrgb) + 4; } } if (h < 0) h += 6; if (h >= 6) h -= 6; *H = clip_colorval(int(h*256/6)); *S = clip_colorval(int(s*255)); *L = clip_colorval(int(l*255)); } /// Formulas from http://en.wikipedia.org/wiki/HSV_color_space void rgb_to_hsv(int R, int G, int B, unsigned char *H, unsigned char *S, unsigned char *V) { float r = R/255.f, g = G/255.f, b = B/255.f; float h, s, v; float maxrgb = std::max(r, std::max(g, b)), minrgb = std::min(r, std::min(g, b)); v = maxrgb; if (maxrgb < .001f) { s = 1; } else { s = (maxrgb - minrgb) / maxrgb; } if (minrgb == maxrgb) { h = 0; } else if (maxrgb == r) { h = (g-b) / (maxrgb-minrgb) + 0; } else if (maxrgb == g) { h = (b-r) / (maxrgb-minrgb) + 2; } else { // if maxrgb == b h = (r-g) / (maxrgb-minrgb) + 4; } if (h < 0) h += 6; if (h >= 6) h -= 6; *H = clip_colorval(int(h*256/6)); *S = clip_colorval(int(s*255)); *V = clip_colorval(int(v*255)); } void hsv_to_hsl(int iH, int iS, int iV, unsigned char *oH, unsigned char *oS, unsigned char *oL) { int p = iV * (255 - iS); *oH = iH; *oL = clip_colorval((p + iV*255)/255/2); if (*oL == 0) { *oS = iS; // oS is actually undefined, so any value should work ;) } else if (*oL <= 128) { *oS = clip_colorval((iV*255 - p) / (2 * *oL)); } else { *oS = clip_colorval((iV*255 - p) / (511 - 2 * *oL)); } } void hsl_to_hsv(int iH, int iS, int iL, unsigned char *oH, unsigned char *oS, unsigned char *oV) { *oH = iH; if (iS == 0) { *oS = 0; *oV = iL; return; } if (iL < 128) { *oV = iL * (255 + iS) / 255; *oS = 2 * 255 * iS / (255 + iS); } else { *oV = (iL*255 + iS*255 - iL*iS)/255; *oS = 2 * 255 * iS * (255 - iL) / (iL*255 + iS*255 - iL*iS); } } wxString color_to_html(wxColour color) { return wxString::Format("#%02X%02X%02X", color.Red(), color.Green(), color.Blue()); } wxColour html_to_color(wxString html) { html.Trim(true); html.Trim(false); if (html.StartsWith("#")) { html.Remove(0, 1); } if (html.size() == 6) { // 8 bit per channel long r, g, b; wxString sr, sg, sb; sr = html.Mid(0, 2); sg = html.Mid(2, 2); sb = html.Mid(4, 2); if (sr.ToLong(&r, 16) && sg.ToLong(&g, 16) && sb.ToLong(&b, 16)) { return wxColour(r, g, b); } else { return wxColour(*wxBLACK); } } else if (html.size() == 3) { // 4 bit per channel long r, g, b; wxString sr, sg, sb; sr = html.Mid(0, 1); sg = html.Mid(1, 1); sb = html.Mid(2, 1); if (sr.ToLong(&r, 16) && sg.ToLong(&g, 16) && sb.ToLong(&b, 16)) { return wxColour(r*16+r, g*16+g, b*16+b); } else { return wxColour(*wxBLACK); } } else { // only care about valid colors return wxColour(*wxBLACK); } }