1 files changed, 124 insertions, 0 deletions
diff --git a/src/libslic3r/SLA/RasterBase.hpp b/src/libslic3r/SLA/RasterBase.hpp
new file mode 100644
index 000000000..431c731a6
--- /dev/null
+++ b/src/libslic3r/SLA/RasterBase.hpp
@@ -0,0 +1,124 @@
+#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