generate.clj

Pinchord · pinchord.git · Tickets · Changelog

Branches > build-process > 717fc20 > files / src / org / pinchord / generate.clj

(ns org.pinchord.generate
  (:require [clj-yaml.core :as yaml]
            [clojure.string :as str]
            [clojure.edn :as edn]))

(def banks [:prefixes :initials :vowels :finals :suffixes])

(def bank->priority
  "Map of banks to priorities used for javelin orthospelling."
  (zipmap banks (rest (range))))

(defn finalize [m]
  (update-keys m (partial str "-")))

(defn mask [s1 s2]
  (apply str (remove (set s1) s2)))

(defn abbreviate
  "Abbreviates chords that are more than 6 characters,
   as Javelin entries can only be 6 characters long."
  [s]
  (if (> (count s) 6)
    (let [new-s (-> s
                    (str/replace #"TSPW" "G")
                    (str/replace #"PKS" "C")
                    (str/replace #"PKFS" "G")
                    (str/replace #"KFS" "SH")
                    (str/replace #"LR" "N"))]
      (if (> (count new-s) 6)
        (throw (ex-info "Too long!" {:s s :new-s new-s}))
        new-s))
    s))

(defn all-chords [s]
  (if (empty? s)
    #{""}
    (let [rest-combos (all-chords (rest s))]
      (set (concat
             rest-combos
             (map #(str (first s) %) rest-combos))))))

(defn abbreviate-long-chords [s]
  (update-vals s abbreviate))

(defn with-gaps-filled [bank chords]
  (let [s (remove (bank chords) (all-chords (bank (:banks chords))))]
    (merge (bank chords) (abbreviate-long-chords (zipmap s s)))))

(defn fill-gaps-in-banks [chords]
  (assoc chords
         :initials (with-gaps-filled :initials chords)
         :vowels   (with-gaps-filled :vowels chords)
         :finals   (with-gaps-filled :finals chords)))

(defn build-javelin-dictionary [chords]
  {:name (str "Pinchord Base " (:version chords))
   :maximumOutlineLength 1
   :starters [{:stroke "#+" :definition "{^}{-|}" :mask "#+"}
              {:stroke "" :definition "{}" :mask ""}]
   :letters (->> (reduce (fn [acc [bank entries]]
                           (if-let [bank-keys ((:banks chords) bank)]
                             (into acc (mapv (fn [[stroke translation]]
                                               {:stroke stroke :data translation :order (bank->priority bank)
                                                :avoidMask (str (when (= bank :finals)
                                                                  "-")
                                                                (mask stroke bank-keys))})
                                             entries))
                             (into acc (mapv (fn [[stroke translation]]
                                               {:stroke stroke :data translation :order (bank->priority bank)})
                                             entries))))
                         #{} (select-keys chords banks))
                 (sort-by (comp - count :stroke)))})

(defn -main [& _args]
  (let [chords (-> (edn/read-string (slurp "chords.edn"))
                   fill-gaps-in-banks
                   (update :finals finalize)
                   (update :suffixes finalize))
        dict (build-javelin-dictionary chords)
        dict-path (str "pinchord-javelin-dictionary-" (:version chords) ".yaml")]
    (spit dict-path (yaml/generate-string dict))))