mirror of
https://github.com/alibaba/GCanvas.git
synced 2025-12-08 17:36:42 +00:00
132 lines
3.8 KiB
C++
132 lines
3.8 KiB
C++
#include "NodeBindingUtil.h"
|
||
#include "lodepng.h"
|
||
#include <curl/curl.h>
|
||
#include <string>
|
||
#include <vector>
|
||
#include <iostream>
|
||
namespace NodeBinding
|
||
{
|
||
static size_t
|
||
writeMemoryCallback(void *contents, size_t size, size_t nmemb, void *userp)
|
||
{
|
||
size_t realsize = size * nmemb;
|
||
struct ImageContent *mem = (struct ImageContent *)userp;
|
||
|
||
char *ptr = (char *)realloc(mem->memory, mem->size + realsize + 1);
|
||
if (ptr == NULL)
|
||
{
|
||
/* out of memory! */
|
||
printf("not enough memory (realloc returned NULL)\n");
|
||
return 0;
|
||
}
|
||
|
||
mem->memory = ptr;
|
||
memcpy(&(mem->memory[mem->size]), contents, realsize);
|
||
mem->size += realsize;
|
||
mem->memory[mem->size] = 0;
|
||
return realsize;
|
||
}
|
||
bool checkArgs(const Napi::CallbackInfo &info, int exectedNumber)
|
||
{
|
||
if (info.Length() != exectedNumber)
|
||
{
|
||
throwError(info, "wrong argument number");
|
||
return false;
|
||
}
|
||
return true;
|
||
}
|
||
|
||
void throwError(const Napi::CallbackInfo &info, const std::string &exception)
|
||
{
|
||
Napi::TypeError::New(info.Env(), exception)
|
||
.ThrowAsJavaScriptException();
|
||
}
|
||
|
||
void throwError(const Napi::Env &env, const std::string &exception)
|
||
{
|
||
Napi::TypeError::New(env, exception)
|
||
.ThrowAsJavaScriptException();
|
||
}
|
||
unsigned int downloadImage(const std::string &src, ImageContent *content)
|
||
{
|
||
CURL *curl_handle;
|
||
CURLcode res;
|
||
|
||
content->memory = (char *)malloc(1); /* will be grown as needed by the realloc above */
|
||
content->size = 0; /* no data at this point */
|
||
|
||
curl_global_init(CURL_GLOBAL_ALL);
|
||
curl_handle = curl_easy_init();
|
||
curl_easy_setopt(curl_handle, CURLOPT_URL, src.c_str());
|
||
curl_easy_setopt(curl_handle, CURLOPT_WRITEFUNCTION, writeMemoryCallback);
|
||
curl_easy_setopt(curl_handle, CURLOPT_WRITEDATA, (void *)content);
|
||
curl_easy_setopt(curl_handle, CURLOPT_USERAGENT, "libcurl-agent/1.0");
|
||
curl_easy_setopt(curl_handle, CURLOPT_TIMEOUT, TIMEOUT_VALUE);
|
||
res = curl_easy_perform(curl_handle);
|
||
if (res != CURLE_OK)
|
||
{
|
||
fprintf(stderr, "curl_easy_perform() failed: %s\n",
|
||
curl_easy_strerror(res));
|
||
content->size = 0;
|
||
}
|
||
curl_easy_cleanup(curl_handle);
|
||
curl_global_cleanup();
|
||
|
||
return content->size;
|
||
}
|
||
|
||
void encodePixelsToFile(std::string filename, uint8_t *buffer, int width, int height)
|
||
{
|
||
//write the pixles to file
|
||
unsigned error = lodepng::encode(filename.c_str(), buffer, width, height);
|
||
if (error)
|
||
{
|
||
std::cout << "encoder error " << error << ": " << lodepng_error_text(error) << filename << std::endl;
|
||
}
|
||
}
|
||
|
||
void decodeFile2Pixels(std::string filename, std::vector<unsigned char> &image)
|
||
{
|
||
unsigned width, height;
|
||
//decode image
|
||
unsigned error = lodepng::decode(image, width, height, filename.c_str());
|
||
//if there's an error, display it
|
||
if (error)
|
||
std::cout << "decoder error " << error << ": " << lodepng_error_text(error) << std::endl;
|
||
}
|
||
|
||
int readLocalImage(const std::string &path, ImageContent *content)
|
||
{
|
||
FILE *pFile;
|
||
size_t result; // 返回值是读取的内容数量
|
||
|
||
pFile = fopen(path.c_str(), "rb");
|
||
|
||
if (pFile == NULL)
|
||
{
|
||
printf("file not exist \n");
|
||
return -1;
|
||
}
|
||
fseek(pFile, 0, SEEK_END); // 指针移到文件末位
|
||
content->size = ftell(pFile); // 获得文件长度
|
||
rewind(pFile);
|
||
content->memory = (char *)malloc(sizeof(char) * content->size); // 分配缓冲区，按前面的 lSize
|
||
|
||
if (content->memory == nullptr)
|
||
{
|
||
printf("memory allocate fail\n");
|
||
free(content->memory);
|
||
return -1;
|
||
} // 内存分配错误，退出2
|
||
|
||
result = fread(content->memory, 1, content->size, pFile); // 返回值是读取的内容数量
|
||
|
||
if (result != content->size)
|
||
{
|
||
printf("read file error\n");
|
||
return -1;
|
||
}
|
||
fclose(pFile);
|
||
return content->size;
|
||
}
|
||
} // namespace NodeBinding
|