You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
pollen/readability.rkt

191 lines
6.9 KiB
Racket

#lang racket/base
11 years ago
(require racket/contract)
11 years ago
(require (only-in racket/list empty? range))
11 years ago
(require (only-in racket/format ~a))
11 years ago
(require (only-in racket/string string-join))
(require (only-in racket/vector vector-member))
11 years ago
(require (only-in racket/set set set->list set?))
11 years ago
(module+ test (require rackunit))
11 years ago
(require "debug.rkt")
(provide (all-defined-out))
11 years ago
;; general way of coercing to string
11 years ago
(define/contract (->string x)
11 years ago
(any/c . -> . string?)
(cond
11 years ago
[(string? x) x]
11 years ago
[(empty? x) ""]
[(symbol? x) (symbol->string x)]
[(number? x) (number->string x)]
[(path? x) (path->string x)]
[(char? x) (~a x)]
[else (error (format "Can't make ~a into string" x))]))
(module+ test
11 years ago
(check-equal? (->string "foo") "foo")
(check-equal? (->string '()) "")
(check-equal? (->string 'foo) "foo")
(check-equal? (->string 123) "123")
11 years ago
(define file-name-as-text "foo.txt")
11 years ago
(check-equal? (->string (string->path file-name-as-text)) file-name-as-text)
(check-equal? (->string #\¶) ""))
11 years ago
11 years ago
;; general way of coercing to symbol
11 years ago
(define (->symbol thing)
11 years ago
; todo: on bad input, it will pop a string error rather than symbol error
11 years ago
(string->symbol (->string thing)))
11 years ago
11 years ago
;; general way of coercing to a list
11 years ago
(define/contract (->list x)
11 years ago
(any/c . -> . list?)
(cond
[(list? x) x]
[(vector? x) (vector->list x)]
11 years ago
[(set? x) (set->list x)]
11 years ago
[else (list x)]))
(module+ test
11 years ago
(check-equal? (->list '(1 2 3)) '(1 2 3))
(check-equal? (->list (list->vector '(1 2 3))) '(1 2 3))
11 years ago
(check-equal? (->list (set 1 2 3)) '(1 2 3))
11 years ago
(check-equal? (->list "foo") (list "foo")))
11 years ago
11 years ago
;; general way of coercing to boolean
11 years ago
(define/contract (->boolean x)
11 years ago
(any/c . -> . boolean?)
;; in Racket, everything but #f is true
(if x #t #f))
(module+ test
11 years ago
(check-true (->boolean #t))
(check-false (->boolean #f))
(check-true (->boolean "#f"))
(check-true (->boolean "foo"))
(check-true (->boolean '()))
(check-true (->boolean '(1 2 3))))
11 years ago
11 years ago
(define/contract (has-length? x)
(any/c . -> . boolean?)
(ormap (λ(proc) (proc x)) (list list? string? symbol? vector? hash? set?)))
11 years ago
11 years ago
;; general way of asking for length
11 years ago
(define/contract (len x)
(has-length? . -> . integer?)
11 years ago
(cond
[(list? x) (length x)]
[(string? x) (string-length x)]
11 years ago
[(symbol? x) (len (->string x))]
11 years ago
[(vector? x) (len (->list x))]
11 years ago
[(hash? x) (len (hash-keys x))]
11 years ago
[(set? x) (len (->list x))]
11 years ago
[else #f]))
(module+ test
(check-equal? (len '(1 2 3)) 3)
(check-not-equal? (len '(1 2)) 3) ; len 2
(check-equal? (len "foo") 3)
(check-not-equal? (len "fo") 3) ; len 2
(check-equal? (len 'foo) 3)
(check-not-equal? (len 'fo) 3) ; len 2
(check-equal? (len (list->vector '(1 2 3))) 3)
(check-not-equal? (len (list->vector '(1 2))) 3) ; len 2
11 years ago
(check-equal? (len (set 1 2 3)) 3)
(check-not-equal? (len (set 1 2)) 3) ; len 2
11 years ago
(check-equal? (len (make-hash '((a . 1) (b . 2) (c . 3)))) 3)
(check-not-equal? (len (make-hash '((a . 1) (b . 2)))) 3)) ; len 2
11 years ago
(define/contract (sliceable-container? x)
(any/c . -> . boolean?)
(ormap (λ(proc) (proc x)) (list list? string? symbol? vector?)))
(define/contract (container? x)
(any/c . -> . boolean?)
(ormap (λ(proc) (proc x)) (list sliceable-container? hash?)))
11 years ago
;; general way of fetching an item from a container
11 years ago
(define/contract (get container start [end #f])
11 years ago
((container? any/c) ((λ(i)(or (integer? i) (and (symbol? i) (equal? i 'end)))))
. ->* . any/c)
11 years ago
11 years ago
(set! end
(if (sliceable-container? container)
(cond
;; treat negative lengths as offset from end (Python style)
[(and (integer? end) (< end 0)) (+ (len container) end)]
;; 'end slices to the end
[(equal? end 'end) (len container)]
;; default to slice length of 1 (i.e, single-item retrieval)
[(equal? end #f) (add1 start)]
[else end])
end))
11 years ago
(define result (cond
;; for sliceable containers, make a slice
11 years ago
[(list? container) (for/list ([i (range start end)])
11 years ago
(list-ref container i))]
11 years ago
[(vector? container) (for/vector ([i (range start end)])
11 years ago
(vector-ref container i))]
11 years ago
[(string? container) (substring container start end)]
[(symbol? container) (->symbol (get (->string container) start end))]
11 years ago
;; for hash, just get item
11 years ago
[(hash? container) (hash-ref container start)]
11 years ago
[else #f]))
;; don't return single-item results inside a list
(if (and (sliceable-container? result) (= (len result) 1))
11 years ago
(car (->list result))
11 years ago
result))
(module+ test
(check-equal? (get '(0 1 2 3 4 5) 2) 2)
(check-equal? (get '(0 1 2 3 4 5) 0 2) '(0 1))
(check-equal? (get '(0 1 2 3 4 5) 2 -1) '(2 3 4))
(check-equal? (get '(0 1 2 3 4 5) 2 'end) '(2 3 4 5))
(check-equal? (get (list->vector '(0 1 2 3 4 5)) 2) 2)
(check-equal? (get (list->vector'(0 1 2 3 4 5)) 0 2) (list->vector '(0 1)))
(check-equal? (get (list->vector'(0 1 2 3 4 5)) 2 -1) (list->vector '(2 3 4)))
(check-equal? (get (list->vector'(0 1 2 3 4 5)) 2 'end) (list->vector '(2 3 4 5)))
(check-equal? (get "purple" 2) "r")
(check-equal? (get "purple" 0 2) "pu")
(check-equal? (get "purple" 2 -1) "rpl")
(check-equal? (get "purple" 2 'end) "rple")
(check-equal? (get 'purple 2) 'r)
(check-equal? (get 'purple 0 2) 'pu)
(check-equal? (get 'purple 2 -1) 'rpl)
(check-equal? (get 'purple 2 'end) 'rple)
(check-equal? (get (make-hash '((a . 1) (b . 2) (c . 3))) 'a) 1))
;; general way of testing for membership (à la Python 'in')
11 years ago
;; put item as first arg so function can use infix notation
;; (item . in . container)
11 years ago
(define/contract (in? item container)
(any/c any/c . -> . boolean?)
(->boolean (cond
[(list? container) (member item container)] ; returns #f or sublist beginning with item
[(vector? container) (vector-member item container)] ; returns #f or zero-based item index
[(hash? container)
(and (hash-has-key? container item) (get container item))] ; returns #f or hash value
[(string? container) ((->string item) . in? . (map ->string (string->list container)))] ; returns #f or substring beginning with item
[(symbol? container) ((->string item) . in? . (->string container))] ; returns #f or subsymbol (?!) beginning with item
[else #f])))
11 years ago
(module+ test
11 years ago
(check-true (2 . in? . '(1 2 3)))
(check-false (4 . in? . '(1 2 3)))
(check-true (2 . in? . (list->vector '(1 2 3))))
(check-false (4 . in? . (list->vector '(1 2 3))))
(check-true ('a . in? . (make-hash '((a . 1) (b . 2) (c . 3)))))
(check-false ('x . in? . (make-hash '((a . 1) (b . 2) (c . 3)))))
(check-true ("o" . in? . "foobar"))
(check-false ("z" . in? . "foobar"))
(check-true ('o . in? . 'foobar))
(check-false ('z . in? . 'foobar))
(check-false ("F" . in? . #\F)))