BaseTools: Enhance StrDefs.h to include ImageDefs.h

[mirror_edk2.git] / BaseTools / Source / Python / AutoGen / GenC.py

diff --git a/BaseTools/Source/Python/AutoGen/GenC.py b/BaseTools/Source/Python/AutoGen/GenC.py
index 96b1459253fdebebc4ae82cc7a6b50b97425b42f..0fb6b9fc4eb608d570827eef6ba9aee7954829dd 100644 (file)
--- a/BaseTools/Source/Python/AutoGen/GenC.py
+++ b/BaseTools/Source/Python/AutoGen/GenC.py
@@ -1981,6 +1981,9 @@ def CreateCode(Info, AutoGenC, AutoGenH, StringH, UniGenCFlag, UniGenBinBuffer,
                             break\r
                     GuidMacros.append('#define %s %s' % ('_PCD_VALUE_'+TokenCName, Value))\r
 \r
+        if Info.IdfFileList:\r
+            GuidMacros.append('#include "%sImgDefs.h"' % Info.Name)\r
+\r
         if GuidMacros:\r
             StringH.Append('\n#ifdef VFRCOMPILE\n%s\n#endif\n' % '\n'.join(GuidMacros))\r
 \r