Skip to content

Commit

Permalink
Added string formatting library
Browse files Browse the repository at this point in the history
  • Loading branch information
Izaakwltn committed Nov 20, 2024
1 parent 19965eb commit 99a164b
Show file tree
Hide file tree
Showing 6 changed files with 232 additions and 7 deletions.
1 change: 1 addition & 0 deletions coalton.asd
Original file line number Diff line number Diff line change
Expand Up @@ -76,6 +76,7 @@
(:file "vector")
(:file "char")
(:file "string")
(:file "format")
(:file "slice")
(:file "hashtable")
(:file "queue")
Expand Down
3 changes: 2 additions & 1 deletion examples/quil-coalton/src/package.lisp
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,8 @@
(:use #:coalton
#:coalton-prelude)
(:local-nicknames
(#:string #:coalton-library/string))
(#:string #:coalton-library/string)
(#:fmt #:coalton-library/format))
(:import-from
#:coalton-library/functions
#:asum)
Expand Down
7 changes: 3 additions & 4 deletions examples/quil-coalton/src/value-parsers.lisp
Original file line number Diff line number Diff line change
Expand Up @@ -10,8 +10,7 @@
(Parser
(fn (str)
(match (next-char str)
((Some (Tuple read-char _)) (Err (ParseError (lisp String (read-char)
(cl:format cl:nil "Unexpected character '~A' expected EOF" read-char)))))
((Some (Tuple read-char _)) (Err (ParseError (fmt:format "Unexpected character " (into read-char) " expected EOF"))))
((None) (Ok (Tuple Unit str)))))))

(declare take (Parser coalton:Char))
Expand All @@ -32,7 +31,7 @@
(let ((read-char (fst t_)))
(if (== c read-char)
(Ok t_)
(Err (ParseError (lisp String (read-char c) (cl:format cl:nil "Unexpected character '~A' expected '~A'" read-char c)))))))
(Err (ParseError (fmt:format "Unexpected character " (into read-char) " expected " (into c)))))))
((None) (Err parse-error-eof))))))

(declare not-char (coalton:Char -> (Parser coalton:Char)))
Expand All @@ -43,7 +42,7 @@
((Some t_)
(let ((read-char (fst t_)))
(if (== c read-char)
(Err (ParseError (lisp String (read-char c) (cl:format cl:nil "Unexpected character '~A' expected not '~A'" read-char c))))
(Err (ParseError (fmt:format "Unexpected character " (into read-char) " expected not " (into c))))
(Ok t_))))
((None) (Err parse-error-eof))))))

Expand Down
223 changes: 223 additions & 0 deletions library/format.lisp
Original file line number Diff line number Diff line change
@@ -0,0 +1,223 @@
(defpackage #:coalton-library/format
(:use
#:coalton
#:coalton-library/classes)
(:local-nicknames
(#:str #:coalton-library/string)
(#:list #:coalton-library/list)
(#:iter #:coalton-library/iterator)
(#:math #:coalton-library/math))
(:export
#:format
#:\n
#:as-string
#:format-with-delimiter
#:as-string-with-delimiter
#:as-quoted
#:Radix
#:radix-directive
#:radix
#:radix-padded
#:bin
#:bin-padded
#:oct
#:oct-padded
#:dec
#:dec-padded
#:hex
#:hex-padded
#:eng
#:rom
#:fixed-point
#:fixed-point-digits
#:exponential
#:exponential-digits
#:currency))

(in-package #:coalton-library/format)

(cl:defmacro format (cl:&rest forms)
"Format takes strings as arguments and combines them into one string."
(cl:let* ((forms (cl:append '(make-list) forms)))
`(mconcat ,forms)))
;;;
;;; Special characters
;;;

(coalton-toplevel

(define \n
"A new line."
"
"))

;;;
;;; Formatting objects and collections of objects
;;;

(coalton-toplevel

(declare as-string ((Into :a String) => :a -> String))
(define (as-string x)
"Returns the object as a string."
(into x))

(declare format-with-delimiter ((Functor :c)
(iter:IntoIterator (:c :a) :a) =>
(:a -> String) -> String -> (:c :a) -> String))
(define (format-with-delimiter f delimiter collection)
"Formats `collection` according to a string function `f` separated by a string `delimiter`."
;; This is a little bit gross, but necessary for checking iterator length without losing the iter.
(let lst = (iter:collect! (iter:into-iter collection)))
(let delim-length = (list:length lst))
(iter:fold! str:concat "" (iter:interleave!
(map f (iter:into-iter lst))
(iter:repeat-for delimiter
delim-length))))

(declare as-string-with-delimiter ((Functor :c)
(iter:IntoIterator (:c :a) :a)
(Into :a String) =>
String -> (:c :a) -> String))
(define (as-string-with-delimiter delimiter collection)
"Formats `collection` as a string with elements separated by a string `delimiter`."
(format-with-delimiter as-string delimiter collection))

(declare as-quoted ((Into :a String) => :a -> String))
(define (as-quoted x)
"Formats the object as a string quoted within a string."
(let str = (as String x))
(lisp String (str)
(cl:format cl:nil "~S" str))))

;;;
;;; Formatting Integer types
;;;

(coalton-toplevel

(define-type FormatFlag
"Flag for string formatting."
Binary "Binary (base 2)"
Octal "Octal (base 8)"
Decimal "Decimal (base 10"
Hexadecimal "Hexadecimal (base 16)"
English "English number word"
Roman "Roman numeral")

(declare flag-directive (FormatFlag -> String))
(define (flag-directive flag)
"Returns the format directive for the given flag."
(match flag
((Binary) "B")
((Octal) "O")
((Decimal) "D")
((Hexadecimal) "X")
((English) "R")
((Roman) "@R")))

(declare flag ((math:Integral :a) => FormatFlag -> :a -> String))
(define (flag flag x)
"Prints an integral type according to the given flag."
(let control-string = (format "~" (flag-directive flag)))
(lisp String (control-string x)
(cl:format cl:nil control-string x)))

(declare flag-padded ((math:Integral :a) => FormatFlag -> UFix -> :a -> String))
(define (flag-padded flag width x)
"Prints an integral type according to the given flag, preserving leading zeroes."
(let control-string = (format "~" (into width) ",'0" (flag-directive flag)))
(lisp String (control-string x)
(cl:format cl:nil control-string x)))

;;;
;;; Toplevel flag formatting functions
;;;

(declare bin ((math:Integral :a) => :a -> String))
(define (bin x)
"Prints an integral type in radix 2 (binary)."
(flag Binary x))

(declare bin-padded ((math:Integral :a) => UFix -> :a -> String))
(define (bin-padded width x)
"Prints an integral type in radix 2 (binary), with leading zeroes."
(flag-padded Binary width x))

(declare oct ((math:Integral :a) => :a -> String))
(define (oct x)
"Prints an integral type in radix 8 (octal)."
(flag Octal x))

(declare oct-padded ((math:Integral :a) => UFix -> :a -> String))
(define (oct-padded width x)
"Prints an integral type in radix 8 (octal), with leading zeroes."
(flag-padded Octal width x))

(declare dec ((math:Integral :a) => :a -> String))
(define (dec x)
"Prints an integral type in radix 10 (decimal)."
(flag Decimal x))

(declare dec-padded ((math:Integral :a) => UFix -> :a -> String))
(define (dec-padded width x)
"Prints an integral type in radix 10 (decimal), with leading zeroes."
(flag-padded Decimal width x))

(declare hex ((math:Integral :a) => :a -> String))
(define (hex x)
"Prints an integral type in radix 16 (hexadecimal)."
(flag Hexadecimal x))

(declare hex-padded ((math:Integral :a) => UFix -> :a -> String))
(define (hex-padded width x)
"Prints an integral type in radix 16 (hexadecimal), with leading zeroes."
(flag-padded Hexadecimal width x))

(declare eng ((math:Integral :a) => :a -> String))
(define (eng x)
"Prints an integral type as an english word."
(flag English x))

(declare rom ((math:Integral :a) => :a -> String))
(define (rom x)
"Prints an integral type as a roman numeral."
(flag Roman x)))

;;;
;;; Formatting floating point numbers
;;;

(coalton-toplevel

(declare fixed-point ((math:Real :a) => :a -> String))
(define (fixed-point x)
"Prints a floating point number in fixed-point notation."
(lisp String (x)
(cl:format cl:nil "~F" x)))

(declare fixed-point-digits ((math:Real :a) => Ufix -> :a -> String))
(define (fixed-point-digits digits x)
"Prints a floating point number in fixed-point notation with the desired number of digits after the point."
(let control-string = (format "~0," (into digits) ",F"))
(lisp String (control-string x)
(cl:format cl:nil control-string x)))

(declare exponential ((math:Real :a) => :a -> String))
(define (exponential x)
"Prints a floating point number in exponential notation."
(lisp String (x)
(cl:format cl:nil "~E" x)))

(declare exponential-digits ((math:Real :a) => Ufix -> :a -> String))
(define (exponential-digits digits x)
"Prints a floating point number in exponential notation with the desired number of digits after the point."
(let control-string = (format "~0," (into digits) ",E"))
(lisp String (control-string x)
(cl:format cl:nil control-string x)))

(declare currency ((math:Real :a) => :a -> String))
(define (currency x)
"Prints a floating point number as currency."
(lisp String (x)
(cl:format cl:nil "~$" x))))
3 changes: 2 additions & 1 deletion tests/package.lisp
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,8 @@
(#:red-black/map #:coalton-library/ord-map)
(#:result #:coalton-library/result)
(#:seq #:coalton-library/seq)
(#:file #:coalton-library/file)))
(#:file #:coalton-library/file)
(#:fmt #:coalton-library/format)))

(in-package #:coalton-native-tests)

Expand Down
2 changes: 1 addition & 1 deletion tests/seq-tests.lisp
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@
(declare legible-seq (UFix -> seq:Seq String))
(define (legible-seq n)
(iter:collect!
(map (fn (i) (lisp String (i) (cl:format cl:nil "~r" i)))
(map fmt:eng
(iter:up-to n)))))

(define-test seq-push-and-pop-implementation ()
Expand Down

0 comments on commit 99a164b

Please sign in to comment.