rust-clippy/src/approx_const.rs

use rustc::lint::*;
use rustc::metadata::cstore::crate_metadata;
use rustc_front::hir::*;
use syntax::codemap::Span;
use std::borrow::Borrow;
use std::f64::consts as f64;
use utils::span_lint;

declare_lint! {
    pub APPROX_CONSTANT,
    Warn,
    "the approximate of a known float constant (in `std::f64::consts` or `std::f32::consts`) \
     is found; suggests to use the constant"
}

const KNOWN_CONSTS : &'static [(f64, &'static str)] = &[(f64::E, "E"), (f64::FRAC_1_PI, "FRAC_1_PI"),
    (f64::FRAC_1_SQRT_2, "FRAC_1_SQRT_2"), (f64::FRAC_2_PI, "FRAC_2_PI"),
    (f64::FRAC_2_SQRT_PI, "FRAC_2_SQRT_PI"), (f64::FRAC_PI_2, "FRAC_PI_2"), (f64::FRAC_PI_3, "FRAC_PI_3"),
    (f64::FRAC_PI_4, "FRAC_PI_4"), (f64::FRAC_PI_6, "FRAC_PI_6"), (f64::FRAC_PI_8, "FRAC_PI_8"),
    (f64::LN_10, "LN_10"), (f64::LN_2, "LN_2"), (f64::LOG10_E, "LOG10_E"), (f64::LOG2_E, "LOG2_E"),
    (f64::PI, "PI"), (f64::SQRT_2, "SQRT_2")];

const EPSILON_DIVISOR : f64 = 8192f64; //TODO: test to find a good value

#[derive(Copy,Clone)]
pub struct ApproxConstant;

impl LintPass for ApproxConstant {
    fn get_lints(&self) -> LintArray {
        lint_array!(APPROX_CONSTANT)
    }

    fn check_expr(&mut self, cx: &Context, e: &Expr) {
        if let &ExprLit(ref lit) = &e.node {
            if let Some(res) = cx.tcx.def_map.borrow().get(&e.id) {
                let krate = res.def_id().krate;
                let cdata = &cx.sess().cstore.get_crate_data(krate);
                let crate_data : &crate_metadata = cdata.borrow();
                let name = &crate_data.name;
                if name == "f32" || name == "f64" { return; }
            }
            check_lit(cx, lit, e.span);
        }
    }
}

fn check_lit(cx: &Context, lit: &Lit, span: Span) {
    match lit.node {
        LitFloat(ref str, TyF32) => check_known_consts(cx, span, str, "f32"),
        LitFloat(ref str, TyF64) => check_known_consts(cx, span, str, "f64"),
        LitFloatUnsuffixed(ref str) =>
            check_known_consts(cx, span, str, "f{32, 64}"),
        _ => ()
    }
}

fn check_known_consts(cx: &Context, span: Span, str: &str, module: &str) {
    if let Ok(value) = str.parse::<f64>() {
        for &(constant, name) in KNOWN_CONSTS {
            if within_epsilon(constant, value) {
                span_lint(cx, APPROX_CONSTANT, span, &format!(
                    "approximate value of `{}::{}` found. \
                    Consider using it directly", module, &name));
            }
        }
    }
}

fn within_epsilon(target: f64, value: f64) -> bool {
    f64::abs(value - target) < f64::abs(if target > value {
                                            target
                                        } else { value }) / EPSILON_DIVISOR
}