mirror of
https://github.com/OrcaSlicer/OrcaSlicer.git
synced 2026-07-24 03:12:07 +00:00
ENH: supporting encoded filament color
jira: [STUDIO-12346] Change-Id: I7f5ce7806acb6fdeb3e3d9db52a0b96e5fadd759 (cherry picked from commit a7bdc2707d3825327258965c90c33836a7da628b)
This commit is contained in:
xin.zhang
243
src/slic3r/GUI/EncodedFilament.hpp
Normal file
243
src/slic3r/GUI/EncodedFilament.hpp
Normal file
@@ -0,0 +1,243 @@
|
||||
#ifndef slic3r_ENCODED_FILAMENT_hpp_
|
||||
#define slic3r_ENCODED_FILAMENT_hpp_
|
||||
|
||||
#include <vector>
|
||||
#include <map>
|
||||
#include <unordered_map>
|
||||
#include <unordered_set>
|
||||
|
||||
#include <wx/colour.h>
|
||||
#include <wx/string.h>
|
||||
|
||||
#include <chrono>
|
||||
|
||||
namespace Slic3r
|
||||
{
|
||||
|
||||
//Previous definitions
|
||||
class FilamentColorCode;
|
||||
class FilamentColorCodes;
|
||||
class FilamentColorCodeQuery;
|
||||
|
||||
// Hasher
|
||||
struct FilamentColorHasher
|
||||
{
|
||||
std::size_t operator()(const wxColour& c) const noexcept {
|
||||
return (static_cast<std::size_t>(c.Red()) << 24) ^
|
||||
(static_cast<std::size_t>(c.Green()) << 16) ^
|
||||
(static_cast<std::size_t>(c.Blue()) << 8) ^
|
||||
(static_cast<std::size_t>(c.Alpha()));
|
||||
}
|
||||
};
|
||||
|
||||
struct FilamentColor
|
||||
{
|
||||
enum class ColorType : char
|
||||
{
|
||||
SINGLE_CLR = 0, // single color filament
|
||||
MULTI_CLR, // multi-color filament
|
||||
GRADIENT_CLR, // gradient filament
|
||||
};
|
||||
|
||||
ColorType m_color_type = ColorType::SINGLE_CLR; // default to single color
|
||||
std::unordered_set<wxColour, FilamentColorHasher> m_colors;
|
||||
|
||||
public:
|
||||
size_t ColorCount() const noexcept { return m_colors.size(); }
|
||||
|
||||
void EndSet(int ctype)
|
||||
{
|
||||
if (m_colors.size() < 2)
|
||||
{
|
||||
m_color_type = ColorType::SINGLE_CLR;
|
||||
}
|
||||
else
|
||||
{
|
||||
if (ctype == 0)
|
||||
{
|
||||
m_color_type = ColorType::GRADIENT_CLR;
|
||||
}
|
||||
else
|
||||
{
|
||||
m_color_type = ColorType::MULTI_CLR;
|
||||
}
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
// Represents a color in HSV format
|
||||
struct ColourHSV
|
||||
{
|
||||
double h, s, v;
|
||||
};
|
||||
|
||||
inline ColourHSV wxColourToHSV(const wxColour& c)
|
||||
{
|
||||
double r = c.Red() / 255.0;
|
||||
double g = c.Green() / 255.0;
|
||||
double b = c.Blue() / 255.0;
|
||||
|
||||
double maxc = std::max({ r, g, b });
|
||||
double minc = std::min({ r, g, b });
|
||||
double delta = maxc - minc;
|
||||
|
||||
double h = 0, s = 0, v = maxc;
|
||||
if (delta > 0.00001) {
|
||||
if (maxc == r) {
|
||||
h = 60.0 * (fmod(((g - b) / delta), 6.0));
|
||||
} else if (maxc == g) {
|
||||
h = 60.0 * (((b - r) / delta) + 2.0);
|
||||
} else {
|
||||
h = 60.0 * (((r - g) / delta) + 4.0);
|
||||
}
|
||||
|
||||
if (h < 0) h += 360.0;
|
||||
s = delta / maxc;
|
||||
} else {
|
||||
h = 0;
|
||||
s = 0;
|
||||
}
|
||||
return { h, s, v };
|
||||
}
|
||||
|
||||
// Compare function for EncodedFilaColor
|
||||
struct EncodedFilaColorEqual
|
||||
{
|
||||
bool operator()(const FilamentColor& lhs, const FilamentColor& rhs) const noexcept
|
||||
{
|
||||
if (lhs.ColorCount() != rhs.ColorCount()) { return lhs.ColorCount() < rhs.ColorCount(); };
|
||||
|
||||
if (lhs.ColorCount() == 1)
|
||||
{
|
||||
ColourHSV ha = wxColourToHSV(*lhs.m_colors.begin());
|
||||
ColourHSV hb = wxColourToHSV(*rhs.m_colors.begin());
|
||||
if (ha.h != hb.h) return ha.h < hb.h;
|
||||
if (ha.s != hb.s) return ha.s < hb.s;
|
||||
if (ha.v != hb.v) return ha.v < hb.v;
|
||||
}
|
||||
|
||||
if (lhs.m_color_type != rhs.m_color_type)
|
||||
{
|
||||
return lhs.m_color_type < rhs.m_color_type;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
private:
|
||||
double hue(const wxColour& colour) const
|
||||
{
|
||||
double r_norm = colour.Red() / 255.0;
|
||||
double g_norm = colour.Green() / 255.0;
|
||||
double b_norm = colour.Blue() / 255.0;
|
||||
|
||||
double max_val = std::max({ r_norm, g_norm, b_norm });
|
||||
double min_val = std::min({ r_norm, g_norm, b_norm });
|
||||
double delta = max_val - min_val;
|
||||
|
||||
if (delta == 0) return 0;
|
||||
|
||||
double h;
|
||||
if (max_val == r_norm) {
|
||||
h = std::fmod(((g_norm - b_norm) / delta), 6.0);
|
||||
} else if (max_val == g_norm) {
|
||||
h = ((b_norm - r_norm) / delta) + 2;
|
||||
} else {
|
||||
h = ((r_norm - g_norm) / delta) + 4;
|
||||
}
|
||||
|
||||
h *= 60;
|
||||
if (h < 0) h += 360;
|
||||
return h;
|
||||
}
|
||||
};
|
||||
using FilamentColor2CodeMap = std::map<FilamentColor, FilamentColorCode*, EncodedFilaColorEqual>;
|
||||
|
||||
|
||||
// FilamentColorCodeQuery class is used to query filament color codes and their associated information
|
||||
class FilamentColorCodeQuery
|
||||
{
|
||||
public:
|
||||
FilamentColorCodeQuery();
|
||||
virtual ~FilamentColorCodeQuery();
|
||||
|
||||
public:
|
||||
FilamentColorCodes* GetFilaInfoMap(const wxString& fila_id) const;
|
||||
wxString GetFilaColorName(const wxString& fila_id, const FilamentColor& colors) const;
|
||||
|
||||
private:
|
||||
FilamentColorCode* GetFilaInfo(const wxString& fila_id, const FilamentColor& colors) const;
|
||||
|
||||
protected:
|
||||
void LoadFromLocal();
|
||||
|
||||
public:
|
||||
void CreateFilaCode(const wxString& fila_id,
|
||||
const wxString& fila_type,
|
||||
const wxString& fila_color_code,
|
||||
FilamentColor&& fila_color,
|
||||
std::unordered_map<wxString, wxString>&& fila_color_names);
|
||||
|
||||
private:
|
||||
/* loaded info*/
|
||||
std::string m_fila_path;
|
||||
|
||||
std::unordered_map<wxString, FilamentColorCodes*>* m_fila_id2colors_map; //
|
||||
};
|
||||
|
||||
// EncodedFilaColorsInfo class holds a mapping of filament codes to specific filamet type
|
||||
class FilamentColorCodes
|
||||
{
|
||||
public:
|
||||
FilamentColorCodes(const wxString& fila_id, const wxString& fila_type);
|
||||
virtual ~FilamentColorCodes();
|
||||
|
||||
public:
|
||||
wxString GetFilaCode() const { return m_fila_id; }
|
||||
wxString GetFilaType() const { return m_fila_type; }
|
||||
|
||||
FilamentColor2CodeMap* GetFilamentColor2CodeMap() const { return m_fila_colors_map; }
|
||||
FilamentColorCode* GetColorCode(const FilamentColor& colors) const;
|
||||
|
||||
void Debug(const char* prefix);
|
||||
|
||||
public:
|
||||
void AddColorCode(FilamentColorCode* code);
|
||||
void RemoveColorCode(FilamentColorCode* code);
|
||||
|
||||
private:
|
||||
wxString m_fila_id;//eg. 54600
|
||||
wxString m_fila_type;//eg. PEBA 90A
|
||||
FilamentColor2CodeMap* m_fila_colors_map; // key is the color set, value is the info
|
||||
};
|
||||
|
||||
// The EncodedFilaColorInfo class holds information about a specific filament color
|
||||
class FilamentColorCode
|
||||
{
|
||||
public:
|
||||
FilamentColorCode() = delete;
|
||||
FilamentColorCode(const wxString& color_code, FilamentColorCodes* owner, FilamentColor&& color, std::unordered_map<wxString, wxString>&& name_map);
|
||||
~FilamentColorCode();
|
||||
|
||||
public:
|
||||
wxString GetFilaCode() const { return m_owner->GetFilaCode(); }
|
||||
wxString GetFilaType() const { return m_owner->GetFilaType(); }
|
||||
|
||||
|
||||
wxString GetFilaColorCode() const { return m_fila_color_code; } // eg. Q01B00
|
||||
FilamentColor GetFilaColor() const { return m_fila_color; }
|
||||
wxString GetFilaColorName() const;
|
||||
|
||||
void Debug(const char* prefix);
|
||||
|
||||
private:
|
||||
FilamentColorCodes* m_owner;
|
||||
|
||||
/* color info*/
|
||||
wxString m_fila_color_code; // eg. Q01B00
|
||||
FilamentColor m_fila_color;
|
||||
std::unordered_map<wxString, wxString> m_fila_color_names; // eg. en -> Red
|
||||
};
|
||||
|
||||
}
|
||||
#endif
|
||||
Reference in New Issue
Block a user