summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--include/mupdf/pdf/object.h1
-rw-r--r--source/pdf/pdf-object.c13
-rw-r--r--source/tools/murun.c539
3 files changed, 553 insertions, 0 deletions
diff --git a/include/mupdf/pdf/object.h b/include/mupdf/pdf/object.h
index c196bece..2d7d8eb5 100644
--- a/include/mupdf/pdf/object.h
+++ b/include/mupdf/pdf/object.h
@@ -143,6 +143,7 @@ fz_rect *pdf_to_rect(fz_context *ctx, pdf_obj *array, fz_rect *rect);
fz_matrix *pdf_to_matrix(fz_context *ctx, pdf_obj *array, fz_matrix *mat);
pdf_document *pdf_get_indirect_document(fz_context *ctx, pdf_obj *obj);
+pdf_document *pdf_get_bound_document(fz_context *ctx, pdf_obj *obj);
void pdf_set_str_len(fz_context *ctx, pdf_obj *obj, int newlen);
void pdf_set_int(fz_context *ctx, pdf_obj *obj, int i);
void pdf_set_int_offset(fz_context *ctx, pdf_obj *obj, fz_off_t i);
diff --git a/source/pdf/pdf-object.c b/source/pdf/pdf-object.c
index 406580b2..10a16c8a 100644
--- a/source/pdf/pdf-object.c
+++ b/source/pdf/pdf-object.c
@@ -388,6 +388,19 @@ pdf_document *pdf_get_indirect_document(fz_context *ctx, pdf_obj *obj)
return REF(obj)->doc;
}
+pdf_document *pdf_get_bound_document(fz_context *ctx, pdf_obj *obj)
+{
+ if (obj < PDF_OBJ__LIMIT)
+ return NULL;
+ if (obj->kind == PDF_INDIRECT)
+ return REF(obj)->doc;
+ if (obj->kind == PDF_ARRAY)
+ return ARRAY(obj)->doc;
+ if (obj->kind == PDF_DICT)
+ return DICT(obj)->doc;
+ return NULL;
+}
+
int pdf_objcmp_resolve(fz_context *ctx, pdf_obj *a, pdf_obj *b)
{
RESOLVE(a);
diff --git a/source/tools/murun.c b/source/tools/murun.c
index 33723b3d..17a91e8a 100644
--- a/source/tools/murun.c
+++ b/source/tools/murun.c
@@ -1,4 +1,5 @@
#include "mupdf/fitz.h"
+#include "mupdf/pdf.h"
#include "mujs.h"
@@ -166,6 +167,18 @@ static void ffi_gc_fz_document(js_State *J, void *doc)
fz_drop_document(ctx, doc);
}
+static void ffi_gc_pdf_document(js_State *J, void *doc)
+{
+ fz_context *ctx = js_getcontext(J);
+ pdf_drop_document(ctx, doc);
+}
+
+static void ffi_gc_pdf_obj(js_State *J, void *obj)
+{
+ fz_context *ctx = js_getcontext(J);
+ pdf_drop_obj(ctx, obj);
+}
+
static void ffi_gc_fz_page(js_State *J, void *page)
{
fz_context *ctx = js_getcontext(J);
@@ -475,6 +488,203 @@ static void ffi_pushimage(js_State *J, fz_image *image)
js_newuserdata(J, "fz_image", fz_keep_image(ctx, image), ffi_gc_fz_image);
}
+static pdf_obj *ffi_toobj(js_State *J, pdf_document *pdf, int idx)
+{
+ fz_context *ctx = js_getcontext(J);
+ pdf_obj *obj;
+
+ /* make sure index is absolute */
+ if (idx < 0)
+ idx += js_gettop(J);
+
+ if (js_isuserdata(J, idx, "pdf_obj"))
+ return pdf_keep_obj(ctx, js_touserdata(J, idx, "pdf_obj"));
+
+ if (js_isnumber(J, idx)) {
+ float f = js_tonumber(J, idx);
+ fz_try(ctx)
+ if (f == (int)f)
+ obj = pdf_new_int(ctx, pdf, f);
+ else
+ obj = pdf_new_real(ctx, pdf, f);
+ fz_catch(ctx)
+ rethrow(J);
+ return obj;
+ }
+
+ if (js_isstring(J, idx)) {
+ const char *s = js_tostring(J, idx);
+ fz_try(ctx)
+ if (s[0] == '(')
+ obj = pdf_new_string(ctx, pdf, s+1, strlen(s)-2);
+ else
+ obj = pdf_new_name(ctx, pdf, s);
+ fz_catch(ctx)
+ rethrow(J);
+ return obj;
+ }
+
+ if (js_isboolean(J, idx)) {
+ int b = js_toboolean(J, idx);
+ fz_try(ctx)
+ obj = pdf_new_bool(ctx, pdf, b);
+ fz_catch(ctx)
+ rethrow(J);
+ return obj;
+ }
+
+ if (js_isnull(J, idx)) {
+ fz_try(ctx)
+ obj = pdf_new_null(ctx, pdf);
+ fz_catch(ctx)
+ rethrow(J);
+ return obj;
+ }
+
+ if (js_isarray(J, idx)) {
+ pdf_obj *val;
+ int i, n = js_getlength(J, idx);
+ fz_try(ctx)
+ obj = pdf_new_array(ctx, pdf, n);
+ fz_catch(ctx)
+ rethrow(J);
+ for (i = 0; i < n; ++i) {
+ js_getindex(J, idx, i);
+ val = ffi_toobj(J, pdf, -1);
+ fz_try(ctx)
+ pdf_array_push(ctx, obj, val);
+ fz_always(ctx)
+ pdf_drop_obj(ctx, val);
+ fz_catch(ctx)
+ rethrow(J);
+ js_pop(J, 1);
+ }
+ return obj;
+ }
+
+ if (js_isobject(J, idx)) {
+ const char *key;
+ pdf_obj *val;
+ fz_try(ctx)
+ obj = pdf_new_dict(ctx, pdf, 0);
+ fz_catch(ctx)
+ rethrow(J);
+ js_pushiterator(J, idx, 1);
+ while ((key = js_nextiterator(J, -1))) {
+ js_getproperty(J, idx, key);
+ val = ffi_toobj(J, pdf, -1);
+ fz_try(ctx)
+ pdf_dict_puts(ctx, obj, key, val);
+ fz_always(ctx)
+ pdf_drop_obj(ctx, val);
+ fz_catch(ctx)
+ rethrow(J);
+ js_pop(J, 1);
+ }
+ js_pop(J, 1);
+ return obj;
+ }
+
+ js_error(J, "cannot convert JS type to PDF");
+}
+
+static void ffi_pushobj(js_State *J, pdf_obj *obj);
+
+static int is_number(const char *key, int *idx)
+{
+ char *end;
+ *idx = strtol(key, &end, 10);
+ return *end == 0;
+}
+
+static int ffi_pdf_obj_has(js_State *J, void *obj, const char *key)
+{
+ fz_context *ctx = js_getcontext(J);
+ pdf_obj *val;
+ int idx;
+
+ if (is_number(key, &idx)) {
+ fz_try(ctx)
+ val = pdf_array_get(ctx, obj, idx);
+ fz_catch(ctx)
+ rethrow(J);
+ } else {
+ fz_try(ctx)
+ val = pdf_dict_gets(ctx, obj, key);
+ fz_catch(ctx)
+ rethrow(J);
+ }
+ if (val) {
+ ffi_pushobj(J, val);
+ return 1;
+ }
+ return 0;
+}
+
+static int ffi_pdf_obj_put(js_State *J, void *obj, const char *key)
+{
+ fz_context *ctx = js_getcontext(J);
+ pdf_document *pdf;
+ pdf_obj *val;
+ int idx;
+
+ fz_try(ctx)
+ pdf = pdf_get_bound_document(ctx, obj);
+ fz_catch(ctx)
+ rethrow(J);
+
+ val = ffi_toobj(J, pdf, -1);
+
+ if (is_number(key, &idx)) {
+ fz_try(ctx)
+ pdf_array_put(ctx, obj, idx, val);
+ fz_always(ctx)
+ pdf_drop_obj(ctx, val);
+ fz_catch(ctx)
+ rethrow(J);
+ } else {
+ fz_try(ctx)
+ pdf_dict_puts(ctx, obj, key, val);
+ fz_always(ctx)
+ pdf_drop_obj(ctx, val);
+ fz_catch(ctx)
+ rethrow(J);
+ }
+ return 1;
+}
+
+static int ffi_pdf_obj_delete(js_State *J, void *obj, const char *key)
+{
+ fz_context *ctx = js_getcontext(J);
+ int idx;
+
+ if (is_number(key, &idx)) {
+ fz_try(ctx)
+ pdf_array_delete(ctx, obj, idx);
+ fz_catch(ctx)
+ rethrow(J);
+ } else {
+ fz_try(ctx)
+ pdf_dict_dels(ctx, obj, key);
+ fz_catch(ctx)
+ rethrow(J);
+ }
+ return 1;
+}
+
+static void ffi_pushobj(js_State *J, pdf_obj *obj)
+{
+ fz_context *ctx = js_getcontext(J);
+ if (obj) {
+ js_getregistry(J, "pdf_obj");
+ js_newuserdatax(J, "pdf_obj", pdf_keep_obj(ctx, obj),
+ ffi_pdf_obj_has, ffi_pdf_obj_put, ffi_pdf_obj_delete,
+ ffi_gc_pdf_obj);
+ } else {
+ js_pushnull(J);
+ }
+}
+
/* device calling into js from c */
typedef struct js_device_s
@@ -1187,6 +1397,25 @@ static void ffi_Document_loadPage(js_State *J)
js_newuserdata(J, "fz_page", page, ffi_gc_fz_page);
}
+static void ffi_Document_toPDF(js_State *J)
+{
+ fz_context *ctx = js_getcontext(J);
+ fz_document *doc = js_touserdata(J, 0, "fz_document");
+ pdf_document *pdf;
+
+ fz_try(ctx)
+ pdf = pdf_specifics(ctx, doc);
+ fz_catch(ctx)
+ rethrow(J);
+
+ if (pdf) {
+ js_getregistry(J, "pdf_document");
+ js_newuserdata(J, "pdf_document", pdf, ffi_gc_pdf_document);
+ } else {
+ js_pushnull(J);
+ }
+}
+
static void ffi_Page_toPixmap(js_State *J)
{
fz_context *ctx = js_getcontext(J);
@@ -1762,6 +1991,282 @@ static void ffi_new_DrawDevice(js_State *J)
js_newuserdata(J, "fz_device", device, ffi_gc_fz_device);
}
+/* PDF specifics */
+
+static void ffi_new_PDFDocument(js_State *J)
+{
+ fz_context *ctx = js_getcontext(J);
+ const char *filename = js_isstring(J, 1) ? js_tostring(J, 1) : NULL;
+ pdf_document *pdf = NULL;
+
+ fz_try(ctx)
+ if (filename)
+ pdf = pdf_open_document(ctx, filename);
+ else
+ pdf = pdf_create_document(ctx);
+ fz_catch(ctx)
+ rethrow(J);
+
+ js_getregistry(J, "pdf_document");
+ js_newuserdata(J, "pdf_document", pdf, ffi_gc_pdf_document);
+}
+
+static void ffi_PDFDocument_getTrailer(js_State *J)
+{
+ fz_context *ctx = js_getcontext(J);
+ pdf_document *pdf = js_touserdata(J, 0, "pdf_document");
+ pdf_obj *trailer;
+
+ fz_try(ctx)
+ trailer = pdf_trailer(ctx, pdf);
+ fz_catch(ctx)
+ rethrow(J);
+
+ ffi_pushobj(J, trailer);
+}
+
+static void ffi_PDFDocument_addObject(js_State *J)
+{
+ fz_context *ctx = js_getcontext(J);
+ pdf_document *pdf = js_touserdata(J, 0, "pdf_document");
+ pdf_obj *obj = ffi_toobj(J, pdf, 1);
+ pdf_obj *ind;
+
+ fz_try(ctx)
+ ind = pdf_add_object(ctx, pdf, obj);
+ fz_catch(ctx)
+ rethrow(J);
+
+ ffi_pushobj(J, ind);
+}
+
+static void ffi_PDFDocument_addStream(js_State *J)
+{
+ fz_context *ctx = js_getcontext(J);
+ pdf_document *pdf = js_touserdata(J, 0, "pdf_document");
+ const char *str = js_tostring(J, 1);
+ fz_buffer *buf = NULL;
+ pdf_obj *ind;
+
+ fz_var(buf);
+
+ fz_try(ctx) {
+ buf = fz_new_buffer_from_shared_data(ctx, (unsigned char*)str, strlen(str));
+ ind = pdf_add_stream(ctx, pdf, buf);
+ }
+ fz_always(ctx)
+ fz_drop_buffer(ctx, buf);
+ fz_catch(ctx)
+ rethrow(J);
+
+ ffi_pushobj(J, ind);
+}
+
+static void ffi_PDFDocument_save(js_State *J)
+{
+ fz_context *ctx = js_getcontext(J);
+ pdf_document *pdf = js_touserdata(J, 0, "pdf_document");
+ const char *filename = js_tostring(J, 1);
+
+ fz_try(ctx)
+ pdf_save_document(ctx, pdf, filename, NULL);
+ fz_catch(ctx)
+ rethrow(J);
+}
+
+static void ffi_PDFDocument_newNull(js_State *J)
+{
+ fz_context *ctx = js_getcontext(J);
+ pdf_document *pdf = js_touserdata(J, 0, "pdf_document");
+ pdf_obj *obj;
+ fz_try(ctx)
+ obj = pdf_new_null(ctx, pdf);
+ fz_catch(ctx)
+ rethrow(J);
+ ffi_pushobj(J, obj);
+}
+
+static void ffi_PDFDocument_newBoolean(js_State *J)
+{
+ fz_context *ctx = js_getcontext(J);
+ pdf_document *pdf = js_touserdata(J, 0, "pdf_document");
+ int val = js_toboolean(J, 1);
+ pdf_obj *obj;
+ fz_try(ctx)
+ obj = pdf_new_bool(ctx, pdf, val);
+ fz_catch(ctx)
+ rethrow(J);
+ ffi_pushobj(J, obj);
+}
+
+static void ffi_PDFDocument_newInteger(js_State *J)
+{
+ fz_context *ctx = js_getcontext(J);
+ pdf_document *pdf = js_touserdata(J, 0, "pdf_document");
+ int val = js_tointeger(J, 1);
+ pdf_obj *obj;
+ fz_try(ctx)
+ obj = pdf_new_int(ctx, pdf, val);
+ fz_catch(ctx)
+ rethrow(J);
+ ffi_pushobj(J, obj);
+}
+
+static void ffi_PDFDocument_newReal(js_State *J)
+{
+ fz_context *ctx = js_getcontext(J);
+ pdf_document *pdf = js_touserdata(J, 0, "pdf_document");
+ float val = js_tonumber(J, 1);
+ pdf_obj *obj;
+ fz_try(ctx)
+ obj = pdf_new_real(ctx, pdf, val);
+ fz_catch(ctx)
+ rethrow(J);
+ ffi_pushobj(J, obj);
+}
+
+static void ffi_PDFDocument_newString(js_State *J)
+{
+ fz_context *ctx = js_getcontext(J);
+ pdf_document *pdf = js_touserdata(J, 0, "pdf_document");
+ // TODO: convert array of numbers
+ const char *val = js_tostring(J, 1);
+ pdf_obj *obj;
+ fz_try(ctx)
+ obj = pdf_new_string(ctx, pdf, val, strlen(val));
+ fz_catch(ctx)
+ rethrow(J);
+ ffi_pushobj(J, obj);
+}
+
+static void ffi_PDFDocument_newName(js_State *J)
+{
+ fz_context *ctx = js_getcontext(J);
+ pdf_document *pdf = js_touserdata(J, 0, "pdf_document");
+ const char *val = js_tostring(J, 1);
+ pdf_obj *obj;
+ fz_try(ctx)
+ obj = pdf_new_name(ctx, pdf, val);
+ fz_catch(ctx)
+ rethrow(J);
+ ffi_pushobj(J, obj);
+}
+
+static void ffi_PDFDocument_newIndirect(js_State *J)
+{
+ fz_context *ctx = js_getcontext(J);
+ pdf_document *pdf = js_touserdata(J, 0, "pdf_document");
+ int num = js_tointeger(J, 1);
+ int gen = js_tointeger(J, 2);
+ pdf_obj *obj;
+ fz_try(ctx)
+ obj = pdf_new_indirect(ctx, pdf, num, gen);
+ fz_catch(ctx)
+ rethrow(J);
+ ffi_pushobj(J, obj);
+}
+
+static void ffi_PDFDocument_newArray(js_State *J)
+{
+ fz_context *ctx = js_getcontext(J);
+ pdf_document *pdf = js_touserdata(J, 0, "pdf_document");
+ pdf_obj *obj;
+ fz_try(ctx)
+ obj = pdf_new_array(ctx, pdf, 0);
+ fz_catch(ctx)
+ rethrow(J);
+ ffi_pushobj(J, obj);
+}
+
+static void ffi_PDFDocument_newDictionary(js_State *J)
+{
+ fz_context *ctx = js_getcontext(J);
+ pdf_document *pdf = js_touserdata(J, 0, "pdf_document");
+ pdf_obj *obj;
+ fz_try(ctx)
+ obj = pdf_new_dict(ctx, pdf, 0);
+ fz_catch(ctx)
+ rethrow(J);
+ ffi_pushobj(J, obj);
+}
+
+static void ffi_PDFObject_get(js_State *J)
+{
+ pdf_obj *obj = js_touserdata(J, 0, "pdf_obj");
+ const char *key = js_tostring(J, 1);
+ if (!ffi_pdf_obj_has(J, obj, key))
+ js_pushundefined(J);
+}
+
+static void ffi_PDFObject_put(js_State *J)
+{
+ pdf_obj *obj = js_touserdata(J, 0, "pdf_obj");
+ const char *key = js_tostring(J, 1);
+ js_copy(J, 2);
+ ffi_pdf_obj_put(J, obj, key);
+}
+
+static void ffi_PDFObject_delete(js_State *J)
+{
+ pdf_obj *obj = js_touserdata(J, 0, "pdf_obj");
+ const char *key = js_tostring(J, 1);
+ ffi_pdf_obj_delete(J, obj, key);
+}
+
+static void ffi_PDFObject_resolve(js_State *J)
+{
+ fz_context *ctx = js_getcontext(J);
+ pdf_obj *obj = js_touserdata(J, 0, "pdf_obj");
+ pdf_obj *res;
+ fz_try(ctx)
+ res = pdf_resolve_indirect(ctx, obj);
+ fz_catch(ctx)
+ rethrow(J);
+ ffi_pushobj(J, res);
+}
+
+static void ffi_PDFObject_toString(js_State *J)
+{
+ fz_context *ctx = js_getcontext(J);
+ pdf_obj *obj = js_touserdata(J, 0, "pdf_obj");
+ int tight = js_isdefined(J, 1) ? js_toboolean(J, 1) : 1;
+ char *s;
+ int n;
+
+ fz_try(ctx) {
+ n = pdf_sprint_obj(ctx, NULL, 0, obj, tight);
+ s = fz_malloc(ctx, n+1);
+ pdf_sprint_obj(ctx, s, n+1, obj, tight);
+ js_pushstring(J, s);
+ } fz_always(ctx) {
+ fz_free(ctx, s);
+ } fz_catch(ctx) {
+ rethrow(J);
+ }
+}
+
+static void ffi_PDFObject_valueOf(js_State *J)
+{
+ fz_context *ctx = js_getcontext(J);
+ pdf_obj *obj = js_touserdata(J, 0, "pdf_obj");
+ if (pdf_is_indirect(ctx, obj))
+ js_pushstring(J, "R");
+ else if (pdf_is_null(ctx, obj))
+ js_pushnull(J);
+ else if (pdf_is_bool(ctx, obj))
+ js_pushboolean(J, pdf_to_bool(ctx, obj));
+ else if (pdf_is_int(ctx, obj))
+ js_pushnumber(J, pdf_to_int(ctx, obj));
+ else if (pdf_is_real(ctx, obj))
+ js_pushnumber(J, pdf_to_real(ctx, obj));
+ else if (pdf_is_string(ctx, obj))
+ js_pushlstring(J, pdf_to_str_buf(ctx, obj), pdf_to_str_len(ctx, obj));
+ else if (pdf_is_name(ctx, obj))
+ js_pushstring(J, pdf_to_name(ctx, obj));
+ else
+ js_copy(J, 0);
+}
+
int murun_main(int argc, char **argv)
{
fz_context *ctx;
@@ -1810,6 +2315,8 @@ int murun_main(int argc, char **argv)
// Document.authenticatePassword
// Document.lookupMetadata
+ jsB_propfun(J, "Document.toPDF", ffi_Document_toPDF, 0);
+
// Document.hasPermission
jsB_propenum(J, "PERMISSION_PRINT", FZ_PERMISSION_PRINT);
jsB_propenum(J, "PERMISSION_COPY", FZ_PERMISSION_COPY);
@@ -1956,6 +2463,38 @@ int murun_main(int argc, char **argv)
js_newobject(J);
{
+ jsB_propfun(J, "PDFDocument.getTrailer", ffi_PDFDocument_getTrailer, 0);
+ jsB_propfun(J, "PDFDocument.addObject", ffi_PDFDocument_addObject, 1);
+ jsB_propfun(J, "PDFDocument.addStream", ffi_PDFDocument_addStream, 1);
+ jsB_propfun(J, "PDFDocument.save", ffi_PDFDocument_save, 2);
+
+ jsB_propfun(J, "PDFDocument.newNull", ffi_PDFDocument_newNull, 0);
+ jsB_propfun(J, "PDFDocument.newBoolean", ffi_PDFDocument_newBoolean, 1);
+ jsB_propfun(J, "PDFDocument.newInteger", ffi_PDFDocument_newInteger, 1);
+ jsB_propfun(J, "PDFDocument.newReal", ffi_PDFDocument_newReal, 1);
+ jsB_propfun(J, "PDFDocument.newString", ffi_PDFDocument_newString, 1);
+ jsB_propfun(J, "PDFDocument.newName", ffi_PDFDocument_newName, 1);
+ jsB_propfun(J, "PDFDocument.newIndirect", ffi_PDFDocument_newIndirect, 2);
+ jsB_propfun(J, "PDFDocument.newArray", ffi_PDFDocument_newArray, 1);
+ jsB_propfun(J, "PDFDocument.newDictionary", ffi_PDFDocument_newDictionary, 1);
+ }
+ js_setregistry(J, "pdf_document");
+
+ js_newobject(J);
+ {
+ jsB_propfun(J, "PDFObject.get", ffi_PDFObject_get, 0);
+ jsB_propfun(J, "PDFObject.put", ffi_PDFObject_put, 0);
+ jsB_propfun(J, "PDFObject.delete", ffi_PDFObject_delete, 0);
+ jsB_propfun(J, "PDFObject.resolve", ffi_PDFObject_resolve, 0);
+ jsB_propfun(J, "PDFObject.toString", ffi_PDFObject_toString, 1);
+ jsB_propfun(J, "PDFObject.valueOf", ffi_PDFObject_valueOf, 0);
+ }
+ js_setregistry(J, "pdf_obj");
+
+ js_newobject(J);
+ {
+ jsB_propcon(J, "pdf_document", "PDFDocument", ffi_new_PDFDocument, 1);
+
jsB_propcon(J, "fz_document", "Document", ffi_new_Document, 1);
jsB_propcon(J, "fz_pixmap", "Pixmap", ffi_new_Pixmap, 2);
jsB_propcon(J, "fz_image", "Image", ffi_new_Image, 1);