Source

OCaml Map Reduce Project / apps / nbody / mapper.ml

Full commit
let (id1, x) = Program.get_input() in
let (mass1, position1, _) = Util.unmarshal x in
let bodies = Util.unmarshal (Program.get_shared_data()) in
let helper acc other =
  let (id2, (mass2, position2, _)) = other in
  let dist = Plane.distance position1 position2 in
  let g_force = (Util.cBIG_G *. mass1) /. (dist *. dist) in
  let direction = Plane.unit_vector position2 position1 in
  if id1 = id2 then acc
  else (other, Util.marshal (Plane.scale_point g_force direction))::acc in
Program.set_output (List.fold_left helper []
                   bodies)