Support to handle IPF assembly files

[mirror_edk2.git] / Tools / Java / Source / MigrationTools / org / tianocore / migration / SourceFileReplacer.java

diff --git a/Tools/Java/Source/MigrationTools/org/tianocore/migration/SourceFileReplacer.java b/Tools/Java/Source/MigrationTools/org/tianocore/migration/SourceFileReplacer.java
index e37f7f46b243ede6a5a45d8634153414ac9616d8..4a4d71cd79418f6b0f83bd570ed0f7ba0581c77e 100644 (file)
--- a/Tools/Java/Source/MigrationTools/org/tianocore/migration/SourceFileReplacer.java
+++ b/Tools/Java/Source/MigrationTools/org/tianocore/migration/SourceFileReplacer.java
@@ -17,6 +17,8 @@ import java.util.*;
 import java.util.regex.Matcher;\r
 import java.util.regex.Pattern;\r
 \r
+import org.tianocore.UsageTypes;\r
+\r
 public final class SourceFileReplacer implements Common.ForDoAll {\r
     private static final SourceFileReplacer SFReplacer = new SourceFileReplacer();\r
     private ModuleInfo mi;\r
@@ -73,7 +75,9 @@ public final class SourceFileReplacer implements Common.ForDoAll {
         }\r
         \r
         public boolean recognize(String filename) {\r
-            return filename.contains(".h") || filename.contains(".H") || filename.contains(".uni");\r
+            return filename.contains(".h") || filename.contains(".H") || filename.contains(".uni") ||\r
+                   filename.contains(".s") || filename.contains(".S") || filename.contains(".asm") ||\r
+                   (!filename.contains(".inf") && filename.contains(".i"));\r
         }\r
 \r
         public String namechange(String oldname) {\r
@@ -297,7 +301,8 @@ public final class SourceFileReplacer implements Common.ForDoAll {
         it = symbolSet.iterator();\r
         while (it.hasNext()) {\r
             r8thing = it.next();\r
-            mi.hashrequiredr9libs.add(MigrationTool.db.getR9Lib(r8thing));                // add a library here\r
+            mi.addLibraryClass(MigrationTool.db.getR9Lib(r8thing), UsageTypes.ALWAYS_CONSUMED);\r
+            //mi.hashrequiredr9libs.add(MigrationTool.db.getR9Lib(r8thing));            // add a library here\r
  \r
             r8tor9 temp;\r
             if ((r9thing = MigrationTool.db.getR9Func(r8thing)) != null) {\r