116 lines
3.6 KiB
OCaml
116 lines
3.6 KiB
OCaml
open Pdfutil
|
|
|
|
type colspec =
|
|
NoCol
|
|
| RGB of float * float * float
|
|
| Grey of float
|
|
| CYMK of float * float * float * float
|
|
|
|
type image =
|
|
JPEG
|
|
|
|
type drawops =
|
|
| Rect of float * float * float * float
|
|
| Bezier of float * float * float * float * float * float
|
|
| To of float * float
|
|
| Line of float * float
|
|
| ClosePath
|
|
| SetFill of colspec
|
|
| SetStroke of colspec
|
|
| SetLineThickness of float
|
|
| SetLineCap of int
|
|
| SetLineJoin of int
|
|
| SetMiterLimit of float
|
|
| SetDashPattern of float list * float
|
|
| Matrix of Pdftransform.transform_matrix
|
|
| Push
|
|
| Pop
|
|
| Fill
|
|
| FillEvenOdd
|
|
| Stroke
|
|
| FillStroke
|
|
| FillStrokeEvenOdd
|
|
| Clip
|
|
| ClipEvenOdd
|
|
| SoftXObject of drawops list
|
|
| HardXObject of drawops list
|
|
| Image of string
|
|
| ImageXObject of string * Pdf.pdfobject
|
|
|
|
(* Hash table of (human name, (resources name, object)) for image xobjects *)
|
|
let images = null_hash ()
|
|
|
|
(* Fresh XObject names. If we are stamping over another page, manage clashes later. *)
|
|
let fresh_xobj_name () = "/Img0"
|
|
|
|
let rec ops_of_drawop pdf = function
|
|
| Push -> [Pdfops.Op_q]
|
|
| Pop -> [Pdfops.Op_Q]
|
|
| Matrix m -> [Pdfops.Op_cm m]
|
|
| Rect (x, y, w, h) -> [Pdfops.Op_re (x, y, w, h)]
|
|
| Bezier (a, b, c, d, e, f) -> [Pdfops.Op_c (a, b, c, d, e, f)]
|
|
| To (x, y) -> [Pdfops.Op_m (x, y)]
|
|
| Line (x, y) -> [Pdfops.Op_l (x, y)]
|
|
| SetFill x ->
|
|
begin match x with
|
|
| RGB (r, g, b) -> [Op_rg (r, g, b)]
|
|
| Grey g -> [Op_g g]
|
|
| CYMK (c, y, m, k) -> [Op_k (c, y, m, k)]
|
|
| NoCol -> []
|
|
end
|
|
| SetStroke x ->
|
|
begin match x with
|
|
| RGB (r, g, b) -> [Op_RG (r, g, b)]
|
|
| Grey g -> [Op_G g]
|
|
| CYMK (c, y, m, k) -> [Op_K (c, y, m, k)]
|
|
| NoCol -> []
|
|
end
|
|
| ClosePath
|
|
| Fill -> [Pdfops.Op_f]
|
|
| FillEvenOdd -> [Pdfops.Op_f']
|
|
| Stroke -> [Pdfops.Op_S]
|
|
| FillStroke -> [Pdfops.Op_B]
|
|
| FillStrokeEvenOdd -> [Pdfops.Op_B']
|
|
| Clip -> [Pdfops.Op_W; Pdfops.Op_n]
|
|
| ClipEvenOdd -> [Pdfops.Op_W']
|
|
| SetLineThickness t -> [Pdfops.Op_w t; Pdfops.Op_n]
|
|
| SetLineCap c -> [Pdfops.Op_J c]
|
|
| SetLineJoin j -> [Pdfops.Op_j j]
|
|
| SetMiterLimit m -> [Pdfops.Op_M m]
|
|
| SetDashPattern (x, y) -> [Pdfops.Op_d (x, y)]
|
|
| SoftXObject l | HardXObject l ->
|
|
[Pdfops.Op_q] @ ops_of_drawops pdf l @ [Pdfops.Op_Q]
|
|
| Image s -> [Pdfops.Op_Do (try fst (Hashtbl.find images s) with _ -> Cpdferror.error ("Image not found: " ^ s))]
|
|
| ImageXObject (s, obj) ->
|
|
Hashtbl.add images s (fresh_xobj_name (), Pdf.addobj pdf obj);
|
|
[]
|
|
|
|
and ops_of_drawops pdf drawops =
|
|
flatten (map (ops_of_drawop pdf) drawops)
|
|
|
|
(* Draw all the accumulated operators. FIXME: Manage name clashes in Xobjects *)
|
|
let draw fast range pdf drawops =
|
|
let s = Pdfops.string_of_ops (ops_of_drawops pdf drawops) in
|
|
let pdf = Cpdftweak.append_page_content s false fast range pdf in
|
|
let images = list_of_hashtbl images in
|
|
let resources = map (fun (_, (n, o)) -> (n, Pdf.Indirect o)) images in
|
|
match images with [] -> pdf | _ ->
|
|
let pages = Pdfpage.pages_of_pagetree pdf in
|
|
let pages =
|
|
map
|
|
(fun p ->
|
|
let new_resources =
|
|
let existing =
|
|
begin match Pdf.lookup_direct pdf "/XObject" p.Pdfpage.resources with
|
|
| Some (Pdf.Dictionary d) -> d
|
|
| _ -> []
|
|
end
|
|
in
|
|
let new_xobjects = fold_right (fun (k, v) d -> add k v d) resources existing in
|
|
Pdf.add_dict_entry p.Pdfpage.resources "/XObject" (Pdf.Dictionary new_xobjects)
|
|
in
|
|
{p with resources = new_resources})
|
|
pages
|
|
in
|
|
Pdfpage.change_pages true pdf pages
|