gcode img

CrealityOfficial · Jan 12, 2022 · 2516420 · 2516420
1 parent cd422db
commit 2516420
Show file tree

Hide file tree

Showing 5 changed files with 225 additions and 3 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -5,7 +5,7 @@ if(NOT GLM_INCLUDE_DIRS)
 	message(FATAL_ERROR "GLM_INCLUDE_DIRS Missing")
 endif()
 
-set(LIBS zlib png cxbin trimesh2)
+set(LIBS png cxbin trimesh2 zlib)
 set(INCS ${GLM_INCLUDE_DIRS})
 set(DEFS GLM_FORCE_LEFT_HANDED THUMBNAIL_DLL)
 
@@ -34,6 +34,8 @@ __add_real_target(thumbnail dll SOURCE
 									thumbnail/stl_thumbnail.cpp
 									thumbnail/thumbnail.h
 									thumbnail/thumbnail.cpp
+									thumbnail/img2gcode.cpp
+									thumbnail/img2gcode.h
 								LIB ${LIBS}
 								INC ${INCS}
 								DEF ${DEFS}
@@ -54,5 +56,5 @@ if(INSTALL_THUMBNAIL)
 endif()
 
 if(CC_INSTALL_THUMBNAIL)
-	__install_directory_specif(${CMAKE_CURRENT_SOURCE_DIR}/ thumbnail)
+	__install_directory_specif(${CMAKE_CURRENT_SOURCE_DIR}/thumbnail thumbnail)
 endif()
diff --git a/thumbnail/img2gcode.cpp b/thumbnail/img2gcode.cpp
@@ -0,0 +1,192 @@
+#include"img2gcode.h"
+
+#include<fstream>
+
+Img2Gcode::Img2Gcode()
+{
+
+}
+Img2Gcode::~Img2Gcode()
+{
+
+}
+
+int getLineStart(const std::vector<unsigned char>& prevData, int line_strlen)
+{
+	int LineStart = 0;
+	int pixelNum = prevData.size();
+	int linePixel = line_strlen / 4 * 3;
+	int startNo = 0;
+	for (int i = 0; i < pixelNum; i++)
+	{
+		if (prevData[i] != 0)
+		{
+			startNo = i;
+			break;
+		}
+	}
+	LineStart = startNo / linePixel;
+	return LineStart;
+}
+
+int getLineEnd(const std::vector<unsigned char>& prevData, int line_strlen)
+{
+	int LineEnd = 0;
+	int pixelNum = prevData.size();
+	int linePixel = line_strlen / 4 * 3;
+	int endNo = pixelNum - 1;
+	for (int i = endNo; i >= 0; i--)
+	{
+		if (prevData[i] != 0)
+		{
+			endNo = i;
+			break;
+		}
+	}
+	LineEnd = endNo / linePixel;
+	return LineEnd;
+}
+
+static const std::string base64_chars =
+	"ABCDEFGHIJKLMNOPQRSTUVWXYZ"
+	"abcdefghijklmnopqrstuvwxyz"
+	"0123456789+/";
+
+
+static inline bool is_base64(unsigned char c) {
+	return (isalnum(c) || (c == '+') || (c == '/'));
+}
+
+std::string base64_encode(std::vector<unsigned char> const& buffer) {
+	size_t in_len = buffer.size();
+	unsigned char const* bytes_to_encode = buffer.data();
+	std::string ret;
+	int i = 0;
+	int j = 0;
+	unsigned char char_array_3[3];
+	unsigned char char_array_4[4];
+
+	while (in_len--) {
+		char_array_3[i++] = *(bytes_to_encode++);
+		if (i == 3) {
+			char_array_4[0] = (char_array_3[0] & 0xfc) >> 2;
+			char_array_4[1] = ((char_array_3[0] & 0x03) << 4) + ((char_array_3[1] & 0xf0) >> 4);
+			char_array_4[2] = ((char_array_3[1] & 0x0f) << 2) + ((char_array_3[2] & 0xc0) >> 6);
+			char_array_4[3] = char_array_3[2] & 0x3f;
+
+			for (i = 0; (i < 4); i++)
+				ret += base64_chars[char_array_4[i]];
+			i = 0;
+		}
+	}
+
+	if (i)
+	{
+		for (j = i; j < 3; j++)
+			char_array_3[j] = '\0';
+
+		char_array_4[0] = (char_array_3[0] & 0xfc) >> 2;
+		char_array_4[1] = ((char_array_3[0] & 0x03) << 4) + ((char_array_3[1] & 0xf0) >> 4);
+		char_array_4[2] = ((char_array_3[1] & 0x0f) << 2) + ((char_array_3[2] & 0xc0) >> 6);
+
+		for (j = 0; (j < i + 1); j++)
+			ret += base64_chars[char_array_4[j]];
+
+		while ((i++ < 3))
+			ret += '=';
+
+	}
+
+	return ret;
+
+}
+
+std::vector<unsigned char> base64_decode(std::string const& encoded_string) {
+	size_t in_len = encoded_string.size();
+	int i = 0;
+	int j = 0;
+	int in_ = 0;
+	unsigned char char_array_4[4], char_array_3[3];
+	std::vector<unsigned char> ret;
+
+	while (in_len-- && (encoded_string[in_] != '=') && is_base64(encoded_string[in_])) {
+		char_array_4[i++] = encoded_string[in_]; in_++;
+		if (i == 4) {
+			for (i = 0; i < 4; i++)
+				char_array_4[i] = base64_chars.find(char_array_4[i]) & 0xff;
+
+			char_array_3[0] = (char_array_4[0] << 2) + ((char_array_4[1] & 0x30) >> 4);
+			char_array_3[1] = ((char_array_4[1] & 0xf) << 4) + ((char_array_4[2] & 0x3c) >> 2);
+			char_array_3[2] = ((char_array_4[2] & 0x3) << 6) + char_array_4[3];
+
+			for (i = 0; (i < 3); i++)
+				ret.push_back(char_array_3[i]);
+			i = 0;
+		}
+	}
+
+	if (i) {
+		for (j = 0; j < i; j++)
+			char_array_4[j] = base64_chars.find(char_array_4[j]) & 0xff;
+
+		char_array_3[0] = (char_array_4[0] << 2) + ((char_array_4[1] & 0x30) >> 4);
+		char_array_3[1] = ((char_array_4[1] & 0xf) << 4) + ((char_array_4[2] & 0x3c) >> 2);
+
+		for (j = 0; (j < i - 1); j++) ret.push_back(char_array_3[j]);
+	}
+
+	return ret;
+}
+
+
+bool Img2Gcode::imgEncode(const std::vector<unsigned char>& prevData, std::vector<std::string>& encodeData, const std::string& imgSizes,
+	const std::string& imgFormat, const int& layerCount, const char* saveFile)
+{
+	std::string prevData_base64 = base64_encode(prevData);
+	int mainDataSize = prevData_base64.size();
+	int line_strlen = 76;
+	int lineNum = mainDataSize % line_strlen == 0 ? mainDataSize / line_strlen : mainDataSize / line_strlen + 1;
+	lineNum += 2;//Í·Î²Á½ÐÐ
+	int imgDataStartLineNo = getLineStart(prevData, line_strlen) + 1;
+	int imgDataEndLineNo = getLineEnd(prevData, line_strlen) + 1;
+	std::string headData = imgFormat + std::string(" begin ") + imgSizes + std::string(" ") + std::to_string(mainDataSize) + std::string(" ") + 
+		std::to_string(imgDataStartLineNo) + std::string(" ") + std::to_string(imgDataEndLineNo) + std::string(" ") + std::to_string(layerCount);
+	std::string endData = imgFormat + std::string(" end");
+
+	encodeData.reserve(lineNum);
+	std::ofstream outfile;
+	outfile.open(saveFile, std::ios::binary | std::ios::trunc | std::ios::in | std::ios::out);
+	for (int i = 0; i < lineNum; i++)
+	{
+		std::string CurLine;
+		if (i == 0) {
+			CurLine = "; " + headData;
+		}
+		else if (i == lineNum - 1) {
+			CurLine = "; " + endData;
+		}
+		else if (i == lineNum - 2) {
+			CurLine = "; " + std::string(prevData_base64, line_strlen * (i - 1));
+		}
+		else {
+			CurLine = "; " + std::string(prevData_base64, line_strlen * (i - 1), line_strlen);
+		}
+		outfile << CurLine << std::endl;
+		encodeData.push_back(CurLine);
+	}
+	outfile.close();
+	return true;
+}
+
+bool Img2Gcode::imgDecode(std::vector<std::string>& prevEncodeData, std::vector<unsigned char>& decodeData)
+{
+	std::string prevData_base64;
+	int linesNum = prevEncodeData.size();
+	for (int i = 1; i < linesNum - 1; i++)
+	{
+		std::string curLineData = std::string(prevEncodeData[i], 2);
+		prevData_base64 += curLineData;
+	}
+	decodeData = base64_decode(prevData_base64);
+	return true;
+}
diff --git a/thumbnail/img2gcode.h b/thumbnail/img2gcode.h
@@ -0,0 +1,15 @@
+#ifndef _IMG2GCODE_H
+#define _IMG2GCODE_H
+#include<vector>
+#include<string>
+class  Img2Gcode
+	{
+	public:
+		Img2Gcode();
+		~Img2Gcode();
+
+		static bool imgEncode(const std::vector<unsigned char>& prevData, std::vector<std::string>& encodeData, const std::string& imgSizes, 
+			const std::string& imgFormat, const int& layerCount, const char* saveFile);
+		static bool imgDecode(std::vector<std::string>& prevData, std::vector<unsigned char>& decodeData);
+	};
+#endif
diff --git a/thumbnail/thumbnail.cpp b/thumbnail/thumbnail.cpp
@@ -3,6 +3,7 @@
 #include "raster_backend/raster.h"
 #include "picture.h"
 #include "trimesh2/TriMesh.h"
+#include"img2gcode.h"
 static Vec3 model_colors[] =
 {
     {0x42 / 255.f, 0xB5 / 255.f, 1.0f}, // Ç³À¶
@@ -139,4 +140,14 @@ bool thumbnail_trimeshs(const std::vector<trimesh::TriMesh*>& meshes, int width,
 bool  thumbnail_trimesh(trimesh::TriMesh* mesh, int width, int height, int model_color_idx, const char* filePath)
 {
     return thumbnail_trimeshs({ mesh }, width, height, model_color_idx,filePath);
+}
+
+bool thumbnail_to_gcode(const char* infilePath, const int inImgSizes, const std::string& inImgFormat, const int& inlayerCount, std::string& outGcodeStr)
+{
+    if (!infilePath) return false;
+
+}
+bool thumbnail_to_gcode(const std::vector<unsigned char>& inPrevData, const std::string& inImgSizes, const std::string& inImgFormat, const int& inlayerCount, std::vector<std::string>& outGcodeStr)
+{
+   return  Img2Gcode::imgEncode(inPrevData,outGcodeStr,inImgSizes, inImgFormat, inlayerCount, nullptr);
 }
diff --git a/thumbnail/thumbnail.h b/thumbnail/thumbnail.h
@@ -2,6 +2,7 @@
 #define _THUMBNAIL_1635927924764_H
 #include "thumbnail/exporter.h"
 #include <vector>
+#include<string>
 
 namespace trimesh
 {
@@ -12,5 +13,6 @@ THUMBNAIL_API void thumbnail_trimeshes(const std::vector<trimesh::TriMesh*>& mes
 THUMBNAIL_API void thumbnail_trimesh(trimesh::TriMesh* mesh, int width, int height, unsigned char* data);
 
 THUMBNAIL_API bool thumbnail_trimeshs(const std::vector<trimesh::TriMesh*>& meshes, int width, int height, int model_color_idx, const char* filePath);
-THUMBNAIL_API bool thumbnail_trimesh(trimesh::TriMesh* mesh, int width, int height, int model_color_idx, const char* filePath);
+
+THUMBNAIL_API bool thumbnail_to_gcode(const std::vector<unsigned char>& inPrevData, const std::string& inImgSizes,const std::string& inImgFormat, const int& inlayerCount, std::vector<std::string>& outGcodeStr);
 #endif // _THUMBNAIL_1635927924764_H