3
// (C) 2007 - 2008 Novell, Inc. http://www.novell.com
4
// (C) 2007 - 2008 Jb Evain http://evain.net
6
// Permission is hereby granted, free of charge, to any person obtaining
7
// a copy of this software and associated documentation files (the
8
// "Software"), to deal in the Software without restriction, including
9
// without limitation the rights to use, copy, modify, merge, publish,
10
// distribute, sublicense, and/or sell copies of the Software, and to
11
// permit persons to whom the Software is furnished to do so, subject to
12
// the following conditions:
14
// The above copyright notice and this permission notice shall be
15
// included in all copies or substantial portions of the Software.
17
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
18
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
19
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
20
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
21
// LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
22
// OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
23
// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
28
using System.Collections.Generic;
32
using Cecil.Decompiler.Ast;
34
namespace Cecil.Decompiler.Steps {
36
public class DeclareVariablesOnFirstAssignment : BaseCodeTransformer, IDecompilationStep {
38
public static readonly IDecompilationStep Instance = new DeclareVariablesOnFirstAssignment ();
40
DecompilationContext context;
41
HashSet<VariableDefinition> not_assigned = new HashSet<VariableDefinition> ();
43
public override ICodeNode VisitVariableReferenceExpression (VariableReferenceExpression node)
45
var variable = (VariableDefinition) node.Variable;
47
if (!TryDiscardVariable (variable))
50
return new VariableDeclarationExpression (variable);
53
public override ICodeNode VisitVariableDeclarationExpression (VariableDeclarationExpression node)
55
TryDiscardVariable (node.Variable);
60
private bool TryDiscardVariable (VariableDefinition variable)
62
if (!not_assigned.Contains (variable))
65
RemoveVariable (variable);
69
void RemoveVariable (VariableDefinition variable)
71
context.RemoveVariable (variable);
72
not_assigned.Remove (variable);
75
public BlockStatement Process (DecompilationContext context, BlockStatement block)
77
this.context = context;
78
PopulateNotAssigned ();
79
return (BlockStatement) VisitBlockStatement (block);
82
void PopulateNotAssigned ()
84
not_assigned.Clear ();
85
foreach (VariableDefinition variable in context.Variables)
86
not_assigned.Add (variable);