clippy_lints/src/excessive_precision.rs

   1 use crate::utils::span_lint_and_sugg;
   2 use if_chain::if_chain;
   3 use rustc::hir;
   4 use rustc::lint::{LateContext, LateLintPass, LintArray, LintPass};
   5 use rustc::ty;
   6 use rustc::{declare_lint_pass, declare_tool_lint};
   7 use rustc_errors::Applicability;
   8 use std::f32;
   9 use std::f64;
  10 use std::fmt;
  11 use syntax::ast::*;
  12 use syntax_pos::symbol::Symbol;
  13
  14 declare_clippy_lint! {
  15     /// **What it does:** Checks for float literals with a precision greater
  16     /// than that supported by the underlying type
  17     ///
  18     /// **Why is this bad?** Rust will truncate the literal silently.
  19     ///
  20     /// **Known problems:** None.
  21     ///
  22     /// **Example:**
  23     ///
  24     /// ```rust
  25     /// // Bad
  26     /// let v: f32 = 0.123_456_789_9;
  27     /// println!("{}", v); //  0.123_456_789
  28     ///
  29     /// // Good
  30     /// let v: f64 = 0.123_456_789_9;
  31     /// println!("{}", v); //  0.123_456_789_9
  32     /// ```
  33     pub EXCESSIVE_PRECISION,
  34     style,
  35     "excessive precision for float literal"
  36 }
  37
  38 declare_lint_pass!(ExcessivePrecision => [EXCESSIVE_PRECISION]);
  39
  40 impl<'a, 'tcx> LateLintPass<'a, 'tcx> for ExcessivePrecision {
  41     fn check_expr(&mut self, cx: &LateContext<'a, 'tcx>, expr: &'tcx hir::Expr) {
  42         if_chain! {
  43             let ty = cx.tables.expr_ty(expr);
  44             if let ty::Float(fty) = ty.kind;
  45             if let hir::ExprKind::Lit(ref lit) = expr.kind;
  46             if let LitKind::Float(sym, _) = lit.node;
  47             if let Some(sugg) = Self::check(sym, fty);
  48             then {
  49                 span_lint_and_sugg(
  50                     cx,
  51                     EXCESSIVE_PRECISION,
  52                     expr.span,
  53                     "float has excessive precision",
  54                     "consider changing the type or truncating it to",
  55                     sugg,
  56                     Applicability::MachineApplicable,
  57                 );
  58             }
  59         }
  60     }
  61 }
  62
  63 impl ExcessivePrecision {
  64     // None if nothing to lint, Some(suggestion) if lint necessary
  65     #[must_use]
  66     fn check(sym: Symbol, fty: FloatTy) -> Option<String> {
  67         let max = max_digits(fty);
  68         let sym_str = sym.as_str();
  69         if dot_zero_exclusion(&sym_str) {
  70             return None;
  71         }
  72         // Try to bail out if the float is for sure fine.
  73         // If its within the 2 decimal digits of being out of precision we
  74         // check if the parsed representation is the same as the string
  75         // since we'll need the truncated string anyway.
  76         let digits = count_digits(&sym_str);
  77         if digits > max as usize {
  78             let formatter = FloatFormat::new(&sym_str);
  79             let sr = match fty {
  80                 FloatTy::F32 => sym_str.parse::<f32>().map(|f| formatter.format(f)),
  81                 FloatTy::F64 => sym_str.parse::<f64>().map(|f| formatter.format(f)),
  82             };
  83             // We know this will parse since we are in LatePass
  84             let s = sr.unwrap();
  85
  86             if sym_str == s {
  87                 None
  88             } else {
  89                 let di = super::literal_representation::DigitInfo::new(&s, true);
  90                 Some(di.grouping_hint())
  91             }
  92         } else {
  93             None
  94         }
  95     }
  96 }
  97
  98 /// Should we exclude the float because it has a `.0` or `.` suffix
  99 /// Ex `1_000_000_000.0`
 100 /// Ex `1_000_000_000.`
 101 #[must_use]
 102 fn dot_zero_exclusion(s: &str) -> bool {
 103     s.split('.').nth(1).map_or(false, |after_dec| {
 104         let mut decpart = after_dec.chars().take_while(|c| *c != 'e' || *c != 'E');
 105
 106         match decpart.next() {
 107             Some('0') => decpart.count() == 0,
 108             Some(_) => false,
 109             None => true,
 110         }
 111     })
 112 }
 113
 114 #[must_use]
 115 fn max_digits(fty: FloatTy) -> u32 {
 116     match fty {
 117         FloatTy::F32 => f32::DIGITS,
 118         FloatTy::F64 => f64::DIGITS,
 119     }
 120 }
 121
 122 /// Counts the digits excluding leading zeros
 123 #[must_use]
 124 fn count_digits(s: &str) -> usize {
 125     // Note that s does not contain the f32/64 suffix, and underscores have been stripped
 126     s.chars()
 127         .filter(|c| *c != '-' && *c != '.')
 128         .take_while(|c| *c != 'e' && *c != 'E')
 129         .fold(0, |count, c| {
 130             // leading zeros
 131             if c == '0' && count == 0 {
 132                 count
 133             } else {
 134                 count + 1
 135             }
 136         })
 137 }
 138
 139 enum FloatFormat {
 140     LowerExp,
 141     UpperExp,
 142     Normal,
 143 }
 144 impl FloatFormat {
 145     #[must_use]
 146     fn new(s: &str) -> Self {
 147         s.chars()
 148             .find_map(|x| match x {
 149                 'e' => Some(Self::LowerExp),
 150                 'E' => Some(Self::UpperExp),
 151                 _ => None,
 152             })
 153             .unwrap_or(Self::Normal)
 154     }
 155     fn format<T>(&self, f: T) -> String
 156     where
 157         T: fmt::UpperExp + fmt::LowerExp + fmt::Display,
 158     {
 159         match self {
 160             Self::LowerExp => format!("{:e}", f),
 161             Self::UpperExp => format!("{:E}", f),
 162             Self::Normal => format!("{}", f),
 163         }
 164     }
 165 }