From f3318e0ed12ebb29fac0a019874f3de00fe02574 Mon Sep 17 00:00:00 2001
From: William JCM <w.jcm59@gmail.com>
Date: Sat, 28 Aug 2021 20:22:04 +0200
Subject: [PATCH] SaveTool: add safety measures in drawMassViewer().

---
 src/SaveTool/SaveTool_MassViewer.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/SaveTool/SaveTool_MassViewer.cpp b/src/SaveTool/SaveTool_MassViewer.cpp
index 7ad9212..e284bba 100644
--- a/src/SaveTool/SaveTool_MassViewer.cpp
+++ b/src/SaveTool/SaveTool_MassViewer.cpp
@@ -19,6 +19,12 @@
 #include "SaveTool.h"
 
 void SaveTool::drawMassViewer() {
+    if(!_currentMass || _currentMass->state() != Mass::State::Valid) {
+        _currentMass = nullptr;
+        _uiState = UiState::MainManager;
+        return;
+    }
+
     ImGui::SetNextWindowPos({0.0f, ImGui::GetItemRectSize().y}, ImGuiCond_Always);
     ImGui::SetNextWindowSize({Float(windowSize().x()), Float(windowSize().y()) - ImGui::GetItemRectSize().y},
                              ImGuiCond_Always);