png supportHEADmaster

Signed-off-by: lns <matzeton@googlemail.com>

1 files changed, 5 insertions, 1 deletions

diff --git a/ImageManipulation.hpp b/ImageManipulation.hpp
index e372d96..784e326 100644
--- a/ImageManipulation.hpp
+++ b/ImageManipulation.hpp
@@ -13,17 +13,21 @@
 #include <string>
 #include <vector>
 
+enum ImageFormat { IF_UNSUPPORTED, IF_JPEG, IF_PNG };
+
 class ImageManipulation {
 public:
   ImageManipulation();
   ~ImageManipulation();
 
-  void SetJpegFromBuffer(std::vector<unsigned char> &image_buffer);
+  bool SetImageFromBuffer(std::vector<unsigned char> &image_buffer);
   void SaveToFile(std::string filename);
   void Ukrainify(float opacity = 0.5f);
 
 private:
   cimg_library::CImg<unsigned char> image;
+
+  enum ImageFormat GetImageFormat(std::vector<unsigned char> &image_buffer);
 };
 
 #endif