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.
84 lines
3.9 KiB
Racket
84 lines
3.9 KiB
Racket
10 years ago
|
#lang typed/racket
|
||
|
|
(require "ocm-typed.rkt")
|
||
|
|
(require typed/rackunit)
|
||
|
|
(require math)
|
||
|
|
|
||
|
|
(define m0 (matrix ((25.0 42.0 57.0 78.0 90.0 103.0 123.0 142.0 151.0)
|
||
|
|
(21.0 35.0 48.0 65.0 76.0 85.0 105.0 123.0 130.0)
|
||
|
|
(13.0 26.0 35.0 51.0 58.0 67.0 86.0 100.0 104.0)
|
||
|
|
(10.0 20.0 28.0 42.0 48.0 56.0 75.0 86.0 88.0)
|
||
|
|
(20.0 29.0 33.0 44.0 49.0 55.0 73.0 82.0 80.0)
|
||
|
|
(13.0 21.0 24.0 35.0 39.0 44.0 59.0 65.0 59.0)
|
||
|
|
(19.0 25.0 28.0 38.0 42.0 44.0 57.0 61.0 52.0)
|
||
|
|
(35.0 37.0 40.0 48.0 48.0 49.0 62.0 62.0 49.0)
|
||
|
|
(37.0 36.0 37.0 42.0 39.0 39.0 51.0 50.0 37.0)
|
||
|
|
(41.0 39.0 37.0 42.0 35.0 33.0 44.0 43.0 29.0)
|
||
|
|
(58.0 56.0 54.0 55.0 47.0 41.0 50.0 47.0 29.0)
|
||
|
|
(66.0 64.0 61.0 61.0 51.0 44.0 52.0 45.0 24.0)
|
||
|
|
(82.0 76.0 72.0 70.0 56.0 49.0 55.0 46.0 23.0)
|
||
|
|
(99.0 91.0 83.0 80.0 63.0 56.0 59.0 46.0 20.0)
|
||
|
|
(124.0 116.0 107.0 100.0 80.0 71.0 72.0 58.0 28.0)
|
||
|
|
(133.0 125.0 113.0 106.0 86.0 75.0 74.0 59.0 25.0)
|
||
|
|
(156.0 146.0 131.0 120.0 97.0 84.0 80.0 65.0 31.0)
|
||
|
|
(178.0 164.0 146.0 135.0 110.0 96.0 92.0 73.0 39.0))))
|
||
|
|
(define m (matrix->list* m0))
|
||
|
|
(define m2 (matrix->list* (matrix-transpose m0)))
|
||
|
|
|
||
|
|
|
||
|
|
(check-true (smawky? m))
|
||
|
|
(check-true (smawky? m2))
|
||
|
|
|
||
|
|
(: simple-entry->value Entry->Value-Type)
|
||
|
|
(define (simple-entry->value e)
|
||
|
|
(fl (cast e Real)))
|
||
|
|
|
||
|
|
;; proc must return a value even for out-of-bounds i and j
|
||
|
|
(: simple-proc Matrix-Proc-Type)
|
||
|
|
(define (simple-proc i j) (cast (fl (with-handlers [(exn:fail? (λ(exn) (* -1 i)))]
|
||
|
|
((inst list-ref Value-Type) ((inst list-ref (Listof Value-Type)) m i) j))) Value-Type))
|
||
|
|
(: simple-proc2 Matrix-Proc-Type)
|
||
|
|
(define (simple-proc2 i j) (cast (fl (with-handlers [(exn:fail? (λ(exn) (* -1 i)))]
|
||
|
|
((inst list-ref Value-Type) ((inst list-ref (Listof Value-Type)) m2 i) j))) Value-Type))
|
||
|
|
(check-equal? (simple-proc 0 2) 57.0) ; 0th row, 2nd col
|
||
|
|
(check-equal? (simple-proc2 2 0) 57.0) ; flipped
|
||
|
|
|
||
|
|
(define row-indices (cast (list->vector (range (length m))) (Vectorof Nonnegative-Integer)))
|
||
|
|
(define col-indices (cast (list->vector (range (length (car m)))) (Vectorof Nonnegative-Integer)))
|
||
|
|
(define result (concave-minima row-indices col-indices simple-proc simple-entry->value))
|
||
|
|
|
||
|
|
|
||
|
|
(check-equal?
|
||
|
|
(for/list : (Listof (List (U Index-Type Value-Type) (U Index-Type Value-Type))) ([j (in-vector col-indices)])
|
||
|
|
(define h (cast (hash-ref result j) (HashTable Symbol (U Index-Type Value-Type))))
|
||
|
|
(list (hash-ref h 'value) (hash-ref h 'row-idx)))
|
||
|
|
'((10.0 3) (20.0 3) (24.0 5) (35.0 5) (35.0 9) (33.0 9) (44.0 9) (43.0 9) (20.0 13))) ; checked against SMAWK.py
|
||
|
|
|
||
|
|
#|
|
||
|
|
(define o (make-ocm simple-proc simple-entry->value))
|
||
|
|
(check-equal?
|
||
|
|
(for/list : (Listof (Pairof Integer Any)) ([j (in-vector col-indices)])
|
||
|
|
(define h (hash-ref result j))
|
||
|
|
(list (hash-ref h 'value) (hash-ref h 'row-idx)))
|
||
|
|
'((10 3) (20 3) (24 5) (35 5) (35 9) (33 9) (44 9) (43 9) (20 13))) ; checked against SMAWK.py
|
||
|
|
(check-equal?
|
||
|
|
(for/list : (Listof (Pairof Integer Any)) ([j (in-vector col-indices)])
|
||
|
|
(list (min-value o j) (min-index o j)))
|
||
|
|
'((0 none) (42 0) (48 1) (51 2) (48 3) (55 4) (59 5) (61 6) (49 7))) ; checked against SMAWK.py
|
||
|
|
|
||
|
|
(define o2 (make-ocm simple-proc2))
|
||
|
|
(define row-indices2 (list->vector (range (length m2))))
|
||
|
|
(define col-indices2 (list->vector (range (length (car m2)))))
|
||
|
|
(define result2 (concave-minima row-indices2 col-indices2 simple-proc2 identity))
|
||
|
|
(check-equal?
|
||
|
|
(for/list : (Listof (Pairof Integer Any)) ([j (in-vector col-indices2)])
|
||
|
|
(define h (hash-ref result2 j))
|
||
|
|
(list (hash-ref h 'value) (hash-ref h 'row-idx)))
|
||
|
|
'((25 0) (21 0) (13 0) (10 0) (20 0) (13 0) (19 0) (35 0) (36 1) (29 8) (29 8) (24 8) (23 8) (20 8) (28 8) (25 8) (31 8) (39 8))) ; checked against SMAWK.py
|
||
|
|
(check-equal?
|
||
|
|
(for/list : (Listof (Pairof Integer Any)) ([j (in-vector col-indices2)])
|
||
|
|
(list (min-value o2 j) (min-index o2 j)))
|
||
|
|
'((0 none) (21 0) (13 0) (10 0) (20 0) (13 0) (19 0) (35 0) (36 1) (29 8) (-9 9) (-10 10) (-11 11) (-12 12) (-13 13) (-14 14) (-15 15) (-16 16))) ; checked against SMAWK.py
|
||
|
|
|#
|
||
|
|
|
||
|
|
|