Merge pull request #439 from mono-soc-2012/garyb/iconfix
[mono.git] / mcs / class / Mono.CodeContracts / Mono.CodeContracts.Static.ControlFlow.Subroutines.Builders / SimpleSubroutineBuilder.cs
1 // 
2 // SimpleSubroutineBuilder.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.Blocks;
31 using Mono.CodeContracts.Static.Providers;
32
33 namespace Mono.CodeContracts.Static.ControlFlow.Subroutines.Builders {
34         class SimpleSubroutineBuilder<TLabel> : SubroutineBuilder<TLabel> {
35                 private readonly HashSet<TLabel> begin_old_start = new HashSet<TLabel> ();
36                 private readonly HashSet<TLabel> end_old_start = new HashSet<TLabel> ();
37
38                 private BlockWithLabels<TLabel> block_prior_to_old;
39                 private OldValueSubroutine<TLabel> current_old_subroutine;
40                 private SubroutineBase<TLabel> current_subroutine;
41
42                 public SimpleSubroutineBuilder (ICodeProvider<TLabel> codeProvider,
43                                                 SubroutineFacade subroutineFacade,
44                                                 TLabel entry)
45                         : base (codeProvider, subroutineFacade, entry)
46                 {
47                         Initialize (entry);
48                 }
49
50                 public override SubroutineBase<TLabel> CurrentSubroutine
51                 {
52                         get
53                         {
54                                 if (this.current_old_subroutine != null)
55                                         return this.current_old_subroutine;
56                                 return this.current_subroutine;
57                         }
58                 }
59
60                 public BlockWithLabels<TLabel> BuildBlocks (TLabel entry, SubroutineBase<TLabel> subroutine)
61                 {
62                         this.current_subroutine = subroutine;
63
64                         return base.BuildBlocks (entry);
65                 }
66
67                 public override BlockWithLabels<TLabel> RecordInformationForNewBlock (TLabel currentLabel, BlockWithLabels<TLabel> previousBlock)
68                 {
69                         TLabel label;
70                         if (previousBlock != null && previousBlock.TryGetLastLabel (out label) && this.end_old_start.Contains (label)) {
71                                 OldValueSubroutine<TLabel> oldValueSubroutine = this.current_old_subroutine;
72                                 oldValueSubroutine.Commit (previousBlock);
73                                 this.current_old_subroutine = null;
74                                 BlockWithLabels<TLabel> result = base.RecordInformationForNewBlock (currentLabel, this.block_prior_to_old);
75                                 CurrentSubroutine.AddEdgeSubroutine (this.block_prior_to_old, result, oldValueSubroutine, EdgeTag.Old);
76                                 return result;
77                         }
78
79                         if (!this.begin_old_start.Contains (currentLabel))
80                                 return base.RecordInformationForNewBlock (currentLabel, previousBlock);
81                         this.current_old_subroutine = new OldValueSubroutine<TLabel> (this.SubroutineFacade,
82                                                                                      ((MethodContractSubroutine<TLabel>) this.current_subroutine).Method,
83                                                                                      this, currentLabel);
84                         this.block_prior_to_old = previousBlock;
85                         BlockWithLabels<TLabel> newBlock = base.RecordInformationForNewBlock (currentLabel, null);
86                         this.current_old_subroutine.RegisterBeginBlock (newBlock);
87                         return newBlock;
88                 }
89
90                 public override void BeginOldHook (TLabel label)
91                 {
92                         this.begin_old_start.Add (label);
93                 }
94
95                 public override void EndOldHook (TLabel label)
96                 {
97                         this.end_old_start.Add (label);
98                 }
99         }
100 }