102 lines
3.5 KiB
C++
102 lines
3.5 KiB
C++
#include <napi.h>
|
|
|
|
#include <vips/vips8>
|
|
|
|
using namespace std;
|
|
using namespace vips;
|
|
|
|
Napi::Value Freeze(const Napi::CallbackInfo &info) {
|
|
Napi::Env env = info.Env();
|
|
Napi::Object result = Napi::Object::New(env);
|
|
|
|
try {
|
|
Napi::Object obj = info[0].As<Napi::Object>();
|
|
Napi::Buffer<char> data = obj.Get("data").As<Napi::Buffer<char>>();
|
|
bool loop =
|
|
obj.Has("loop") ? obj.Get("loop").As<Napi::Boolean>().Value() : false;
|
|
string type = obj.Get("type").As<Napi::String>().Utf8Value();
|
|
int frame = obj.Has("frame")
|
|
? obj.Get("frame").As<Napi::Number>().Int32Value()
|
|
: -1;
|
|
|
|
char *fileData = data.Data();
|
|
char *match = (char *)"\x21\xFF\x0BNETSCAPE2.0\x03\x01";
|
|
char *descriptor = (char *)"\x2C\x00\x00\x00\x00";
|
|
char *lastPos;
|
|
|
|
bool none = true;
|
|
|
|
if (loop) {
|
|
char *newData = (char *)malloc(data.Length() + 19);
|
|
memcpy(newData, fileData, data.Length());
|
|
lastPos = (char *)memchr(newData, '\x2C', data.Length());
|
|
while (lastPos != NULL) {
|
|
if (memcmp(lastPos, descriptor, 5) != 0) {
|
|
lastPos = (char *)memchr(lastPos + 1, '\x2C',
|
|
(data.Length() - (lastPos - newData)) - 1);
|
|
continue;
|
|
}
|
|
|
|
memcpy(lastPos + 19, lastPos, (data.Length() - (lastPos - newData)));
|
|
memcpy(lastPos, match, 16);
|
|
memcpy(lastPos + 16, "\x00\x00\x00", 3);
|
|
result.Set("data",
|
|
Napi::Buffer<char>::Copy(env, newData, data.Length() + 19));
|
|
none = false;
|
|
break;
|
|
}
|
|
if (none)
|
|
result.Set("data",
|
|
Napi::Buffer<char>::Copy(env, newData, data.Length()));
|
|
} else if (frame >= 0 && !loop) {
|
|
VOption *options = VImage::option()->set("access", "sequential");
|
|
|
|
VImage in = VImage::new_from_buffer(
|
|
data.Data(), data.Length(), "",
|
|
type == "gif" ? options->set("n", -1) : options)
|
|
.colourspace(VIPS_INTERPRETATION_sRGB);
|
|
if (!in.has_alpha()) in = in.bandjoin(255);
|
|
|
|
int pageHeight = vips_image_get_page_height(in.get_image());
|
|
int nPages = vips_image_get_n_pages(in.get_image());
|
|
int framePos = clamp(frame, 0, (int)nPages);
|
|
VImage out = in.crop(0, 0, in.width(), pageHeight * (framePos + 1));
|
|
out.set(VIPS_META_PAGE_HEIGHT, pageHeight);
|
|
out.set("loop", loop ? 0 : 1);
|
|
|
|
void *buf;
|
|
size_t length;
|
|
out.write_to_buffer(("." + type).c_str(), &buf, &length);
|
|
|
|
result.Set("data", Napi::Buffer<char>::Copy(env, (char *)buf, length));
|
|
} else {
|
|
lastPos = (char *)memchr(fileData, '\x21', data.Length());
|
|
while (lastPos != NULL) {
|
|
if (memcmp(lastPos, match, 16) != 0) {
|
|
lastPos = (char *)memchr(lastPos + 1, '\x21',
|
|
(data.Length() - (lastPos - fileData)) - 1);
|
|
continue;
|
|
}
|
|
memcpy(lastPos, lastPos + 19,
|
|
(data.Length() - (lastPos - fileData)) - 19);
|
|
result.Set("data",
|
|
Napi::Buffer<char>::Copy(env, fileData, data.Length() - 19));
|
|
none = false;
|
|
break;
|
|
}
|
|
if (none)
|
|
result.Set("data",
|
|
Napi::Buffer<char>::Copy(env, fileData, data.Length()));
|
|
}
|
|
|
|
result.Set("type", type);
|
|
} catch (std::exception const &err) {
|
|
Napi::Error::New(env, err.what()).ThrowAsJavaScriptException();
|
|
} catch (...) {
|
|
Napi::Error::New(env, "Unknown error").ThrowAsJavaScriptException();
|
|
}
|
|
|
|
vips_error_clear();
|
|
vips_thread_shutdown();
|
|
return result;
|
|
}
|