mirror of
https://github.com/OrcaSlicer/OrcaSlicer.git
synced 2026-06-22 19:50:44 +00:00
ENH: clean codes about device
JIRA: [STUDIO-13609] Change-Id: I591de7033360b9570600006cfbce2148a8d031d5 (cherry picked from commit e9c774be8f4c89b8dafa14ef56913612fb68bd0c)
This commit is contained in:
xin.zhang
364
src/slic3r/GUI/DeviceCore/DevMapping.cpp
Normal file
364
src/slic3r/GUI/DeviceCore/DevMapping.cpp
Normal file
@@ -0,0 +1,364 @@
|
||||
#include <nlohmann/json.hpp>
|
||||
#include "DevMapping.h"
|
||||
#include "DevFilaSystem.h"
|
||||
#include "DevUtil.h"
|
||||
|
||||
// TODO: remove this include
|
||||
#include "slic3r/GUI/DeviceManager.hpp"
|
||||
#include "slic3r/GUI/I18N.hpp"
|
||||
#include "slic3r/GUI/GuiColor.hpp"
|
||||
|
||||
using namespace nlohmann;
|
||||
|
||||
namespace Slic3r
|
||||
{
|
||||
bool DevMappingUtil::is_valid_mapping_result(const MachineObject* obj, std::vector<FilamentInfo>& result, bool check_empty_slot)
|
||||
{
|
||||
if (result.empty()) return false;
|
||||
|
||||
for (int i = 0; i < result.size(); i++)
|
||||
{
|
||||
// invalid mapping result
|
||||
if (result[i].tray_id < 0)
|
||||
{
|
||||
if (result[i].ams_id.empty() && result[i].slot_id.empty())
|
||||
{
|
||||
return false;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
|
||||
auto ams_item = obj->GetFilaSystem()->GetAmsById(result[i].ams_id);
|
||||
if (ams_item == nullptr)
|
||||
{
|
||||
if ((result[i].ams_id != std::to_string(VIRTUAL_TRAY_MAIN_ID)) &&
|
||||
(result[i].ams_id != std::to_string(VIRTUAL_TRAY_DEPUTY_ID)))
|
||||
{
|
||||
result[i].tray_id = -1;
|
||||
return false;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (check_empty_slot)
|
||||
{
|
||||
auto tray_item = ams_item->GetTrays().find(result[i].slot_id);
|
||||
if (tray_item == ams_item->GetTrays().end())
|
||||
{
|
||||
result[i].tray_id = -1;
|
||||
return false;
|
||||
}
|
||||
else
|
||||
{
|
||||
if (!tray_item->second->is_exists)
|
||||
{
|
||||
result[i].tray_id = -1;
|
||||
return false;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
// calc distance map
|
||||
struct DisValue {
|
||||
int tray_id;
|
||||
float distance;
|
||||
bool is_same_color = true;
|
||||
bool is_type_match = true;
|
||||
};
|
||||
|
||||
static void _parse_tray_info(int ams_id, int slot_id, DevAmsTray tray, FilamentInfo& result)
|
||||
{
|
||||
result.color = tray.color;
|
||||
result.type = tray.get_filament_type();
|
||||
result.filament_id = tray.setting_id;
|
||||
result.ctype = tray.ctype;
|
||||
result.colors = tray.cols;
|
||||
|
||||
/*for new ams mapping*/
|
||||
result.ams_id = std::to_string(ams_id);
|
||||
result.slot_id = std::to_string(slot_id);
|
||||
|
||||
if (ams_id == VIRTUAL_TRAY_MAIN_ID || ams_id == VIRTUAL_TRAY_DEPUTY_ID)
|
||||
{
|
||||
result.tray_id = atoi(tray.id.c_str());
|
||||
result.id = atoi(tray.id.c_str());
|
||||
}
|
||||
else
|
||||
{
|
||||
result.id = ams_id * 4 + slot_id;
|
||||
}
|
||||
}
|
||||
|
||||
int DevMappingUtil::ams_filament_mapping(const MachineObject* obj, const std::vector<FilamentInfo>& filaments, std::vector<FilamentInfo>& result, std::vector<bool> map_opt, std::vector<int> exclude_id, bool nozzle_has_ams_then_ignore_ext)
|
||||
{
|
||||
if (filaments.empty())
|
||||
return -1;
|
||||
|
||||
/////////////////////////
|
||||
// Step 1: collect filaments in machine
|
||||
std::map<int, FilamentInfo> tray_filaments; // tray_index : tray_color
|
||||
bool left_nozzle_has_ams = false, right_nozzle_has_ams = false;
|
||||
|
||||
const auto& ams_list = obj->GetFilaSystem()->GetAmsList();
|
||||
for (auto ams = ams_list.begin(); ams != ams_list.end(); ams++)
|
||||
{
|
||||
std::string ams_id = ams->second->GetAmsId();
|
||||
auto ams_type = ams->second->GetAmsType();
|
||||
for (auto tray = ams->second->GetTrays().begin(); tray != ams->second->GetTrays().end(); tray++)
|
||||
{
|
||||
int ams_id = atoi(ams->first.c_str());
|
||||
int tray_id = atoi(tray->first.c_str());
|
||||
int tray_index = 0;
|
||||
if (ams_type == DevAms::AMS || ams_type == DevAms::AMS_LITE || ams_type == DevAms::N3F)
|
||||
{
|
||||
tray_index = ams_id * 4 + tray_id;
|
||||
}
|
||||
else if (ams_type == DevAms::N3S)
|
||||
{
|
||||
tray_index = ams_id + tray_id;
|
||||
}
|
||||
else
|
||||
{
|
||||
assert(0);
|
||||
}
|
||||
|
||||
// skip exclude id
|
||||
for (int i = 0; i < exclude_id.size(); i++)
|
||||
{
|
||||
if (tray_index == exclude_id[i])
|
||||
continue;
|
||||
}
|
||||
// push
|
||||
FilamentInfo info;
|
||||
if (tray->second->is_tray_info_ready())
|
||||
{
|
||||
_parse_tray_info(ams_id, tray_id, *(tray->second), info);
|
||||
}
|
||||
|
||||
//first: left,nozzle=1,map=1 second: right,nozzle=0,map=2
|
||||
bool right_ams_valid = ams->second->GetExtruderId() == 0 && map_opt[MappingOption::USE_RIGHT_AMS];
|
||||
bool left_ams_valid = ams->second->GetExtruderId() == 1 && map_opt[MappingOption::USE_LEFT_AMS];
|
||||
if (right_ams_valid || left_ams_valid)
|
||||
{
|
||||
tray_filaments.emplace(std::make_pair(tray_index, info));
|
||||
if (right_ams_valid)
|
||||
{
|
||||
right_nozzle_has_ams = true;
|
||||
}
|
||||
if (left_ams_valid)
|
||||
{
|
||||
left_nozzle_has_ams = true;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (map_opt[MappingOption::USE_RIGHT_EXT] || map_opt[MappingOption::USE_LEFT_EXT])
|
||||
{
|
||||
for (auto tray : obj->vt_slot)
|
||||
{
|
||||
bool right_ext_valid = (tray.id == std::to_string(VIRTUAL_TRAY_MAIN_ID) && map_opt[MappingOption::USE_RIGHT_EXT]);
|
||||
bool left_ext_valid = (tray.id == std::to_string(VIRTUAL_TRAY_DEPUTY_ID) && map_opt[MappingOption::USE_LEFT_EXT]);
|
||||
if (right_ext_valid || left_ext_valid)
|
||||
{
|
||||
if (nozzle_has_ams_then_ignore_ext)
|
||||
{
|
||||
if (right_ext_valid && right_nozzle_has_ams)
|
||||
{
|
||||
continue;
|
||||
}
|
||||
if (left_ext_valid && left_nozzle_has_ams)
|
||||
{
|
||||
continue;
|
||||
}
|
||||
}
|
||||
FilamentInfo info;
|
||||
_parse_tray_info(atoi(tray.id.c_str()), 0, tray, info);
|
||||
tray_filaments.emplace(std::make_pair(info.tray_id, info));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/////////////////////////
|
||||
// Step 2: collect the distances of filaments_in_slicing to filaments_in_machine
|
||||
char buffer[256];
|
||||
std::vector<std::vector<DisValue>> distance_map;
|
||||
|
||||
// print title
|
||||
::sprintf(buffer, "F(id)");
|
||||
std::string line = std::string(buffer);
|
||||
for (auto tray = tray_filaments.begin(); tray != tray_filaments.end(); tray++)
|
||||
{
|
||||
::sprintf(buffer, " AMS%02d", tray->second.id + 1);
|
||||
line += std::string(buffer);
|
||||
}
|
||||
BOOST_LOG_TRIVIAL(info) << "ams_mapping_distance:" << line;// Print the collected filaments
|
||||
|
||||
for (int i = 0; i < filaments.size(); i++)
|
||||
{
|
||||
std::vector<DisValue> rol;
|
||||
::sprintf(buffer, "F(%02d)", filaments[i].id + 1);
|
||||
line = std::string(buffer);
|
||||
for (auto tray = tray_filaments.begin(); tray != tray_filaments.end(); tray++)
|
||||
{
|
||||
DisValue val;
|
||||
val.tray_id = tray->second.id;
|
||||
wxColour c = wxColour(filaments[i].color);
|
||||
wxColour tray_c = DevAmsTray::decode_color(tray->second.color);
|
||||
val.distance = GUI::calc_color_distance(c, tray_c);
|
||||
if (filaments[i].type != tray->second.type)
|
||||
{
|
||||
val.distance = 999999;
|
||||
val.is_type_match = false;
|
||||
}
|
||||
else
|
||||
{
|
||||
if (c.Alpha() != tray_c.Alpha())
|
||||
val.distance = 999999;
|
||||
val.is_type_match = true;
|
||||
}
|
||||
::sprintf(buffer, " %6.0f", val.distance);
|
||||
line += std::string(buffer);
|
||||
rol.push_back(val);
|
||||
}
|
||||
BOOST_LOG_TRIVIAL(info) << "ams_mapping_distance:" << line;
|
||||
distance_map.push_back(rol);
|
||||
}
|
||||
|
||||
/////////////////////////
|
||||
// Step 3: do mapping algorithm
|
||||
|
||||
// setup the mapping result
|
||||
for (int i = 0; i < filaments.size(); i++)
|
||||
{
|
||||
FilamentInfo info;
|
||||
info.id = filaments[i].id;
|
||||
info.tray_id = -1;
|
||||
info.type = filaments[i].type;
|
||||
info.filament_id = filaments[i].filament_id;
|
||||
result.push_back(info);
|
||||
}
|
||||
|
||||
// traverse the mapping
|
||||
std::set<int> picked_src;
|
||||
std::set<int> picked_tar;
|
||||
for (int k = 0; k < distance_map.size(); k++)
|
||||
{
|
||||
float min_val = INT_MAX;
|
||||
int picked_src_idx = -1;
|
||||
int picked_tar_idx = -1;
|
||||
for (int i = 0; i < distance_map.size(); i++)
|
||||
{
|
||||
if (picked_src.find(i) != picked_src.end())
|
||||
continue;
|
||||
|
||||
// try to mapping to different tray
|
||||
for (int j = 0; j < distance_map[i].size(); j++)
|
||||
{
|
||||
if (picked_tar.find(j) != picked_tar.end())
|
||||
{
|
||||
if (distance_map[i][j].is_same_color
|
||||
&& distance_map[i][j].is_type_match
|
||||
&& distance_map[i][j].distance < (float)0.0001)
|
||||
{
|
||||
min_val = distance_map[i][j].distance;
|
||||
picked_src_idx = i;
|
||||
picked_tar_idx = j;
|
||||
tray_filaments[picked_tar_idx].distance = min_val;
|
||||
}
|
||||
continue;
|
||||
}
|
||||
|
||||
if (distance_map[i][j].is_same_color
|
||||
&& distance_map[i][j].is_type_match)
|
||||
{
|
||||
if (min_val > distance_map[i][j].distance)
|
||||
{
|
||||
|
||||
min_val = distance_map[i][j].distance;
|
||||
picked_src_idx = i;
|
||||
picked_tar_idx = j;
|
||||
tray_filaments[picked_tar_idx].distance = min_val;
|
||||
}
|
||||
else if (min_val == distance_map[i][j].distance && filaments[picked_src_idx].filament_id != tray_filaments[picked_tar_idx].filament_id && filaments[i].filament_id == tray_filaments[j].filament_id)
|
||||
{
|
||||
|
||||
picked_src_idx = i;
|
||||
picked_tar_idx = j;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// take a retry to mapping to used tray
|
||||
if (picked_src_idx < 0 || picked_tar_idx < 0)
|
||||
{
|
||||
for (int j = 0; j < distance_map[i].size(); j++)
|
||||
{
|
||||
if (distance_map[i][j].is_same_color && distance_map[i][j].is_type_match)
|
||||
{
|
||||
if (min_val > distance_map[i][j].distance)
|
||||
{
|
||||
min_val = distance_map[i][j].distance;
|
||||
picked_src_idx = i;
|
||||
picked_tar_idx = j;
|
||||
tray_filaments[picked_tar_idx].distance = min_val;
|
||||
}
|
||||
else if (min_val == distance_map[i][j].distance && filaments[picked_src_idx].filament_id != tray_filaments[picked_tar_idx].filament_id && filaments[i].filament_id == tray_filaments[j].filament_id)
|
||||
{
|
||||
picked_src_idx = i;
|
||||
picked_tar_idx = j;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (picked_src_idx >= 0 && picked_tar_idx >= 0)
|
||||
{
|
||||
auto tray = tray_filaments.find(distance_map[k][picked_tar_idx].tray_id);
|
||||
|
||||
if (tray != tray_filaments.end())
|
||||
{
|
||||
result[picked_src_idx].tray_id = tray->first;
|
||||
|
||||
result[picked_src_idx].color = tray->second.color;
|
||||
result[picked_src_idx].type = tray->second.type;
|
||||
result[picked_src_idx].distance = tray->second.distance;
|
||||
result[picked_src_idx].filament_id = tray->second.filament_id;
|
||||
result[picked_src_idx].ctype = tray->second.ctype;
|
||||
result[picked_src_idx].colors = tray->second.colors;
|
||||
|
||||
/*for new ams mapping*/
|
||||
result[picked_src_idx].ams_id = tray->second.ams_id;
|
||||
result[picked_src_idx].slot_id = tray->second.slot_id;
|
||||
}
|
||||
|
||||
::sprintf(buffer, "ams_mapping, picked F(%02d) AMS(%02d), distance=%6.0f", picked_src_idx + 1, picked_tar_idx + 1,
|
||||
distance_map[picked_src_idx][picked_tar_idx].distance);
|
||||
BOOST_LOG_TRIVIAL(info) << std::string(buffer);
|
||||
picked_src.insert(picked_src_idx);
|
||||
picked_tar.insert(picked_tar_idx);
|
||||
}
|
||||
}
|
||||
|
||||
//check ams mapping result
|
||||
if (DevMappingUtil::is_valid_mapping_result(obj, result, true))
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
|
||||
/* for (auto it = result.begin(); it != result.end(); it++) {//This code has never been effective before 2025.03.18
|
||||
if (it->distance >= 6000) {
|
||||
it->tray_id = -1;
|
||||
}
|
||||
}*/
|
||||
return 0;
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user