1 // Copyright 2015 The Rust Project Developers. See the COPYRIGHT
2 // file at the top-level directory of this distribution and at
3 // http://rust-lang.org/COPYRIGHT.
5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
8 // option. This file may not be copied, modified, or distributed
9 // except according to those terms.
13 // After candidates have been simplified, the only match pairs that
14 // remain are those that require some sort of test. The functions here
15 // identify what tests are needed, perform the tests, and then filter
16 // the candidates based on the result.
19 use build::matches::{Candidate, MatchPair, Test, TestKind};
21 use rustc_data_structures::fx::FxHashMap;
22 use rustc_data_structures::bitvec::BitVector;
23 use rustc::middle::const_val::{ConstVal, ConstInt};
24 use rustc::ty::{self, Ty};
25 use rustc::ty::util::IntTypeExt;
27 use rustc::hir::RangeEnd;
29 use std::cmp::Ordering;
31 impl<'a, 'gcx, 'tcx> Builder<'a, 'gcx, 'tcx> {
32 /// Identifies what test is needed to decide if `match_pair` is applicable.
34 /// It is a bug to call this with a simplifyable pattern.
35 pub fn test<'pat>(&mut self, match_pair: &MatchPair<'pat, 'tcx>) -> Test<'tcx> {
36 match *match_pair.pattern.kind {
37 PatternKind::Variant { ref adt_def, substs: _, variant_index: _, subpatterns: _ } => {
39 span: match_pair.pattern.span,
40 kind: TestKind::Switch {
41 adt_def: adt_def.clone(),
42 variants: BitVector::new(self.hir.num_variants(adt_def)),
47 PatternKind::Constant { .. }
48 if is_switch_ty(match_pair.pattern.ty) => {
49 // for integers, we use a SwitchInt match, which allows
50 // us to handle more cases
52 span: match_pair.pattern.span,
53 kind: TestKind::SwitchInt {
54 switch_ty: match_pair.pattern.ty,
56 // these maps are empty to start; cases are
57 // added below in add_cases_to_switch
64 PatternKind::Constant { ref value } => {
66 span: match_pair.pattern.span,
69 ty: match_pair.pattern.ty.clone()
74 PatternKind::Range { ref lo, ref hi, ref end } => {
76 span: match_pair.pattern.span,
77 kind: TestKind::Range {
78 lo: Literal::Value { value: lo.clone() },
79 hi: Literal::Value { value: hi.clone() },
80 ty: match_pair.pattern.ty.clone(),
86 PatternKind::Slice { ref prefix, ref slice, ref suffix }
87 if !match_pair.slice_len_checked => {
88 let len = prefix.len() + suffix.len();
89 let op = if slice.is_some() {
95 span: match_pair.pattern.span,
96 kind: TestKind::Len { len: len as u64, op: op },
100 PatternKind::Array { .. } |
101 PatternKind::Slice { .. } |
103 PatternKind::Binding { .. } |
104 PatternKind::Leaf { .. } |
105 PatternKind::Deref { .. } => {
106 self.error_simplifyable(match_pair)
111 pub fn add_cases_to_switch<'pat>(&mut self,
112 test_lvalue: &Lvalue<'tcx>,
113 candidate: &Candidate<'pat, 'tcx>,
115 options: &mut Vec<ConstVal>,
116 indices: &mut FxHashMap<ConstVal, usize>)
119 let match_pair = match candidate.match_pairs.iter().find(|mp| mp.lvalue == *test_lvalue) {
120 Some(match_pair) => match_pair,
121 _ => { return false; }
124 match *match_pair.pattern.kind {
125 PatternKind::Constant { ref value } => {
126 // if the lvalues match, the type should match
127 assert_eq!(match_pair.pattern.ty, switch_ty);
129 indices.entry(value.clone())
131 options.push(value.clone());
136 PatternKind::Variant { .. } => {
137 panic!("you should have called add_variants_to_switch instead!");
139 PatternKind::Range { .. } |
140 PatternKind::Slice { .. } |
141 PatternKind::Array { .. } |
143 PatternKind::Binding { .. } |
144 PatternKind::Leaf { .. } |
145 PatternKind::Deref { .. } => {
146 // don't know how to add these patterns to a switch
152 pub fn add_variants_to_switch<'pat>(&mut self,
153 test_lvalue: &Lvalue<'tcx>,
154 candidate: &Candidate<'pat, 'tcx>,
155 variants: &mut BitVector)
158 let match_pair = match candidate.match_pairs.iter().find(|mp| mp.lvalue == *test_lvalue) {
159 Some(match_pair) => match_pair,
160 _ => { return false; }
163 match *match_pair.pattern.kind {
164 PatternKind::Variant { adt_def: _ , variant_index, .. } => {
165 // We have a pattern testing for variant `variant_index`
166 // set the corresponding index to true
167 variants.insert(variant_index);
171 // don't know how to add these patterns to a switch
177 /// Generates the code to perform a test.
178 pub fn perform_test(&mut self,
180 lvalue: &Lvalue<'tcx>,
183 let source_info = self.source_info(test.span);
185 TestKind::Switch { adt_def, ref variants } => {
186 // Variants is a BitVec of indexes into adt_def.variants.
187 let num_enum_variants = self.hir.num_variants(adt_def);
188 let used_variants = variants.count();
189 let mut otherwise_block = None;
190 let mut target_blocks = Vec::with_capacity(num_enum_variants);
191 let mut targets = Vec::with_capacity(used_variants + 1);
192 let mut values = Vec::with_capacity(used_variants);
193 let tcx = self.hir.tcx();
194 for (idx, variant) in adt_def.variants.iter().enumerate() {
195 target_blocks.place_back() <- if variants.contains(idx) {
196 let discr = ConstInt::new_inttype(variant.disr_val, adt_def.discr_ty,
197 tcx.sess.target.uint_type,
198 tcx.sess.target.int_type).unwrap();
200 *(targets.place_back() <- self.cfg.start_new_block())
202 if otherwise_block.is_none() {
203 otherwise_block = Some(self.cfg.start_new_block());
205 otherwise_block.unwrap()
208 if let Some(otherwise_block) = otherwise_block {
209 targets.push(otherwise_block);
213 debug!("num_enum_variants: {}, tested variants: {:?}, variants: {:?}",
214 num_enum_variants, values, variants);
215 let discr_ty = adt_def.discr_ty.to_ty(tcx);
216 let discr = self.temp(discr_ty);
217 self.cfg.push_assign(block, source_info, &discr,
218 Rvalue::Discriminant(lvalue.clone()));
219 assert_eq!(values.len() + 1, targets.len());
220 self.cfg.terminate(block, source_info, TerminatorKind::SwitchInt {
221 discr: Operand::Consume(discr),
223 values: From::from(values),
229 TestKind::SwitchInt { switch_ty, ref options, indices: _ } => {
230 let (values, targets, ret) = if switch_ty.sty == ty::TyBool {
231 assert!(options.len() > 0 && options.len() <= 2);
232 let (true_bb, false_bb) = (self.cfg.start_new_block(),
233 self.cfg.start_new_block());
234 let ret = match &options[0] {
235 &ConstVal::Bool(true) => vec![true_bb, false_bb],
236 &ConstVal::Bool(false) => vec![false_bb, true_bb],
237 v => span_bug!(test.span, "expected boolean value but got {:?}", v)
239 (BOOL_SWITCH_TRUE.clone(), vec![true_bb, false_bb], ret)
241 // The switch may be inexhaustive so we
242 // add a catch all block
243 let otherwise = self.cfg.start_new_block();
244 let targets: Vec<_> =
246 .map(|_| self.cfg.start_new_block())
247 .chain(Some(otherwise))
249 let values: Vec<_> = options.iter().map(|v|
250 v.to_const_int().expect("switching on integral")
252 (From::from(values), targets.clone(), targets)
255 self.cfg.terminate(block, source_info, TerminatorKind::SwitchInt {
256 discr: Operand::Consume(lvalue.clone()),
257 switch_ty: switch_ty,
259 targets: targets.clone(),
264 TestKind::Eq { ref value, mut ty } => {
265 let mut val = Operand::Consume(lvalue.clone());
267 // If we're using b"..." as a pattern, we need to insert an
268 // unsizing coercion, as the byte string has the type &[u8; N].
269 let expect = if let ConstVal::ByteStr(ref bytes) = *value {
270 let tcx = self.hir.tcx();
272 // Unsize the lvalue to &[u8], too, if necessary.
273 if let ty::TyRef(region, mt) = ty.sty {
274 if let ty::TyArray(_, _) = mt.ty.sty {
275 ty = tcx.mk_imm_ref(region, tcx.mk_slice(tcx.types.u8));
276 let val_slice = self.temp(ty);
277 self.cfg.push_assign(block, source_info, &val_slice,
278 Rvalue::Cast(CastKind::Unsize, val, ty));
279 val = Operand::Consume(val_slice);
283 assert!(ty.is_slice());
285 let array_ty = tcx.mk_array(tcx.types.u8, bytes.len());
286 let array_ref = tcx.mk_imm_ref(tcx.mk_region(ty::ReStatic), array_ty);
287 let array = self.literal_operand(test.span, array_ref, Literal::Value {
291 let slice = self.temp(ty);
292 self.cfg.push_assign(block, source_info, &slice,
293 Rvalue::Cast(CastKind::Unsize, array, ty));
294 Operand::Consume(slice)
296 self.literal_operand(test.span, ty, Literal::Value {
301 // Use PartialEq::eq for &str and &[u8] slices, instead of BinOp::Eq.
302 let fail = self.cfg.start_new_block();
303 if let ty::TyRef(_, mt) = ty.sty {
304 assert!(ty.is_slice());
305 let eq_def_id = self.hir.tcx().lang_items.eq_trait().unwrap();
307 let (mty, method) = self.hir.trait_method(eq_def_id, "eq", ty, &[ty]);
309 let bool_ty = self.hir.bool_ty();
310 let eq_result = self.temp(bool_ty);
311 let eq_block = self.cfg.start_new_block();
312 let cleanup = self.diverge_cleanup();
313 self.cfg.terminate(block, source_info, TerminatorKind::Call {
314 func: Operand::Constant(Constant {
319 args: vec![val, expect],
320 destination: Some((eq_result.clone(), eq_block)),
325 let block = self.cfg.start_new_block();
326 self.cfg.terminate(eq_block, source_info, TerminatorKind::SwitchInt {
327 discr: Operand::Consume(eq_result),
328 switch_ty: self.hir.bool_ty(),
329 values: BOOL_SWITCH_TRUE.clone(),
330 targets: vec![block, fail],
334 let block = self.compare(block, fail, test.span, BinOp::Eq, expect, val);
339 TestKind::Range { ref lo, ref hi, ty, ref end } => {
340 // Test `val` by computing `lo <= val && val <= hi`, using primitive comparisons.
341 let lo = self.literal_operand(test.span, ty.clone(), lo.clone());
342 let hi = self.literal_operand(test.span, ty.clone(), hi.clone());
343 let val = Operand::Consume(lvalue.clone());
345 let fail = self.cfg.start_new_block();
346 let block = self.compare(block, fail, test.span, BinOp::Le, lo, val.clone());
347 let block = match *end {
348 RangeEnd::Included => self.compare(block, fail, test.span, BinOp::Le, val, hi),
349 RangeEnd::Excluded => self.compare(block, fail, test.span, BinOp::Lt, val, hi),
355 TestKind::Len { len, op } => {
356 let (usize_ty, bool_ty) = (self.hir.usize_ty(), self.hir.bool_ty());
357 let (actual, result) = (self.temp(usize_ty), self.temp(bool_ty));
359 // actual = len(lvalue)
360 self.cfg.push_assign(block, source_info,
361 &actual, Rvalue::Len(lvalue.clone()));
364 let expected = self.push_usize(block, source_info, len);
366 // result = actual == expected OR result = actual < expected
367 self.cfg.push_assign(block, source_info, &result,
369 Operand::Consume(actual),
370 Operand::Consume(expected)));
372 // branch based on result
373 let (false_bb, true_bb) = (self.cfg.start_new_block(),
374 self.cfg.start_new_block());
375 self.cfg.terminate(block, source_info, TerminatorKind::SwitchInt {
376 discr: Operand::Consume(result),
377 switch_ty: self.hir.bool_ty(),
378 values: BOOL_SWITCH_TRUE.clone(),
379 targets: vec![true_bb, false_bb],
381 vec![true_bb, false_bb]
386 fn compare(&mut self,
388 fail_block: BasicBlock,
392 right: Operand<'tcx>) -> BasicBlock {
393 let bool_ty = self.hir.bool_ty();
394 let result = self.temp(bool_ty);
396 // result = op(left, right)
397 let source_info = self.source_info(span);
398 self.cfg.push_assign(block, source_info, &result,
399 Rvalue::BinaryOp(op, left, right));
401 // branch based on result
402 let target_block = self.cfg.start_new_block();
403 self.cfg.terminate(block, source_info, TerminatorKind::SwitchInt {
404 discr: Operand::Consume(result),
405 switch_ty: self.hir.bool_ty(),
406 values: BOOL_SWITCH_TRUE.clone(),
407 targets: vec![target_block, fail_block]
412 /// Given that we are performing `test` against `test_lvalue`,
413 /// this job sorts out what the status of `candidate` will be
414 /// after the test. The `resulting_candidates` vector stores, for
415 /// each possible outcome of `test`, a vector of the candidates
416 /// that will result. This fn should add a (possibly modified)
417 /// clone of candidate into `resulting_candidates` wherever
420 /// So, for example, if this candidate is `x @ Some(P0)` and the
421 /// test is a variant test, then we would add `(x as Option).0 @
422 /// P0` to the `resulting_candidates` entry corresponding to the
425 /// However, in some cases, the test may just not be relevant to
426 /// candidate. For example, suppose we are testing whether `foo.x == 22`,
427 /// but in one match arm we have `Foo { x: _, ... }`... in that case,
428 /// the test for what value `x` has has no particular relevance
429 /// to this candidate. In such cases, this function just returns false
430 /// without doing anything. This is used by the overall `match_candidates`
431 /// algorithm to structure the match as a whole. See `match_candidates` for
434 /// FIXME(#29623). In some cases, we have some tricky choices to
435 /// make. for example, if we are testing that `x == 22`, but the
436 /// candidate is `x @ 13..55`, what should we do? In the event
437 /// that the test is true, we know that the candidate applies, but
438 /// in the event of false, we don't know that it *doesn't*
439 /// apply. For now, we return false, indicate that the test does
440 /// not apply to this candidate, but it might be we can get
441 /// tighter match code if we do something a bit different.
442 pub fn sort_candidate<'pat>(&mut self,
443 test_lvalue: &Lvalue<'tcx>,
445 candidate: &Candidate<'pat, 'tcx>,
446 resulting_candidates: &mut [Vec<Candidate<'pat, 'tcx>>])
448 // Find the match_pair for this lvalue (if any). At present,
449 // afaik, there can be at most one. (In the future, if we
450 // adopted a more general `@` operator, there might be more
451 // than one, but it'd be very unusual to have two sides that
452 // both require tests; you'd expect one side to be simplified
454 let tested_match_pair = candidate.match_pairs.iter()
456 .filter(|&(_, mp)| mp.lvalue == *test_lvalue)
458 let (match_pair_index, match_pair) = match tested_match_pair {
461 // We are not testing this lvalue. Therefore, this
462 // candidate applies to ALL outcomes.
467 match (&test.kind, &*match_pair.pattern.kind) {
468 // If we are performing a variant switch, then this
469 // informs variant patterns, but nothing else.
470 (&TestKind::Switch { adt_def: tested_adt_def, .. },
471 &PatternKind::Variant { adt_def, variant_index, ref subpatterns, .. }) => {
472 assert_eq!(adt_def, tested_adt_def);
474 self.candidate_after_variant_switch(match_pair_index,
479 resulting_candidates[variant_index].push(new_candidate);
482 (&TestKind::Switch { .. }, _) => false,
484 // If we are performing a switch over integers, then this informs integer
485 // equality, but nothing else.
487 // FIXME(#29623) we could use PatternKind::Range to rule
488 // things out here, in some cases.
489 (&TestKind::SwitchInt { switch_ty: _, options: _, ref indices },
490 &PatternKind::Constant { ref value })
491 if is_switch_ty(match_pair.pattern.ty) => {
492 let index = indices[value];
493 let new_candidate = self.candidate_without_match_pair(match_pair_index,
495 resulting_candidates[index].push(new_candidate);
498 (&TestKind::SwitchInt { .. }, _) => false,
501 (&TestKind::Len { len: test_len, op: BinOp::Eq },
502 &PatternKind::Slice { ref prefix, ref slice, ref suffix }) => {
503 let pat_len = (prefix.len() + suffix.len()) as u64;
504 match (test_len.cmp(&pat_len), slice) {
505 (Ordering::Equal, &None) => {
506 // on true, min_len = len = $actual_length,
507 // on false, len != $actual_length
508 resulting_candidates[0].push(
509 self.candidate_after_slice_test(match_pair_index,
517 (Ordering::Less, _) => {
518 // test_len < pat_len. If $actual_len = test_len,
519 // then $actual_len < pat_len and we don't have
521 resulting_candidates[1].push(candidate.clone());
524 (Ordering::Equal, &Some(_)) | (Ordering::Greater, &Some(_)) => {
525 // This can match both if $actual_len = test_len >= pat_len,
526 // and if $actual_len > test_len. We can't advance.
529 (Ordering::Greater, &None) => {
530 // test_len != pat_len, so if $actual_len = test_len, then
531 // $actual_len != pat_len.
532 resulting_candidates[1].push(candidate.clone());
538 (&TestKind::Len { len: test_len, op: BinOp::Ge },
539 &PatternKind::Slice { ref prefix, ref slice, ref suffix }) => {
540 // the test is `$actual_len >= test_len`
541 let pat_len = (prefix.len() + suffix.len()) as u64;
542 match (test_len.cmp(&pat_len), slice) {
543 (Ordering::Equal, &Some(_)) => {
544 // $actual_len >= test_len = pat_len,
546 resulting_candidates[0].push(
547 self.candidate_after_slice_test(match_pair_index,
555 (Ordering::Less, _) | (Ordering::Equal, &None) => {
556 // test_len <= pat_len. If $actual_len < test_len,
557 // then it is also < pat_len, so the test passing is
558 // necessary (but insufficient).
559 resulting_candidates[0].push(candidate.clone());
562 (Ordering::Greater, &None) => {
563 // test_len > pat_len. If $actual_len >= test_len > pat_len,
564 // then we know we won't have a match.
565 resulting_candidates[1].push(candidate.clone());
568 (Ordering::Greater, &Some(_)) => {
569 // test_len < pat_len, and is therefore less
570 // strict. This can still go both ways.
576 (&TestKind::Eq { .. }, _) |
577 (&TestKind::Range { .. }, _) |
578 (&TestKind::Len { .. }, _) => {
579 // These are all binary tests.
581 // FIXME(#29623) we can be more clever here
582 let pattern_test = self.test(&match_pair);
583 if pattern_test.kind == test.kind {
584 let new_candidate = self.candidate_without_match_pair(match_pair_index,
586 resulting_candidates[0].push(new_candidate);
595 fn candidate_without_match_pair<'pat>(&mut self,
596 match_pair_index: usize,
597 candidate: &Candidate<'pat, 'tcx>)
598 -> Candidate<'pat, 'tcx> {
599 let other_match_pairs =
600 candidate.match_pairs.iter()
602 .filter(|&(index, _)| index != match_pair_index)
603 .map(|(_, mp)| mp.clone())
606 span: candidate.span,
607 match_pairs: other_match_pairs,
608 bindings: candidate.bindings.clone(),
609 guard: candidate.guard.clone(),
610 arm_index: candidate.arm_index,
614 fn candidate_after_slice_test<'pat>(&mut self,
615 match_pair_index: usize,
616 candidate: &Candidate<'pat, 'tcx>,
617 prefix: &'pat [Pattern<'tcx>],
618 opt_slice: Option<&'pat Pattern<'tcx>>,
619 suffix: &'pat [Pattern<'tcx>])
620 -> Candidate<'pat, 'tcx> {
621 let mut new_candidate =
622 self.candidate_without_match_pair(match_pair_index, candidate);
623 self.prefix_slice_suffix(
624 &mut new_candidate.match_pairs,
625 &candidate.match_pairs[match_pair_index].lvalue,
633 fn candidate_after_variant_switch<'pat>(&mut self,
634 match_pair_index: usize,
635 adt_def: &'tcx ty::AdtDef,
636 variant_index: usize,
637 subpatterns: &'pat [FieldPattern<'tcx>],
638 candidate: &Candidate<'pat, 'tcx>)
639 -> Candidate<'pat, 'tcx> {
640 let match_pair = &candidate.match_pairs[match_pair_index];
642 // So, if we have a match-pattern like `x @ Enum::Variant(P1, P2)`,
643 // we want to create a set of derived match-patterns like
644 // `(x as Variant).0 @ P1` and `(x as Variant).1 @ P1`.
645 let elem = ProjectionElem::Downcast(adt_def, variant_index);
646 let downcast_lvalue = match_pair.lvalue.clone().elem(elem); // `(x as Variant)`
647 let consequent_match_pairs =
650 // e.g., `(x as Variant).0`
651 let lvalue = downcast_lvalue.clone().field(subpattern.field,
652 subpattern.pattern.ty);
653 // e.g., `(x as Variant).0 @ P1`
654 MatchPair::new(lvalue, &subpattern.pattern)
657 // In addition, we need all the other match pairs from the old candidate.
658 let other_match_pairs =
659 candidate.match_pairs.iter()
661 .filter(|&(index, _)| index != match_pair_index)
662 .map(|(_, mp)| mp.clone());
664 let all_match_pairs = consequent_match_pairs.chain(other_match_pairs).collect();
667 span: candidate.span,
668 match_pairs: all_match_pairs,
669 bindings: candidate.bindings.clone(),
670 guard: candidate.guard.clone(),
671 arm_index: candidate.arm_index,
675 fn error_simplifyable<'pat>(&mut self, match_pair: &MatchPair<'pat, 'tcx>) -> ! {
676 span_bug!(match_pair.pattern.span,
677 "simplifyable pattern found: {:?}",
682 fn is_switch_ty<'tcx>(ty: Ty<'tcx>) -> bool {
683 ty.is_integral() || ty.is_char() || ty.is_bool()