Source

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

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)
Tip: Filter by directory path e.g. /media app.js to search for public/media/app.js.
Tip: Use camelCasing e.g. ProjME to search for ProjectModifiedEvent.java.
Tip: Filter by extension type e.g. /repo .js to search for all .js files in the /repo directory.
Tip: Separate your search with spaces e.g. /ssh pom.xml to search for src/ssh/pom.xml.
Tip: Use ↑ and ↓ arrow keys to navigate and return to view the file.
Tip: You can also navigate files with Ctrl+j (next) and Ctrl+k (previous) and view the file with Ctrl+o.
Tip: You can also navigate files with Alt+j (next) and Alt+k (previous) and view the file with Alt+o.