dimitri/uniplot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Support wide axis labels

Browse source
This commit is contained in:
Dimitri Lozeve 2021-04-30 22:55:19 +02:00
parent dee804454c
commit 4ff1f01e5c
1 changed files with 17 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

26
uniplot/lineplot.ss
View file

@ -1,4 +1,4 @@
(export #t) (export line-plot)
(import :std/format (import :std/format
:std/iter :std/iter
@ -25,33 +25,41 @@
canvas) canvas)
(def (add-border! plot xmin xmax ymin ymax) (def (add-border! plot xmin xmax ymin ymax)
(def ymin-label (format "~5,1F" ymin))
(def ymax-label (format "~5,1F" ymax))
(def label-width (max (string-length ymin-label) (string-length ymax-label)))
(def padding (make-string label-width #\ ))
(set! (plot-str plot) (set! (plot-str plot)
(str (str
(format " ┌─~a─┐\n~5,1F ┤ " (make-string (plot-hsize plot) #\─) ymax) padding " ┌─" (make-string (plot-hsize plot) #\─) "─┐\n"
(string-subst (plot-str plot) "\n" " │\n │ ") (make-string (- label-width (string-length ymax-label)) #\ ) ymax-label " ┤ "
(format " │\n~5,1F ┤~a │\n" ymin (make-string (+ 1 (plot-hsize plot)) #\ )) (string-subst (plot-str plot) "\n" (format " │\n~a │ " padding))
(format " └─┬~a┬─┘\n" (make-string (- (plot-hsize plot) 2) #\─)) " │\n"
(format "\n ~5,1F~a~5,1F\n" xmin (make-string (- (plot-hsize plot) 6) #\ ) xmax)))) (make-string (- label-width (string-length ymax-label)) #\ ) ymax-label " ┤ "
(make-string (+ 1 (plot-hsize plot)) #\ ) "│\n"
(make-string label-width #\ ) " └─┬" (make-string (- (plot-hsize plot) 2) #\─) "┬─┘\n"
"\n" padding (format "~5,1F~a~5,1F\n" xmin (make-string (- (plot-hsize plot) 6) #\ ) xmax)))
(set! (plot-hsize plot) (+ (plot-hsize plot) label-width 5)))
(def (add-legend! plot names colors) (def (add-legend! plot names colors)
(set! (plot-str plot) (set! (plot-str plot)
(str (plot-str plot) (str (plot-str plot)
(cursor-up (+ 4 (plot-vsize plot))) (cursor-up (+ 4 (plot-vsize plot)))
(cursor-forward (+ 12 (plot-hsize plot))) (cursor-forward (+ 1 (plot-hsize plot)))
(apply str (apply str
(for/collect ((name names) (color colors)) (for/collect ((name names) (color colors))
(str (graphics-style [color]) name (graphics-style) (str (graphics-style [color]) name (graphics-style)
(cursor-down 1) (cursor-down 1)
(cursor-back (string-length name))))) (cursor-back (string-length name)))))
(cursor-down (- (+ 4 (plot-vsize plot)) (length names))) (cursor-down (- (+ 4 (plot-vsize plot)) (length names)))
(cursor-back (+ 12 (plot-hsize plot)))))) (cursor-back (+ 1 (plot-hsize plot))))))
(def (add-xlabel! plot xlabel) (def (add-xlabel! plot xlabel)
(set! (plot-str plot) (set! (plot-str plot)
(str (plot-str plot) (make-string (quotient (plot-hsize plot) 2) #\ ) xlabel "\n"))) (str (plot-str plot) (make-string (quotient (plot-hsize plot) 2) #\ ) xlabel "\n")))
(def (line-plot lsts (colors +default-colors+) (def (line-plot lsts (colors +default-colors+)
width: (width 160) height: (height 80) width: (width 160) height: (height 100)
xlabel: (xlabel "") names: (names [])) xlabel: (xlabel "") names: (names []))
(define-values (xmin xmax ymin ymax canvases) (define-values (xmin xmax ymin ymax canvases)
(match lsts (match lsts