125 lines
3.9 KiB
C++
125 lines
3.9 KiB
C++
#ifndef SLA_RASTERBASE_HPP
|
|
#define SLA_RASTERBASE_HPP
|
|
|
|
#include <ostream>
|
|
#include <memory>
|
|
#include <vector>
|
|
#include <array>
|
|
#include <utility>
|
|
#include <cstdint>
|
|
|
|
#include <libslic3r/ExPolygon.hpp>
|
|
#include <libslic3r/SLA/Concurrency.hpp>
|
|
|
|
namespace ClipperLib { struct Polygon; }
|
|
|
|
namespace Slic3r {
|
|
|
|
template<class T> using uqptr = std::unique_ptr<T>;
|
|
template<class T> using shptr = std::shared_ptr<T>;
|
|
template<class T> using wkptr = std::weak_ptr<T>;
|
|
|
|
namespace sla {
|
|
|
|
// Raw byte buffer paired with its size. Suitable for compressed image data.
|
|
class EncodedRaster {
|
|
protected:
|
|
std::vector<uint8_t> m_buffer;
|
|
std::string m_ext;
|
|
public:
|
|
EncodedRaster() = default;
|
|
explicit EncodedRaster(std::vector<uint8_t> &&buf, std::string ext)
|
|
: m_buffer(std::move(buf)), m_ext(std::move(ext))
|
|
{}
|
|
|
|
size_t size() const { return m_buffer.size(); }
|
|
const void * data() const { return m_buffer.data(); }
|
|
const char * extension() const { return m_ext.c_str(); }
|
|
};
|
|
|
|
using RasterEncoder =
|
|
std::function<EncodedRaster(const void *ptr, size_t w, size_t h, size_t num_components)>;
|
|
|
|
class RasterBase {
|
|
public:
|
|
|
|
enum Orientation { roLandscape, roPortrait };
|
|
|
|
using TMirroring = std::array<bool, 2>;
|
|
static const TMirroring NoMirror;
|
|
static const TMirroring MirrorX;
|
|
static const TMirroring MirrorY;
|
|
static const TMirroring MirrorXY;
|
|
|
|
struct Trafo {
|
|
bool mirror_x = false, mirror_y = false, flipXY = false;
|
|
coord_t center_x = 0, center_y = 0;
|
|
|
|
// Portrait orientation will make sure the drawed polygons are rotated
|
|
// by 90 degrees.
|
|
Trafo(Orientation o = roLandscape, const TMirroring &mirror = NoMirror)
|
|
// XY flipping implicitly does an X mirror
|
|
: mirror_x(o == roPortrait ? !mirror[0] : mirror[0])
|
|
, mirror_y(!mirror[1]) // Makes raster origin to be top left corner
|
|
, flipXY(o == roPortrait)
|
|
{}
|
|
|
|
TMirroring get_mirror() const { return { (roPortrait ? !mirror_x : mirror_x), mirror_y}; }
|
|
Orientation get_orientation() const { return flipXY ? roPortrait : roLandscape; }
|
|
Point get_center() const { return {center_x, center_y}; }
|
|
};
|
|
|
|
/// Type that represents a resolution in pixels.
|
|
struct Resolution {
|
|
size_t width_px = 0;
|
|
size_t height_px = 0;
|
|
|
|
Resolution(size_t w = 0, size_t h = 0) : width_px(w), height_px(h) {}
|
|
size_t pixels() const { return width_px * height_px; }
|
|
};
|
|
|
|
/// Types that represents the dimension of a pixel in millimeters.
|
|
struct PixelDim {
|
|
double w_mm = 0.;
|
|
double h_mm = 0.;
|
|
|
|
PixelDim(double px_width_mm = 0.0, double px_height_mm = 0.0)
|
|
: w_mm(px_width_mm), h_mm(px_height_mm)
|
|
{}
|
|
};
|
|
|
|
virtual ~RasterBase() = default;
|
|
|
|
/// Draw a polygon with holes.
|
|
virtual void draw(const ExPolygon& poly) = 0;
|
|
virtual void draw(const ClipperLib::Polygon& poly) = 0;
|
|
|
|
/// Get the resolution of the raster.
|
|
virtual Resolution resolution() const = 0;
|
|
virtual PixelDim pixel_dimensions() const = 0;
|
|
virtual Trafo trafo() const = 0;
|
|
|
|
virtual EncodedRaster encode(RasterEncoder encoder) const = 0;
|
|
};
|
|
|
|
struct PNGRasterEncoder {
|
|
EncodedRaster operator()(const void *ptr, size_t w, size_t h, size_t num_components);
|
|
};
|
|
|
|
struct PPMRasterEncoder {
|
|
EncodedRaster operator()(const void *ptr, size_t w, size_t h, size_t num_components);
|
|
};
|
|
|
|
std::ostream& operator<<(std::ostream &stream, const EncodedRaster &bytes);
|
|
|
|
// If gamma is zero, thresholding will be performed which disables AA.
|
|
uqptr<RasterBase> create_raster_grayscale_aa(
|
|
const RasterBase::Resolution &res,
|
|
const RasterBase::PixelDim & pxdim,
|
|
double gamma = 1.0,
|
|
const RasterBase::Trafo & tr = {});
|
|
|
|
}} // namespace Slic3r::sla
|
|
|
|
#endif // SLARASTERBASE_HPP
|