Timer Queries

frontend/services/profiling_service/Profiling_Service.cpp

@@ -1,5 +1,7 @@
 #include "Profiling_Service.hpp"
 
+#include "LuaCallbacksCollection.h"
+
 namespace megamol {
 namespace frontend {
 
@@ -31,6 +33,12 @@ bool Profiling_Service::init(void* configPtr) {
  });
  }
 #endif
+
+ this->_requestedResourcesNames = {"RegisterLuaCallbacks"};
+
+ distro_ = std::uniform_int_distribution<int64_t>(1);
+ rng_ = std::mt19937_64(42);
+
  return true;
 }
 
@@ -44,10 +52,91 @@ void Profiling_Service::close() {
 
 void Profiling_Service::updateProvidedResources() {
  _perf_man.startFrame();
+ start_timer_queries();
 }
 
 void Profiling_Service::resetProvidedResources() {
  _perf_man.endFrame();
+ notify_timer_queries();
+}
+
+void Profiling_Service::fill_lua_callbacks() {
+ frontend_resources::LuaCallbacksCollection callbacks;
+
+ callbacks.add<frontend_resources::LuaCallbacksCollection::LongResult>("mmCreateTimeQuery",
+ "(void)\n\tCreates a time query to time a specified number of frames.\n\tReturn UID of the query.",
+ {[&]() -> frontend_resources::LuaCallbacksCollection::LongResult {
+ //auto uid = distro_(rng_);
+ auto uid = ++timer_id_;
+ auto fit = timer_map_.find(uid);
+ if (fit != timer_map_.end()) {
+ return frontend_resources::LuaCallbacksCollection::LongResult(0);
+ }
+ timer_map_[uid] = std::tuple<int, int64_t, int64_t>(-1, -1, -1);
+ return frontend_resources::LuaCallbacksCollection::LongResult(uid);
+ }});
+
+ callbacks.add<frontend_resources::LuaCallbacksCollection::BoolResult, int64_t, int>("mmStartTimeQuery",
+ "(long UID, int num_frames)\n\tStart the query specified by the UID. After num_frames a timestamp is "
+ "recorded.\n\tReturns true if successful.",
+ {[&](int64_t uid, int num_frames) -> frontend_resources::LuaCallbacksCollection::BoolResult {
+ auto fit = timer_map_.find(uid);
+ if (fit != timer_map_.end()) {
+ std::get<0>(fit->second) = num_frames;
+ /*std::get<1>(fit->second) = std::chrono::system_clock::to_time_t(std::chrono::system_clock::now());
+ std::get<2>(fit->second) = -1;*/
+ return frontend_resources::LuaCallbacksCollection::BoolResult(true);
+ }
+ return frontend_resources::LuaCallbacksCollection::BoolResult(false);
+ }});
+
+ callbacks.add<frontend_resources::LuaCallbacksCollection::BoolResult, int64_t>("mmPokeTimeQuery",
+ "(long UID)\n\tPokes the query specified by UID if num_frames have passed.",
+ {[&](int64_t uid) -> frontend_resources::LuaCallbacksCollection::BoolResult {
+ auto fit = timer_map_.find(uid);
+ if (fit != timer_map_.end() && std::get<0>(fit->second) == 0) {
+ /*auto val = std::get<2>(fit->second);
+ timer_map_.erase(fit->first);*/
+ return frontend_resources::LuaCallbacksCollection::BoolResult(true);
+ }
+ return frontend_resources::LuaCallbacksCollection::BoolResult(false);
+ }});
+
+ callbacks.add<frontend_resources::LuaCallbacksCollection::LongResult, int64_t>("mmEndTimeQuery",
+ "(long UID)\n\tDestroys the query.\n\tReturns timestamp at end of query.",
+ {[&](int64_t uid) -> frontend_resources::LuaCallbacksCollection::LongResult {
+ auto fit = timer_map_.find(uid);
+ if (fit != timer_map_.end() && std::get<0>(fit->second) == 0) {
+ auto val = std::get<2>(fit->second);
+ timer_map_.erase(fit->first);
+ return frontend_resources::LuaCallbacksCollection::LongResult(val);
+ }
+ return frontend_resources::LuaCallbacksCollection::LongResult(-1);
+ }});
+
+ auto& register_callbacks =
+ _requestedResourcesReferences[0]
+ .getResource<std::function<void(frontend_resources::LuaCallbacksCollection const&)>>();
+
+ register_callbacks(callbacks);
+}
+
+void Profiling_Service::start_timer_queries() {
+ for (auto& query : timer_map_) {
+ if (std::get<0>(query.second) != 0 && std::get<1>(query.second) == -1) {
+ std::get<1>(query.second) = std::chrono::system_clock::to_time_t(std::chrono::system_clock::now());
+ }
+ }
+}
+
+void Profiling_Service::notify_timer_queries() {
+ for (auto& query : timer_map_) {
+ if (std::get<0>(query.second) == 0) {
+ std::get<2>(query.second) = std::chrono::system_clock::to_time_t(std::chrono::system_clock::now());
+ } else if (std::get<0>(query.second) > 0) {
+ --std::get<0>(query.second);
+ }
+ }
 }
 } // namespace frontend
 } // namespace megamol

frontend/services/profiling_service/Profiling_Service.hpp

@@ -8,6 +8,8 @@
 #pragma once
 
 #include <fstream>
+#include <random>
+#include <tuple>
 
 #include "AbstractFrontendService.hpp"
 #include "PerformanceManager.h"
@@ -39,13 +41,31 @@ class Profiling_Service final : public AbstractFrontendService {
  const std::vector<std::string> getRequestedResourceNames() const override {
  return _requestedResourcesNames;
  }
- void setRequestedResources(std::vector<FrontendResource> resources) override {}
+ void setRequestedResources(std::vector<FrontendResource> resources) override {
+ _requestedResourcesReferences = resources;
+ }
 
  std::vector<FrontendResource> _providedResourceReferences;
  std::vector<std::string> _requestedResourcesNames;
+ std::vector<FrontendResource> _requestedResourcesReferences;
 
  megamol::frontend_resources::PerformanceManager _perf_man;
  std::ofstream log_file;
+
+private:
+ void fill_lua_callbacks();
+
+ void start_timer_queries();
+
+ void notify_timer_queries();
+
+ std::unordered_map<int64_t, std::tuple<int, int64_t, int64_t>> timer_map_;
+
+ std::mt19937_64 rng_;
+
+ std::uniform_int_distribution<int64_t> distro_;
+
+ int64_t timer_id_ = 0;
 };
 
 } // namespace frontend