src/Pure/Concurrent/par_list.ML
author huffman
Mon, 20 Dec 2010 07:50:47 -0800
changeset 41320 4953e21ac76c
parent 40700 4b4dfe05b5d7
child 41672 2f70b1ddd09f
permissions -rw-r--r--
replace list_map function with an abbreviation

(*  Title:      Pure/Concurrent/par_list.ML
    Author:     Makarius

Parallel list combinators.

Notes:

  * These combinators only make sense if the operator (function or
    predicate) applied to the list of operands takes considerable
    time.  The overhead of scheduling is significantly higher than
    just traversing the list of operands sequentially.

  * The order of operator application is non-deterministic.  Watch out
    for operators that have side-effects or raise exceptions!
*)

signature PAR_LIST =
sig
  val map: ('a -> 'b) -> 'a list -> 'b list
  val get_some: ('a -> 'b option) -> 'a list -> 'b option
  val find_some: ('a -> bool) -> 'a list -> 'a option
  val exists: ('a -> bool) -> 'a list -> bool
  val forall: ('a -> bool) -> 'a list -> bool
end;

structure Par_List: PAR_LIST =
struct

fun managed_results f xs =
  if Multithreading.enabled () andalso not (Multithreading.self_critical ()) then
    let
      val shared_group = Task_Queue.new_group (Future.worker_group ());
      val results =
        Future.join_results (map (fn x => Future.fork_group shared_group (fn () => f x)) xs)
          handle exn =>
            (if Exn.is_interrupt exn then Future.cancel_group shared_group else (); reraise exn);
    in results end
  else map (Exn.capture f) xs;

fun map f xs = Exn.release_first (managed_results f xs);

fun get_some f xs =
  let
    exception FOUND of 'b option;
    fun found (Exn.Exn (FOUND some)) = some
      | found _ = NONE;
    val results =
      managed_results (fn x => (case f x of NONE => () | some => raise FOUND some)) xs;
  in
    (case get_first found results of
      SOME y => SOME y
    | NONE => (Exn.release_first results; NONE))
  end;

fun find_some P = get_some (fn x => if P x then SOME x else NONE);

fun exists P = is_some o get_some (fn x => if P x then SOME () else NONE);
fun forall P = not o exists (not o P);

end;