From 61e70674374aa06ee8aa3d8bc350a25017978f62 Mon Sep 17 00:00:00 2001
From: John Whitington <john@coherentgraphics.co.uk>
Date: Fri, 17 Mar 2023 19:04:20 +0000
Subject: [PATCH] more

---
 cpdfpng.ml  | 55 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 cpdfpng.mli |  3 +++
 2 files changed, 58 insertions(+)

diff --git a/cpdfpng.ml b/cpdfpng.ml
index b9f3702..d15e810 100644
--- a/cpdfpng.ml
+++ b/cpdfpng.ml
@@ -72,3 +72,58 @@ let read_png i =
       {width = i32toi width; height = i32toi height; idat = concat_bytes (rev !idat)}
   with
     e -> raise (Pdf.PDFError (Printf.sprintf "read_png: failed on %s" (Printexc.to_string e)))
+
+let (>>) = Int32.shift_right_logical
+let (&)  = Int32.logand
+let (^)  = Int32.logxor
+
+let crc_table =
+  let elem n =
+    let c = ref (Int32.of_int n) in
+    for _ = 0 to 7 do
+      c := (!c >> 1) ^ (0xedb88320l & (Int32.succ (Int32.lognot (!c & 1l))))
+    done; !c
+  in Array.init 256 elem
+
+let update_crc crc buf len =
+  let c = ref crc in
+  for n = 0 to len - 1 do
+    let e = Int32.of_int (int_of_char buf.[n]) in
+    c := crc_table.(Int32.to_int ((!c ^ e) & 0xffl)) ^ (!c >> 8)
+  done; !c
+
+let png_crc buf len =
+  Int32.lognot (update_crc 0xffffffffl buf len)
+
+let write_crc o ctype cdata =
+  o.output_byte 0;
+  o.output_byte 0;
+  o.output_byte 0;
+  o.output_byte 0
+
+let write_chunk o ctype data =
+  for x = 0 to 4 do o.output_byte (int_of_char ctype.[x]) done;
+  o.output_string (Bytes.unsafe_to_string data);
+  write_crc o ctype data
+
+let write_word b p n =
+  Bytes.set b p ' ';
+  Bytes.set b (p + 1) ' ';
+  Bytes.set b (p + 2) ' ';
+  Bytes.set b (p + 3) ' '
+
+let write_png png o =
+  if bytes_size png.idat > 2147483647 then raise (Invalid_argument "write_png: too large") else
+  (* Signature *)
+  o.output_string "\137\080\078\071\013\010\026\010";
+  let ihdr = Bytes.create 13 in
+  write_word ihdr 0 png.width;
+  write_word ihdr 4 png.height;
+  Bytes.set ihdr 8 (char_of_int 8); (* bit depth *)
+  Bytes.set ihdr 9 (char_of_int 2); (* colour type *)
+  Bytes.set ihdr 10 (char_of_int 0); (* compression method *)
+  Bytes.set ihdr 11 (char_of_int 0); (* filter method *)
+  Bytes.set ihdr 12 (char_of_int 0); (* interlace method *)
+  write_chunk o "IHDR" ihdr;
+  write_chunk o "IDAT" (Bytes.unsafe_of_string (Pdfio.string_of_bytes png.idat));
+  write_chunk o "IEND" (Bytes.create 0)
diff --git a/cpdfpng.mli b/cpdfpng.mli
index 5a8313c..16d6ac4 100644
--- a/cpdfpng.mli
+++ b/cpdfpng.mli
@@ -6,3 +6,6 @@ type t =
 (* Read a non-interlaced, non-transparent 24 bit PNG for inclusion in a PDF
    file. Raises BadPNG on failure. *)
 val read_png : Pdfio.input -> t
+
+(* Write a non-interlaced, non-transparent 24 bit PNG from PDF image contents *)
+val write_png : t -> Pdfio.output -> unit