diff --git a/Tools/Java/Source/MigrationTools/org/tianocore/migration/ModuleReader.java b/Tools/Java/Source/MigrationTools/org/tianocore/migration/ModuleReader.java index d65d72fa44..c72399ea2c 100644 --- a/Tools/Java/Source/MigrationTools/org/tianocore/migration/ModuleReader.java +++ b/Tools/Java/Source/MigrationTools/org/tianocore/migration/ModuleReader.java @@ -361,11 +361,32 @@ public final class ModuleReader implements Common.ForDoAll { .group(1))) { templine = mtrinclude.group(); } else { - templine = MigrationTool.MIGRATIONCOMMENT - + mtrinclude.group(); - } - mtrinclude.appendReplacement(wholebuffer, templine); - } + String line = mtrinclude.group().toLowerCase(); + if (line.contains("pal.h")) { + templine = "#include \n"; + } else if (line.contains("sal.h")) { + templine = "#include \n"; + } else if (line.contains("pci22.h")) { + templine = "#include \n"; + } else if (line.contains("pci23.h")) { + templine = "#include \n"; + } else if (line.contains("pci30.h")) { + templine = "#include \n"; + } else if (line.contains("pci.h")) { + templine = "#include \n"; + } else if (line.contains("acpi.h")) { + templine = "#include \n"; + } else if (line.contains("scsi.h")) { + templine = "#include \n"; + } else if (line.contains("usb.h")) { + templine = "#include \n"; + } else { + templine = MigrationTool.MIGRATIONCOMMENT + + mtrinclude.group(); + } + } + mtrinclude.appendReplacement(wholebuffer, templine); + } mtrinclude.appendTail(wholebuffer); return wholebuffer.toString(); }