Subversion Repositories tpanel

/*

 * Copyright (C) 2020, 2021 by Andreas Theofilu <andreas@theosys.at>

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation; either version 3 of the License, or

 * (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301  USA

 */

#include <iostream>

#include <iomanip>

#include "tcolor.h"

#include "terror.h"

TPalette *TColor::mPalette = nullptr;

using std::string;

using std::vector;

using std::setw;

using std::setfill;

using std::hex;

using std::dec;

using std::resetiosflags;

TColor::COLOR_T TColor::getAMXColor(const string& color)

{

    DECL_TRACER("TColor::getAMXColor(const string& color)");

    if (color.empty())

        return TColor::COLOR_T();

    size_t pos = color.find('#');

    if (pos == string::npos)

    {

        if (!mPalette)

        {

            MSG_ERROR("No palette was set! First set a palette to be able to get any color!");

            TError::setError();

            return TColor::COLOR_T();

        }

        PDATA_T pd = mPalette->findColor(color);

        if (!pd.name.empty())

            return splitColors(pd);

        return TColor::COLOR_T();

    }

    // Check if we've an index number

    if (color.length() <= 2 && isdigit(color[0]))

    {

        int idx = atoi(color.c_str());

        if (idx < 0 || idx > 88)

            return TColor::COLOR_T();

        PDATA_T pd = mPalette->findColor(idx);

        if (!pd.name.empty())

            return splitColors(pd);

        return TColor::COLOR_T();

    }

    TColor::COLOR_T ct;

    ct.red = (int)strtol(color.substr(pos+1, 2).c_str(), NULL, 16);

    ct.green = (int)strtol(color.substr(pos+3, 2).c_str(), NULL, 16);

    ct.blue = (int)strtol(color.substr(pos+5, 2).c_str(), NULL, 16);

    if (color.length() > 7)

        ct.alpha = (int)strtol(color.substr(pos+7).c_str(), NULL, 16);

    else

        ct.alpha = 0x00ff;

    return ct;

}

TColor::COLOR_T TColor::splitColors(PDATA_T& pd)

{

    DECL_TRACER("TColor::splitColors(PDATA_T& pd)");

    TColor::COLOR_T ct;

    if (pd.color > 0x00ffffff)

    {

        ct.red = (pd.color & 0xff000000) >> 24;

        ct.green = (pd.color & 0x00ff0000) >> 16;

        ct.blue = (pd.color & 0x0000ff00) >> 8;

        ct.alpha = (pd.color & 0x000000ff);

    }

    else

    {

        ct.red = (pd.color & 0xff000000) >> 24;

        ct.green = (pd.color & 0x00ff0000) >> 16;

        ct.blue = (pd.color & 0x0000ff00) >> 8;

        ct.alpha = 0x00ff;

    }

    return ct;

}

SkColor TColor::getSkiaColor(const std::string& color)

{

    DECL_TRACER("TColor::getSkiaColor(const std::string& color)");

    COLOR_T col = getAMXColor(color);

#ifdef __ANDROID__

    return SkColorSetARGB(col.alpha, col.blue, col.green, col.red); // Workaround for a bug in Skia

#else

    return SkColorSetARGB(col.alpha, col.red, col.green, col.blue);

#endif

}

ulong TColor::getColor(const std::string& color)

{

    DECL_TRACER("TColor::getColor(const std::string& color)");

    TColor::COLOR_T ct = getAMXColor(color);

    ulong col = ((ct.red << 24) & 0xff000000) | ((ct.green << 16) & 0x00ff0000) | ((ct.blue << 8) & 0x0000ff00) | (ct.alpha & 0x000000ff);

    return col;

}

std::string TColor::colorToString(ulong color)

{

    PDATA_T pd;

    pd.color = color;

    TColor::COLOR_T col = splitColors(pd);

    std::stringstream s;

    s << std::setw(2) << std::setfill('0') << std::hex << col.red << ":";

    s << std::setw(2) << std::setfill('0') << std::hex << col.green << ":";

    s << std::setw(2) << std::setfill('0') << std::hex << col.blue << ":";

    s << std::setw(2) << std::setfill('0') << std::hex << col.alpha;

    std::string ret = s.str();

    return ret;

}

std::string TColor::colorToString(SkColor color)

{

    std::stringstream s;

    s << std::setw(2) << std::setfill('0') << std::hex << SkColorGetR(color) << ":";

    s << std::setw(2) << std::setfill('0') << std::hex << SkColorGetG(color) << ":";

    s << std::setw(2) << std::setfill('0') << std::hex << SkColorGetB(color) << ":";

    s << std::setw(2) << std::setfill('0') << std::hex << SkColorGetA(color);

    std::string ret = s.str();

    return ret;

}

std::vector<SkColor> TColor::colorRange(SkColor col, int width, int bandwidth, DIRECTION_t dir)

