rustc_utils 0.15.2-nightly-2026-05-01

Utilities for working with the Rust compiler
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81

use either::Either;
use rustc_middle::mir::{Body, Local, Location, Place};

use crate::PlaceExt;

/// Used to represent dependencies of places.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub enum LocationOrArg {
  Location(Location),
  Arg(Local),
}

impl LocationOrArg {
  pub fn from_place<'tcx>(place: Place<'tcx>, body: &Body<'tcx>) -> Option<Self> {
    place
      .is_arg(body)
      .then_some(LocationOrArg::Arg(place.local))
  }

  pub fn to_string(self, body: &Body<'_>) -> String {
    match self {
      LocationOrArg::Arg(local) => format!("{local:?}"),
      LocationOrArg::Location(location) => match body.stmt_at(location) {
        Either::Left(stmt) => format!("{:?}", stmt.kind),
        Either::Right(terminator) => format!("{:?}", terminator.kind),
      },
    }
  }
}

impl From<Location> for LocationOrArg {
  fn from(location: Location) -> Self {
    LocationOrArg::Location(location)
  }
}

impl From<Local> for LocationOrArg {
  fn from(local: Local) -> Self {
    LocationOrArg::Arg(local)
  }
}

#[cfg(feature = "indexical")]
pub mod index {
  use indexical::{
    IndexedDomain, ToIndex, bitset::rustc::ArcIndexSet, define_index_type,
  };

  use super::*;

  define_index_type! {
    pub struct LocationOrArgIndex for LocationOrArg = u32;
  }

  pub type LocationOrArgSet = ArcIndexSet<LocationOrArg>;
  pub type LocationOrArgDomain = IndexedDomain<LocationOrArg>;

  pub struct CustomMarker;

  impl ToIndex<LocationOrArg, CustomMarker> for Location {
    fn to_index(self, domain: &IndexedDomain<LocationOrArg>) -> LocationOrArgIndex {
      LocationOrArg::Location(self).to_index(domain)
    }
  }

  impl ToIndex<LocationOrArg, CustomMarker> for Local {
    fn to_index(self, domain: &IndexedDomain<LocationOrArg>) -> LocationOrArgIndex {
      LocationOrArg::Arg(self).to_index(domain)
    }
  }

  impl rustc_index::Idx for LocationOrArgIndex {
    fn new(idx: usize) -> Self {
      LocationOrArgIndex::new(idx)
    }

    fn index(self) -> usize {
      self.index()
    }
  }
}