379 lines
12 KiB
Rust
379 lines
12 KiB
Rust
// Copyright 2016 The Rust Project Developers. See the COPYRIGHT
|
|
// file at the top-level directory of this distribution and at
|
|
// http://rust-lang.org/COPYRIGHT.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
|
// option. This file may not be copied, modified, or distributed
|
|
// except according to those terms.
|
|
|
|
// The visitors in this module collect sizes and counts of the most important
|
|
// pieces of AST and HIR. The resulting numbers are good approximations but not
|
|
// completely accurate (some things might be counted twice, others missed).
|
|
|
|
use rustc::hir;
|
|
use rustc::hir::intravisit as hir_visit;
|
|
use rustc::util::common::to_readable_str;
|
|
use rustc::util::nodemap::{FxHashMap, FxHashSet};
|
|
use syntax::ast::{self, NodeId, AttrId};
|
|
use syntax::visit as ast_visit;
|
|
use syntax_pos::Span;
|
|
|
|
#[derive(Copy, Clone, PartialEq, Eq, Hash)]
|
|
enum Id {
|
|
Node(NodeId),
|
|
Attr(AttrId),
|
|
None,
|
|
}
|
|
|
|
struct NodeData {
|
|
count: usize,
|
|
size: usize,
|
|
}
|
|
|
|
struct StatCollector<'k> {
|
|
krate: Option<&'k hir::Crate>,
|
|
data: FxHashMap<&'static str, NodeData>,
|
|
seen: FxHashSet<Id>,
|
|
}
|
|
|
|
pub fn print_hir_stats(krate: &hir::Crate) {
|
|
let mut collector = StatCollector {
|
|
krate: Some(krate),
|
|
data: FxHashMap(),
|
|
seen: FxHashSet(),
|
|
};
|
|
hir_visit::walk_crate(&mut collector, krate);
|
|
collector.print("HIR STATS");
|
|
}
|
|
|
|
pub fn print_ast_stats<'v>(krate: &'v ast::Crate, title: &str) {
|
|
let mut collector = StatCollector {
|
|
krate: None,
|
|
data: FxHashMap(),
|
|
seen: FxHashSet(),
|
|
};
|
|
ast_visit::walk_crate(&mut collector, krate);
|
|
collector.print(title);
|
|
}
|
|
|
|
impl<'k> StatCollector<'k> {
|
|
|
|
fn record<T>(&mut self, label: &'static str, id: Id, node: &T) {
|
|
if id != Id::None {
|
|
if !self.seen.insert(id) {
|
|
return
|
|
}
|
|
}
|
|
|
|
let entry = self.data.entry(label).or_insert(NodeData {
|
|
count: 0,
|
|
size: 0,
|
|
});
|
|
|
|
entry.count += 1;
|
|
entry.size = ::std::mem::size_of_val(node);
|
|
}
|
|
|
|
fn print(&self, title: &str) {
|
|
let mut stats: Vec<_> = self.data.iter().collect();
|
|
|
|
stats.sort_by_key(|&(_, ref d)| d.count * d.size);
|
|
|
|
let mut total_size = 0;
|
|
|
|
println!("\n{}\n", title);
|
|
|
|
println!("{:<18}{:>18}{:>14}{:>14}",
|
|
"Name", "Accumulated Size", "Count", "Item Size");
|
|
println!("----------------------------------------------------------------");
|
|
|
|
for (label, data) in stats {
|
|
println!("{:<18}{:>18}{:>14}{:>14}",
|
|
label,
|
|
to_readable_str(data.count * data.size),
|
|
to_readable_str(data.count),
|
|
to_readable_str(data.size));
|
|
|
|
total_size += data.count * data.size;
|
|
}
|
|
println!("----------------------------------------------------------------");
|
|
println!("{:<18}{:>18}\n",
|
|
"Total",
|
|
to_readable_str(total_size));
|
|
}
|
|
}
|
|
|
|
impl<'v> hir_visit::Visitor<'v> for StatCollector<'v> {
|
|
fn nested_visit_map<'this>(&'this mut self) -> hir_visit::NestedVisitorMap<'this, 'v> {
|
|
panic!("visit_nested_xxx must be manually implemented in this visitor")
|
|
}
|
|
|
|
fn visit_nested_item(&mut self, id: hir::ItemId) {
|
|
let nested_item = self.krate.unwrap().item(id.id);
|
|
self.visit_item(nested_item)
|
|
}
|
|
|
|
fn visit_nested_trait_item(&mut self, trait_item_id: hir::TraitItemId) {
|
|
let nested_trait_item = self.krate.unwrap().trait_item(trait_item_id);
|
|
self.visit_trait_item(nested_trait_item)
|
|
}
|
|
|
|
fn visit_nested_impl_item(&mut self, impl_item_id: hir::ImplItemId) {
|
|
let nested_impl_item = self.krate.unwrap().impl_item(impl_item_id);
|
|
self.visit_impl_item(nested_impl_item)
|
|
}
|
|
|
|
fn visit_item(&mut self, i: &'v hir::Item) {
|
|
self.record("Item", Id::Node(i.id), i);
|
|
hir_visit::walk_item(self, i)
|
|
}
|
|
|
|
///////////////////////////////////////////////////////////////////////////
|
|
|
|
fn visit_mod(&mut self, m: &'v hir::Mod, _s: Span, n: NodeId) {
|
|
self.record("Mod", Id::None, m);
|
|
hir_visit::walk_mod(self, m, n)
|
|
}
|
|
fn visit_foreign_item(&mut self, i: &'v hir::ForeignItem) {
|
|
self.record("ForeignItem", Id::Node(i.id), i);
|
|
hir_visit::walk_foreign_item(self, i)
|
|
}
|
|
fn visit_local(&mut self, l: &'v hir::Local) {
|
|
self.record("Local", Id::Node(l.id), l);
|
|
hir_visit::walk_local(self, l)
|
|
}
|
|
fn visit_block(&mut self, b: &'v hir::Block) {
|
|
self.record("Block", Id::Node(b.id), b);
|
|
hir_visit::walk_block(self, b)
|
|
}
|
|
fn visit_stmt(&mut self, s: &'v hir::Stmt) {
|
|
self.record("Stmt", Id::Node(s.node.id()), s);
|
|
hir_visit::walk_stmt(self, s)
|
|
}
|
|
fn visit_arm(&mut self, a: &'v hir::Arm) {
|
|
self.record("Arm", Id::None, a);
|
|
hir_visit::walk_arm(self, a)
|
|
}
|
|
fn visit_pat(&mut self, p: &'v hir::Pat) {
|
|
self.record("Pat", Id::Node(p.id), p);
|
|
hir_visit::walk_pat(self, p)
|
|
}
|
|
fn visit_decl(&mut self, d: &'v hir::Decl) {
|
|
self.record("Decl", Id::None, d);
|
|
hir_visit::walk_decl(self, d)
|
|
}
|
|
fn visit_expr(&mut self, ex: &'v hir::Expr) {
|
|
self.record("Expr", Id::Node(ex.id), ex);
|
|
hir_visit::walk_expr(self, ex)
|
|
}
|
|
|
|
fn visit_ty(&mut self, t: &'v hir::Ty) {
|
|
self.record("Ty", Id::Node(t.id), t);
|
|
hir_visit::walk_ty(self, t)
|
|
}
|
|
|
|
fn visit_fn(&mut self,
|
|
fk: hir_visit::FnKind<'v>,
|
|
fd: &'v hir::FnDecl,
|
|
b: hir::BodyId,
|
|
s: Span,
|
|
id: NodeId) {
|
|
self.record("FnDecl", Id::None, fd);
|
|
hir_visit::walk_fn(self, fk, fd, b, s, id)
|
|
}
|
|
|
|
fn visit_where_predicate(&mut self, predicate: &'v hir::WherePredicate) {
|
|
self.record("WherePredicate", Id::None, predicate);
|
|
hir_visit::walk_where_predicate(self, predicate)
|
|
}
|
|
|
|
fn visit_trait_item(&mut self, ti: &'v hir::TraitItem) {
|
|
self.record("TraitItem", Id::Node(ti.id), ti);
|
|
hir_visit::walk_trait_item(self, ti)
|
|
}
|
|
fn visit_impl_item(&mut self, ii: &'v hir::ImplItem) {
|
|
self.record("ImplItem", Id::Node(ii.id), ii);
|
|
hir_visit::walk_impl_item(self, ii)
|
|
}
|
|
|
|
fn visit_ty_param_bound(&mut self, bounds: &'v hir::TyParamBound) {
|
|
self.record("TyParamBound", Id::None, bounds);
|
|
hir_visit::walk_ty_param_bound(self, bounds)
|
|
}
|
|
|
|
fn visit_struct_field(&mut self, s: &'v hir::StructField) {
|
|
self.record("StructField", Id::Node(s.id), s);
|
|
hir_visit::walk_struct_field(self, s)
|
|
}
|
|
|
|
fn visit_variant(&mut self,
|
|
v: &'v hir::Variant,
|
|
g: &'v hir::Generics,
|
|
item_id: NodeId) {
|
|
self.record("Variant", Id::None, v);
|
|
hir_visit::walk_variant(self, v, g, item_id)
|
|
}
|
|
fn visit_lifetime(&mut self, lifetime: &'v hir::Lifetime) {
|
|
self.record("Lifetime", Id::Node(lifetime.id), lifetime);
|
|
hir_visit::walk_lifetime(self, lifetime)
|
|
}
|
|
fn visit_lifetime_def(&mut self, lifetime: &'v hir::LifetimeDef) {
|
|
self.record("LifetimeDef", Id::None, lifetime);
|
|
hir_visit::walk_lifetime_def(self, lifetime)
|
|
}
|
|
fn visit_qpath(&mut self, qpath: &'v hir::QPath, id: NodeId, span: Span) {
|
|
self.record("QPath", Id::None, qpath);
|
|
hir_visit::walk_qpath(self, qpath, id, span)
|
|
}
|
|
fn visit_path(&mut self, path: &'v hir::Path, _id: NodeId) {
|
|
self.record("Path", Id::None, path);
|
|
hir_visit::walk_path(self, path)
|
|
}
|
|
fn visit_path_segment(&mut self,
|
|
path_span: Span,
|
|
path_segment: &'v hir::PathSegment) {
|
|
self.record("PathSegment", Id::None, path_segment);
|
|
hir_visit::walk_path_segment(self, path_span, path_segment)
|
|
}
|
|
fn visit_assoc_type_binding(&mut self, type_binding: &'v hir::TypeBinding) {
|
|
self.record("TypeBinding", Id::Node(type_binding.id), type_binding);
|
|
hir_visit::walk_assoc_type_binding(self, type_binding)
|
|
}
|
|
fn visit_attribute(&mut self, attr: &'v ast::Attribute) {
|
|
self.record("Attribute", Id::Attr(attr.id), attr);
|
|
}
|
|
fn visit_macro_def(&mut self, macro_def: &'v hir::MacroDef) {
|
|
self.record("MacroDef", Id::Node(macro_def.id), macro_def);
|
|
hir_visit::walk_macro_def(self, macro_def)
|
|
}
|
|
}
|
|
|
|
impl<'v> ast_visit::Visitor<'v> for StatCollector<'v> {
|
|
|
|
fn visit_mod(&mut self, m: &'v ast::Mod, _s: Span, _n: NodeId) {
|
|
self.record("Mod", Id::None, m);
|
|
ast_visit::walk_mod(self, m)
|
|
}
|
|
|
|
fn visit_foreign_item(&mut self, i: &'v ast::ForeignItem) {
|
|
self.record("ForeignItem", Id::None, i);
|
|
ast_visit::walk_foreign_item(self, i)
|
|
}
|
|
|
|
fn visit_item(&mut self, i: &'v ast::Item) {
|
|
self.record("Item", Id::None, i);
|
|
ast_visit::walk_item(self, i)
|
|
}
|
|
|
|
fn visit_local(&mut self, l: &'v ast::Local) {
|
|
self.record("Local", Id::None, l);
|
|
ast_visit::walk_local(self, l)
|
|
}
|
|
|
|
fn visit_block(&mut self, b: &'v ast::Block) {
|
|
self.record("Block", Id::None, b);
|
|
ast_visit::walk_block(self, b)
|
|
}
|
|
|
|
fn visit_stmt(&mut self, s: &'v ast::Stmt) {
|
|
self.record("Stmt", Id::None, s);
|
|
ast_visit::walk_stmt(self, s)
|
|
}
|
|
|
|
fn visit_arm(&mut self, a: &'v ast::Arm) {
|
|
self.record("Arm", Id::None, a);
|
|
ast_visit::walk_arm(self, a)
|
|
}
|
|
|
|
fn visit_pat(&mut self, p: &'v ast::Pat) {
|
|
self.record("Pat", Id::None, p);
|
|
ast_visit::walk_pat(self, p)
|
|
}
|
|
|
|
fn visit_expr(&mut self, ex: &'v ast::Expr) {
|
|
self.record("Expr", Id::None, ex);
|
|
ast_visit::walk_expr(self, ex)
|
|
}
|
|
|
|
fn visit_ty(&mut self, t: &'v ast::Ty) {
|
|
self.record("Ty", Id::None, t);
|
|
ast_visit::walk_ty(self, t)
|
|
}
|
|
|
|
fn visit_fn(&mut self,
|
|
fk: ast_visit::FnKind<'v>,
|
|
fd: &'v ast::FnDecl,
|
|
s: Span,
|
|
_: NodeId) {
|
|
self.record("FnDecl", Id::None, fd);
|
|
ast_visit::walk_fn(self, fk, fd, s)
|
|
}
|
|
|
|
fn visit_trait_item(&mut self, ti: &'v ast::TraitItem) {
|
|
self.record("TraitItem", Id::None, ti);
|
|
ast_visit::walk_trait_item(self, ti)
|
|
}
|
|
|
|
fn visit_impl_item(&mut self, ii: &'v ast::ImplItem) {
|
|
self.record("ImplItem", Id::None, ii);
|
|
ast_visit::walk_impl_item(self, ii)
|
|
}
|
|
|
|
fn visit_ty_param_bound(&mut self, bounds: &'v ast::TyParamBound) {
|
|
self.record("TyParamBound", Id::None, bounds);
|
|
ast_visit::walk_ty_param_bound(self, bounds)
|
|
}
|
|
|
|
fn visit_struct_field(&mut self, s: &'v ast::StructField) {
|
|
self.record("StructField", Id::None, s);
|
|
ast_visit::walk_struct_field(self, s)
|
|
}
|
|
|
|
fn visit_variant(&mut self,
|
|
v: &'v ast::Variant,
|
|
g: &'v ast::Generics,
|
|
item_id: NodeId) {
|
|
self.record("Variant", Id::None, v);
|
|
ast_visit::walk_variant(self, v, g, item_id)
|
|
}
|
|
|
|
fn visit_lifetime(&mut self, lifetime: &'v ast::Lifetime) {
|
|
self.record("Lifetime", Id::None, lifetime);
|
|
ast_visit::walk_lifetime(self, lifetime)
|
|
}
|
|
|
|
fn visit_lifetime_def(&mut self, lifetime: &'v ast::LifetimeDef) {
|
|
self.record("LifetimeDef", Id::None, lifetime);
|
|
ast_visit::walk_lifetime_def(self, lifetime)
|
|
}
|
|
|
|
fn visit_mac(&mut self, mac: &'v ast::Mac) {
|
|
self.record("Mac", Id::None, mac);
|
|
}
|
|
|
|
fn visit_path_list_item(&mut self,
|
|
prefix: &'v ast::Path,
|
|
item: &'v ast::PathListItem) {
|
|
self.record("PathListItem", Id::None, item);
|
|
ast_visit::walk_path_list_item(self, prefix, item)
|
|
}
|
|
|
|
fn visit_path_segment(&mut self,
|
|
path_span: Span,
|
|
path_segment: &'v ast::PathSegment) {
|
|
self.record("PathSegment", Id::None, path_segment);
|
|
ast_visit::walk_path_segment(self, path_span, path_segment)
|
|
}
|
|
|
|
fn visit_assoc_type_binding(&mut self, type_binding: &'v ast::TypeBinding) {
|
|
self.record("TypeBinding", Id::None, type_binding);
|
|
ast_visit::walk_assoc_type_binding(self, type_binding)
|
|
}
|
|
|
|
fn visit_attribute(&mut self, attr: &'v ast::Attribute) {
|
|
self.record("Attribute", Id::None, attr);
|
|
}
|
|
}
|