]>
Commit | Line | Data |
---|---|---|
ba9703b0 | 1 | use rustc_middle::mir; |
f2b60f7d | 2 | use rustc_middle::mir::NonDivergingIntrinsic; |
3157f602 | 3 | |
2c00a5a8 | 4 | use super::FunctionCx; |
3157f602 | 5 | use super::LocalRef; |
dfeec247 | 6 | use crate::traits::BuilderMethods; |
9fa01778 | 7 | use crate::traits::*; |
92a42be0 | 8 | |
dc9dc135 | 9 | impl<'a, 'tcx, Bx: BuilderMethods<'a, 'tcx>> FunctionCx<'a, 'tcx, Bx> { |
064997fb | 10 | #[instrument(level = "debug", skip(self, bx))] |
dfeec247 | 11 | pub fn codegen_statement(&mut self, mut bx: Bx, statement: &mir::Statement<'tcx>) -> Bx { |
a1dfa0c6 | 12 | self.set_debug_loc(&mut bx, statement.source_info); |
92a42be0 | 13 | match statement.kind { |
dfeec247 | 14 | mir::StatementKind::Assign(box (ref place, ref rvalue)) => { |
e74abb32 XL |
15 | if let Some(index) = place.as_local() { |
16 | match self.locals[index] { | |
dfeec247 | 17 | LocalRef::Place(cg_dest) => self.codegen_rvalue(bx, cg_dest, rvalue), |
b7449926 XL |
18 | LocalRef::UnsizedPlace(cg_indirect_dest) => { |
19 | self.codegen_rvalue_unsized(bx, cg_indirect_dest, rvalue) | |
20 | } | |
3157f602 | 21 | LocalRef::Operand(None) => { |
e1599b0c | 22 | let (mut bx, operand) = self.codegen_rvalue_operand(bx, rvalue); |
e74abb32 XL |
23 | self.locals[index] = LocalRef::Operand(Some(operand)); |
24 | self.debug_introduce_local(&mut bx, index); | |
2c00a5a8 | 25 | bx |
3157f602 | 26 | } |
ff7c6d11 XL |
27 | LocalRef::Operand(Some(op)) => { |
28 | if !op.layout.is_zst() { | |
dfeec247 XL |
29 | span_bug!( |
30 | statement.source_info.span, | |
31 | "operand {:?} already assigned", | |
32 | rvalue | |
33 | ); | |
92a42be0 | 34 | } |
ff7c6d11 XL |
35 | |
36 | // If the type is zero-sized, it's already been set here, | |
94b46f34 XL |
37 | // but we still need to make sure we codegen the operand |
38 | self.codegen_rvalue_operand(bx, rvalue).0 | |
92a42be0 SL |
39 | } |
40 | } | |
3157f602 | 41 | } else { |
74b04a01 | 42 | let cg_dest = self.codegen_place(&mut bx, place.as_ref()); |
94b46f34 | 43 | self.codegen_rvalue(bx, cg_dest, rvalue) |
92a42be0 SL |
44 | } |
45 | } | |
dfeec247 | 46 | mir::StatementKind::SetDiscriminant { box ref place, variant_index } => { |
74b04a01 | 47 | self.codegen_place(&mut bx, place.as_ref()) |
a1dfa0c6 | 48 | .codegen_set_discr(&mut bx, variant_index); |
2c00a5a8 | 49 | bx |
5bcae85e | 50 | } |
04454e1e FG |
51 | mir::StatementKind::Deinit(..) => { |
52 | // For now, don't codegen this to anything. In the future it may be worth | |
53 | // experimenting with what kind of information we can emit to LLVM without hurting | |
54 | // perf here | |
55 | bx | |
56 | } | |
ea8adc8c | 57 | mir::StatementKind::StorageLive(local) => { |
94b46f34 | 58 | if let LocalRef::Place(cg_place) = self.locals[local] { |
a1dfa0c6 | 59 | cg_place.storage_live(&mut bx); |
b7449926 | 60 | } else if let LocalRef::UnsizedPlace(cg_indirect_place) = self.locals[local] { |
a1dfa0c6 | 61 | cg_indirect_place.storage_live(&mut bx); |
ff7c6d11 | 62 | } |
2c00a5a8 | 63 | bx |
5bcae85e | 64 | } |
ea8adc8c | 65 | mir::StatementKind::StorageDead(local) => { |
94b46f34 | 66 | if let LocalRef::Place(cg_place) = self.locals[local] { |
a1dfa0c6 | 67 | cg_place.storage_dead(&mut bx); |
b7449926 | 68 | } else if let LocalRef::UnsizedPlace(cg_indirect_place) = self.locals[local] { |
a1dfa0c6 | 69 | cg_indirect_place.storage_dead(&mut bx); |
ff7c6d11 | 70 | } |
2c00a5a8 | 71 | bx |
5bcae85e | 72 | } |
3dfed10e | 73 | mir::StatementKind::Coverage(box ref coverage) => { |
6a06907d XL |
74 | self.codegen_coverage(&mut bx, coverage.clone(), statement.source_info.scope); |
75 | bx | |
76 | } | |
f2b60f7d FG |
77 | mir::StatementKind::Intrinsic(box NonDivergingIntrinsic::Assume(ref op)) => { |
78 | let op_val = self.codegen_operand(&mut bx, op); | |
79 | bx.assume(op_val.immediate()); | |
80 | bx | |
81 | } | |
82 | mir::StatementKind::Intrinsic(box NonDivergingIntrinsic::CopyNonOverlapping( | |
83 | mir::CopyNonOverlapping { ref count, ref src, ref dst }, | |
84 | )) => { | |
6a06907d XL |
85 | let dst_val = self.codegen_operand(&mut bx, dst); |
86 | let src_val = self.codegen_operand(&mut bx, src); | |
87 | let count = self.codegen_operand(&mut bx, count).immediate(); | |
88 | let pointee_layout = dst_val | |
89 | .layout | |
3c0e092e | 90 | .pointee_info_at(&bx, rustc_target::abi::Size::ZERO) |
6a06907d XL |
91 | .expect("Expected pointer"); |
92 | let bytes = bx.mul(count, bx.const_usize(pointee_layout.size.bytes())); | |
93 | ||
94 | let align = pointee_layout.align; | |
95 | let dst = dst_val.immediate(); | |
96 | let src = src_val.immediate(); | |
97 | bx.memcpy(dst, align, src, align, bytes, crate::MemFlags::empty()); | |
3dfed10e XL |
98 | bx |
99 | } | |
dfeec247 XL |
100 | mir::StatementKind::FakeRead(..) |
101 | | mir::StatementKind::Retag { .. } | |
102 | | mir::StatementKind::AscribeUserType(..) | |
103 | | mir::StatementKind::Nop => bx, | |
5bcae85e SL |
104 | } |
105 | } | |
92a42be0 | 106 | } |