{

    DECL_TRACER("TColor::colorRange(SkColor col, int width, int bandwidth)");

    int red, green, blue, alpha, colStep;

    vector<SkColor> colRange;

    SkColor color;

    alpha = SkColorGetA(col);

    // Define the start color

    if (dir == DIR_LIGHT_DARK || dir == DIR_LIGHT_DARK_LIGHT)

    {

        red = std::max((int)SkColorGetR(col) - bandwidth, 0);

        green = std::max((int)SkColorGetG(col) - bandwidth, 0);

        blue = std::max((int)SkColorGetB(col) - bandwidth, 0);

        if (dir == DIR_LIGHT_DARK)

            colStep = bandwidth / width;

        else

            colStep = bandwidth / (width / 2);

    }

    else

    {

        red = std::min((int)SkColorGetR(col) + bandwidth, 255);

        green = std::min((int)SkColorGetG(col) + bandwidth, 255);

        blue = std::min((int)SkColorGetB(col) + bandwidth, 255);

        if (dir == DIR_DARK_LIGHT)

            colStep = bandwidth / width;

        else

            colStep = bandwidth / (width / 2);

    }

    if (colStep <= 1)

    {

        colRange.push_back(col);

        return colRange;

    }

    switch(dir)

    {

        case DIR_LIGHT_DARK:

            for (int i = 0; i < width; i++)

            {

                red -= colStep * i;

                green -= colStep * i;

                blue -= colStep * i;

                color = SkColorSetARGB(alpha, red, green, blue);

                colRange.push_back(color);

            }

        break;

        case DIR_DARK_LIGHT:

            for (int i = 0; i < width; i++)

            {

                red += colStep * i;

                green += colStep * i;

                blue += colStep * i;

                color = SkColorSetARGB(alpha, red, green, blue);

                colRange.push_back(color);

            }

        break;

        case DIR_LIGHT_DARK_LIGHT:

            for (int i = 0; i < (width / 2); i++)

            {

                red -= colStep * i;

                green -= colStep * i;

                blue -= colStep * i;

                color = SkColorSetARGB(alpha, red, green, blue);

                colRange.push_back(color);

            }

            for (int i = 0; i < (width / 2); i++)

            {

                red += colStep * i;

                green += colStep * i;

                blue += colStep * i;

                color = SkColorSetARGB(alpha, red, green, blue);

                colRange.push_back(color);

            }

        break;

        case DIR_DARK_LIGHT_DARK:

            for (int i = 0; i < (width / 2); i++)

            {

                red += colStep * i;

                green += colStep * i;

                blue += colStep * i;

                color = SkColorSetARGB(alpha, red, green, blue);

                colRange.push_back(color);

            }

            for (int i = 0; i < (width / 2); i++)

            {

                red -= colStep * i;

                green -= colStep * i;

                blue -= colStep * i;

                color = SkColorSetARGB(alpha, red, green, blue);

                colRange.push_back(color);

            }

        break;

    }

    return colRange;

}

bool TColor::isValidAMXcolor(const string& color)

{

    DECL_TRACER("TColor::isValidAMXcolor(const string& color)");

    if (color.empty())

        return false;

    size_t pos = color.find('#');

    if (pos == string::npos)

    {

        if (!mPalette)

        {

            MSG_ERROR("No palette was set! First set a palette to be able to get any color!");

            return false;

        }

        PDATA_T pd = mPalette->findColor(color);

        if (!pd.name.empty())

            return true;

        return false;

    }

    // Check if we've an index number

    if (color.length() <= 2 && isdigit(color[0]))

    {

        int idx = atoi(color.c_str());

        if (idx < 0 || idx > 88)

            return false;

        return true;

    }

    if (color.length() >= 7)    // #RRGGBBAA

        return true;

    return false;

}

int TColor::setAlpha(ulong color, int alpha)

{

    DECL_TRACER("TColor::setAlpha(ulong color, int alpha)");

    if (alpha > 0x00ff)

        alpha = 0x00ff;

    else if (alpha < 0)

        alpha = 0;

    return ((color & 0xffffff00) | (alpha & 0x00ff));

}

SkColor TColor::setAlpha(SkColor color, int alpha)

{

    DECL_TRACER("TColor::setAlpha(SkColor color, int alpha)");

    if (alpha > 0x00ff)

        alpha = 0x00ff;

    else if (alpha < 0)

        alpha = 0;

    SkColor red = SkColorGetR(color);

    SkColor green = SkColorGetG(color);

    SkColor blue = SkColorGetB(color);

    return SkColorSetARGB(alpha, red, green, blue);

}

int TColor::calcAlpha(int alpha1, int alpha2)

{

    DECL_TRACER("TColor::calcAlpha(int alpha1, int alpha2)");

    return (alpha1 + alpha2) / 2;

}

ulong TColor::setAlphaTreshold(ulong color, int alpha)

{

    DECL_TRACER("TColor::setAlphaTreshold(ulong color, int alpha)");

    uint al = color & 0x000000ff;

    if (al > (uint)alpha)

        return (color & 0xffffff00) | (alpha & 0x00ff);

    return color;

}

SkColor TColor::setAlphaTreshold(SkColor color, int alpha)

{

    DECL_TRACER("TColor::setAlphaTreshold(SkColor color, int alpha)");

    int al = SkColorGetA(color);

    if (al > alpha)

        return SkColorSetA(color, alpha);

    return color;

}