aboutsummaryrefslogtreecommitdiff
path: root/stdlib
diff options
context:
space:
mode:
Diffstat (limited to 'stdlib')
-rw-r--r--stdlib/source/lux/control/parser/binary.lux7
-rw-r--r--stdlib/source/lux/data/binary.lux59
-rw-r--r--stdlib/source/lux/data/format/tar.lux2
-rw-r--r--stdlib/source/program/aedifex/repository/remote.lux2
-rw-r--r--stdlib/source/test/lux/data/binary.lux27
-rw-r--r--stdlib/source/test/lux/world/file.lux4
6 files changed, 48 insertions, 53 deletions
diff --git a/stdlib/source/lux/control/parser/binary.lux b/stdlib/source/lux/control/parser/binary.lux
index 423cff74f..7221f5c65 100644
--- a/stdlib/source/lux/control/parser/binary.lux
+++ b/stdlib/source/lux/control/parser/binary.lux
@@ -147,10 +147,9 @@
(function (_ [offset binary])
(case size
0 (#try.Success [[offset binary] (/.create 0)])
- _ (do try.monad
- [#let [end (n.+ size offset)]
- output (/.slice offset (.dec end) binary)]
- (wrap [[end binary] output])))))
+ _ (|> binary
+ (/.slice offset size)
+ (\ try.monad map (|>> [[(n.+ size offset) binary]]))))))
(template [<name> <bits>]
[(def: #export <name>
diff --git a/stdlib/source/lux/data/binary.lux b/stdlib/source/lux/data/binary.lux
index ce8deb1d2..c706ec4cb 100644
--- a/stdlib/source/lux/data/binary.lux
+++ b/stdlib/source/lux/data/binary.lux
@@ -26,16 +26,11 @@
["Size" (%.nat size)]
["Index" (%.nat index)]))
-(template [<name>]
- [(exception: #export (<name> {size Nat} {from Nat} {to Nat})
- (exception.report
- ["Size" (%.nat size)]
- ["From" (%.nat from)]
- ["To" (%.nat to)]))]
-
- [slice_out_of_bounds]
- [inverted_slice]
- )
+(exception: #export (slice_out_of_bounds {size Nat} {offset Nat} {length Nat})
+ (exception.report
+ ["Size" (%.nat size)]
+ ["Offset" (%.nat offset)]
+ ["Length" (%.nat length)]))
(with_expansions [<jvm> (as_is (type: #export Binary (ffi.type [byte]))
@@ -328,32 +323,30 @@
(#try.Success target)))
(exception.throw ..cannot_copy_bytes [bytes source_input target_output]))))))
-(def: #export (slice from to binary)
+(def: #export (slice offset length binary)
(-> Nat Nat Binary (Try Binary))
- (let [size (..!size binary)]
- (if (n.<= to from)
- (if (and (n.< size from)
- (n.< size to))
- (with_expansions [<jvm> (as_is (#try.Success (java/util/Arrays::copyOfRange binary (.int from) (.int (inc to)))))]
- (for {@.old <jvm>
- @.jvm <jvm>}
-
- ## Default
- (let [how_many (inc (n.- from to))]
- (..copy how_many from binary 0 (..create how_many)))))
- (exception.throw ..slice_out_of_bounds [size from to]))
- (exception.throw ..inverted_slice [size from to]))))
-
-(def: #export (drop from binary)
+ (let [size (..!size binary)
+ limit (n.+ length offset)]
+ (if (n.<= size limit)
+ (with_expansions [<jvm> (as_is (#try.Success (java/util/Arrays::copyOfRange binary (.int offset) (.int limit))))]
+ (for {@.old <jvm>
+ @.jvm <jvm>}
+
+ ## Default
+ (..copy length offset binary 0 (..create length))))
+ (exception.throw ..slice_out_of_bounds [size offset length]))))
+
+(def: #export (drop offset binary)
(-> Nat Binary Binary)
- (case from
+ (case offset
0 binary
- _ (case (..slice from (dec (..!size binary)) binary)
- (#try.Success slice)
- slice
-
- (#try.Failure _)
- (..create 0))))
+ _ (let [distance (n.- offset (..!size binary))]
+ (case (..slice offset distance binary)
+ (#try.Success slice)
+ slice
+
+ (#try.Failure _)
+ (..create 0)))))
(structure: #export monoid
(Monoid Binary)
diff --git a/stdlib/source/lux/data/format/tar.lux b/stdlib/source/lux/data/format/tar.lux
index 598b52be6..f796b7a6c 100644
--- a/stdlib/source/lux/data/format/tar.lux
+++ b/stdlib/source/lux/data/format/tar.lux
@@ -235,7 +235,7 @@
(recur (dec end))
_
- (binary.slice 0 end string))))))))
+ (binary.slice 0 (dec end) string))))))))
(template [<type> <representation> <size> <exception> <in> <out> <writer> <parser> <none>]
[(abstract: #export <type>
diff --git a/stdlib/source/program/aedifex/repository/remote.lux b/stdlib/source/program/aedifex/repository/remote.lux
index fd0d65d6f..eb285bb67 100644
--- a/stdlib/source/program/aedifex/repository/remote.lux
+++ b/stdlib/source/program/aedifex/repository/remote.lux
@@ -111,7 +111,7 @@
_ (if (n.= ..buffer_size bytes_read)
(recur (\ binary.monoid compose output buffer))
(do !
- [chunk (\ io.monad wrap (binary.slice 0 (dec (.nat bytes_read)) buffer))]
+ [chunk (\ io.monad wrap (binary.slice 0 (.nat bytes_read) buffer))]
(recur (\ binary.monoid compose output chunk)))))))]
(wrap output)))
diff --git a/stdlib/source/test/lux/data/binary.lux b/stdlib/source/test/lux/data/binary.lux
index 89237babc..11d4a8889 100644
--- a/stdlib/source/test/lux/data/binary.lux
+++ b/stdlib/source/test/lux/data/binary.lux
@@ -84,8 +84,8 @@
sample (..random size)
value random.nat
#let [gen_idx (|> random.nat (\ ! map (n.% size)))]
- [from to] (random.and gen_idx gen_idx)
- #let [[from to] [(n.min from to) (n.max from to)]]]
+ offset gen_idx
+ length (\ ! map (n.% (n.- offset size)) random.nat)]
(_.for [/.Binary]
($_ _.and
(_.for [/.equivalence]
@@ -113,12 +113,15 @@
(_.cover [/.read/64 /.write/64]
(..binary_io 3 /.read/64 /.write/64 value))))
(_.cover [/.slice]
- (let [slice_size (|> to (n.- from) inc)
- random_slice (try.assume (/.slice from to sample))
- idxs (enum.range n.enum 0 (dec slice_size))
- reader (function (_ binary idx) (/.read/8 idx binary))]
- (and (n.= slice_size (/.size random_slice))
- (case [(monad.map try.monad (|>> (n.+ from) (reader sample)) idxs)
+ (let [random_slice (try.assume (/.slice offset length sample))
+ idxs (: (List Nat)
+ (case length
+ 0 (list)
+ _ (enum.range n.enum 0 (dec length))))
+ reader (function (_ binary idx)
+ (/.read/8 idx binary))]
+ (and (n.= length (/.size random_slice))
+ (case [(monad.map try.monad (|>> (n.+ offset) (reader sample)) idxs)
(monad.map try.monad (reader random_slice) idxs)]
[(#try.Success binary_vals) (#try.Success slice_vals)]
(\ (list.equivalence n.equivalence) = binary_vals slice_vals)
@@ -127,10 +130,10 @@
#0))))
(_.cover [/.slice_out_of_bounds]
(and (throws? /.slice_out_of_bounds (/.slice size size sample))
- (throws? /.slice_out_of_bounds (/.slice from size sample))))
- (_.cover [/.inverted_slice]
- (or (throws? /.inverted_slice (/.slice to from sample))
- (n.= to from)))
+ (let [verdict (throws? /.slice_out_of_bounds (/.slice offset size sample))]
+ (case offset
+ 0 (not verdict)
+ _ verdict))))
(_.cover [/.drop]
(and (\ /.equivalence = sample (/.drop 0 sample))
(\ /.equivalence = (/.create 0) (/.drop size sample))
diff --git a/stdlib/source/test/lux/world/file.lux b/stdlib/source/test/lux/world/file.lux
index 002d76c42..c7f546a1b 100644
--- a/stdlib/source/test/lux/world/file.lux
+++ b/stdlib/source/test/lux/world/file.lux
@@ -110,10 +110,10 @@
## (wrap (and (n.= (n.* 2 file_size) read_size)
## (\ binary.equivalence =
## dataL
- ## (try.assume (binary.slice 0 (dec file_size) content)))
+ ## (try.assume (binary.slice 0 file_size content)))
## (\ binary.equivalence =
## dataR
- ## (try.assume (binary.slice file_size (dec read_size) content)))))))]
+ ## (try.assume (binary.slice file_size (n.- file_size read_size) content)))))))]
## (_.assert "Can append to files."
## (try.default #0 result))))
## (wrap (do promise.monad
generated by cgit v1.2.3 (git 2.25.1) at 2024-12-23 10:13:47 +0000