310 lines
11 KiB
Clojure
310 lines
11 KiB
Clojure
(ns ^{:doc " Utility functions needed by MicroWorld and, specifically, in the
|
|
interpretation of MicroWorld rule."
|
|
:author "Simon Brooke"}
|
|
mw-engine.utils
|
|
(:require
|
|
[clojure.math.combinatorics :as combo]))
|
|
|
|
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
|
|
;;;;
|
|
;;;; mw-engine: the state/transition engine of MicroWorld.
|
|
;;;;
|
|
;;;; This program is free software; you can redistribute it and/or
|
|
;;;; modify it under the terms of the GNU General Public License
|
|
;;;; as published by the Free Software Foundation; either version 2
|
|
;;;; of the License, or (at your option) any later version.
|
|
;;;;
|
|
;;;; This program is distributed in the hope that it will be useful,
|
|
;;;; but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
;;;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
;;;; GNU General Public License for more details.
|
|
;;;;
|
|
;;;; You should have received a copy of the GNU General Public License
|
|
;;;; along with this program; if not, write to the Free Software
|
|
;;;; Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
|
|
;;;; USA.
|
|
;;;;
|
|
;;;; Copyright (C) 2014 Simon Brooke
|
|
;;;;
|
|
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
|
|
|
|
(defn abs
|
|
"Surprisingly, Clojure doesn't seem to have an abs function, or else I've
|
|
missed it. So here's one of my own. Maps natural numbers onto themselves,
|
|
and negative integers onto natural numbers. Also maps negative real numbers
|
|
onto positive real numbers.
|
|
|
|
* `n` a number, on the set of real numbers."
|
|
[n]
|
|
(if (neg? n) (- 0 n) n))
|
|
|
|
|
|
(defn member?
|
|
"True if elt is a member of col."
|
|
[elt col] (some #(= elt %) col))
|
|
|
|
|
|
(defn get-int-or-zero
|
|
"Return the value of this `property` from this `map` if it is a integer;
|
|
otherwise return zero."
|
|
[map property]
|
|
(let [value (map property)]
|
|
(if (integer? value) value 0)))
|
|
|
|
|
|
(defn init-generation
|
|
"Return a cell like this `cell`, but having a value for :generation, zero if
|
|
the cell passed had no integer value for generation, otherwise the value
|
|
taken from the cell passed. The `world` argument is present only for
|
|
consistency with the rule engine and is ignored."
|
|
[world cell]
|
|
(merge cell {:generation (get-int-or-zero cell :generation)}))
|
|
|
|
|
|
(defn in-bounds
|
|
"True if x, y are in bounds for this world (i.e., there is a cell at x, y)
|
|
else false.
|
|
|
|
* `world` a world as defined above;
|
|
* `x` a number which may or may not be a valid x coordinate within that world;
|
|
* `y` a number which may or may not be a valid y coordinate within that world."
|
|
[world x y]
|
|
(and (>= x 0)(>= y 0)(< y (count world))(< x (count (first world)))))
|
|
|
|
|
|
(defn map-world-n-n
|
|
"Wholly non-parallel map world implementation; see documentation for `map-world`."
|
|
([world function]
|
|
(map-world-n-n world function nil))
|
|
([world function additional-args]
|
|
(into []
|
|
(map (fn [row]
|
|
(into [] (map
|
|
#(apply function
|
|
(cons world (cons % additional-args)))
|
|
row)))
|
|
world))))
|
|
|
|
|
|
(defn map-world-p-p
|
|
"Wholly parallel map-world implementation; see documentation for `map-world`."
|
|
([world function]
|
|
(map-world-p-p world function nil))
|
|
([world function additional-args]
|
|
(into []
|
|
(pmap (fn [row]
|
|
(into [] (pmap
|
|
#(apply function
|
|
(cons world (cons % additional-args)))
|
|
row)))
|
|
world))))
|
|
|
|
|
|
(defn map-world
|
|
"Apply this `function` to each cell in this `world` to produce a new world.
|
|
the arguments to the function will be the world, the cell, and any
|
|
`additional-args` supplied. Note that we parallel map over rows but
|
|
just map over cells within a row. That's because it isn't worth starting
|
|
a new thread for each cell, but there may be efficiency gains in
|
|
running rows in parallel."
|
|
([world function]
|
|
(map-world world function nil))
|
|
([world function additional-args]
|
|
(into []
|
|
(pmap (fn [row]
|
|
(into [] (mapv
|
|
#(apply function
|
|
(cons world (cons % additional-args)))
|
|
row)))
|
|
world))))
|
|
|
|
|
|
(defn get-cell
|
|
"Return the cell a x, y in this world, if any.
|
|
|
|
* `world` a world as defined above;
|
|
* `x` a number which may or may not be a valid x coordinate within that world;
|
|
* `y` a number which may or may not be a valid y coordinate within that world."
|
|
[world x y]
|
|
(cond (in-bounds world x y)
|
|
(nth (nth world y) x)))
|
|
|
|
|
|
(defn get-int
|
|
"Get the value of a property expected to be an integer from a map; if not present (or not an integer) return 0.
|
|
|
|
* `map` a map;
|
|
* `key` a symbol or keyword, presumed to be a key into the `map`."
|
|
[map key]
|
|
(cond (map? map)
|
|
(let [v (map key)]
|
|
(cond (and v (integer? v)) v
|
|
true 0))
|
|
true (throw (Exception. "No map passed?"))))
|
|
|
|
|
|
(defn population
|
|
"Return the population of this species in this cell. Currently a synonym for
|
|
`get-int`, but may not always be (depending whether species are later
|
|
implemented as actors)
|
|
|
|
* `cell` a map;
|
|
* `species` a keyword representing a species which may populate that cell."
|
|
[cell species]
|
|
(get-int cell species))
|
|
|
|
|
|
(def memo-get-neighbours
|
|
"Memoised get neighbours is more efficient when running deeply recursive
|
|
algorithms on the same world. But it's less efficient when running the
|
|
engine in its normal iterative style, because then we will rarely call
|
|
get naighbours on the same cell of the same world twice."
|
|
(memoize
|
|
(fn [world x y depth]
|
|
(remove nil?
|
|
(map #(get-cell world (first %) (first (rest %)))
|
|
(remove #(= % (list x y))
|
|
(combo/cartesian-product
|
|
(range (- x depth) (+ x depth 1))
|
|
(range (- y depth) (+ y depth 1)))))))))
|
|
|
|
|
|
(defn get-neighbours
|
|
"Get the neighbours to distance depth of a cell in this world.
|
|
|
|
Several overloads:
|
|
* `world` a world, as described in world.clj;
|
|
* `cell` a cell within that world
|
|
Gets immediate neighbours of the specified cell.
|
|
|
|
* `world` a world, as described in world.clj;
|
|
* `cell` a cell within that world
|
|
* `depth` an integer representing the depth to search from the
|
|
`cell`
|
|
Gets neighbours within the specified distance of the cell.
|
|
|
|
* `world` a world, as described in world.clj;
|
|
* `x` an integer representing an x coordinate in that world;
|
|
* `y` an integer representing an y coordinate in that world;
|
|
* `depth` an integer representing the distance from [x,y] that
|
|
should be searched
|
|
Gets the neighbours within the specified distance of the cell at
|
|
coordinates [x,y] in this world."
|
|
([world x y depth]
|
|
(remove nil?
|
|
(map #(get-cell world (first %) (first (rest %)))
|
|
(remove #(= % (list x y))
|
|
(combo/cartesian-product
|
|
(range (- x depth) (+ x depth 1))
|
|
(range (- y depth) (+ y depth 1)))))))
|
|
([world cell depth]
|
|
(memo-get-neighbours world (:x cell) (:y cell) depth))
|
|
([world cell]
|
|
(get-neighbours world cell 1)))
|
|
|
|
|
|
(defn get-neighbours-with-property-value
|
|
"Get the neighbours to distance depth of the cell at x, y in this world which
|
|
have this value for this property.
|
|
|
|
* `world` a world, as described in `world.clj`;
|
|
* `cell` a cell within that world;
|
|
* `depth` an integer representing the distance from [x,y] that
|
|
should be searched (optional);
|
|
* `property` a keyword representing a property of the neighbours;
|
|
* `value` a value of that property (or, possibly, the name of another);
|
|
* `op` a comparator function to use in place of `=` (optional).
|
|
|
|
It gets messy."
|
|
([world x y depth property value op]
|
|
(filter
|
|
#(eval
|
|
(list op
|
|
(or (get % property) (get-int % property))
|
|
value))
|
|
(get-neighbours world x y depth)))
|
|
([world x y depth property value]
|
|
(get-neighbours-with-property-value world x y depth property value =))
|
|
([world cell depth property value]
|
|
(get-neighbours-with-property-value world (:x cell) (:y cell) depth
|
|
property value))
|
|
([world cell property value]
|
|
(get-neighbours-with-property-value world cell 1
|
|
property value)))
|
|
|
|
(defn get-neighbours-with-state
|
|
"Get the neighbours to distance depth of the cell at x, y in this world which
|
|
have this state.
|
|
|
|
* `world` a world, as described in `world.clj`;
|
|
* `cell` a cell within that world;
|
|
* `depth` an integer representing the distance from [x,y] that
|
|
should be searched;
|
|
* `state` a keyword representing a state in the world."
|
|
([world x y depth state]
|
|
(filter #(= (:state %) state) (get-neighbours world x y depth)))
|
|
([world cell depth state]
|
|
(get-neighbours-with-state world (:x cell) (:y cell) depth state))
|
|
([world cell state]
|
|
(get-neighbours-with-state world cell 1 state)))
|
|
|
|
|
|
(defn get-least-cell
|
|
"Return the cell from among these `cells` which has the lowest numeric value
|
|
for this `property`; if the property is absent or not a number, use this
|
|
`default`"
|
|
([cells property default]
|
|
(cond
|
|
(empty? cells) nil
|
|
true (let [downstream (get-least-cell (rest cells) property default)]
|
|
(cond (<
|
|
(or (property (first cells)) default)
|
|
(or (property downstream) default)) (first cells)
|
|
true downstream))))
|
|
([cells property]
|
|
(get-least-cell cells property (Integer/MAX_VALUE))))
|
|
|
|
|
|
(defn- set-cell-property
|
|
"If this `cell`s x and y properties are equal to these `x` and `y` values,
|
|
return a cell like this cell but with the value of this `property` set to
|
|
this `value`. Otherwise, just return this `cell`."
|
|
[cell x y property value]
|
|
(cond
|
|
(and (= x (:x cell)) (= y (:y cell)))
|
|
(merge cell {property value :rule "Set by user"})
|
|
true
|
|
cell))
|
|
|
|
|
|
(defn set-property
|
|
"Return a world like this `world` but with the value of exactly one `property`
|
|
of one `cell` changed to this `value`"
|
|
([world cell property value]
|
|
(set-property world (:x cell) (:y cell) property value))
|
|
([world x y property value]
|
|
(apply
|
|
vector ;; we want a vector of vectors, not a list of lists, for efficiency
|
|
(map
|
|
(fn [row]
|
|
(apply
|
|
vector
|
|
(map #(set-cell-property % x y property value)
|
|
row)))
|
|
world))))
|
|
|
|
|
|
(defn merge-cell
|
|
"Return a world like this `world`, but merge the values from this `cell` with
|
|
those from the cell in the world with the same co-ordinates"
|
|
[world cell]
|
|
(if (in-bounds world (:x cell) (:y cell))
|
|
(map-world world
|
|
#(if
|
|
(and
|
|
(= (:x cell)(:x %2))
|
|
(= (:y cell)(:y %2)))
|
|
(merge %2 cell)
|
|
%2))
|
|
world))
|