Redacting with process_struct_trees

cpdfcommand.ml

@@ -4650,7 +4650,7 @@ let go () =
   | Some Redact ->
       let pdf = get_single_pdf args.op false in
       let range = parse_pagespec_allow_empty pdf (get_pagespec ()) in
-        write_pdf false (Cpdfpage.redact pdf range)
+        write_pdf false (Cpdfpage.redact ~process_struct_tree:args.process_struct_trees pdf range)
 
 (* Advise the user if a combination of command line flags makes little sense,
 or error out if it make no sense at all. *)

cpdfpage.ml

@@ -996,8 +996,8 @@ let copy_box f t mediabox_if_missing pdf range =
     pdf
     range
 
-(* TODO: Trim structure tree *)
-let redact pdf range =
+let redact ~process_struct_tree pdf range =
+  let pdf =
     process_pages
       (Pdfpage.ppstub
          (fun pnum page ->
@@ -1010,3 +1010,6 @@ let redact pdf range =
               page))
       pdf
       range
+  in
+    if process_struct_tree then Pdfst.trim_structure_tree pdf (Cpdfpagespec.invert_range (Pdfpage.endpage pdf) range);
+    pdf

cpdfpage.mli

@@ -131,4 +131,4 @@ val alluprightonly : int list -> Pdf.t -> bool
 (** When a page is transformed, its patterns must be too. *)
 val change_pattern_matrices_page : Pdf.t -> Pdftransform.transform_matrix -> Pdfpage.t -> Pdfpage.t
 
-val redact : Pdf.t -> int list -> Pdf.t
+val redact : process_struct_tree:bool -> Pdf.t -> int list -> Pdf.t

cpdfpagespec.mli

@@ -32,3 +32,5 @@ val validate_pagespec : string -> bool
 (** Return a string for the given range. Knows how to identify all, odd, even,
 x-y ranges etc. *)
 val string_of_pagespec : Pdf.t -> int list -> string
+
+val invert_range : int -> int list -> int list