Allow logging of image loading

1 files changed, 9 insertions, 4 deletions

diff --git a/src/vm/perfmap.h b/src/vm/perfmap.h
index ca86f5b028..fe38ed3ad5 100644
--- a/src/vm/perfmap.h
+++ b/src/vm/perfmap.h
@@ -10,6 +10,8 @@
 #include "sstring.h"
 #include "fstream.h"
 
+class PerfInfo;
+
 // Generates a perfmap file.
 class PerfMap
 {
@@ -20,6 +22,9 @@ private:
     // The file stream to write the map to.
     CFileStream * m_FileStream;
 
+    // The perfinfo file to log images to.
+    PerfInfo* m_PerfInfo;
+
     // Set to true if an error is encountered when writing to the file.
     bool m_ErrorEncountered;
 
@@ -43,10 +48,10 @@ protected:
     // Does the actual work to log a method to the map.
     void LogMethod(MethodDesc * pMethod, PCODE pCode, size_t codeSize);
 
-    // Does the actual work to log a native image load to the map.
-    void LogNativeImage(PEFile * pFile);
+    // Does the actual work to log an image
+    void LogImage(PEFile * pFile);
 
-    // Get the native image signature and store it as a string.
+    // Get the image signature and store it as a string.
     static void GetNativeImageSignature(PEFile * pFile, WCHAR * pwszSig, unsigned int nSigSize);
 
 public:
@@ -54,7 +59,7 @@ public:
     static void Initialize();
 
     // Log a native image load to the map.
-    static void LogNativeImageLoad(PEFile * pFile);
+    static void LogImageLoad(PEFile * pFile);
 
     // Log a JIT compiled method to the map.
     static void LogJITCompiledMethod(MethodDesc * pMethod, PCODE pCode, size_t codeSize);