2
0

containerManager.cpp 31 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205
  1. #include "containerManager.h"
  2. #include "containerManager.h"
  3. #include "containerManager.h"
  4. #include <globalAllocator/globalAllocator.h>
  5. #include <unordered_set>
  6. #include <chrono>
  7. #include <thread>
  8. #include <unordered_map>
  9. #include <fileManipulation/fileManipulation.h>
  10. #include <stringManipulation/stringManipulation.h>
  11. #include <filesystem>
  12. #include <imgui.h>
  13. #include <containersWindow/containersWindow.h>
  14. #include <imgui_internal.h>
  15. #include <fstream>
  16. pika::containerId_t pika::ContainerManager::createContainer(std::string containerName,
  17. pika::LoadedDll &loadedDll, pika::LogManager &logManager,
  18. pika::pikaImgui::ImGuiIdsManager &imguiIDsManager, ConsoleWindow *consoleWindow, size_t memoryPos)
  19. {
  20. for(auto &i : loadedDll.containerInfo)
  21. {
  22. if (i.containerName == containerName)
  23. {
  24. return createContainer(i, loadedDll, logManager, imguiIDsManager, consoleWindow, memoryPos);
  25. }
  26. }
  27. logManager.log(("Couldn't create container, couldn't find the name: " + containerName).c_str(), pika::logError);
  28. return 0;
  29. }
  30. bool pika::ContainerManager::setSnapshotToContainer(pika::containerId_t containerId, const char *snapshotName, pika::LogManager &logManager
  31. , pika::pikaImgui::ImGuiIdsManager &imguiIdManager)
  32. {
  33. auto c = runningContainers.find(containerId);
  34. if (c == runningContainers.end())
  35. {
  36. logManager.log((std::string("Couldn't find container for setting snapshot: #")
  37. + std::to_string(containerId)).c_str(),
  38. pika::logError);
  39. return false;
  40. }
  41. if (!checkIfSnapshotIsCompatible(c->second, snapshotName))
  42. {
  43. logManager.log((std::string("Snapshot incompatible: ") + snapshotName).c_str(),
  44. pika::logError);
  45. return false;
  46. }
  47. std::string file = PIKA_ENGINE_RESOURCES_PATH;
  48. file += snapshotName;
  49. file += ".snapshot";
  50. auto s = pika::getFileSize(file.c_str());
  51. if (s != (c->second.totalSize + sizeof(pika::RuntimeContainer)) )
  52. {
  53. logManager.log((std::string("Snapshot corrupted probably, file size incorrect: ")
  54. + snapshotName).c_str(),
  55. pika::logError);
  56. return false;
  57. }
  58. pika::readEntireFile(file.c_str(), c->second.getBaseAdress(),
  59. c->second.totalSize, sizeof(pika::RuntimeContainer));
  60. //c->second.requestedContainerInfo.requestedImguiIds
  61. // = imguiIdManager.getImguiIds(c->second.requestedContainerInfo.imguiTotalRequestedIds);
  62. logManager.log("Loaded snapshot");
  63. return true;
  64. }
  65. bool pika::ContainerManager::setRecordingToContainer(pika::containerId_t containerId, const char *recordingName,
  66. pika::LogManager &logManager, pika::pikaImgui::ImGuiIdsManager &imguiIdManager)
  67. {
  68. auto c = runningContainers.find(containerId);
  69. if (c == runningContainers.end())
  70. {
  71. logManager.log((std::string("Couldn't find container for setting recording: #")
  72. + std::to_string(containerId)).c_str(),
  73. pika::logError);
  74. return false;
  75. }
  76. //todo not reset imgui id on input recording
  77. if (!setSnapshotToContainer(containerId, recordingName, logManager, imguiIdManager))
  78. {
  79. return false;
  80. }
  81. c->second.flags.status = pika::RuntimeContainer::FLAGS::STATUS_BEING_PLAYBACK;
  82. c->second.flags.frameNumber = 0;
  83. pika::strlcpy(c->second.flags.recordingName, recordingName, sizeof(c->second.flags.recordingName));
  84. }
  85. //todo use regions further appart in production
  86. void* pika::ContainerManager::allocateContainerMemory(pika::RuntimeContainer &container,
  87. pika::ContainerInformation containerInformation, void *memPos)
  88. {
  89. size_t memoryRequired = containerInformation.calculateMemoryRequirements();
  90. void * baseMemory = allocateOSMemory(memoryRequired, memPos);
  91. if (baseMemory == nullptr) { return 0; }
  92. container.totalSize = memoryRequired;
  93. allocateContainerMemoryAtBuffer(container,
  94. containerInformation, baseMemory);
  95. return baseMemory;
  96. }
  97. void pika::ContainerManager::allocateContainerMemoryAtBuffer(pika::RuntimeContainer &container,
  98. pika::ContainerInformation containerInformation, void *buffer)
  99. {
  100. const size_t staticMemory = containerInformation.containerStructBaseSize;
  101. const size_t heapMemory = containerInformation.containerStaticInfo.defaultHeapMemorySize;
  102. char *currentMemoryAdress = (char *)buffer;
  103. container.arena.containerStructMemory.size = staticMemory;
  104. container.arena.containerStructMemory.block = currentMemoryAdress;
  105. currentMemoryAdress += staticMemory;
  106. pika::align64(currentMemoryAdress);
  107. container.allocator.init(currentMemoryAdress, heapMemory);
  108. currentMemoryAdress += heapMemory;
  109. for (int i = 0; i < containerInformation.containerStaticInfo.bonusAllocators.size(); i++)
  110. {
  111. pika::align64(currentMemoryAdress);
  112. pika::memory::FreeListAllocator allocator;
  113. allocator.init(
  114. currentMemoryAdress,
  115. containerInformation.containerStaticInfo.bonusAllocators[i]
  116. );
  117. container.bonusAllocators.push_back(allocator);
  118. currentMemoryAdress += containerInformation.containerStaticInfo.bonusAllocators[i];
  119. }
  120. }
  121. void pika::ContainerManager::freeContainerStuff(pika::RuntimeContainer &container)
  122. {
  123. deallocateOSMemory(container.arena.containerStructMemory.block);
  124. //container.arena.dealocateStaticMemory(); //static memory
  125. //deallocateOSMemory(container.allocator.originalBaseMemory); //heap memory
  126. //
  127. //for (auto &i : container.bonusAllocators)
  128. //{
  129. // deallocateOSMemory(i.originalBaseMemory);
  130. //}
  131. }
  132. pika::containerId_t pika::ContainerManager::createContainer
  133. (pika::ContainerInformation containerInformation,
  134. pika::LoadedDll &loadedDll, pika::LogManager &logManager, pika::pikaImgui::ImGuiIdsManager &imguiIDsManager,
  135. ConsoleWindow *consoleWindow,
  136. size_t memoryPos)
  137. {
  138. containerId_t id = ++idCounter;
  139. //not necessary if this is the only things that assigns ids.
  140. //if (runningContainers.find(id) != runningContainers.end())
  141. //{
  142. // logManager.log((std::string("Container id already exists: #") + std::to_string(id)).c_str(), pika::logError);
  143. // return false;
  144. //}
  145. //todo a create and destruct wrapper
  146. pika::RuntimeContainer container = {};
  147. pika::strlcpy(container.baseContainerName, containerInformation.containerName,
  148. sizeof(container.baseContainerName));
  149. if (!allocateContainerMemory(container, containerInformation, (void*)memoryPos))
  150. {
  151. logManager.log((std::string("Couldn't allocate memory for constructing container: #")
  152. + std::to_string(id)).c_str(), pika::logError);
  153. return 0;
  154. }
  155. if (containerInformation.containerStaticInfo.requestImguiFbo)
  156. {
  157. container.requestedContainerInfo.requestedFBO.createFramebuffer(400, 400); //todo resize small or sthing
  158. container.imguiWindowId = imguiIDsManager.getImguiIds();
  159. }
  160. loadedDll.bindAllocatorDllRealm(&container.allocator);
  161. //this calls the constructors (from the dll realm)
  162. if (!loadedDll.constructRuntimeContainer(container, containerInformation.containerName.c_str()))
  163. {
  164. loadedDll.resetAllocatorDllRealm();
  165. logManager.log((std::string("Couldn't construct container: #") + std::to_string(id)).c_str(), pika::logError);
  166. freeContainerStuff(container);
  167. container.requestedContainerInfo.requestedFBO.deleteFramebuffer();
  168. return 0;
  169. }
  170. loadedDll.resetAllocatorDllRealm();
  171. #pragma region setup requested container info
  172. container.requestedContainerInfo.mainAllocator = &container.allocator;
  173. container.requestedContainerInfo.bonusAllocators = &container.bonusAllocators;
  174. container.requestedContainerInfo.requestedImguiIds =
  175. imguiIDsManager.getImguiIds(containerInformation.containerStaticInfo.requestImguiIds);
  176. container.requestedContainerInfo.imguiTotalRequestedIds = containerInformation.containerStaticInfo.requestImguiIds;
  177. container.requestedContainerInfo.consoleWindow = consoleWindow;
  178. #pragma endregion
  179. loadedDll.bindAllocatorDllRealm(&container.allocator);
  180. container.pointer->create(container.requestedContainerInfo); //this calls create() (from the dll realm)
  181. loadedDll.resetAllocatorDllRealm();//sets the global allocator back to standard (used for runtime realm)
  182. runningContainers[id] = container;
  183. logManager.log(("Created container: " + std::string(container.baseContainerName) ).c_str());
  184. return id;
  185. }
  186. void pika::ContainerManager::init()
  187. {
  188. }
  189. void pika::ContainerManager::update(pika::LoadedDll &loadedDll, pika::PikaWindow &window,
  190. pika::LogManager &logs, pika::pikaImgui::ImGuiIdsManager &imguiIdManager, pika::ConsoleWindow *consoleWindow)
  191. {
  192. PIKA_DEVELOPMENT_ONLY_ASSERT(loadedDll.dllHand != 0, "dll not loaded when trying to update containers");
  193. #pragma region reload dll
  194. //todo try to recover from a failed load
  195. if (loadedDll.shouldReloadDll())
  196. {
  197. reloadDll(loadedDll, window, logs); //todo return 0 on fail
  198. //todo mark shouldCallReaload or just call reload
  199. }
  200. #pragma endregion
  201. #pragma region running containers
  202. for (auto &c : runningContainers)
  203. {
  204. c.second.requestedContainerInfo.consoleWindow = consoleWindow;
  205. if (c.second.flags.status == pika::RuntimeContainer::FLAGS::STATUS_RUNNING
  206. ||
  207. c.second.flags.status == pika::RuntimeContainer::FLAGS::STATUS_BEING_RECORDED
  208. ||
  209. c.second.flags.status == pika::RuntimeContainer::FLAGS::STATUS_BEING_PLAYBACK
  210. )
  211. {
  212. PIKA_DEVELOPMENT_ONLY_ASSERT(
  213. (c.second.requestedContainerInfo.requestedFBO.fbo == 0 &&
  214. c.second.imguiWindowId == 0) ||
  215. (
  216. c.second.requestedContainerInfo.requestedFBO.fbo != 0 &&
  217. c.second.imguiWindowId != 0), "we have a fbo but no imguiwindow id"
  218. );
  219. auto windowInput = window.input;
  220. #if PIKA_DEVELOPMENT
  221. if (c.second.flags.status == pika::RuntimeContainer::FLAGS::STATUS_BEING_RECORDED)
  222. {
  223. if (!makeRecordingStep(c.first, logs, windowInput))
  224. {
  225. c.second.flags.status = pika::RuntimeContainer::FLAGS::STATUS_RUNNING;
  226. logs.log((std::string("Stopped container recording because we couldn't oppen file")
  227. + std::to_string(c.first)).c_str(),
  228. pika::logError);
  229. }
  230. }
  231. if (c.second.flags.status == pika::RuntimeContainer::FLAGS::STATUS_BEING_PLAYBACK)
  232. {
  233. pika::Input readInput = {};
  234. std::string fileName = c.second.flags.recordingName;
  235. fileName += ".recording";
  236. fileName = PIKA_ENGINE_RESOURCES_PATH + fileName;
  237. auto s = pika::getFileSize(fileName.c_str());
  238. if (c.second.flags.frameNumber * sizeof(pika::Input) >= s && s != 0)
  239. {
  240. //todo optional logs here
  241. if (!setSnapshotToContainer(c.first, c.second.flags.recordingName, logs, imguiIdManager))
  242. {
  243. logs.log((std::string("Stopped container playback because we couldn't assign it's snapshot on frame 0")
  244. + std::to_string(c.first)).c_str(),
  245. pika::logError);
  246. c.second.flags.status = pika::RuntimeContainer::FLAGS::STATUS_RUNNING;
  247. goto endContainerErrorChecking;
  248. }
  249. c.second.flags.frameNumber = 0;
  250. }
  251. if (s == 0)
  252. {
  253. logs.log((std::string("Stopped container playback because we couldn't oppen file or its content is empty")
  254. + std::to_string(c.first)).c_str(),
  255. pika::logError);
  256. c.second.flags.status = pika::RuntimeContainer::FLAGS::STATUS_RUNNING;
  257. }
  258. else if (s % sizeof(pika::Input) != 0)
  259. {
  260. logs.log((std::string("Stopped container playback because the file content is corrupt")
  261. + std::to_string(c.first)).c_str(),
  262. pika::logError);
  263. c.second.flags.status = pika::RuntimeContainer::FLAGS::STATUS_RUNNING;
  264. }
  265. if (!pika::readEntireFile(fileName.c_str(), &readInput, sizeof(pika::Input), sizeof(pika::Input) * c.second.flags.frameNumber))
  266. {
  267. logs.log((std::string("Stopped container playback because we couldn't oppen file")
  268. + std::to_string(c.first)).c_str(),
  269. pika::logError);
  270. c.second.flags.status = pika::RuntimeContainer::FLAGS::STATUS_RUNNING;
  271. }
  272. else
  273. {
  274. windowInput = readInput;
  275. c.second.flags.frameNumber++;
  276. }
  277. }
  278. #endif
  279. endContainerErrorChecking:
  280. #if PIKA_PRODUCTION
  281. constexpr bool isProduction = 1;
  282. #else
  283. constexpr bool isProduction = 0;
  284. #endif
  285. auto callUpdate = [&](pika::WindowState &windowState)
  286. {
  287. auto t1 = std::chrono::high_resolution_clock::now();
  288. loadedDll.bindAllocatorDllRealm(&c.second.allocator);
  289. c.second.pointer->update(windowInput, windowState, c.second.requestedContainerInfo);
  290. loadedDll.resetAllocatorDllRealm();
  291. auto t2 = std::chrono::high_resolution_clock::now();
  292. auto milliseconds = (std::chrono::duration_cast<std::chrono::microseconds>(t2 - t1)).count()/1000.f;
  293. c.second.frameTimer += milliseconds;
  294. c.second.frameCounter++;
  295. if (c.second.frameCounter >= 100)
  296. {
  297. c.second.currentMs = c.second.frameTimer/100.f;
  298. c.second.frameTimer = 0;
  299. c.second.frameCounter = 0;
  300. }
  301. };
  302. if (c.second.imguiWindowId && !isProduction)
  303. {
  304. #pragma region imguiwindow
  305. ImGui::PushID(c.second.imguiWindowId);
  306. ImGui::PushStyleColor(ImGuiCol_WindowBg, ImVec4(0.f, 0.f, 0.f, 1.0f));
  307. ImGui::SetNextWindowSize({200,200}, ImGuiCond_Once);
  308. ImGui::Begin( (std::string("gameplay window id: ") + std::to_string(c.first)).c_str(),
  309. 0, ImGuiWindowFlags_NoScrollbar | ImGuiWindowFlags_NoScrollWithMouse);
  310. //mouse pos and focus
  311. if(c.second.flags.status != pika::RuntimeContainer::FLAGS::STATUS_BEING_PLAYBACK)
  312. {
  313. auto windowPos = ImGui::GetWindowPos();
  314. ImVec2 globalMousePos = {};
  315. {
  316. ImGuiContext *g = ImGui::GetCurrentContext();
  317. globalMousePos = g->IO.MousePos;
  318. }
  319. windowInput.mouseX = globalMousePos.x;
  320. windowInput.mouseY = globalMousePos.y;
  321. ImVec2 vMin = ImGui::GetWindowContentRegionMin();
  322. windowInput.mouseX -= windowPos.x + vMin.x;
  323. windowInput.mouseY -= windowPos.y + vMin.y;
  324. //https://github.com/ocornut/imgui/issues/5882
  325. ImGuiViewport *viewPort = ImGui::GetWindowViewport();
  326. auto io = ImGui::GetIO();
  327. windowInput.hasFocus = viewPort->PlatformUserData && ImGui::GetPlatformIO().Platform_GetWindowFocus(viewPort) && !io.AppFocusLost;
  328. //windowInput.hasFocus = windowInput.hasFocus && !io.AppFocusLost;
  329. }
  330. auto s = ImGui::GetContentRegionMax();
  331. ImGui::Image((void *)c.second.requestedContainerInfo.requestedFBO.texture, s, {0, 1}, {1, 0},
  332. {1,1,1,1}, {0,0,0,1});
  333. ImGui::End();
  334. ImGui::PopStyleColor();
  335. ImGui::PopID();
  336. #pragma endregion
  337. auto windowState = window.windowState;
  338. windowState.w = s.x;
  339. windowState.h = s.y;
  340. c.second.requestedContainerInfo.requestedFBO.resizeFramebuffer(windowState.w, windowState.h);
  341. glBindFramebuffer(GL_FRAMEBUFFER, c.second.requestedContainerInfo.requestedFBO.fbo);
  342. callUpdate(windowState);
  343. glBindFramebuffer(GL_FRAMEBUFFER, 0);
  344. }
  345. else
  346. {
  347. callUpdate(window.windowState);
  348. }
  349. }
  350. else //on pause
  351. {
  352. //still keep it running on the same frame mabe
  353. }
  354. }
  355. #pragma endregion
  356. }
  357. void pika::ContainerManager::reloadDll(pika::LoadedDll &loadedDll, pika::PikaWindow &window, pika::LogManager &logs)
  358. {
  359. std::this_thread::sleep_for(std::chrono::milliseconds(200)); // make sure that the compiler had enough time
  360. //to get hold onto the dll
  361. //pika::LoadedDll newDll;
  362. //if (!newDll.tryToloadDllUntillPossible(loadedDll.id + 1, logs, std::chrono::seconds(1)))
  363. //{
  364. // logs.log("Dll reload attemp failed", pika::logWarning);
  365. // newDll.unloadDll();
  366. // return;
  367. //}
  368. //else
  369. //{
  370. // newDll.unloadDll();
  371. //}
  372. //std::this_thread::sleep_for(std::chrono::milliseconds(10)); // make sure that the dll is unloaded
  373. auto oldContainerInfo = loadedDll.containerInfo;
  374. if (!loadedDll.tryToloadDllUntillPossible(loadedDll.id, logs, std::chrono::seconds(5)))
  375. {
  376. logs.log("Couldn't reloaded dll", pika::logWarning);
  377. return;
  378. }
  379. //todo pospone dll reloading
  380. std::unordered_map<std::string, pika::ContainerInformation> containerNames;
  381. for (auto &c : loadedDll.containerInfo)
  382. {
  383. containerNames[c.containerName] = c;
  384. }
  385. std::unordered_map<std::string, pika::ContainerInformation> oldContainerNames;
  386. for (auto &c : oldContainerInfo)
  387. {
  388. oldContainerNames[c.containerName] = c;
  389. }
  390. //clear containers that dissapeared
  391. {
  392. std::vector<pika::containerId_t> containersToClean;
  393. for (auto &i : runningContainers)
  394. {
  395. if (containerNames.find(i.second.baseContainerName) ==
  396. containerNames.end())
  397. {
  398. std::string l = "Killed container because it does not exist anymore in dll: " +
  399. std::string(i.second.baseContainerName)
  400. + " #" + std::to_string(i.first);
  401. logs.log(l.c_str(), pika::logError);
  402. containersToClean.push_back(i.first);
  403. }
  404. }
  405. for (auto i : containersToClean)
  406. {
  407. forceTerminateContainer(i, loadedDll, logs);
  408. }
  409. }
  410. //clear containers that changed static info
  411. {
  412. std::vector<pika::containerId_t> containersToClean;
  413. for (auto &i : runningContainers)
  414. {
  415. auto &newContainer = containerNames[i.second.baseContainerName];
  416. auto &oldContainer = oldContainerNames[i.second.baseContainerName];
  417. if (newContainer != oldContainer)
  418. {
  419. std::string l = "Killed container because its static container info\nhas changed: "
  420. + std::string(i.second.baseContainerName)
  421. + " #" + std::to_string(i.first);
  422. logs.log(l.c_str(), pika::logError);
  423. containersToClean.push_back(i.first);
  424. }
  425. }
  426. for (auto i : containersToClean)
  427. {
  428. forceTerminateContainer(i, loadedDll, logs);
  429. }
  430. }
  431. loadedDll.gameplayReload_(window.context);
  432. logs.log("Reloaded dll");
  433. }
  434. bool pika::ContainerManager::destroyContainer(containerId_t id, pika::LoadedDll &loadedDll,
  435. pika::LogManager &logManager)
  436. {
  437. PIKA_DEVELOPMENT_ONLY_ASSERT(loadedDll.dllHand != 0, "dll not loaded when trying to destroy container");
  438. auto c = runningContainers.find(id);
  439. if (c == runningContainers.end())
  440. {
  441. logManager.log((std::string("Couldn't find container for destruction: #") + std::to_string(id)).c_str(),
  442. pika::logError);
  443. return false;
  444. }
  445. auto name = c->second.baseContainerName;
  446. loadedDll.bindAllocatorDllRealm(&c->second.allocator);
  447. loadedDll.destructContainer_(&(c->second.pointer), &c->second.arena);
  448. loadedDll.resetAllocatorDllRealm();
  449. freeContainerStuff(c->second);
  450. c->second.requestedContainerInfo.requestedFBO.deleteFramebuffer();
  451. runningContainers.erase(c);
  452. logManager.log((std::string("Destroyed continer: ") + name + " #" + std::to_string(id)).c_str());
  453. return true;
  454. }
  455. //todo remove some of this functions in production
  456. //snapshot file format:
  457. //
  458. // binary
  459. //
  460. // pika::RuntimeContainer
  461. //
  462. // static memory
  463. //
  464. // heap memory
  465. //
  466. //
  467. bool pika::ContainerManager::makeSnapshot(containerId_t id, pika::LogManager &logManager, const char *fileName)
  468. {
  469. auto c = runningContainers.find(id);
  470. if (c == runningContainers.end())
  471. {
  472. logManager.log((std::string("Couldn't find container for making snapshot: #") + std::to_string(id)).c_str(),
  473. pika::logError);
  474. return false;
  475. }
  476. std::string filePath = PIKA_ENGINE_RESOURCES_PATH;
  477. filePath += fileName;
  478. filePath += ".snapshot";
  479. if(!pika::writeEntireFile(filePath.c_str(), &c->second, sizeof(c->second)))
  480. {
  481. logManager.log(("Couldn't write to file for making snapshot: " + filePath).c_str(),
  482. pika::logError);
  483. return false;
  484. }
  485. if (!pika::appendToFile(filePath.c_str(),
  486. c->second.getBaseAdress(), c->second.totalSize))
  487. {
  488. pika::deleteFile(filePath.c_str());
  489. logManager.log(("Couldn't write to file for making snapshot: " + filePath).c_str(),
  490. pika::logError);
  491. return false;
  492. }
  493. //if (!pika::appendToFile(filePath.c_str(),
  494. // c->second.arena.containerStructMemory.block, c->second.arena.containerStructMemory.size))
  495. //{
  496. // pika::deleteFile(filePath.c_str());
  497. // logManager.log(("Couldn't write to file for making snapshot: " + filePath).c_str(),
  498. // pika::logError);
  499. // return false;
  500. //}
  501. //
  502. //if (!pika::appendToFile(filePath.c_str(),
  503. // c->second.allocator.originalBaseMemory, c->second.allocatorSize))
  504. //{
  505. // pika::deleteFile(filePath.c_str());
  506. // logManager.log(("Couldn't write to file for making snapshot: " + filePath).c_str(),
  507. // pika::logError);
  508. // return false;
  509. //}
  510. return true;
  511. }
  512. bool pika::ContainerManager::startRecordingContainer(containerId_t id, pika::LogManager &logManager, const char *fileName)
  513. {
  514. auto c = runningContainers.find(id);
  515. if (c == runningContainers.end())
  516. {
  517. logManager.log((std::string("Couldn't find container for making recording: #") + std::to_string(id)).c_str(),
  518. pika::logError);
  519. return false;
  520. }
  521. if(c->second.flags.status != pika::RuntimeContainer::FLAGS::STATUS_RUNNING)
  522. {
  523. logManager.log((std::string("Trying to record a container that is not running (on status): #") + std::to_string(id)).c_str(),
  524. pika::logError);
  525. return false;
  526. }
  527. std::string filePath = PIKA_ENGINE_RESOURCES_PATH;
  528. filePath += fileName;
  529. filePath += ".recording";
  530. if (filePath.size() > sizeof(c->second.flags.recordingName) - 1)
  531. {
  532. logManager.log((std::string("File path too big (on trying to record)") + std::to_string(id)).c_str(),
  533. pika::logError);
  534. return 0;
  535. }
  536. if (!makeSnapshot(id, logManager, fileName))
  537. {
  538. logManager.log((std::string("Couldn't make snapshot for starting recording") + std::to_string(id)).c_str(),
  539. pika::logError);
  540. return 0;
  541. }
  542. //cear file
  543. {
  544. std::ofstream record(filePath);
  545. if (!record.is_open())
  546. {
  547. logManager.log((std::string("Couldn't open recording file for starting recording") + std::to_string(id)).c_str(),
  548. pika::logError);
  549. return 0;
  550. }
  551. record.close();
  552. }
  553. c->second.flags.status = pika::RuntimeContainer::FLAGS::STATUS_BEING_RECORDED;
  554. pika::strlcpy(c->second.flags.recordingName, filePath, sizeof(c->second.flags.recordingName));
  555. }
  556. bool pika::ContainerManager::stopRecordingContainer(containerId_t id, pika::LogManager &logManager)
  557. {
  558. auto c = runningContainers.find(id);
  559. if (c == runningContainers.end())
  560. {
  561. logManager.log((std::string("Couldn't find container for stopping recording: #") + std::to_string(id)).c_str(),
  562. pika::logError);
  563. return false;
  564. }
  565. c->second.flags.status = pika::RuntimeContainer::FLAGS::STATUS_RUNNING;
  566. }
  567. bool pika::ContainerManager::makeRecordingStep(containerId_t id, pika::LogManager &logManager,
  568. pika::Input &input)
  569. {
  570. auto c = runningContainers.find(id);
  571. if (c == runningContainers.end())
  572. {
  573. logManager.log((std::string("Couldn't find container for making recording step: #") + std::to_string(id)).c_str(),
  574. pika::logError);
  575. return false;
  576. }
  577. if (!pika::appendToFile(c->second.flags.recordingName, &input, sizeof(input)))
  578. {
  579. logManager.log((std::string("Couldn't append to file for recording container") + std::to_string(id)).c_str(),
  580. pika::logError);
  581. return false;
  582. }
  583. return true;
  584. }
  585. bool pika::ContainerManager::forceTerminateContainer(containerId_t id, pika::LoadedDll &loadedDll, pika::LogManager &logManager)
  586. {
  587. PIKA_DEVELOPMENT_ONLY_ASSERT(loadedDll.dllHand != 0, "dll not loaded when trying to destroy container");
  588. auto c = runningContainers.find(id);
  589. if (c == runningContainers.end())
  590. {
  591. logManager.log((std::string("Couldn't find container for destruction: #") + std::to_string(id)).c_str(),
  592. pika::logError);
  593. return false;
  594. }
  595. auto name = c->second.baseContainerName;
  596. freeContainerStuff(c->second);
  597. runningContainers.erase(c);
  598. logManager.log((std::string("Force terminated continer: ") + name + " #" + std::to_string(id)).c_str());
  599. c->second.requestedContainerInfo.requestedFBO.deleteFramebuffer();
  600. return true;
  601. }
  602. void pika::ContainerManager::destroyAllContainers(pika::LoadedDll &loadedDll,
  603. pika::LogManager &logManager)
  604. {
  605. std::vector < pika::containerId_t> containersId;
  606. containersId.reserve(runningContainers.size());
  607. for (auto &c : runningContainers)
  608. {
  609. containersId.push_back(c.first);
  610. }
  611. for (auto i : containersId)
  612. {
  613. destroyContainer(i, loadedDll, logManager);
  614. }
  615. }
  616. #ifdef PIKA_PRODUCTION
  617. void *pika::ContainerManager::allocateOSMemory(size_t size, void *baseAdress)
  618. {
  619. PIKA_PERMA_ASSERT(baseAdress == nullptr, "can't allocate fixed memory in production");
  620. return malloc(size);
  621. }
  622. void pika::ContainerManager::deallocateOSMemory(void *baseAdress)
  623. {
  624. free(baseAdress);
  625. }
  626. #else
  627. #include <Windows.h>
  628. void *pika::ContainerManager::allocateOSMemory(size_t size, void *baseAdress)
  629. {
  630. return VirtualAlloc(baseAdress, size, MEM_RESERVE | MEM_COMMIT, PAGE_READWRITE);
  631. }
  632. void pika::ContainerManager::deallocateOSMemory(void *baseAdress)
  633. {
  634. VirtualFree(baseAdress, 0, MEM_RELEASE);
  635. }
  636. #endif
  637. std::vector<std::string> pika::getAvailableSnapshots(pika::RuntimeContainer &info)
  638. {
  639. std::vector<std::string> files;
  640. auto curDir = std::filesystem::directory_iterator(PIKA_ENGINE_RESOURCES_PATH);
  641. for (const auto &iter : curDir)
  642. {
  643. if (std::filesystem::is_regular_file(iter)
  644. && iter.path().extension() == ".snapshot"
  645. )
  646. {
  647. if (pika::checkIfSnapshotIsCompatible(info, iter.path().stem().string().c_str()))
  648. {
  649. files.push_back(iter.path().stem().string());
  650. }
  651. }
  652. }
  653. return files;
  654. }
  655. std::vector<std::string> pika::getAvailableRecordings(pika::RuntimeContainer &info)
  656. {
  657. auto snapshots = getAvailableSnapshots(info);
  658. std::vector<std::string> returnVec;
  659. returnVec.reserve(snapshots.size());
  660. auto curDir = std::filesystem::directory_iterator(PIKA_ENGINE_RESOURCES_PATH);
  661. for (const auto &iter : curDir)
  662. {
  663. if (std::filesystem::is_regular_file(iter)
  664. && iter.path().extension() == ".recording"
  665. )
  666. {
  667. if (std::find(snapshots.begin(), snapshots.end(), iter.path().stem().string()) != snapshots.end())
  668. {
  669. returnVec.push_back(iter.path().stem().string());
  670. }
  671. }
  672. }
  673. return returnVec;
  674. }
  675. std::vector<std::string> pika::getAvailableSnapshotsAnyMemoryPosition(pika::RuntimeContainer &info)
  676. {
  677. std::vector<std::string> files;
  678. auto curDir = std::filesystem::directory_iterator(PIKA_ENGINE_RESOURCES_PATH);
  679. for (const auto &iter : curDir)
  680. {
  681. if (std::filesystem::is_regular_file(iter)
  682. && iter.path().extension() == ".snapshot"
  683. )
  684. {
  685. if (pika::checkIfSnapshotIsCompatibleAnyMemoryPosition(info, iter.path().stem().string().c_str()))
  686. {
  687. files.push_back(iter.path().stem().string());
  688. }
  689. }
  690. }
  691. return files;
  692. }
  693. std::vector<std::string> pika::getAvailableRecordingAnyMemoryPosition(pika::RuntimeContainer &info)
  694. {
  695. auto snapshots = getAvailableSnapshotsAnyMemoryPosition(info);
  696. std::vector<std::string> returnVec;
  697. returnVec.reserve(snapshots.size());
  698. auto curDir = std::filesystem::directory_iterator(PIKA_ENGINE_RESOURCES_PATH);
  699. for (const auto &iter : curDir)
  700. {
  701. if (std::filesystem::is_regular_file(iter)
  702. && iter.path().extension() == ".recording"
  703. )
  704. {
  705. if (std::find(snapshots.begin(), snapshots.end(), iter.path().stem().string()) != snapshots.end())
  706. {
  707. returnVec.push_back(iter.path().stem().string());
  708. }
  709. }
  710. }
  711. return returnVec;
  712. }
  713. std::vector<std::string> pika::getAvailableSnapshotsAnyMemoryPosition(pika::ContainerInformation &info)
  714. {
  715. std::vector<std::string> files;
  716. auto curDir = std::filesystem::directory_iterator(PIKA_ENGINE_RESOURCES_PATH);
  717. for (const auto &iter : curDir)
  718. {
  719. if (std::filesystem::is_regular_file(iter)
  720. && iter.path().extension() == ".snapshot"
  721. )
  722. {
  723. if (pika::checkIfSnapshotIsCompatibleAnyMemoryPosition(info, iter.path().stem().string().c_str()))
  724. {
  725. files.push_back(iter.path().stem().string());
  726. }
  727. }
  728. }
  729. return files;
  730. }
  731. std::vector<std::string> pika::getAvailableRecordingsAnyMemoryPosition(pika::ContainerInformation &info)
  732. {
  733. auto snapshots = getAvailableSnapshotsAnyMemoryPosition(info);
  734. std::vector<std::string> returnVec;
  735. returnVec.reserve(snapshots.size());
  736. auto curDir = std::filesystem::directory_iterator(PIKA_ENGINE_RESOURCES_PATH);
  737. for (const auto &iter : curDir)
  738. {
  739. if (std::filesystem::is_regular_file(iter)
  740. && iter.path().extension() == ".recording"
  741. )
  742. {
  743. if (std::find(snapshots.begin(), snapshots.end(), iter.path().stem().string()) != snapshots.end())
  744. {
  745. returnVec.push_back(iter.path().stem().string());
  746. }
  747. }
  748. }
  749. return returnVec;
  750. }
  751. bool pika::checkIfSnapshotIsCompatible(pika::RuntimeContainer &info, const char *snapshotName)
  752. {
  753. std::string file = PIKA_ENGINE_RESOURCES_PATH;
  754. file += snapshotName;
  755. file += ".snapshot";
  756. pika::RuntimeContainer loadedInfo = {};
  757. auto s = pika::readEntireFile(file.c_str(), &loadedInfo, sizeof(loadedInfo));
  758. if (s != sizeof(loadedInfo))
  759. {
  760. return 0;
  761. }
  762. //todo mabe a method here?
  763. if (loadedInfo.allocator.originalBaseMemory != info.allocator.originalBaseMemory)
  764. {
  765. return false;
  766. }
  767. if (loadedInfo.requestedContainerInfo.imguiTotalRequestedIds != info.requestedContainerInfo.imguiTotalRequestedIds)
  768. {
  769. return false;
  770. }
  771. if (loadedInfo.arena.containerStructMemory.block != info.arena.containerStructMemory.block)
  772. {
  773. return false;
  774. }
  775. if (loadedInfo.arena.containerStructMemory.size != info.arena.containerStructMemory.size)
  776. {
  777. return false;
  778. }
  779. if (std::strcmp(loadedInfo.baseContainerName, info.baseContainerName) != 0)
  780. {
  781. return false;
  782. }
  783. if (loadedInfo.requestedContainerInfo.bonusAllocators != info.requestedContainerInfo.bonusAllocators)
  784. {
  785. return false;
  786. }
  787. //check if user requested an imgui window
  788. if (
  789. !(
  790. (info.imguiWindowId == 0&&
  791. loadedInfo.imguiWindowId == 0
  792. )||
  793. (
  794. info.imguiWindowId != 0 &&
  795. loadedInfo.imguiWindowId != 0
  796. )
  797. )
  798. )
  799. {
  800. return false;
  801. }
  802. if (loadedInfo.totalSize != info.totalSize)
  803. {
  804. return false;
  805. }
  806. return true;
  807. }
  808. bool pika::checkIfSnapshotIsCompatibleAnyMemoryPosition(pika::RuntimeContainer &info, const char *snapshotName)
  809. {
  810. std::string file = PIKA_ENGINE_RESOURCES_PATH;
  811. file += snapshotName;
  812. file += ".snapshot";
  813. pika::RuntimeContainer loadedInfo = {};
  814. auto s = pika::readEntireFile(file.c_str(), &loadedInfo, sizeof(loadedInfo));
  815. if (s != sizeof(loadedInfo))
  816. {
  817. return 0;
  818. }
  819. if (loadedInfo.arena.containerStructMemory.size != info.arena.containerStructMemory.size)
  820. {
  821. return false;
  822. }
  823. if (std::strcmp(loadedInfo.baseContainerName, info.baseContainerName) != 0)
  824. {
  825. return false;
  826. }
  827. if (loadedInfo.requestedContainerInfo.bonusAllocators != info.requestedContainerInfo.bonusAllocators)
  828. {
  829. return false;
  830. }
  831. //check if user requested an imgui window
  832. if (
  833. !(
  834. (info.imguiWindowId == 0 &&
  835. loadedInfo.imguiWindowId == 0
  836. ) ||
  837. (
  838. info.imguiWindowId != 0 &&
  839. loadedInfo.imguiWindowId != 0
  840. )
  841. )
  842. )
  843. {
  844. return false;
  845. }
  846. if (loadedInfo.totalSize != info.totalSize)
  847. {
  848. return false;
  849. }
  850. return true;
  851. }
  852. bool pika::checkIfSnapshotIsCompatibleAnyMemoryPosition(pika::ContainerInformation &info, const char *snapshotName)
  853. {
  854. std::string file = PIKA_ENGINE_RESOURCES_PATH;
  855. file += snapshotName;
  856. file += ".snapshot";
  857. pika::RuntimeContainer loadedInfo = {};
  858. auto s = pika::readEntireFile(file.c_str(), &loadedInfo, sizeof(loadedInfo));
  859. if (s != sizeof(loadedInfo))
  860. {
  861. return 0;
  862. }
  863. if (loadedInfo.arena.containerStructMemory.size != info.containerStructBaseSize)
  864. {
  865. return false;
  866. }
  867. if (std::strcmp(loadedInfo.baseContainerName, info.containerName.c_str()) != 0)
  868. {
  869. return false;
  870. }
  871. if (loadedInfo.bonusAllocators.size() != info.containerStaticInfo.bonusAllocators.size())
  872. {
  873. return false;
  874. }
  875. for (int i = 0; i < loadedInfo.bonusAllocators.size(); i++)
  876. {
  877. if (loadedInfo.bonusAllocators[i].heapSize != info.containerStaticInfo.bonusAllocators[i]) //todo this doesn't seem to work
  878. {
  879. return false;
  880. }
  881. }
  882. //check if user requested an imgui window
  883. if (
  884. !(
  885. (info.containerStaticInfo.requestImguiFbo == false &&
  886. loadedInfo.imguiWindowId == 0
  887. ) ||
  888. (
  889. info.containerStaticInfo.requestImguiFbo == true &&
  890. loadedInfo.imguiWindowId != 0
  891. )
  892. )
  893. )
  894. {
  895. return false;
  896. }
  897. if (loadedInfo.allocator.heapSize != info.containerStaticInfo.defaultHeapMemorySize)
  898. {
  899. return false;
  900. }
  901. return true;
  902. }
  903. void *pika::getSnapshotMemoryPosition(const char *snapshotName)
  904. {
  905. std::string file = PIKA_ENGINE_RESOURCES_PATH;
  906. file += snapshotName;
  907. file += ".snapshot";
  908. pika::RuntimeContainer loadedInfo = {};
  909. auto s = pika::readEntireFile(file.c_str(), &loadedInfo, sizeof(loadedInfo));
  910. if (s != sizeof(loadedInfo))
  911. {
  912. return nullptr;
  913. }
  914. return loadedInfo.getBaseAdress();
  915. }