Compare commits

..

No commits in common. "9dd6bee0e350a89be3eec67caefc1bb6ff089685" and "8084e34cc02a2a2031e52f0fab84c7bfc2cf931a" have entirely different histories.

2 changed files with 11 additions and 39 deletions

View file

@ -17,7 +17,6 @@ pub enum Expression {
Mul(ExpBox, ExpBox), Mul(ExpBox, ExpBox),
Div(ExpBox, ExpBox), Div(ExpBox, ExpBox),
IntDiv(ExpBox, ExpBox), IntDiv(ExpBox, ExpBox),
Pow(ExpBox, ExpBox),
} }
#[derive(Debug, PartialEq, Clone)] #[derive(Debug, PartialEq, Clone)]
@ -122,6 +121,9 @@ impl Expression {
// Exploding dice are early, they add rolls // Exploding dice are early, they add rolls
if let Some(fragments) = x { if let Some(fragments) = x {
write!(w, "x")?; write!(w, "x")?;
if size < 2 {
bail!("Infinite explosion.")
}
let mut comparers = Vec::with_capacity(fragments.len()); let mut comparers = Vec::with_capacity(fragments.len());
for fragment in fragments { for fragment in fragments {
comparers.push(fragment.comparer(rng, w)?); comparers.push(fragment.comparer(rng, w)?);
@ -263,12 +265,6 @@ impl Expression {
let rhs = rhs_node.evaluate(rng, w, precedence)?; let rhs = rhs_node.evaluate(rng, w, precedence)?;
lhs.div_euclid(rhs) lhs.div_euclid(rhs)
} }
Pow(lhs_node, rhs_node) => {
let lhs = lhs_node.evaluate(rng, w, precedence)?;
write!(w, " ^ ")?;
let rhs = rhs_node.evaluate(rng, w, precedence)?;
lhs.powf(rhs)
}
}; };
if needs_parens { if needs_parens {
write!(w, ")")?; write!(w, ")")?;
@ -298,7 +294,7 @@ impl Expression {
} else { } else {
1 1
}; };
if count > DICE_POOL_LIMIT { if count > 1_000_000 {
bail!("Too many dice.") bail!("Too many dice.")
}; };
let size = size.avg()? as i64; let size = size.avg()? as i64;
@ -330,7 +326,6 @@ impl Expression {
Expression::Mul(lhs, rhs) => lhs.avg()? * rhs.avg()?, Expression::Mul(lhs, rhs) => lhs.avg()? * rhs.avg()?,
Expression::Div(lhs, rhs) => lhs.avg()? / rhs.avg()?, Expression::Div(lhs, rhs) => lhs.avg()? / rhs.avg()?,
Expression::IntDiv(lhs, rhs) => lhs.avg()?.div_euclid(rhs.avg()?), Expression::IntDiv(lhs, rhs) => lhs.avg()?.div_euclid(rhs.avg()?),
Expression::Pow(lhs, rhs) => lhs.avg()?.powf(rhs.avg()?),
}; };
Ok(result) Ok(result)
} }
@ -382,7 +377,6 @@ impl Expression {
match self { match self {
Const(_) => 1, Const(_) => 1,
Dice { .. } => 2, Dice { .. } => 2,
Pow(_, _) => 4,
Neg(_) => 5, Neg(_) => 5,
Mul(_, _) | Div(_, _) | IntDiv(_, _) => 7, Mul(_, _) | Div(_, _) | IntDiv(_, _) => 7,
Add(_, _) | Sub(_, _) => 8, Add(_, _) | Sub(_, _) => 8,
@ -486,7 +480,7 @@ impl<'a> WitnessSet for DiscordMdWitnessSet<'a> {
type Error = DiscordMdWitnessError; type Error = DiscordMdWitnessError;
fn witness_roll(&mut self, dice: DiceRoll) -> Result<Self::Ok, Self::Error> { fn witness_roll(&mut self, dice: DiceRoll) -> Result<Self::Ok, Self::Error> {
if self.parent.dice_written >= 50 { if self.parent.dice_written >= 100 {
if !self.dice_elided { if !self.dice_elided {
write!(self.parent.buffer, "…")?; write!(self.parent.buffer, "…")?;
self.dice_elided = true; self.dice_elided = true;

View file

@ -1,10 +1,5 @@
use crate::dice::{ use crate::dice::{CompareFragment, DiceFormula, Expression::{self, Const}};
CompareFragment, DiceFormula,
Expression::{self, Const},
};
use Assoc::Left; use Assoc::Left;
use nom::character::complete::digit0;
use nom::combinator::recognize;
use nom::{ use nom::{
Parser, Parser,
branch::alt, branch::alt,
@ -94,7 +89,6 @@ fn expr<'c, 'i>(
complete(unary_op(5, tag("-"))), complete(unary_op(5, tag("-"))),
fail(), fail(),
complete(alt(( complete(alt((
binary_op(4, Left, spaced_op("^")),
binary_op(7, Left, spaced_op("*")), binary_op(7, Left, spaced_op("*")),
binary_op(7, Left, spaced_op("//")), binary_op(7, Left, spaced_op("//")),
binary_op(7, Left, spaced_op("/")), binary_op(7, Left, spaced_op("/")),
@ -107,7 +101,6 @@ fn expr<'c, 'i>(
Ok(match op { Ok(match op {
Prefix("-", x) => Neg(Box::new(x)), Prefix("-", x) => Neg(Box::new(x)),
Binary(lhs, "^", rhs) => Pow(Box::new(lhs), Box::new(rhs)),
Binary(lhs, "*", rhs) => Mul(Box::new(lhs), Box::new(rhs)), Binary(lhs, "*", rhs) => Mul(Box::new(lhs), Box::new(rhs)),
Binary(lhs, "//", rhs) => IntDiv(Box::new(lhs), Box::new(rhs)), Binary(lhs, "//", rhs) => IntDiv(Box::new(lhs), Box::new(rhs)),
Binary(lhs, "/", rhs) => Div(Box::new(lhs), Box::new(rhs)), Binary(lhs, "/", rhs) => Div(Box::new(lhs), Box::new(rhs)),
@ -122,10 +115,7 @@ fn expr<'c, 'i>(
} }
fn number(i: &str) -> IResult<&str, Expression> { fn number(i: &str) -> IResult<&str, Expression> {
map_res( map_res(digit1(), |s: &str| s.parse::<f64>())
alt((recognize((digit0, tag("."), digit1())), digit1())),
|s: &str| s.parse::<f64>(),
)
.map(Const) .map(Const)
.parse_complete(i) .parse_complete(i)
} }
@ -164,18 +154,6 @@ mod test {
assert_matches!(expression, Expression::Dice { .. }); assert_matches!(expression, Expression::Dice { .. });
} }
#[test]
pub fn parse_decimal() {
let expression = parse("2.5+2.5").unwrap();
assert_matches!(expression, Expression::Add { .. });
}
#[test]
pub fn parse_decimal2() {
let expression = parse(".4*.8").unwrap();
assert_matches!(expression, Expression::Mul { .. });
}
#[test] #[test]
pub fn parse_bare_dice() { pub fn parse_bare_dice() {
let expression = parse("d8").unwrap(); let expression = parse("d8").unwrap();
@ -184,7 +162,7 @@ mod test {
#[test] #[test]
pub fn parse_math() { pub fn parse_math() {
let expression = parse("8 * 2 + 2 * 5 ^ 4").unwrap(); let expression = parse("8 * 2 + 2 * 5").unwrap();
assert_matches!(expression, Expression::Add { .. }); assert_matches!(expression, Expression::Add { .. });
} }