BambuStudio/src/libslic3r/FilamentGroup.hpp

#ifndef FILAMENT_GROUP_HPP
#define FILAMENT_GROUP_HPP

#include <chrono>
#include <memory>
#include <numeric>
#include <set>
#include <map>
#include <vector>
#include "GCode/ToolOrderUtils.hpp"

const static int DEFAULT_CLUSTER_SIZE = 16;

namespace Slic3r
{
    std::vector<unsigned int>collect_sorted_used_filaments(const std::vector<std::vector<unsigned int>>& layer_filaments);

    struct FlushTimeMachine
    {
    private:
        std::chrono::high_resolution_clock::time_point start;

    public:
        void time_machine_start()
        {
            start = std::chrono::high_resolution_clock::now();
        }

        int time_machine_end()
        {
            auto end = std::chrono::high_resolution_clock::now();
            auto duration = std::chrono::duration_cast<std::chrono::milliseconds>(end - start);
            return duration.count();
        }
    };

    enum FGStrategy {
        BestCost,
        BestFit
    };

    struct FilamentGroupContext
    {
        std::vector<FlushMatrix> flush_matrix;
        std::vector<std::set<int>>physical_unprintables;
        std::vector<std::set<int>>geometric_unprintables;
        std::vector<int>max_group_size;
        int total_filament_num;
    };


    class FlushDistanceEvaluator
    {
    public:
        FlushDistanceEvaluator(const FlushMatrix& flush_matrix,const std::vector<unsigned int>&used_filaments,const std::vector<std::vector<unsigned int>>& layer_filaments, double p = 0.65);
        ~FlushDistanceEvaluator() = default;
        double get_distance(int idx_a, int idx_b) const;
    private:
        std::vector<std::vector<float>>m_distance_matrix;

    };

    class FilamentGroup
    {
    public:
        FilamentGroup(const FilamentGroupContext& context);
        std::vector<int> calc_filament_group(const std::vector<std::vector<unsigned int>>& layer_filaments, const FGStrategy& g_strategy = FGStrategy::BestFit, int* cost = nullptr);
    public:
        std::vector<int> calc_filament_group_by_enum(const std::vector<std::vector<unsigned int>>& layer_filaments, const std::vector<unsigned int>& used_filaments, const FGStrategy& g_strategy, int* cost = nullptr);
        std::vector<int> calc_filament_group_by_pam2(const std::vector<std::vector<unsigned int>>& layer_filaments, const std::vector<unsigned int>& used_filaments, const FGStrategy& g_strategy, int* cost = nullptr, int timeout_ms = 300);
    private:
        FilamentGroupContext m_context;
    public:
        std::optional<std::function<bool(int, std::vector<int>&)>> get_custom_seq;
    };


    class KMediods2
    {
        enum INIT_TYPE
        {
            Random = 0,
            Farthest
        };
    public:
        KMediods2(const int elem_count, const std::shared_ptr<FlushDistanceEvaluator>& evaluator) :
            m_evaluator{ evaluator },
            m_elem_count{ elem_count }
        {
            m_max_cluster_size = std::vector<int>(m_k, DEFAULT_CLUSTER_SIZE);
        }

        // set max group size
        void set_max_cluster_size(const std::vector<int>& group_size) { m_max_cluster_size = group_size; }

        // key stores elem idx, value stores the cluster id that elem cnanot be placed
        void set_unplaceable_limits(const std::map<int, int>& placeable_limits) { m_unplaceable_limits = placeable_limits; }

        void do_clustering(const FGStrategy& g_strategy,int timeout_ms = 100);
        std::vector<int>get_cluster_labels()const { return m_cluster_labels; }

    private:
        std::vector<int>cluster_small_data(const std::map<int, int>& unplaceable_limits, const std::vector<int>& group_size);
        std::vector<int>assign_cluster_label(const std::vector<int>& center, const std::map<int, int>& unplaceable_limits, const std::vector<int>& group_size, const FGStrategy& strategy);
        int calc_cost(const std::vector<int>& labels, const std::vector<int>& medoids);
    private:
        std::shared_ptr<FlushDistanceEvaluator> m_evaluator;
        std::map<int, int>m_unplaceable_limits;
        std::vector<int>m_max_cluster_size;
        int m_elem_count;
        const int m_k = 2;

        std::vector<int>m_cluster_labels;
    };
}
#endif // !FILAMENT_GROUP_HPP