BambuSrc/libslic3r/FilamentGroupUtils.cpp

278 lines
11 KiB
C++

#include "FilamentGroupUtils.hpp"
#include <regex>
#include <sstream>
namespace Slic3r
{
namespace FilamentGroupUtils
{
Color::Color(const std::string& hexstr) {
if (hexstr.empty() || (hexstr.length() != 9 && hexstr.length() != 7) || hexstr[0] != '#')
{
assert(false);
r = 0, g = 0, b = 0, a = 255;
return;
}
auto hexToByte = [](const std::string& hex)->unsigned char
{
unsigned int byte;
std::istringstream(hex) >> std::hex >> byte;
return static_cast<unsigned char>(byte);
};
r = hexToByte(hexstr.substr(1, 2));
g = hexToByte(hexstr.substr(3, 2));
b = hexToByte(hexstr.substr(5, 2));
if (hexstr.size() == 9)
a = hexToByte(hexstr.substr(7, 2));
}
bool Color::operator<(const Color& other) const
{
if (r != other.r) return r < other.r;
if (g != other.g) return g < other.g;
if (b != other.b) return b < other.b;
return a < other.a;
}
bool Color::operator==(const Color& other) const
{
return r == other.r && g == other.g && b == other.b && a == other.a;
}
bool Color::operator!=(const Color& other) const
{
return r != other.r || g != other.g || b != other.b || a != other.a;
}
std::string Color::to_hex_str(bool include_alpha) const {
std::ostringstream oss;
oss << "#" << std::hex << std::setfill('0')
<< std::setw(2) << static_cast<int>(r)
<< std::setw(2) << static_cast<int>(g)
<< std::setw(2) << static_cast<int>(b);
if (include_alpha) {
oss << std::setw(2) << static_cast<int>(a);
}
return oss.str();
}
bool MachineFilamentInfo::operator<(const MachineFilamentInfo& other) const
{
if (color != other.color) return color < other.color;
if (type != other.type) return type < other.type;
return is_support <other.is_support;
}
// TODO: add explanation
std::vector<int> calc_max_group_size(const std::vector<std::map<int, int>>& ams_counts, bool ignore_ext_filament) {
// add default value to 2
std::vector<int>group_size(2, 0);
for (size_t idx = 0; idx < ams_counts.size(); ++idx) {
const auto& ams_count = ams_counts[idx];
for (auto iter = ams_count.begin(); iter != ams_count.end(); ++iter) {
group_size[idx] += iter->first * iter->second;
}
}
for (size_t idx = 0; idx < group_size.size(); ++idx) {
if (!ignore_ext_filament && group_size[idx] == 0)
group_size[idx] = 1;
}
return group_size;
}
static std::vector<std::vector<MachineFilamentInfo>> build_full_machine_filaments(const std::vector<std::vector<DynamicPrintConfig>>& filament_configs)
{
auto extract_filament_type = [](const std::string& s)->std::string {
std::regex r1(R"(^Sup.(\w+)$)");
std::regex r2(R"(^(\w+)-S$)");
std::smatch m;
if (std::regex_match(s, m, r1))
return m[1].str();
if (std::regex_match(s, m, r2))
return m[1].str();
return s;
};
// change filament type to type format in preset
// defualt size set to 2
std::vector<std::vector<MachineFilamentInfo>> machine_filaments(2);
for (size_t idx = 0; idx < filament_configs.size(); ++idx) {
auto& arr = filament_configs[idx];
for (auto& item : arr) {
MachineFilamentInfo temp;
std::string type;
std::string color;
std::string tray_name;
bool is_support_filament = false;
if (auto color_ptr = item.option<ConfigOptionStrings>("filament_colour"); color_ptr)
color = color_ptr->get_at(0);
if (auto type_ptr = item.option<ConfigOptionStrings>("filament_type"); type_ptr) {
type = type_ptr->get_at(0);
type = extract_filament_type(type);
}
if (auto tray_ptr = item.option<ConfigOptionStrings>("tray_name"); tray_ptr)
tray_name = tray_ptr->get_at(0);
if (auto support_ptr = item.option<ConfigOptionBools>("filament_is_support"); support_ptr)
is_support_filament = support_ptr->get_at(0);
if (color.empty() || type.empty() || tray_name.empty())
continue;
temp.color = Color(color);
temp.type =type;
temp.extruder_id = idx;
temp.is_extended = tray_name == "Ext"; // hard-coded ext flag
temp.is_support = is_support_filament;
machine_filaments[idx].emplace_back(std::move(temp));
}
}
return machine_filaments;
}
std::vector<std::vector<MachineFilamentInfo>> build_machine_filaments(const std::vector<std::vector<DynamicPrintConfig>>& filament_configs, const std::vector<std::map<int, int>>& ams_counts, bool ignore_ext_filament)
{
std::vector<std::vector<MachineFilamentInfo>> ret(2);
std::vector<int> ams_size(2, 0);
std::vector<std::vector<MachineFilamentInfo>> full_machine_filaments = build_full_machine_filaments(filament_configs);
assert(full_machine_filaments.size() == 2);
for (size_t idx = 0; idx < std::min(ams_counts.size(),ams_size.size()); ++idx) {
const auto& ams_count = ams_counts[idx];
for (auto iter = ams_count.begin(); iter != ams_count.end(); ++iter) {
ams_size[idx] += iter->first * iter->second;
}
}
assert(full_machine_filaments.size() == ams_size.size());
for (size_t idx = 0; idx < std::min(ams_size.size(), full_machine_filaments.size()); ++idx) {
std::vector<MachineFilamentInfo> tmp;
for (size_t j = 0; j < full_machine_filaments[idx].size(); ++j) {
auto& machine_filament = full_machine_filaments[idx][j];
if (!machine_filament.is_extended)
tmp.emplace_back(machine_filament);
}
// if do not have valid ams filament, try to use ext filament
if (tmp.empty() && !ignore_ext_filament) {
for (size_t j = 0; j < full_machine_filaments[idx].size(); ++j) {
auto& machine_filament = full_machine_filaments[idx][j];
if (machine_filament.is_extended)
tmp.emplace_back(machine_filament);
}
}
ret[idx] = std::move(tmp);
}
return ret;
}
bool collect_unprintable_limits(const std::vector<std::set<int>>& physical_unprintables, const std::vector<std::set<int>>& geometric_unprintables, std::vector<std::set<int>>& unprintable_limits)
{
unprintable_limits.clear();
unprintable_limits.resize(2);
// resize unprintables to 2
auto resized_physical_unprintables = physical_unprintables;
resized_physical_unprintables.resize(2);
auto resized_geometric_unprintables = geometric_unprintables;
resized_geometric_unprintables.resize(2);
bool conflict = false;
conflict |= remove_intersection(resized_physical_unprintables[0], resized_physical_unprintables[1]);
conflict |= remove_intersection(resized_geometric_unprintables[0], resized_geometric_unprintables[1]);
std::map<int, int>filament_unprintable_exts;
for (auto& ext_unprintables : { resized_physical_unprintables,resized_geometric_unprintables }) {
for (int eid = 0; eid < ext_unprintables.size(); ++eid) {
for (int fid : ext_unprintables[eid]) {
if (auto iter = filament_unprintable_exts.find(fid); iter != filament_unprintable_exts.end() && iter->second != eid)
conflict = true;
else
filament_unprintable_exts[fid] = eid;
}
}
}
for (auto& elem : filament_unprintable_exts)
unprintable_limits[elem.second].insert(elem.first);
return !conflict;
}
bool remove_intersection(std::set<int>& a, std::set<int>& b) {
std::vector<int>intersection;
std::set_intersection(a.begin(), a.end(), b.begin(), b.end(), std::back_inserter(intersection));
bool have_intersection = !intersection.empty();
for (auto& item : intersection) {
a.erase(item);
b.erase(item);
}
return have_intersection;
}
void extract_indices(const std::vector<unsigned int>& used_filaments, const std::vector<std::set<int>>& unprintable_elems, std::vector<std::set<int>>& unprintable_idxs)
{
std::vector<std::set<int>>(unprintable_elems.size()).swap(unprintable_idxs);
for (size_t gid = 0; gid < unprintable_elems.size(); ++gid) {
for (auto& f : unprintable_elems[gid]) {
auto iter = std::find(used_filaments.begin(), used_filaments.end(), (unsigned)f);
if (iter != used_filaments.end())
unprintable_idxs[gid].insert(iter - used_filaments.begin());
}
}
}
void extract_unprintable_limit_indices(const std::vector<std::set<int>>& unprintable_elems, const std::vector<unsigned int>& used_filaments, std::map<int, int>& unplaceable_limits)
{
unplaceable_limits.clear();
// map the unprintable filaments to idx of used filaments , if not used ,just ignore
std::vector<std::set<int>> unprintable_idxs;
extract_indices(used_filaments, unprintable_elems, unprintable_idxs);
if (unprintable_idxs.size() > 1)
remove_intersection(unprintable_idxs[0], unprintable_idxs[1]);
for (size_t idx = 0; idx < unprintable_idxs.size(); ++idx) {
for (auto f : unprintable_idxs[idx])
if (unplaceable_limits.count(f) == 0)
unplaceable_limits[f] = idx;
}
}
void extract_unprintable_limit_indices(const std::vector<std::set<int>>& unprintable_elems, const std::vector<unsigned int>& used_filaments, std::unordered_map<int, std::vector<int>>& unplaceable_limits)
{
unplaceable_limits.clear();
std::vector<std::set<int>>unprintable_idxs;
// map the unprintable filaments to idx of used filaments , if not used ,just ignore
extract_indices(used_filaments, unprintable_elems, unprintable_idxs);
// remove elems that cannot be printed in both extruder
if (unprintable_idxs.size() > 1)
remove_intersection(unprintable_idxs[0], unprintable_idxs[1]);
for (size_t group_id = 0; group_id < unprintable_idxs.size(); ++group_id)
for (auto f : unprintable_idxs[group_id])
unplaceable_limits[f].emplace_back(group_id);
for (auto& elem : unplaceable_limits)
sort_remove_duplicates(elem.second);
}
bool check_printable(const std::vector<std::set<int>>& groups, const std::map<int,int>& unprintable)
{
for (size_t i = 0; i < groups.size(); ++i) {
auto& group = groups[i];
for (auto& filament : group) {
if (auto iter = unprintable.find(filament); iter != unprintable.end() && i == iter->second)
return false;
}
}
return true;
}
}
}