mirror of
https://github.com/ProgramSnail/Lama.git
synced 2025-12-26 16:48:46 +00:00
Assertions in runtime
This commit is contained in:
Dmitry Boulytchev
parent
27c091129a
commit
6322ee6bed
3 changed files with 171 additions and 173 deletions
|
|
@ -1,141 +0,0 @@
|
|||
-- MNode (key, list of values, balance factor, left subtree, right subtree)
|
||||
-- balance factor = height (left subtree) - height (right subtree)
|
||||
fun insert (m, k, v) {
|
||||
fun rot (left, node) {
|
||||
return
|
||||
if left
|
||||
then case node of
|
||||
MNode (k, v, _, l, MNode (rk, rv, _, ll, rr)) ->
|
||||
MNode (rk, rv, 0, MNode (k, v, 0, l, ll), rr)
|
||||
esac
|
||||
else case node of
|
||||
MNode (k, v, _, MNode (lk, lv, _, ll, rr), r) ->
|
||||
MNode (lk, lv, 0, ll, MNode (k, v, 0, rr, r))
|
||||
esac
|
||||
fi
|
||||
}
|
||||
|
||||
fun factor (x) {
|
||||
return x [2]
|
||||
}
|
||||
|
||||
fun inner (m, k, v) {
|
||||
return
|
||||
case m of
|
||||
{} -> [true, MNode (k, {v}, 0, {}, {})]
|
||||
| MNode (kk, vv, bf, l, r) ->
|
||||
local c = compare (k, kk);
|
||||
if c == 0
|
||||
then [false, MNode (kk, v : vv, bf, l, r)]
|
||||
else if c < 0
|
||||
then
|
||||
case inner (l, k, v) of
|
||||
[true, ll] -> if bf < 0
|
||||
then [false, MNode (kk, vv, bf + 1, ll, r)]
|
||||
elif bf == 1
|
||||
then if ll.factor > 0
|
||||
then [false, rot (false, MNode (kk, vv, bf, ll, r))]
|
||||
else [false, rot (false, MNode (kk, vv, bf, rot (true, ll), r))]
|
||||
fi
|
||||
else [true, MNode (kk, vv, bf + 1, ll, r)]
|
||||
fi
|
||||
| [false, ll] -> [false, MNode (kk, vv, bf, ll, r)]
|
||||
esac
|
||||
else
|
||||
case inner (r, k, v) of
|
||||
[true, rr] -> if bf > 0
|
||||
then [false, MNode (kk, vv, bf - 1, l, rr)]
|
||||
elif bf == -1
|
||||
then if rr.factor < 0
|
||||
then [false, rot (true, MNode (kk, vv, bf, l, rr))]
|
||||
else [false, rot (true, MNode (kk, vv, bf, l, rot (false, rr)))]
|
||||
fi
|
||||
else [true, MNode (kk, vv, bf - 1, l, rr)]
|
||||
fi
|
||||
| [false, rr] -> [false, MNode (kk, vv, bf, l, rr)]
|
||||
esac
|
||||
fi
|
||||
fi
|
||||
esac
|
||||
}
|
||||
|
||||
return m.inner (k, v).snd
|
||||
}
|
||||
|
||||
fun find (m, k) {
|
||||
return
|
||||
case m of
|
||||
{} -> None
|
||||
| MNode (kk, vv, _, l, r) ->
|
||||
local c = compare (k, kk);
|
||||
if c == 0
|
||||
then case vv of v : _ -> Some (v) | _ -> None esac
|
||||
else find (if c < 0 then l else r fi, k)
|
||||
fi
|
||||
esac
|
||||
}
|
||||
|
||||
fun remove (m, k) {
|
||||
return
|
||||
case m of
|
||||
{} -> m
|
||||
| MNode (kk, vv, bf, l, r) ->
|
||||
local c = compare (k, kk);
|
||||
if c == 0
|
||||
then case vv of {} -> m | _ : vt -> MNode (kk, vt, l, r) esac
|
||||
else if c < 0
|
||||
then MNode (kk, vv, bf, remove (l, k), r)
|
||||
else MNode (kk, vv, bf, l, remove (r, k))
|
||||
fi
|
||||
fi
|
||||
esac
|
||||
}
|
||||
|
||||
fun validate (t) {
|
||||
fun inner (t, verify) {
|
||||
return
|
||||
case t of
|
||||
{} -> 0
|
||||
| MNode (k, _, bf, l, r) ->
|
||||
if verify (k)
|
||||
then
|
||||
local lh = validate (l, fun (x) {return x < k}),
|
||||
rh = validate (r, fun (x) {return x > k});
|
||||
|
||||
if bf == lh - rh
|
||||
then 1 + if lh > rh then lh else rh fi
|
||||
else failure ("Balance violation on key %s\n", k.string)
|
||||
fi
|
||||
else failure ("Order violation on key %s\n", k.string)
|
||||
fi
|
||||
esac
|
||||
}
|
||||
|
||||
inner (t, fun (x) {return true})
|
||||
}
|
||||
|
||||
local tree = {}, i;
|
||||
|
||||
for i := 100, i >= 1, i := i-1 do
|
||||
validate (tree);
|
||||
tree := insert (tree, i, i);
|
||||
printf ("Inserting: %s\n", i.string);
|
||||
printf ("Result : %s\n", tree.string)
|
||||
od;
|
||||
|
||||
tree := {};
|
||||
|
||||
for i := 1, i <= 100, i := i+2 do
|
||||
validate (tree);
|
||||
tree := insert (tree, i, i);
|
||||
printf ("Inserting: %s\n", i.string);
|
||||
printf ("Result : %s\n", tree.string)
|
||||
od;
|
||||
|
||||
for i := 2, i <= 100, i := i+2 do
|
||||
validate (tree);
|
||||
tree := insert (tree, i, i);
|
||||
printf ("Inserting: %s\n", i.string);
|
||||
printf ("Result : %s\n", tree.string)
|
||||
od
|
||||
|
||||
|
|
@ -1,7 +1,6 @@
|
|||
-- (C) Dmitry Boulytchev, St. Petersburg State University, JetBrains Research, 2020
|
||||
-- Matcher: simple string matching library.
|
||||
|
||||
|
||||
-- Create a regular expression representation.
|
||||
-- Arguments:
|
||||
-- r --- a string representation for regular expression (as per GNU regexp)
|
||||
|
|
@ -30,9 +29,19 @@ fun matcherCreate (buf, pos, line, col) {
|
|||
return buf.length - pos
|
||||
}
|
||||
|
||||
-- Moves the position pointer on given number of characters within one line (i.e.
|
||||
-- Moves the position pointer on given number of characters.
|
||||
fun shift (n) {
|
||||
return matcherCreate (buf, pos + n, line, col + n)
|
||||
local i, l = line, c = col;
|
||||
|
||||
for i := pos, i < n, i := i+1 do
|
||||
case buf [i] of
|
||||
'\n' -> l := l + 1; c := 1
|
||||
| '\t' -> c := c + 8
|
||||
| _ -> c := c + 1
|
||||
esac
|
||||
od;
|
||||
|
||||
return matcherCreate (buf, pos + n, l, c)
|
||||
}
|
||||
|
||||
fun matchString (s) {
|
||||
|
|
@ -53,7 +62,7 @@ fun matcherCreate (buf, pos, line, col) {
|
|||
else Fail (sprintf ("%s expected at %d:%d", r[1], line, col))
|
||||
fi
|
||||
}
|
||||
|
||||
|
||||
fun eof () {
|
||||
return rest () == 0
|
||||
}
|
||||
|
|
@ -70,7 +79,7 @@ fun show (m) {
|
|||
return m [0] ()
|
||||
}
|
||||
|
||||
fun eof (m) {
|
||||
fun endOf (m) {
|
||||
return m [1] ()
|
||||
}
|
||||
|
||||
|
|
@ -78,11 +87,13 @@ fun matchString (m, s) {
|
|||
return m [2] (s)
|
||||
}
|
||||
|
||||
-- Matches against a regexp
|
||||
fun matchRegexp (m, r) {
|
||||
return m [3] (r)
|
||||
}
|
||||
|
||||
fun matcherInit (buf) {
|
||||
-- Creates a fresh matcher from a string buffer
|
||||
public fun matcherInit (buf) {
|
||||
return matcherCreate (buf, 0, 1, 1)
|
||||
}
|
||||
|
||||
|
|
@ -90,10 +101,76 @@ fun matcherInit (buf) {
|
|||
|
||||
--}
|
||||
|
||||
local m = matcherInit ("abc");
|
||||
local m = matcherInit (" -- asdasdakm ,m.,msd .,m.,asd\n \n\n abc");
|
||||
|
||||
local
|
||||
lident = createRegexp ("[a-z][a-zA-Z_]*", "lowercase identifier"),
|
||||
uident = createRegexp ("[A-Z][a-zA-Z_]*", "uppercase identifier"),
|
||||
ws = createRegexp ("\\([ \t\n]\\|--[^\n]*\n\\)*", "whitespace"),
|
||||
str = createRegexp ("""\([^""]\|""""\)*""", "string literal"),
|
||||
decimal = createRegexp ("[0-9]+", "decimal literal"),
|
||||
chr = createRegexp ("'[^']'", "character literal");
|
||||
|
||||
fun token (s) {
|
||||
return fun (m) {return m.matchString (s)}
|
||||
}
|
||||
|
||||
fun lid (m) {
|
||||
return m.matchRegexp (lident)
|
||||
}
|
||||
|
||||
fun uid (m) {
|
||||
return m.matchRegexp (uident)
|
||||
}
|
||||
|
||||
fun const (m) {
|
||||
return m.matchRegexp (decimal)
|
||||
}
|
||||
|
||||
infixl "@" before "*" (p, f) {
|
||||
return fun (m) {
|
||||
return
|
||||
case p (m) of
|
||||
Succ (m, x) -> Succ (m, f (x))
|
||||
| err -> err
|
||||
esac
|
||||
}
|
||||
}
|
||||
|
||||
infixr "|>" after "!!" (l, r) {
|
||||
return fun (m) {
|
||||
return
|
||||
case l (m) of
|
||||
Succ (m, s) -> r (s) (m)
|
||||
| err -> err
|
||||
esac
|
||||
}
|
||||
}
|
||||
|
||||
infixr "||" after "|>" (l, r) {
|
||||
return fun (m) {
|
||||
return
|
||||
case l (m) of
|
||||
s@Succ (_, _) -> s
|
||||
| err -> r (m)
|
||||
esac
|
||||
}
|
||||
}
|
||||
|
||||
local expr = lid @ fun (s) {return Lid (s)} ||
|
||||
const @ fun (s) {return Dec (s)},
|
||||
assn = lid |> fun (id) {return token (":=") |> fun (s) {return expr @ fun (e) {return Assn (id, e)}}};
|
||||
|
||||
printf ("%s\n", assn (matcherInit ("x:=3")).string)
|
||||
|
||||
--local ident = createRegexp ("[a-z][a-zA-Z_]*", "identifier");
|
||||
--local ws = createRegexp ("[ \n\t]+", "whitespace");
|
||||
|
||||
--local ws = createRegexp ("\\([ \t\n]\\|--[^\n]*\n\\)*", "whitespace");
|
||||
|
||||
--local str = createRegexp ("""\([^""]\|""""\)*""", "string literal");
|
||||
--local lineComment = createRegexp ("--[^\n]*\n", "line comment");
|
||||
|
||||
--printf ("ws: %s\n", case m.matchRegexp (ws) of Succ (m, s) -> "(" ++ m.show ++ ", " ++ s ++ ")" | Fail (err) -> err.string esac);
|
||||
|
||||
local ident = createRegexp ("[a-z][a-zA-Z_]*", "identifier");
|
||||
local ws = createRegexp ("[ \n\t]+", "whitespace");
|
||||
local str = createRegexp ("""\([^""]\|""""\)*""", "string literal");
|
||||
|
||||
printf ("ident: %s\n", case m.matchRegexp (ident) of Succ (m, s) -> "(" ++ m.show ++ ", " ++ s ++ ")" | Fail (err) -> err.string esac)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue