1
0
mirror of https://github.com/janet-lang/janet synced 2024-12-25 16:00:27 +00:00

take-drop performance tweaks

Increase efficiency for `take` and `drop` with slices.
Check indexed types before bytes types.
This commit is contained in:
primo-ppcg 2023-06-08 13:26:13 +07:00
parent b5407ac708
commit a238391b36

View File

@ -1083,9 +1083,9 @@
(defn- take-n-slice (defn- take-n-slice
[f n ind] [f n ind]
(def len (length ind)) (def len (length ind))
(def negn (< n 0)) (def m (+ len n))
(def start (if negn (max 0 (+ len n)) 0)) (def start (if (< n 0 m) m 0))
(def end (if negn len (min n len))) (def end (if (<= 0 n len) n len))
(f ind start end)) (f ind start end))
(defn take (defn take
@ -1093,8 +1093,8 @@
respectively. If `n` is negative, takes the last `n` elements instead.`` respectively. If `n` is negative, takes the last `n` elements instead.``
[n ind] [n ind]
(cond (cond
(bytes? ind) (take-n-slice string/slice n ind)
(indexed? ind) (take-n-slice tuple/slice n ind) (indexed? ind) (take-n-slice tuple/slice n ind)
(bytes? ind) (take-n-slice string/slice n ind)
(dictionary? ind) (do (dictionary? ind) (do
(var left n) (var left n)
(tabseq [[i x] :pairs ind :until (< (-- left) 0)] i x)) (tabseq [[i x] :pairs ind :until (< (-- left) 0)] i x))
@ -1117,8 +1117,8 @@
"Same as `(take-while (complement pred) ind)`." "Same as `(take-while (complement pred) ind)`."
[pred ind] [pred ind]
(cond (cond
(bytes? ind) (take-until-slice string/slice pred ind)
(indexed? ind) (take-until-slice tuple/slice pred ind) (indexed? ind) (take-until-slice tuple/slice pred ind)
(bytes? ind) (take-until-slice string/slice pred ind)
(dictionary? ind) (tabseq [[i x] :pairs ind :until (pred x)] i x) (dictionary? ind) (tabseq [[i x] :pairs ind :until (pred x)] i x)
(seq [x :in ind :until (pred x)] x))) (seq [x :in ind :until (pred x)] x)))
@ -1131,10 +1131,10 @@
(defn- drop-n-slice (defn- drop-n-slice
[f n ind] [f n ind]
(def len (length ind)) (def len (length ind))
(def negn (< n 0)) (cond
(def start (if negn 0 (min n len))) (<= 0 n len) (f ind n)
(def end (if negn (max 0 (+ len n)) len)) (< (- len) n 0) (f ind 0 (+ len n))
(f ind start end)) (f ind 0 0)))
(defn- drop-n-dict (defn- drop-n-dict
[f n ind] [f n ind]
@ -1148,8 +1148,8 @@
instance, respectively. If `n` is negative, drops the last `n` elements instead.`` instance, respectively. If `n` is negative, drops the last `n` elements instead.``
[n ind] [n ind]
(cond (cond
(bytes? ind) (drop-n-slice string/slice n ind)
(indexed? ind) (drop-n-slice tuple/slice n ind) (indexed? ind) (drop-n-slice tuple/slice n ind)
(bytes? ind) (drop-n-slice string/slice n ind)
(struct? ind) (drop-n-dict struct/to-table n ind) (struct? ind) (drop-n-dict struct/to-table n ind)
(table? ind) (drop-n-dict table/clone n ind) (table? ind) (drop-n-dict table/clone n ind)
(do (do
@ -1175,8 +1175,8 @@
"Same as `(drop-while (complement pred) ind)`." "Same as `(drop-while (complement pred) ind)`."
[pred ind] [pred ind]
(cond (cond
(bytes? ind) (drop-until-slice string/slice pred ind)
(indexed? ind) (drop-until-slice tuple/slice pred ind) (indexed? ind) (drop-until-slice tuple/slice pred ind)
(bytes? ind) (drop-until-slice string/slice pred ind)
(struct? ind) (drop-until-dict struct/to-table pred ind) (struct? ind) (drop-until-dict struct/to-table pred ind)
(table? ind) (drop-until-dict table/clone pred ind) (table? ind) (drop-until-dict table/clone pred ind)
(do (find pred ind) ind))) (do (find pred ind) ind)))