From cdf0950af1fa9e457563e762093747fd811c5a05 Mon Sep 17 00:00:00 2001
From: John Whitington <john@coherentgraphics.co.uk>
Date: Tue, 9 Jul 2019 16:31:45 +0100
Subject: [PATCH] Scaffolding for -draft-remove-only

---
 cpdf.ml        |  2 +-
 cpdf.mli       |  2 +-
 cpdfcommand.ml | 17 +++++++++++++----
 3 files changed, 15 insertions(+), 6 deletions(-)

diff --git a/cpdf.ml b/cpdf.ml
index 49a38ce..a6c2dc8 100644
--- a/cpdf.ml
+++ b/cpdf.ml
@@ -3988,7 +3988,7 @@ and remove_images_page boxes pdf page =
           Pdfpage.resources = resources'}, pdf
 
 (* Remove images from all pages in a document. *)
-let draft boxes range pdf =
+let draft onlyremove boxes range pdf =
   let pages = Pdfpage.pages_of_pagetree pdf in
     let pagenums = indx pages in
     let pdf = ref pdf
diff --git a/cpdf.mli b/cpdf.mli
index 6c36791..ff84b09 100644
--- a/cpdf.mli
+++ b/cpdf.mli
@@ -421,7 +421,7 @@ val blacklines : float * float * float -> int list -> Pdf.t -> Pdf.t
 val blackfills : float * float * float -> int list -> Pdf.t -> Pdf.t
 
 (** Remove images from a PDF, optionally adding crossed boxes. *)
-val draft : bool -> int list -> Pdf.t -> Pdf.t
+val draft : string option -> bool -> int list -> Pdf.t -> Pdf.t
 
 (** Squeeze a PDF *)
 val squeeze : ?logto:string -> Pdf.t -> unit
diff --git a/cpdfcommand.ml b/cpdfcommand.ml
index cee724d..95cad8e 100644
--- a/cpdfcommand.ml
+++ b/cpdfcommand.ml
@@ -420,7 +420,8 @@ type args =
    mutable merge_add_bookmarks : bool;
    mutable merge_add_bookmarks_use_titles : bool;
    mutable createpdf_pages : int;
-   mutable createpdf_pagesize : Pdfpaper.t}
+   mutable createpdf_pagesize : Pdfpaper.t;
+   mutable removeonly : string option}
 
 let args =
   {op = None;
@@ -513,7 +514,8 @@ let args =
    merge_add_bookmarks = false;
    merge_add_bookmarks_use_titles = false;
    createpdf_pages = 1;
-   createpdf_pagesize = Pdfpaper.a4}
+   createpdf_pagesize = Pdfpaper.a4;
+   removeonly = None}
 
 let reset_arguments () =
   args.op <- None;
@@ -596,7 +598,8 @@ let reset_arguments () =
   args.merge_add_bookmarks <- false;
   args.merge_add_bookmarks_use_titles <- false;
   args.createpdf_pages <- 1;
-  args.createpdf_pagesize <- Pdfpaper.a4
+  args.createpdf_pagesize <- Pdfpaper.a4;
+  args.removeonly <- None
   (* Do not reset original_filename or cpdflin or was_encrypted or
    * was_decrypted_with_owner or recrypt or producer or creator or
    * path_to_ghostscript or gs_malformed, since we want these to work across
@@ -1659,6 +1662,9 @@ let setcreatepdfpapersize s =
     let w, h = parse_coordinate (Pdf.empty ()) s in
     Pdfpaper.make Pdfunits.PdfPoint w h
 
+let setdraftremoveonly s =
+  args.removeonly <- Some s
+
 (* Parse a control file, make an argv, and then make Arg parse it. *)
 let rec make_control_argv_and_parse filename =
   control_args := !control_args @ parse_control_file filename
@@ -2190,6 +2196,9 @@ and specs =
    ("-draft",
       Arg.Unit (setop Draft),
       " Remove images from the file");
+   ("-draft-remove-only",
+      Arg.String setdraftremoveonly,
+      " Only remove named image");
    ("-boxes",
       Arg.Unit setboxes,
       " Add crossed boxes to -draft option");
@@ -4218,7 +4227,7 @@ let go () =
   | Some Draft ->
       let pdf = get_single_pdf args.op false in
         let range = parse_pagespec pdf (get_pagespec ()) in
-          write_pdf false (Cpdf.draft args.boxes range pdf)
+          write_pdf false (Cpdf.draft args.removeonly args.boxes range pdf)
   | Some (AddText text) ->
       let pdf = get_single_pdf args.op false in
         let range = parse_pagespec pdf (get_pagespec ()) in