Function main

pulley/examples/profiler-html.rs:64–199

()

}

fn main() -> Result<()> {
    let args = ProfilerHtml::parse();
    let profile = std::fs::read(&args.profile)
        .with_context(|| format!("failed to read {:?}", args.profile))?;

    // All known functions and the total of all samples taken.
    let mut functions = BTreeMap::new();
    let mut total = 0;

    let mut found_samples = false;
    for event in decode(&profile) {
        match event? {
            Event::Function(addr, name, body) => {
                let prev = functions.insert(
                    addr,
                    Function {
                        addr,
                        name,
                        body,
                        hits: 0,
                        instructions: BTreeMap::new(),
                    },
                );
                assert!(prev.is_none());
            }
            Event::Samples(samples) => {
                found_samples = true;
                for sample in samples {
                    let addr = sample.0;
                    let (_, function) = functions.range_mut(..=addr).next_back().unwrap();
                    assert!(addr >= function.addr);
                    assert!(addr < function.addr + (function.body.len() as u64));

                    total += 1;
                    function.hits += 1;
                    *function
                        .instructions
                        .entry(u32::try_from(addr - function.addr).unwrap())
                        .or_insert(0) += 1;
                }
            }
        }
    }

    if functions.is_empty() {
        bail!("no functions found in profile");
    }
    if !found_samples {
        bail!("no samples found in profile");
    }

    let mut funcs = functions
        .into_iter()
        .map(|(_, func)| func)
        .collect::<Vec<_>>();
    funcs.sort_by_key(|f| f.hits);

    let mut term = StandardStream::stdout(ColorChoice::Auto);