feat: ImageBuffer 支持传入 encoded

Author: MistEO

include/MaaUtils/Buffer/BufferTypes.hpp

@@ -22,6 +22,8 @@ struct MaaStringBuffer
     virtual void set(std::string str) = 0;
 };
 
+using ImageEncodedBuffer = std::vector<uint8_t>;
+
 struct MaaImageBuffer
 {
 public:
@@ -42,4 +44,5 @@ struct MaaImageBuffer
     virtual const cv::Mat& get() const = 0;
 
     virtual void set(cv::Mat image) = 0;
+    virtual void set(ImageEncodedBuffer buffer) = 0;
 };

include/MaaUtils/Buffer/ImageBuffer.hpp

@@ -1,8 +1,8 @@
 #pragma once
 
 #include "BufferTypes.hpp"
-#include "MaaUtils/Conf.h"
 #include "ListBuffer.hpp"
+#include "MaaUtils/Conf.h"
 
 MAA_SUPPRESS_CV_WARNINGS_BEGIN
 #include <opencv2/core/mat.hpp>
@@ -21,6 +21,8 @@ class ImageBuffer : public MaaImageBuffer
     {
     }
 
+    ImageBuffer(ImageEncodedBuffer buffer) { from_encoded(std::move(buffer)); }
+
     virtual ~ImageBuffer() override = default;
 
     virtual bool empty() const override { return image_.empty(); }
@@ -61,7 +63,16 @@ class ImageBuffer : public MaaImageBuffer
         image_ = image.clone();
     }
 
+    virtual void set(ImageEncodedBuffer buffer) override { from_encoded(std::move(buffer)); }
+
 private:
+    void from_encoded(ImageEncodedBuffer buffer)
+    {
+        image_ = cv::imdecode(buffer, cv::IMREAD_COLOR);
+        encoded_cache_ = std::move(buffer);
+        dirty_ = false;
+    }
+
     void encode() const
     {
         if (!dirty_) {
@@ -79,7 +90,7 @@ class ImageBuffer : public MaaImageBuffer
 
     cv::Mat image_;
     mutable bool dirty_ = true;
-    mutable std::vector<uint8_t> encoded_cache_;
+    mutable ImageEncodedBuffer encoded_cache_;
 };
 
 MAA_NS_END