Skip to content

Commit dd3772e

Browse files
A few more code cleanups
1 parent 3ed189e commit dd3772e

File tree

1 file changed

+18
-34
lines changed

1 file changed

+18
-34
lines changed

compiler/rustc_driver/src/pretty.rs

+18-34
Original file line numberDiff line numberDiff line change
@@ -391,16 +391,13 @@ pub fn print_after_parsing(
391391
) {
392392
let (src, src_name) = get_source(input, sess);
393393

394-
let mut out = String::new();
395-
396-
if let Source(s) = ppm {
394+
let out = if let Source(s) = ppm {
397395
// Silently ignores an identified node.
398-
let out = &mut out;
399396
call_with_pp_support(&s, sess, None, move |annotation| {
400397
debug!("pretty printing source code {:?}", s);
401398
let sess = annotation.sess();
402399
let parse = &sess.parse_sess;
403-
*out = pprust::print_crate(
400+
pprust::print_crate(
404401
sess.source_map(),
405402
krate,
406403
src_name,
@@ -411,7 +408,7 @@ pub fn print_after_parsing(
411408
)
412409
})
413410
} else {
414-
unreachable!();
411+
unreachable!()
415412
};
416413

417414
write_or_print(&out, ofile);
@@ -431,17 +428,14 @@ pub fn print_after_hir_lowering<'tcx>(
431428

432429
let (src, src_name) = get_source(input, tcx.sess);
433430

434-
let mut out = String::new();
435-
436-
match ppm {
431+
let out = match ppm {
437432
Source(s) => {
438433
// Silently ignores an identified node.
439-
let out = &mut out;
440434
call_with_pp_support(&s, tcx.sess, Some(tcx), move |annotation| {
441435
debug!("pretty printing source code {:?}", s);
442436
let sess = annotation.sess();
443437
let parse = &sess.parse_sess;
444-
*out = pprust::print_crate(
438+
pprust::print_crate(
445439
sess.source_map(),
446440
krate,
447441
src_name,
@@ -453,26 +447,20 @@ pub fn print_after_hir_lowering<'tcx>(
453447
})
454448
}
455449

456-
Hir(s) => {
457-
let out = &mut out;
458-
call_with_pp_support_hir(&s, tcx, move |annotation, krate| {
459-
debug!("pretty printing HIR {:?}", s);
460-
let sess = annotation.sess();
461-
let sm = sess.source_map();
462-
*out = pprust_hir::print_crate(sm, krate, src_name, src, annotation.pp_ann())
463-
})
464-
}
450+
Hir(s) => call_with_pp_support_hir(&s, tcx, move |annotation, krate| {
451+
debug!("pretty printing HIR {:?}", s);
452+
let sess = annotation.sess();
453+
let sm = sess.source_map();
454+
pprust_hir::print_crate(sm, krate, src_name, src, annotation.pp_ann())
455+
}),
465456

466-
HirTree => {
467-
let out = &mut out;
468-
call_with_pp_support_hir(&PpHirMode::Normal, tcx, move |_annotation, krate| {
469-
debug!("pretty printing HIR tree");
470-
*out = format!("{:#?}", krate);
471-
});
472-
}
457+
HirTree => call_with_pp_support_hir(&PpHirMode::Normal, tcx, move |_annotation, krate| {
458+
debug!("pretty printing HIR tree");
459+
format!("{:#?}", krate)
460+
}),
473461

474462
_ => unreachable!(),
475-
}
463+
};
476464

477465
write_or_print(&out, ofile);
478466
}
@@ -491,14 +479,10 @@ fn print_with_analysis(
491479
tcx.analysis(LOCAL_CRATE)?;
492480

493481
match ppm {
494-
Mir | MirCFG => match ppm {
495-
Mir => write_mir_pretty(tcx, None, &mut out),
496-
MirCFG => write_mir_graphviz(tcx, None, &mut out),
497-
_ => unreachable!(),
498-
},
482+
Mir => write_mir_pretty(tcx, None, &mut out).unwrap(),
483+
MirCFG => write_mir_graphviz(tcx, None, &mut out).unwrap(),
499484
_ => unreachable!(),
500485
}
501-
.unwrap();
502486

503487
let out = std::str::from_utf8(&out).unwrap();
504488
write_or_print(out, ofile);

0 commit comments

Comments
 (0)