QIDIStudio/src/libslic3r/GCode/Thumbnails.hpp

#ifndef slic3r_GCodeThumbnails_hpp_
#define slic3r_GCodeThumbnails_hpp_

#include "../Point.hpp"
#include "../PrintConfig.hpp"
#include "ThumbnailData.hpp"

#include <vector>
#include <memory>
#include <string_view>

#include <boost/beast/core/detail/base64.hpp>

namespace Slic3r {
    enum class ThumbnailError : int { InvalidVal, OutOfRange, InvalidExt };
    using ThumbnailErrors = enum_bitmask<ThumbnailError>;
    ENABLE_ENUM_BITMASK_OPERATORS(ThumbnailError);
}

#ifndef U16
typedef		unsigned short		U16;
#endif
#ifndef U8
typedef		unsigned char		U8;
#endif
#ifndef U32
typedef		unsigned int		U32;
#endif
#ifndef null
#define null 0
#endif
typedef struct
{
    U16 colo16;
    U8  A0;
    U8  A1;
    U8  A2;
    U8  res0;
    U16 res1;
    U32 qty;
} U16HEAD;
typedef struct
{
    U8  encodever;
    U8  res0;
    U16 oncelistqty;
    U32 PicW;
    U32 PicH;
    U32 mark;
    U32 ListDataSize;
    U32 ColorDataSize;
    U32 res1;
    U32 res2;
} ColPicHead3;

namespace Slic3r::GCodeThumbnails {

    struct CompressedImageBuffer
    {
        void* data{ nullptr };
        size_t      size{ 0 };
        virtual ~CompressedImageBuffer() {}
        virtual std::string_view tag() const = 0;
    };


    std::unique_ptr<CompressedImageBuffer> compress_thumbnail(const ThumbnailData& data, GCodeThumbnailsFormat format);
    std::string get_error_string(const ThumbnailErrors& errors);
    std::string compress_qidi_thumbnail(const ThumbnailData& data, GCodeThumbnailsFormat format);


    typedef std::vector<std::pair<GCodeThumbnailsFormat, Vec2d>> GCodeThumbnailDefinitionsList;
    using namespace std::literals;
    std::pair<GCodeThumbnailDefinitionsList, ThumbnailErrors> make_and_check_thumbnail_list(const std::string& thumbnails_string, const std::string_view def_ext = "PNG"sv);
    std::pair<GCodeThumbnailDefinitionsList, ThumbnailErrors> make_and_check_thumbnail_list(const ConfigBase& config);


    template<typename WriteToOutput, typename ThrowIfCanceledCallback>
    inline void export_thumbnails_to_file(ThumbnailsGeneratorCallback& thumbnail_cb,
        int                                                         plate_id,
        const std::vector<std::pair<GCodeThumbnailsFormat, Vec2d>>& thumbnails_list,
        WriteToOutput                                               output,
        ThrowIfCanceledCallback                                     throw_if_canceled)
    {
        // Write thumbnails using base64 encoding
        if (thumbnail_cb == nullptr)
            return;
        short i = 0;
        bool first_Qidi = true;
        for (const auto& [format, size] : thumbnails_list) {
            static constexpr const size_t max_row_length = 78;
            ThumbnailsList                thumbnails = thumbnail_cb(ThumbnailsParams{ {size}, true, true, true, true, plate_id });
            for (const ThumbnailData& data : thumbnails) {
                if (data.is_valid()) {
                    auto compressed = compress_thumbnail(data, format);
                    if (compressed->data && compressed->size) {
                        if (format == GCodeThumbnailsFormat::Qidi) {
                            ;
                        }
                        else {
                            output("; THUMBNAIL_BLOCK_START\n");
                            std::string encoded;
                            encoded.resize(boost::beast::detail::base64::encoded_size(compressed->size));
                            encoded.resize(boost::beast::detail::base64::encode((void*)encoded.data(), (const void*)compressed->data,
                                compressed->size));
                            output((boost::format("\n;\n; %s begin %dx%d %d\n") % compressed->tag() % data.width % data.height % encoded.size())
                                .str()
                                .c_str());
                            while (encoded.size() > max_row_length) {
                                output((boost::format("; %s\n") % encoded.substr(0, max_row_length)).str().c_str());
                                encoded = encoded.substr(max_row_length);
                            }

                            if (encoded.size() > 0)
                                output((boost::format("; %s\n") % encoded).str().c_str());

                            output((boost::format("; %s end\n") % compressed->tag()).str().c_str());
                            output("; THUMBNAIL_BLOCK_END\n\n");
                        }
                        throw_if_canceled();
                    }
                }
            }
            i++;
        }
    }
    template<typename WriteToOutput, typename ThrowIfCanceledCallback>
    inline void export_qidi_thumbnails_to_file(ThumbnailsGeneratorCallback& thumbnail_cb,
        int                                                         plate_id,
        const std::vector<std::pair<GCodeThumbnailsFormat, Vec2d>>& thumbnails_list,
        WriteToOutput                                               output,
        ThrowIfCanceledCallback                                     throw_if_canceled)
    {
        // Write thumbnails using base64 encoding
        if (thumbnail_cb == nullptr)
            return;
        short i = 0;
        bool first_Qidi = true;
        for (const auto& [format, size] : thumbnails_list) {
            static constexpr const size_t max_row_length = 78;
            ThumbnailsList                thumbnails = thumbnail_cb(ThumbnailsParams{ {size}, true, true, true, true, plate_id });
            for (const ThumbnailData& data : thumbnails) {
                if (data.is_valid()) {
                    auto compressed = compress_thumbnail(data, format);
                    if (compressed->data && compressed->size) {
                        if (format == GCodeThumbnailsFormat::Qidi) {
                            auto compressed_qidi = compress_qidi_thumbnail(data, format);
                            if (first_Qidi) {
                                output((boost::format("\n\n;gimage:%s\n\n") % compressed_qidi).str().c_str());
                            }
                            else {
                                output((boost::format("\n\n;simage:%s\n\n") % compressed_qidi).str().c_str());
                            }
                            first_Qidi = false;
                        }
                        else {
                            ;
                        }
                        throw_if_canceled();
                    }
                }
            }
            i++;
        }
    }

} // namespace Slic3r::GCodeThumbnails

#endif // slic3r_GCodeThumbnails_hpp_