mcs/class/Mono.CodeContracts/Mono.CodeContracts.Static.Analysis.StackAnalysis/APCMap.cs

   1 //
   2 // APCMap.cs
   3 //
   4 // Authors:
   5 //      Alexander Chebaturkin (chebaturkin@gmail.com)
   6 //
   7 // Copyright (C) 2011 Alexander Chebaturkin
   8 //
   9 // Permission is hereby granted, free of charge, to any person obtaining
  10 // a copy of this software and associated documentation files (the
  11 // "Software"), to deal in the Software without restriction, including
  12 // without limitation the rights to use, copy, modify, merge, publish,
  13 // distribute, sublicense, and/or sell copies of the Software, and to
  14 // permit persons to whom the Software is furnished to do so, subject to
  15 // the following conditions:
  16 //
  17 // The above copyright notice and this permission notice shall be
  18 // included in all copies or substantial portions of the Software.
  19 //
  20 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  21 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  22 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  23 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  24 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  25 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  26 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  27 //
  28
  29 using System.Collections.Generic;
  30 using Mono.CodeContracts.Static.ControlFlow;
  31 using Mono.CodeContracts.Static.DataStructures;
  32
  33 namespace Mono.CodeContracts.Static.Analysis.StackAnalysis {
  34         class APCMap<T> {
  35                 private readonly Dictionary<int, T>[] block_map;
  36                 private IImmutableIntMap<bool> call_on_this_map;
  37
  38                 public APCMap (Subroutine parent)
  39                 {
  40                         this.block_map = new Dictionary<int, T>[parent.BlockCount];
  41                         this.call_on_this_map = ImmutableIntMap<bool>.Empty;
  42                 }
  43
  44                 public T this [APC key]
  45                 {
  46                         get
  47                         {
  48                                 T value;
  49                                 if (!TryGetValue (key, out value))
  50                                         throw new KeyNotFoundException ();
  51                                 return value;
  52                         }
  53                 }
  54
  55                 public void Add (APC pc, T value)
  56                 {
  57                         Dictionary<int, T> pcBlockMap = this.block_map [pc.Block.Index];
  58                         if (pcBlockMap == null)
  59                                 this.block_map [pc.Block.Index] = pcBlockMap = new Dictionary<int, T> ();
  60
  61                         pcBlockMap.Add (pc.Index, value);
  62                 }
  63
  64                 public bool TryGetValue (APC pc, out T obj)
  65                 {
  66                         Dictionary<int, T> dictionary = this.block_map [pc.Block.Index];
  67                         if (dictionary != null)
  68                                 return dictionary.TryGetValue (pc.Index, out obj);
  69
  70                         obj = default(T);
  71                         return false;
  72                 }
  73
  74                 public bool ContainsKey (APC pc)
  75                 {
  76                         Dictionary<int, T> dictionary = this.block_map [pc.Block.Index];
  77                         if (dictionary == null)
  78                                 return false;
  79                         return dictionary.ContainsKey (pc.Index);
  80                 }
  81
  82                 public bool IsCallOnThis (APC key)
  83                 {
  84                         return this.call_on_this_map [key.Block.Index];
  85                 }
  86
  87                 public void AddCallOnThis (APC pc)
  88                 {
  89                         this.call_on_this_map = this.call_on_this_map.Add (pc.Block.Index, true);
  90                 }
  91         }
  92 }