1 files changed, 81 insertions, 0 deletions
diff --git a/src/vm/stacksampler.h b/src/vm/stacksampler.h
new file mode 100644
index 0000000000..33fc6b93ce
--- /dev/null
+++ b/src/vm/stacksampler.h
@@ -0,0 +1,81 @@
+// Licensed to the .NET Foundation under one or more agreements.
+// The .NET Foundation licenses this file to you under the MIT license.
+// See the LICENSE file in the project root for more information.
+
+/*++
+
+Module Name:
+
+    StackSampler.h
+
+--*/
+
+#ifndef __STACK_SAMPLER_H
+#define __STACK_SAMPLER_H
+#endif
+
+#ifdef FEATURE_STACK_SAMPLING
+
+class StackSampler
+{
+public:
+    // Interface
+    static void Init();
+    static void RecordJittingInfo(MethodDesc* pMD, DWORD dwFlags, DWORD dwFlags2);
+
+private:
+
+    // Methods
+    StackSampler();
+    ~StackSampler();
+
+    static DWORD __stdcall SamplingThreadProc(void* arg);
+
+    static StackWalkAction StackWalkCallback(CrawlFrame* pCf, VOID* data);
+    
+    StackWalkAction CrawlFrameVisitor(CrawlFrame* pCf, Thread* pMdThread);
+
+    void ThreadProc();
+
+    void JitFrequentMethodsInSamples();
+
+    void JitAndCollectTrace(MethodDesc* pMD, const ADID& adId);
+
+    void RecordJittingInfoInternal(MethodDesc* pMD, DWORD flags);
+    ADID GetDomainId(MethodDesc* pMD, const ADID& defaultId);
+
+
+    // Constants
+    static const int s_knDefaultSamplingIntervalMsec = 100;
+    static const int s_knDefaultNumMethods = 32;
+    static const int s_knDefaultCountForImportance = 0;    // TODO: Set to some reasonable value.
+
+    // Typedefs
+    struct CountInfo;
+    typedef MapSHash<MethodDesc*, CountInfo> CountInfoHash;
+    typedef CountInfoHash::element_t CountInfoHashEntry;
+
+    typedef MapSHash<MethodDesc*, ADID> JitInfoHash;
+    typedef JitInfoHash::element_t JitInfoHashEntry;
+
+    // Nested types
+    struct CountInfo
+    {
+        unsigned uCount;
+        bool fJitted;
+        ADID adDomainId;
+        CountInfo(const ADID& adId) : adDomainId(adId), fJitted(false), uCount(0) {}
+        CountInfo() {} // SHash doesn't like it
+    };
+
+    // Fields
+    Crst m_crstJitInfo;
+    CountInfoHash m_countInfo;
+    JitInfoHash m_jitInfo;
+    Thread* m_pThread;
+    unsigned m_nSampleEvery;
+    unsigned m_nSampleAfter;
+    unsigned m_nNumMethods;
+};
+#endif // FEATURE_STACK_SAMPLING
+