unikernel/duniverse/ocaml-re/lib/group.ml

@@ -0,0 +1,103 @@
+(* Result of a successful match. *)
+type t =
+  { (* Input string. Matched strings are substrings of s *)
+    s : string
+      (* Mapping from group indices to positions in gpos. group i has positions 2*i
+         - 1, 2*i + 1 in gpos. If the group wasn't matched, then its corresponding
+           values in marks will be -1,-1 *)
+  ; marks : Mark_infos.t
+  ; (* Marks positions. i.e. those marks created with Re.marks *)
+    pmarks : Pmark.Set.t
+  ; (* Group positions. Adjacent elements are (start, stop) of group match.
+       indexed by the values in marks. So group i in an re would be the substring:
+
+       start = t.gpos.(marks.(2*i)) - 1
+       stop = t.gpos.(marks.(2*i + 1)) - 1 *)
+    gpos : int array
+  ; (* Number of groups the regular expression contains. Matched or not *)
+    gcount : int
+  }
+
+let create s ~gcount ~gpos marks pmarks = { s; gcount; gpos; marks; pmarks }
+
+module Offset = struct
+  type t = int
+
+  let absent = -1
+  let is_present t = t >= 0
+  let get_no_check t = t
+end
+
+let start_offset t i =
+  let i = Mark_infos.start_offset t.marks i in
+  if Mark_infos.Offset.is_present i
+  then t.gpos.(Mark_infos.Offset.get_no_check i)
+  else Offset.absent
+;;
+
+let stop_offset t i =
+  let i = Mark_infos.stop_offset t.marks i in
+  if Mark_infos.Offset.is_present i
+  then t.gpos.(Mark_infos.Offset.get_no_check i)
+  else Offset.absent
+;;
+
+let offset_opt t i =
+  Mark_infos.offset t.marks i
+  |> Option.map (fun (start, stop) -> t.gpos.(start), t.gpos.(stop))
+;;
+
+let or_not_found = function
+  | None -> raise Not_found
+  | Some s -> s
+;;
+
+let offset t i = offset_opt t i |> or_not_found
+
+let get_opt t i =
+  offset_opt t i |> Option.map (fun (p1, p2) -> String.sub t.s p1 (p2 - p1))
+;;
+
+let pmarks t = t.pmarks
+let get t i = get_opt t i |> or_not_found
+let start_opt subs i = offset_opt subs i |> Option.map fst
+let start subs i = start_opt subs i |> or_not_found
+let stop_opt subs i = offset_opt subs i |> Option.map snd
+let stop subs i = stop_opt subs i |> or_not_found
+let test t i = Mark_infos.test t.marks i
+let get_opt t i = if test t i then Some (get t i) else None
+let dummy_offset = -1, -1
+
+let all_offset t =
+  let res = Array.make t.gcount dummy_offset in
+  Mark_infos.iteri t.marks ~f:(fun i start stop ->
+    let p1 = t.gpos.(start) in
+    let p2 = t.gpos.(stop) in
+    res.(i) <- p1, p2);
+  res
+;;
+
+let dummy_string = ""
+
+let all t =
+  let res = Array.make t.gcount dummy_string in
+  Mark_infos.iteri t.marks ~f:(fun i start stop ->
+    let p1 = t.gpos.(start) in
+    let p2 = t.gpos.(stop) in
+    res.(i) <- String.sub t.s p1 (p2 - p1));
+  res
+;;
+
+let pp fmt t =
+  let matches =
+    let offsets = all_offset t in
+    let strs = all t in
+    Array.to_list (Array.init (Array.length strs) (fun i -> strs.(i), offsets.(i)))
+  in
+  let open Format in
+  let open Fmt in
+  let pp_match fmt (str, (start, stop)) = fprintf fmt "@[(%s (%d %d))@]" str start stop in
+  sexp fmt "Group" (list pp_match) matches
+;;
+
+let nb_groups t = t.gcount