Protect against multiple runs; optimize il

Source/OpenTK.Rewrite/OpenTK.Rewrite.csproj

@@ -52,6 +52,9 @@
     <Reference Include="Mono.Cecil.Pdb">
       <HintPath>Dependencies\Mono.Cecil.Pdb.dll</HintPath>
     </Reference>
+    <Reference Include="Mono.Cecil.Rocks">
+      <HintPath>Dependencies\Mono.Cecil.Rocks.dll</HintPath>
+    </Reference>
   </ItemGroup>
   <ItemGroup>
     <Compile Include="Program.cs" />
@@ -60,16 +63,14 @@
   <ItemGroup>
     <Content Include="Dependencies\Mono.Cecil.dll" />
     <Content Include="Dependencies\Mono.Cecil.Mdb.dll" />
-    <Content Include="Dependencies\Mono.Cecil.Mdb.pdb" />
-    <Content Include="Dependencies\Mono.Cecil.pdb" />
     <Content Include="Dependencies\Mono.Cecil.Pdb.dll" />
-    <Content Include="Dependencies\Mono.Cecil.Pdb.pdb" />
     <Content Include="Dependencies\IKVM.Reflection.dll" />
   </ItemGroup>
   <ItemGroup>
     <None Include="..\..\OpenTK.snk">
       <Link>OpenTK.snk</Link>
     </None>
+    <None Include="Dependencies\Mono.Cecil.Rocks.dll" />
   </ItemGroup>
   <Import Project="$(MSBuildToolsPath)\Microsoft.CSharp.targets" />
   <!-- To modify your build process, add your task inside one of the targets below and uncomment it. 

Source/OpenTK.Rewrite/Program.cs

@@ -22,6 +22,7 @@ using System.Text;
 
 using Mono.Cecil;
 using Mono.Cecil.Cil;
+using Mono.Cecil.Rocks;
 
 namespace OpenTK.Rewrite
 {
@@ -81,7 +82,12 @@ namespace OpenTK.Rewrite
 
             // Load assembly and process all modules
             var assembly = AssemblyDefinition.ReadAssembly(file, read_params);
+            var rewritten = assembly.CustomAttributes.FirstOrDefault(a => a.AttributeType.Name == "RewrittenAttribute");
+            if (rewritten == null)
+            {
                 foreach (var module in assembly.Modules)
+                {
+                    if (rewritten == null)
                     {
                         foreach (var reference in module.AssemblyReferences)
                         {
@@ -93,6 +99,12 @@ namespace OpenTK.Rewrite
                             Rewrite(type);
                         }
                     }
+                }
+            }
+            else
+            {
+                Console.Error.WriteLine("Error: assembly has already been rewritten");
+            }
 
             // Save rewritten assembly
             assembly.Write(file, write_params);
@@ -111,6 +123,13 @@ namespace OpenTK.Rewrite
                     }
                 }
             }
+
+            if (type.Name == "RewrittenAttribute")
+            {
+                var rewritten_constructor = type.GetConstructors().First();
+                var rewritten = new CustomAttribute(rewritten_constructor, BitConverter.GetBytes(true));
+                type.Module.Assembly.CustomAttributes.Add(rewritten);
+            }
         }
 
         // Create body for method
@@ -138,6 +157,8 @@ namespace OpenTK.Rewrite
 
             // return
             il.Emit(OpCodes.Ret);
+
+            body.OptimizeMacros();
         }
 
         static void EmitParameters(MethodDefinition method, TypeReference nint, MethodBody body, ILProcessor il)
@@ -145,51 +166,14 @@ namespace OpenTK.Rewrite
             for (int i = 0; i < method.Parameters.Count; i++)
             {
                 var p = method.Parameters[i];
-                switch (i)
+                il.Emit(OpCodes.Ldarg, i);
-                {
+
-                    case 0:
-                        il.Emit(OpCodes.Ldarg_0);
-                        break;
-                    case 1:
-                        il.Emit(OpCodes.Ldarg_1);
-                        break;
-                    case 2:
-                        il.Emit(OpCodes.Ldarg_2);
-                        break;
-                    case 3:
-                        il.Emit(OpCodes.Ldarg_3);
-                        break;
-                    default:
-                        il.Emit(OpCodes.Ldarg_S, (byte)i);
-                        break;
-                }
                 if (p.ParameterType.IsArray || p.ParameterType.IsByReference)
                 {
                     body.Variables.Add(new VariableDefinition(new PinnedType(nint)));
                     var index = body.Variables.Count - 1;
-                    switch (index)
+                    il.Emit(OpCodes.Stloc, index);
-                    {
+                    il.Emit(OpCodes.Ldloc, index);
-                        case 0:
-                            il.Emit(OpCodes.Stloc_0);
-                            il.Emit(OpCodes.Ldloc_0);
-                            break;
-                        case 1:
-                            il.Emit(OpCodes.Stloc_1);
-                            il.Emit(OpCodes.Ldloc_1);
-                            break;
-                        case 2:
-                            il.Emit(OpCodes.Stloc_2);
-                            il.Emit(OpCodes.Ldloc_2);
-                            break;
-                        case 3:
-                            il.Emit(OpCodes.Stloc_3);
-                            il.Emit(OpCodes.Ldloc_3);
-                            break;
-                        default:
-                            il.Emit(OpCodes.Stloc_S, (byte)index);
-                            il.Emit(OpCodes.Ldloc_S, (byte)index);
-                            break;
-                    }
                     //il.Emit(OpCodes.Conv_I);
                 }
             }
@@ -198,42 +182,7 @@ namespace OpenTK.Rewrite
         static void EmitEntryPoint(FieldDefinition entry_points, ILProcessor il, int slot)
         {
             il.Emit(OpCodes.Ldsfld, entry_points);
-            switch (slot)
-            {
-                case 0:
-                    il.Emit(OpCodes.Ldc_I4_0);
-                    break;
-                case 1:
-                    il.Emit(OpCodes.Ldc_I4_1);
-                    break;
-                case 2:
-                    il.Emit(OpCodes.Ldc_I4_2);
-                    break;
-                case 3:
-                    il.Emit(OpCodes.Ldc_I4_3);
-                    break;
-                case 4:
-                    il.Emit(OpCodes.Ldc_I4_4);
-                    break;
-                case 5:
-                    il.Emit(OpCodes.Ldc_I4_5);
-                    break;
-                case 6:
-                    il.Emit(OpCodes.Ldc_I4_6);
-                    break;
-                case 7:
-                    il.Emit(OpCodes.Ldc_I4_7);
-                    break;
-                case 8:
-                    il.Emit(OpCodes.Ldc_I4_8);
-                    break;
-                default:
-                    if (slot < 128)
-                        il.Emit(OpCodes.Ldc_I4_S, (byte)slot);
-                    else
             il.Emit(OpCodes.Ldc_I4, slot);
-                    break;
-            }
             il.Emit(OpCodes.Ldelem_I);
         }