Implement pattern ranges for all numeric types.

This commit is contained in:
Josh Matthews 2011-09-21 03:00:32 -04:00 committed by Marijn Haverbeke
parent e6a84f252a
commit ce0f054f9d
14 changed files with 451 additions and 25 deletions

View file

@ -84,6 +84,162 @@ fn local_rhs_span(l: @ast::local, def: span) -> span {
alt l.node.init { some(i) { ret i.expr.span; } _ { ret def; } }
}
fn lit_is_numeric(l: @ast::lit) -> bool {
alt l.node {
ast::lit_int(_) | ast::lit_char(_) | ast::lit_uint(_) |
ast::lit_mach_int(_, _) | ast::lit_float(_) | ast::lit_mach_float(_,_) {
true
}
_ { false }
}
}
fn lit_type_eq(l: @ast::lit, m: @ast::lit) -> bool {
alt l.node {
ast::lit_str(_) {
alt m.node { ast::lit_str(_) { true } _ { false } }
}
ast::lit_char(_) {
alt m.node { ast::lit_char(_) { true } _ { false } }
}
ast::lit_int(_) {
alt m.node { ast::lit_int(_) { true } _ { false } }
}
ast::lit_uint(_) {
alt m.node { ast::lit_uint(_) { true } _ { false } }
}
ast::lit_mach_int(_, _) {
alt m.node { ast::lit_mach_int(_, _) { true } _ { false } }
}
ast::lit_float(_) {
alt m.node { ast::lit_float(_) { true } _ { false } }
}
ast::lit_mach_float(_, _) {
alt m.node { ast::lit_mach_float(_, _) { true } _ { false } }
}
ast::lit_nil. {
alt m.node { ast::lit_nil. { true } _ { false } }
}
ast::lit_bool(_) {
alt m.node { ast::lit_bool(_) { true } _ { false } }
}
}
}
fn lit_in_range(l: @ast::lit, m1: @ast::lit, m2: @ast::lit) -> bool {
alt lits_to_range(m1, m2) {
irange(i1, i2) {
alt l.node {
ast::lit_int(i3) | ast::lit_mach_int(_, i3) {
i3 >= *min(i1, i2) && i3 <= *max(i1, i2)
}
_ { fail }
}
}
urange(u1, u2) {
alt l.node {
ast::lit_uint(u3) {
u3 >= *min(u1, u2) && u3 <= *max(u1, u2)
}
_ { fail }
}
}
crange(c1, c2) {
alt l.node {
ast::lit_char(c3) {
(c3 as uint) >= *min(c1 as uint, c2 as uint) &&
(c3 as uint) <= *max(c1 as uint, c2 as uint)
}
_ { fail }
}
}
frange(f1, f2) {
alt l.node {
ast::lit_float(f3) | ast::lit_mach_float(_, f3) {
str_to_float(f3) >= *min(f1, f2) &&
str_to_float(f3) <= *max(f1, f2)
}
_ { fail }
}
}
}
}
fn min<T>(x: T, y: T) -> @T {
ret @(if x > y { y } else { x });
}
fn max<T>(x: T, y: T) -> @T {
ret @(if x > y { x } else { y });
}
fn ranges_overlap<T>(a1: T, a2: T, b1: T, b2: T) -> bool {
let min1 = *min(a1, a2);
let max1 = *max(a1, a2);
let min2 = *min(b1, b2);
let max2 = *max(b1, b2);
ret (min1 >= min2 && max1 <= max2) || (min1 <= min2 && max1 >= min2) ||
(min1 >= min2 && min1 <= max2) || (max1 >= min2 && max1 <= max2);
}
fn lit_ranges_overlap(a1: @ast::lit, a2: @ast::lit,
b1: @ast::lit, b2: @ast::lit) -> bool {
alt lits_to_range(a1, a2) {
irange(i1, i2) {
alt lits_to_range(b1, b2) {
irange(i3, i4) { ranges_overlap(i1, i2, i3, i4) }
_ { fail }
}
}
urange(u1, u2) {
alt lits_to_range(b1, b2) {
urange(u3, u4) { ranges_overlap(u1, u2, u3, u4) }
_ { fail }
}
}
crange(c1, c2) {
alt lits_to_range(b1, b2) {
crange(c3, c4) { ranges_overlap(c1, c2, c3, c4) }
_ { fail }
}
}
frange(f1, f2) {
alt lits_to_range(b1, b2) {
frange(f3, f4) { ranges_overlap(f1, f2, f3, f4) }
_ { fail }
}
}
}
}
tag range {
irange(int, int);
urange(uint, uint);
crange(char, char);
frange(float, float);
}
fn lits_to_range(l: @ast::lit, r: @ast::lit) -> range {
alt l.node {
ast::lit_int(i1) | ast::lit_mach_int(_, i1) {
alt r.node { ast::lit_int(i2) { irange(i1, i2) } _ { fail } }
}
ast::lit_uint(u1) {
alt r.node { ast::lit_uint(u2) { urange(u1, u2) } _ { fail } }
}
ast::lit_char(c1) {
alt r.node { ast::lit_char(c2) { crange(c1, c2) } _ { fail } }
}
ast::lit_float(f1) | ast::lit_mach_float(_, f1) {
alt r.node { ast::lit_float(f2) | ast::lit_mach_float(_, f2) {
frange(str_to_float(f1), str_to_float(f2))
}
_ { fail } }
}
_ { fail }
}
}
fn lit_eq(l: @ast::lit, m: @ast::lit) -> bool {
alt l.node {
ast::lit_str(s) {
@ -156,6 +312,22 @@ fn float_to_str(num: float, digits: uint) -> str {
ret accum;
}
fn str_to_float(num: str) -> float {
let digits = str::split(num, '.' as u8);
let total = int::from_str(digits[0]) as float;
fn dec_val(c: char) -> int { ret (c as int) - ('0' as int); }
let right = digits[1];
let len = str::char_len(digits[1]);
let i = 1u;
while (i < len) {
total += dec_val(str::pop_char(right)) as float /
(int::pow(10, i) as float);
i += 1u;
}
ret total;
}
//
// Local Variables: