Haskell Quiz/Bytecode Compiler/Solution Michael Sloan
From HaskellWiki
(Difference between revisions)
| Line 1: | Line 1: | ||
<haskell>import Char(digitToInt) | <haskell>import Char(digitToInt) | ||
| - | data Tok = C Char | N Int | Op Char | App Tok Tok Tok | G {members :: [Tok]} | + | data Tok = C Char |
| + | | N Int | ||
| + | | Op Char | ||
| + | | App Tok Tok Tok | ||
| + | | G {members :: [Tok]} | ||
--Debugging porpoises | --Debugging porpoises | ||
| Line 13: | Line 17: | ||
compile = bytes . parse | compile = bytes . parse | ||
| - | parse | + | parse = stripG . head . foldr ((.) . prec) id "+-*/%^" . pToks . parseNOp . map C |
parseNOp [] = [] | parseNOp [] = [] | ||
| - | parseNOp ((N num):(C ch):inp) | ch `elem` "0123456789" = parseNOp (N (num * 10 + digitToInt ch) : inp) | + | parseNOp ((N num):(C ch):inp) |
| - | parseNOp | + | | ch `elem` "0123456789" = parseNOp (N (num * 10 + digitToInt ch) : inp) |
| - | parseNOp | + | parseNOp ((C ch):inp) | ch `elem` "0123456789" = parseNOp (N (digitToInt ch) : inp) |
| - | parseNOp | + | parseNOp ((C ch):inp) | ch `elem` "+-*/^%" = parseNOp (Op ch : inp) |
| - | parseNOp (a:inp) = a : | + | parseNOp ((C ch):inp) | ch `elem` " \t\r\n" = parseNOp inp |
| + | parseNOp (a:inp) = a : parseNOp inp | ||
--Groups an array of tokens if it needs to be | --Groups an array of tokens if it needs to be | ||
| Line 28: | Line 33: | ||
--Returns parsed tokens/stream remainder | --Returns parsed tokens/stream remainder | ||
pToks = fst . parseG | pToks = fst . parseG | ||
| - | pRem = snd . parseG | + | pRem = snd . parseG |
parseG [] = ([], []) | parseG [] = ([], []) | ||
| Line 35: | Line 40: | ||
parseG ((C ')') : inp) = ([], inp) | parseG ((C ')') : inp) = ([], inp) | ||
parseG (i:inp) = (i : (pToks inp), pRem inp) | parseG (i:inp) = (i : (pToks inp), pRem inp) | ||
| + | |||
| + | aPrec mo a (Op o) b = App (g $ prec mo [a]) (Op o) (g $ prec mo [b]) | ||
--Traverses groups, applies operators to their immediate arguments | --Traverses groups, applies operators to their immediate arguments | ||
prec mo [] = [] | prec mo [] = [] | ||
prec mo ((G xs):inp) = (g $ prec mo xs) : (prec mo inp) | prec mo ((G xs):inp) = (g $ prec mo xs) : (prec mo inp) | ||
| - | prec mo (a:(Op o):b:inp) | o == mo = prec mo (( | + | prec mo (a:(Op o):b:inp) | o == mo = prec mo ((aPrec mo a (Op o) b) : inp) |
| - | prec mo ((App a o b):inp) = ( | + | prec mo ((App a o b):inp) = (aPrec mo a o b) : (prec mo inp) |
prec mo (x:inp) = x : (prec mo inp) | prec mo (x:inp) = x : (prec mo inp) | ||
--Removes any vestigial grouping | --Removes any vestigial grouping | ||
stripG (App a o b) = App (stripG a) o (stripG b) | stripG (App a o b) = App (stripG a) o (stripG b) | ||
| - | stripG (G (x:xs)) | null xs = error $ "error: " ++ show (x:xs) ++ " <- improper infix nesting here" | + | stripG (G (x:xs)) | null xs = stripG x |
| - | + | stripG (G (x:xs)) = error $ "error: " ++ show (x:xs) ++ " <- improper infix nesting here" | |
stripG x = x | stripG x = x | ||
--Converts to a byte format | --Converts to a byte format | ||
bytes (C x) = error ("Invalid character: " ++ [x]) | bytes (C x) = error ("Invalid character: " ++ [x]) | ||
| - | bytes (N n) = if abs n < 2^15 then 1 : | + | bytes (N n) = if abs n < 2^15 then 1 : toBytes 2 n else 2 : toBytes 4 n |
| - | bytes (Op ch) = case ch of '+' -> [10]; ' | + | bytes (Op ch) = case ch of '+' -> [10]; '-' -> [11]; '*' -> [12]; '^' -> [13]; '/' -> [14]; '%' -> [15] |
bytes (App a o b) = (bytes a) ++ (bytes b) ++ (bytes o) | bytes (App a o b) = (bytes a) ++ (bytes b) ++ (bytes o) | ||
bytes x = error ("Error, invalid: " ++ show x) | bytes x = error ("Error, invalid: " ++ show x) | ||
| - | + | toBytes n x = map (\i -> (x `div` 2^(i*8)) `mod` 256 ) [(n-1),(n-2)..0]</haskell> | |
Revision as of 08:32, 6 November 2006
import Char(digitToInt) data Tok = C Char | N Int | Op Char | App Tok Tok Tok | G {members :: [Tok]} --Debugging porpoises instance Show Tok where show (C ch) = ch : "" show (N i) = show i show (Op ch) = ch : "" show (App a o b) = '(' : (show a) ++ (show o) ++ (show b) ++ ")" show (G xs) = '[' : (concatMap (\x -> show x) xs) ++ "]" compile = bytes . parse parse = stripG . head . foldr ((.) . prec) id "+-*/%^" . pToks . parseNOp . map C parseNOp [] = [] parseNOp ((N num):(C ch):inp) | ch `elem` "0123456789" = parseNOp (N (num * 10 + digitToInt ch) : inp) parseNOp ((C ch):inp) | ch `elem` "0123456789" = parseNOp (N (digitToInt ch) : inp) parseNOp ((C ch):inp) | ch `elem` "+-*/^%" = parseNOp (Op ch : inp) parseNOp ((C ch):inp) | ch `elem` " \t\r\n" = parseNOp inp parseNOp (a:inp) = a : parseNOp inp --Groups an array of tokens if it needs to be g (x:xs) | null xs = x g xs = G xs --Returns parsed tokens/stream remainder pToks = fst . parseG pRem = snd . parseG parseG [] = ([], []) parseG [(G ts)] = (ts, []) parseG ((C '(') : inp) = (g (pToks inp) : (pToks $ pRem inp), []) parseG ((C ')') : inp) = ([], inp) parseG (i:inp) = (i : (pToks inp), pRem inp) aPrec mo a (Op o) b = App (g $ prec mo [a]) (Op o) (g $ prec mo [b]) --Traverses groups, applies operators to their immediate arguments prec mo [] = [] prec mo ((G xs):inp) = (g $ prec mo xs) : (prec mo inp) prec mo (a:(Op o):b:inp) | o == mo = prec mo ((aPrec mo a (Op o) b) : inp) prec mo ((App a o b):inp) = (aPrec mo a o b) : (prec mo inp) prec mo (x:inp) = x : (prec mo inp) --Removes any vestigial grouping stripG (App a o b) = App (stripG a) o (stripG b) stripG (G (x:xs)) | null xs = stripG x stripG (G (x:xs)) = error $ "error: " ++ show (x:xs) ++ " <- improper infix nesting here" stripG x = x --Converts to a byte format bytes (C x) = error ("Invalid character: " ++ [x]) bytes (N n) = if abs n < 2^15 then 1 : toBytes 2 n else 2 : toBytes 4 n bytes (Op ch) = case ch of '+' -> [10]; '-' -> [11]; '*' -> [12]; '^' -> [13]; '/' -> [14]; '%' -> [15] bytes (App a o b) = (bytes a) ++ (bytes b) ++ (bytes o) bytes x = error ("Error, invalid: " ++ show x) toBytes n x = map (\i -> (x `div` 2^(i*8)) `mod` 256 ) [(n-1),(n-2)..0]
