|
|
#lang restructure/racket
|
|
|
(require "decodestream.rkt")
|
|
|
|
|
|
;; approximates https://github.com/mbutterick/restructure/blob/master/src/Number.coffee
|
|
|
|
|
|
(define (ends-with-8? type)
|
|
|
(define str (symbol->string type))
|
|
|
(equal? (substring str (sub1 (string-length str))) "8"))
|
|
|
|
|
|
(define (unsigned-type? type)
|
|
|
(equal? "U" (substring (symbol->string type) 0 1)))
|
|
|
|
|
|
(test-module
|
|
|
(check-true (unsigned-type? 'UInt16))
|
|
|
(check-false (unsigned-type? 'Int16)))
|
|
|
|
|
|
(define-subclass RBase (Number [type 'UInt16] [endian (if (system-big-endian?) 'BE 'LE)])
|
|
|
(getter-field [fn (string->symbol (format "~a~a" type (if (ends-with-8? type) "" endian)))])
|
|
|
|
|
|
(unless (hash-has-key? type-sizes fn)
|
|
|
(raise-argument-error 'Number "valid type and endian" (format "~v ~v" type endian)))
|
|
|
|
|
|
(getter-field [size (hash-ref type-sizes fn)])
|
|
|
|
|
|
(define/override (decode stream)
|
|
|
(unless (input-port? stream)
|
|
|
(raise-argument-error 'decode "input port" stream))
|
|
|
(define bstr (read-bytes-exact size stream))
|
|
|
(if (= 1 size)
|
|
|
(bytes-ref bstr 0)
|
|
|
(integer-bytes->integer bstr (unsigned-type? type) (eq? endian 'BE))))
|
|
|
|
|
|
(define/override (encode stream val)
|
|
|
(unless (output-port? stream)
|
|
|
(raise-argument-error 'encode "output port" stream))
|
|
|
(if (= 1 size)
|
|
|
(bytes val)
|
|
|
(integer->integer-bytes val size (unsigned-type? type) (eq? endian 'BE)))))
|
|
|
|
|
|
|
|
|
(test-module
|
|
|
(let ([o (make-object Number 'UInt16 'LE)]
|
|
|
[ip (open-input-bytes (bytes 1 2 3 4))]
|
|
|
[op (open-output-bytes)])
|
|
|
(check-equal? (send o decode ip) 513) ;; 1000 0000 0100 0000
|
|
|
(check-equal? (send o decode ip) 1027) ;; 1100 0000 0010 0000
|
|
|
(check-equal? (send o encode op 513) (bytes 1 2))
|
|
|
(check-equal? (send o encode op 1027) (bytes 3 4)))
|
|
|
|
|
|
(let ([o (make-object Number 'UInt16 'BE)]
|
|
|
[ip (open-input-bytes (bytes 1 2 3 4))]
|
|
|
[op (open-output-bytes)])
|
|
|
(check-equal? (send o decode ip) 258) ;; 0100 0000 1000 0000
|
|
|
(check-equal? (send o decode ip) 772) ;; 0010 0000 1100 0000
|
|
|
(check-equal? (send o encode op 258) (bytes 1 2))
|
|
|
(check-equal? (send o encode op 772) (bytes 3 4))))
|
|
|
|
|
|
|
|
|
(test-module
|
|
|
(check-equal? (send (make-object Number 'UInt8) size) 1)
|
|
|
(check-equal? (send (make-object Number) size) 2)
|
|
|
(check-equal? (send (make-object Number 'UInt32) size) 4)
|
|
|
(check-equal? (send (make-object Number 'Double) size) 8))
|
|
|
|
|
|
|
|
|
(require (for-syntax "decodestream.rkt" racket/match))
|
|
|
|
|
|
;; use keys of type-sizes hash to generate corresponding number definitions
|
|
|
(define-macro (make-int-types)
|
|
|
(with-pattern ([((ID BASE ENDIAN) ...) (for/list ([k (in-hash-keys type-sizes)])
|
|
|
(define kstr (format "~a" k))
|
|
|
(match-define (list* prefix suffix _)
|
|
|
(regexp-split #rx"(?=[BL]E|$)" kstr))
|
|
|
(map string->symbol
|
|
|
(list (string-downcase kstr)
|
|
|
prefix
|
|
|
(if (positive? (string-length suffix))
|
|
|
suffix
|
|
|
(if (system-big-endian?) "BE" "LE")))))]
|
|
|
[(ID ...) (suffix-id #'(ID ...) #:context caller-stx)])
|
|
|
#'(begin (define+provide ID (make-object Number 'BASE 'ENDIAN)) ...)))
|
|
|
|
|
|
|
|
|
(make-int-types)
|
|
|
|
|
|
|
|
|
(test-module
|
|
|
(check-equal? (send uint8 size) 1)
|
|
|
(check-equal? (send uint16 size) 2)
|
|
|
(check-equal? (send uint32 size) 4)
|
|
|
(check-equal? (send double size) 8))
|
|
|
|