136 lines
4.4 KiB
OCaml
136 lines
4.4 KiB
OCaml
|
(* Copy of buffer.ml from the ocaml 3.12 source code, extended with a
|
||
|
new function write, for integrating with Lwt. Removed a few
|
||
|
functions which for some reason don't compile using the Lwt camlp4
|
||
|
extensions (?). *)
|
||
|
(***********************************************************************)
|
||
|
(* *)
|
||
|
(* Objective Caml *)
|
||
|
(* *)
|
||
|
(* Pierre Weis and Xavier Leroy, projet Cristal, INRIA Rocquencourt *)
|
||
|
(* *)
|
||
|
(* Copyright 1999 Institut National de Recherche en Informatique et *)
|
||
|
(* en Automatique. All rights reserved. This file is distributed *)
|
||
|
(* under the terms of the GNU Library General Public License, with *)
|
||
|
(* the special exception on linking described in file ../LICENSE. *)
|
||
|
(* *)
|
||
|
(***********************************************************************)
|
||
|
|
||
|
(* $Id: buffer.ml 10216 2010-03-28 08:16:45Z xleroy $ *)
|
||
|
|
||
|
(* Extensible buffers *)
|
||
|
|
||
|
type t =
|
||
|
{mutable buffer : string;
|
||
|
mutable position : int;
|
||
|
mutable length : int;
|
||
|
initial_buffer : string}
|
||
|
|
||
|
let create n =
|
||
|
let n = if n < 1 then 1 else n in
|
||
|
let n = if n > Sys.max_string_length then Sys.max_string_length else n in
|
||
|
let s = String.create n in
|
||
|
{buffer = s; position = 0; length = n; initial_buffer = s}
|
||
|
|
||
|
let contents b = String.sub b.buffer 0 b.position
|
||
|
|
||
|
let sub b ofs len =
|
||
|
if ofs < 0 || len < 0 || ofs > b.position - len
|
||
|
then invalid_arg "Obuffer.sub"
|
||
|
else begin
|
||
|
let r = String.create len in
|
||
|
String.blit b.buffer ofs r 0 len;
|
||
|
r
|
||
|
end
|
||
|
;;
|
||
|
|
||
|
let blit src srcoff dst dstoff len =
|
||
|
if len < 0 || srcoff < 0 || srcoff > src.position - len
|
||
|
|| dstoff < 0 || dstoff > (String.length dst) - len
|
||
|
then invalid_arg "Obuffer.blit"
|
||
|
else
|
||
|
String.blit src.buffer srcoff dst dstoff len
|
||
|
;;
|
||
|
|
||
|
let nth b ofs =
|
||
|
if ofs < 0 || ofs >= b.position then
|
||
|
invalid_arg "Obuffer.nth"
|
||
|
else String.get b.buffer ofs
|
||
|
;;
|
||
|
|
||
|
let length b = b.position
|
||
|
|
||
|
let clear b = b.position <- 0
|
||
|
|
||
|
let reset b =
|
||
|
b.position <- 0; b.buffer <- b.initial_buffer;
|
||
|
b.length <- String.length b.buffer
|
||
|
|
||
|
let resize b more =
|
||
|
let len = b.length in
|
||
|
let new_len = ref len in
|
||
|
while b.position + more > !new_len do new_len := 2 * !new_len done;
|
||
|
if !new_len > Sys.max_string_length then begin
|
||
|
if b.position + more <= Sys.max_string_length
|
||
|
then new_len := Sys.max_string_length
|
||
|
else failwith "Obuffer.add: cannot grow buffer"
|
||
|
end;
|
||
|
let new_buffer = String.create !new_len in
|
||
|
String.blit b.buffer 0 new_buffer 0 b.position;
|
||
|
b.buffer <- new_buffer;
|
||
|
b.length <- !new_len
|
||
|
|
||
|
let add_char b c =
|
||
|
let pos = b.position in
|
||
|
if pos >= b.length then resize b 1;
|
||
|
b.buffer.[pos] <- c;
|
||
|
b.position <- pos + 1
|
||
|
|
||
|
let add_substring b s offset len =
|
||
|
if offset < 0 || len < 0 || offset > String.length s - len
|
||
|
then invalid_arg "Obuffer.add_substring";
|
||
|
let new_position = b.position + len in
|
||
|
if new_position > b.length then resize b len;
|
||
|
String.blit s offset b.buffer b.position len;
|
||
|
b.position <- new_position
|
||
|
|
||
|
let add_string b s =
|
||
|
let len = String.length s in
|
||
|
let new_position = b.position + len in
|
||
|
if new_position > b.length then resize b len;
|
||
|
String.blit s 0 b.buffer b.position len;
|
||
|
b.position <- new_position
|
||
|
|
||
|
let add_buffer b bs =
|
||
|
add_substring b bs.buffer 0 bs.position
|
||
|
|
||
|
let add_channel b ic len =
|
||
|
if len < 0 || len > Sys.max_string_length then (* PR#5004 *)
|
||
|
invalid_arg "Obuffer.add_channel";
|
||
|
if b.position + len > b.length then resize b len;
|
||
|
really_input ic b.buffer b.position len;
|
||
|
b.position <- b.position + len
|
||
|
|
||
|
let output_buffer oc b =
|
||
|
output oc b.buffer 0 b.position
|
||
|
|
||
|
let write c b =
|
||
|
Lwt_io.write_from_exactly c b.buffer 0 b.position
|
||
|
|
||
|
let closing = function
|
||
|
| '(' -> ')'
|
||
|
| '{' -> '}'
|
||
|
| _ -> assert false;;
|
||
|
|
||
|
(* opening and closing: open and close characters, typically ( and )
|
||
|
k: balance of opening and closing chars
|
||
|
s: the string where we are searching
|
||
|
start: the index where we start the search. *)
|
||
|
let advance_to_closing opening closing k s start =
|
||
|
let rec advance k i lim =
|
||
|
if i >= lim then raise Not_found else
|
||
|
if s.[i] = opening then advance (k + 1) (i + 1) lim else
|
||
|
if s.[i] = closing then
|
||
|
if k = 0 then i else advance (k - 1) (i + 1) lim
|
||
|
else advance k (i + 1) lim in
|
||
|
advance k start (String.length s);;
|