aboutsummaryrefslogtreecommitdiff
path: root/src/macroknight/utils.hy
blob: 5de88872836df67f4912e77547b41c14cc584fba (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
41
42
43
44
45
46
47
48
49
50
51
(import enum [Enum]
        math [sqrt])

(defclass Direction [Enum]
  (setv UP "UP")
  (setv DOWN "DOWN")
  (setv LEFT "LEFT")
  (setv RIGHT "RIGHT")

  (defn [staticmethod] from-move [move]
    #((.x-from-move Direction move)
       (.y-from-move Direction move)))

  (defn [staticmethod] x-from-move [move]
    (if (>= (get move 0) 0)
        Direction.RIGHT
        Direction.LEFT))

  (defn [staticmethod] y-from-move [move]
    (if (>= (get move 0) 0)
        Direction.DOWN
        Direction.UP))

  (defn [staticmethod] x-between-points [point1 point2]
    (let [delta-x (abs (- (get point1 0) (get point2 0)))]
      (if (>= delta-x 0)
          Direction.RIGHT
          Direction.LEFT))))

(defn neg [value]
  (* -1 value))

(defn invert [move]
  #((neg (get move 0))
     (neg (get move 1))))

(defn merge-moves [moves]
  (let [end-move #(0 0)]
    (for [move moves]
      (setv end-move
            #((+ (get end-move 0) (get move 0))
               (+ (get end-move 1) (get move 1)))))
    end-move))

(defn sub-points [point1 point2]
  #((- (get point2 0) (get point1 0))
     (- (get point2 1) (get point2 1))))

(defn distance [point1 point2]
  (sqrt (+ (** (- (get point1 0) (get point2 0)) 2)
           (** (- (get point1 1) (get point2 1)) 2))))