mirror of
https://github.com/OrcaSlicer/OrcaSlicer.git
synced 2026-06-14 16:02:55 +00:00
f5c5f322d2
* add missing includes Building without precompiled header support revealed some missing includes that are added with this change for code health purposes. Those missing includes were hidden by precompiled header support due to the fact that the precompiled header file provides some definitions despite not being defined in directly or indirectly included header files.
109 lines
5.1 KiB
C++
109 lines
5.1 KiB
C++
#ifndef slic3r_GCodeThumbnails_hpp_
|
|
#define slic3r_GCodeThumbnails_hpp_
|
|
|
|
#include "../Point.hpp"
|
|
#include "../PrintConfig.hpp"
|
|
#include "ThumbnailData.hpp"
|
|
#include "../enum_bitmask.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);
|
|
}
|
|
|
|
namespace Slic3r::GCodeThumbnails {
|
|
|
|
struct CompressedImageBuffer
|
|
{
|
|
void *data { nullptr };
|
|
size_t size { 0 };
|
|
virtual ~CompressedImageBuffer() {}
|
|
virtual std::string_view tag() const = 0;
|
|
};
|
|
|
|
std::string get_hex(const unsigned int input);
|
|
std::string rjust(std::string input, unsigned int width, char fill_char);
|
|
std::unique_ptr<CompressedImageBuffer> compress_thumbnail(const ThumbnailData &data, GCodeThumbnailsFormat format);
|
|
std::string get_error_string(const ThumbnailErrors& errors);
|
|
|
|
|
|
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_ColPic = 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::BTT_TFT) {
|
|
// write BTT_TFT header
|
|
output((";" + rjust(get_hex(data.width), 4, '0') + rjust(get_hex(data.height), 4, '0') + "\r\n").c_str());
|
|
output((char *) compressed->data);
|
|
if (i == (thumbnails_list.size() - 1))
|
|
output("; bigtree thumbnail end\r\n\r\n");
|
|
}
|
|
else if (format == GCodeThumbnailsFormat::ColPic) {
|
|
if (first_ColPic) {
|
|
output((boost::format("\n\n;gimage:%s\n\n") % reinterpret_cast<char*>(compressed->data)).str().c_str());
|
|
} else {
|
|
output((boost::format("\n\n;simage:%s\n\n") % reinterpret_cast<char*>(compressed->data)).str().c_str());
|
|
}
|
|
first_ColPic = false;
|
|
}
|
|
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);
|
|
}
|
|
|
|
// Orca write remaining ecoded data
|
|
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++;
|
|
}
|
|
}
|
|
|
|
} // namespace Slic3r::GCodeThumbnails
|
|
|
|
#endif // slic3r_GCodeThumbnails_hpp_
|