1. Fix EDKT339 Library Class Search need to get info for selected library class

[mirror_edk2.git] / Tools / Java / Source / FrameworkWizard / src / org / tianocore / frameworkwizard / module / Identifications / LibraryClass / LibraryClassVector.java

diff --git a/Tools/Java/Source/FrameworkWizard/src/org/tianocore/frameworkwizard/module/Identifications/LibraryClass/LibraryClassVector.java b/Tools/Java/Source/FrameworkWizard/src/org/tianocore/frameworkwizard/module/Identifications/LibraryClass/LibraryClassVector.java
index fa03258bdc33ef86643200b2c24d04becc21404a..3155cf63299644f62eba7e9da8b6eb990b567989 100644 (file)
--- a/Tools/Java/Source/FrameworkWizard/src/org/tianocore/frameworkwizard/module/Identifications/LibraryClass/LibraryClassVector.java
+++ b/Tools/Java/Source/FrameworkWizard/src/org/tianocore/frameworkwizard/module/Identifications/LibraryClass/LibraryClassVector.java
@@ -23,7 +23,14 @@ public class LibraryClassVector {
         for (int index = 0; index < vLibraryClass.size(); index++) {\r
             if (vLibraryClass.elementAt(index).getLibraryClassName().equals(lib.getLibraryClassName())\r
                 && vLibraryClass.elementAt(index).getUsage().equals(lib.getUsage())) {\r
-                return index;\r
+                if (vLibraryClass.elementAt(index).getBelongModule() != null && lib.getBelongModule() != null) {\r
+                    if (vLibraryClass.elementAt(index).getBelongModule().equals(lib.getBelongModule())) {\r
+                        return index;\r
+                    }\r
+                }\r
+                if (vLibraryClass.elementAt(index).getBelongModule() == null && lib.getBelongModule() == null) {\r
+                    return index;\r
+                }\r
             }\r
         }\r
         return -1;\r