blob: 97a65e3b1c1ce239e385f636930ad76d796debfe (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
;; datastructures.cljs - functions for organizing data
;; Copyright (C) 2014 David Barksdale <amatus@amatus.name>
;;
;; 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 3 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, see <http://www.gnu.org/licenses/>.
(ns amatus.datastructures)
(defn nested-group-by
([fs coll] (nested-group-by fs identity coll))
([fs inner-fn coll]
(if (empty? fs)
(inner-fn coll)
(persistent!
(reduce
(fn [ret [k v]]
(assoc! ret k (nested-group-by (rest fs) inner-fn v)))
(transient {}) (group-by (first fs) coll))))))
(defn flatten-nested-maps
([nested-maps level-keys] (flatten-nested-maps nested-maps level-keys {}))
([nested-maps [level-key & level-keys] level-accumulator]
(if (empty? level-keys)
(for [v nested-maps]
(assoc level-accumulator level-key v))
(mapcat #(flatten-nested-maps
(val %)
level-keys
(assoc level-accumulator level-key (key %)))
nested-maps))))
|
