feat add profiling

2024-03-16 22:56:10 +08:00
parent e05c1f4894
commit 719fecd4bc
30 changed files with 4673 additions and 1 deletions
--- a/3party/cppuprofile/lib/uprofileimpl.h
+++ b/3party/cppuprofile/lib/uprofileimpl.h
@@ -0,0 +1,91 @@
+// Software Name : cppuprofile
+// SPDX-FileCopyrightText: Copyright (c) 2022 Orange
+// SPDX-License-Identifier: BSD-3-Clause
+//
+// This software is distributed under the BSD License;
+// see the LICENSE file for more details.
+//
+// Author: Cédric CHEDALEUX <cedric.chedaleux@orange.com> et al.
+
+#ifndef UPROFILEIMPL_H_
+#define UPROFILEIMPL_H_
+
+#include "igpumonitor.h"
+#include "timestampunit.h"
+#include "util/cpumonitor.h"
+#include "util/timer.h"
+#include <fstream>
+#include <list>
+#include <map>
+#include <mutex>
+#include <string>
+
+namespace uprofile
+{
+class UProfileImpl
+{
+public:
+    enum class ProfilingType {
+        TIME_EXEC,
+        TIME_EVENT,
+        PROCESS_MEMORY,
+        SYSTEM_MEMORY,
+        CPU,
+        GPU_USAGE,
+        GPU_MEMORY
+    };
+
+    static UProfileImpl* getInstance();
+    static void destroyInstance();
+    virtual ~UProfileImpl();
+
+    // Implementation
+    void start(const char* file);
+    void stop();
+    void addGPUMonitor(IGPUMonitor* monitor);
+    void removeGPUMonitor();
+    void setTimestampUnit(TimestampUnit tsUnit);
+    void timeBegin(const std::string& title);
+    void timeEnd(const std::string& title);
+    void startProcessMemoryMonitoring(int period);
+    void startSystemMemoryMonitoring(int period);
+    void startCPUUsageMonitoring(int period);
+    void startGPUUsageMonitoring(int period);
+    void startGPUMemoryMonitoring(int period);
+    void getProcessMemory(int& rss, int& shared);
+    void getSystemMemory(int& totalMem, int& availableMem, int& freeMem);
+    vector<float> getInstantCpuUsage();
+
+private:
+    static UProfileImpl* m_uprofiler;
+    UProfileImpl();
+
+    void write(ProfilingType type, const std::list<std::string>& data);
+    unsigned long long getTimestamp() const;
+    static unsigned long long getEpochTime();
+    static unsigned long long getTimeSinceBoot();
+
+    void dumpCpuUsage();
+    void dumpProcessMemory();
+    void dumpSystemMemory();
+    void dumpGpuUsage();
+    void dumpGpuMemory();
+
+    TimestampUnit m_tsUnit;
+    std::map<std::string, unsigned long long> m_steps; // Store steps (title, start time)
+    std::ofstream m_file;
+    Timer m_processMemoryMonitorTimer;
+    Timer m_systemMemoryMonitorTimer;
+    Timer m_cpuMonitorTimer;
+    Timer m_gpuUsageMonitorTimer;
+    Timer m_gpuMemoryMonitorTimer;
+    CpuMonitor m_cpuMonitor;
+    IGPUMonitor* m_gpuMonitor;
+
+    std::mutex m_fileMutex;
+    std::mutex m_stepsMutex;
+};
+
+}
+
+#endif /* UPROFILEIMPL_H_ */