#3507 Undo and Redo Meshes

[creaVtk.git] / lib / creaVtk / MeshManagerModel.cpp

diff --git a/lib/creaVtk/MeshManagerModel.cpp b/lib/creaVtk/MeshManagerModel.cpp
index 677727c6057a054b06ecb097298ba36977c1f78a..c03e9e8b2268567199c8b62237d72bd8fb469b35 100644 (file)
--- a/lib/creaVtk/MeshManagerModel.cpp
+++ b/lib/creaVtk/MeshManagerModel.cpp
@@ -27,6 +27,82 @@
 
 #include "MeshManagerModel.h"
 
+template <class T>
+HistoryHandler<T>::HistoryHandler(int maxElements)
+{
+       this->maxElements = maxElements;
+}
+template <class T>
+HistoryHandler<T>::~HistoryHandler()
+{
+       CleanHistory();
+}
+
+template <class T>
+void HistoryHandler<T>::CleanHistory()
+{
+       for (T* element : redoStack)
+       {
+               delete element;
+       }
+       for (T* element : undoStack)
+       {
+               delete element;
+       }
+}
+
+template <class T>
+T* HistoryHandler<T>::Undo()
+{
+       if(!undoStack.empty() && undoStack.size()>1)
+       {
+               auto lastElem = undoStack.back();
+               undoStack.pop_back();
+               redoStack.push_back(lastElem);
+               return undoStack.back();
+       }
+       return NULL;
+}
+
+template <class T>
+T* HistoryHandler<T>::Redo()
+{
+       if(!redoStack.empty())
+       {
+               auto lastElem = redoStack.back();
+               redoStack.pop_back();
+               undoStack.push_back(lastElem);
+               return lastElem;
+       }
+       return NULL;
+}
+
+template <class T>
+void HistoryHandler<T>::Save(T* state)
+{
+       undoStack.push_back(state);
+       if(!redoStack.empty())
+       {
+               for (T* element : redoStack)
+               {
+                       delete element;
+               } 
+               redoStack.clear();
+       }
+}
+
+template <class T>
+int HistoryHandler<T>::UndoSize()
+{
+       return undoStack.size();
+}
+
+template <class T>
+int HistoryHandler<T>::RedoSize()
+{
+       return redoStack.size();
+}
+
 ////////////////////
 /////////////////////// MESH MODEL
 ////////////////////
@@ -50,6 +126,20 @@ MeshModel::MeshModel(vtkPolyData* mesh, int id)
        }
 }
 
+MeshModel::MeshModel(MeshModel* meshModel)
+{
+       _meshBase = NULL;
+    _meshTemp = NULL;
+       if(meshModel->GetMeshBase() != NULL)
+       {
+               _meshBase = vtkPolyData::New();
+               _meshBase->DeepCopy(meshModel->GetMeshBase());
+               ResetMeshTemp_();
+       }
+       _meshId = meshModel->GetId();
+       _name = "mesh-" + std::to_string(meshModel->GetId());
+}
+
 MeshModel::~MeshModel(){
        if(_meshBase != NULL){
                _meshBase->Delete();
@@ -70,7 +160,11 @@ void MeshModel::ResetMeshTemp_()
         _meshTemp = vtkPolyData::New();
         _meshTemp->DeepCopy(_meshBase);
     } else {
-        _meshTemp=NULL;
+        if (_meshTemp!=NULL)
+        {
+            _meshTemp->Delete();
+            _meshTemp = NULL;
+        }
     }
 }
 
@@ -78,6 +172,9 @@ void MeshModel::SetMeshBase(vtkPolyData* mesh)
 {
     if (mesh!=NULL)
     {
+       if(_meshBase != NULL){
+               _meshBase->Delete();
+       }
         _meshBase = mesh;
         ResetMeshTemp_();
     } // if mesh
@@ -85,7 +182,17 @@ void MeshModel::SetMeshBase(vtkPolyData* mesh)
 
 void MeshModel::SetMeshMemoryMode(vtkPolyData* mesh)
 {
-    _meshBase = mesh;
+       //if(_meshBase != NULL)
+       //{
+       //      _meshBase->Delete();
+       //      _meshBase = NULL;
+       //}
+       //if (_meshTemp != NULL)
+    //{
+    //    _meshTemp->Delete();
+    //    _meshTemp = NULL;
+       //}
+       _meshBase = mesh;
 }
 
 void MeshModel::ResetMeshTemp()
@@ -104,12 +211,12 @@ void MeshModel::CopySetMeshBase(vtkPolyData* mesh)
 }
 
 
-vtkPolyData*  MeshModel::GetMeshBase()
+vtkPolyData* MeshModel::GetMeshBase()
 {
    return _meshBase;
 }
 
-vtkPolyData*  MeshModel::GetMeshTemp()
+vtkPolyData* MeshModel::GetMeshTemp()
 {
    return _meshTemp;
 }
@@ -130,10 +237,9 @@ std::string MeshModel::GetName()
 
 MeshManagerModel::MeshManagerModel()
 {
-       //MeshModel* firstMesh = new MeshModel();
-       //_meshes.push_back(firstMesh);
        currentMesh = 0;
        meshId = 0;
+       history = new HistoryHandler<ManagerState>(20);
 }
 
 MeshManagerModel::~MeshManagerModel()
@@ -145,14 +251,27 @@ void MeshManagerModel::RefreshOutputs(bool signalBox) // virtual
 {
 }
 
-int MeshManagerModel::GetNumberOfMeshes(){
+void MeshManagerModel::ResetHistory()
+{
+       history->CleanHistory();
+       Save();
+}
+
+void MeshManagerModel::UpdateMeshReference(vtkPolyData* mesh)
+{
+       _meshes.at(currentMesh)->SetMeshBase(mesh);
+       RefreshOutputs(true);
+}
+
+int MeshManagerModel::GetNumberOfMeshes()
+{
        return _meshes.size();
 }
 
-void MeshManagerModel::AddMesh_(vtkPolyData* mesh){
+void MeshManagerModel::AddMesh_(vtkPolyData* mesh)
+{
        if(mesh != NULL){
-               MeshModel *meshModel = new MeshModel(mesh, meshId);
-               _meshes.push_back(meshModel);
+               _meshes.push_back(std::make_shared<MeshModel>(mesh, meshId));
                meshId++;
        }
        else{
@@ -160,18 +279,20 @@ void MeshManagerModel::AddMesh_(vtkPolyData* mesh){
        }
 }
 
-void MeshManagerModel::AddMesh(vtkPolyData* mesh){
+void MeshManagerModel::AddMesh(vtkPolyData* mesh)
+{
        AddMesh_(mesh);
+       Save();
        RefreshOutputs(true);
 }
 
-void MeshManagerModel::AddMeshes_(std::vector<vtkPolyData*> meshList){
+void MeshManagerModel::AddMeshes_(std::vector<vtkPolyData*> meshList)
+{
        if(!meshList.empty())
        {
                MeshModel *meshModel;
                for(int i = 0; i < meshList.size(); i++){
-                       meshModel = new MeshModel(meshList[i], meshId);
-                       _meshes.push_back(meshModel);
+                       _meshes.push_back(std::make_shared<MeshModel>(meshList[i], meshId));
                        meshId++;
                }
        }else{
@@ -179,19 +300,23 @@ void MeshManagerModel::AddMeshes_(std::vector<vtkPolyData*> meshList){
        }
 }
 
-void MeshManagerModel::AddMeshes(std::vector<vtkPolyData*> meshList){
+void MeshManagerModel::AddMeshes(std::vector<vtkPolyData*> meshList)
+{
        AddMeshes_(meshList);
+       Save();
        RefreshOutputs(true);
 }
 
-void MeshManagerModel::AddEmptyMesh_(){
-       MeshModel *meshModel = new MeshModel(meshId);
-       _meshes.push_back(meshModel);
+void MeshManagerModel::AddEmptyMesh_()
+{
+       _meshes.push_back(std::make_shared<MeshModel>(meshId));
        meshId++;
 }
 
-void MeshManagerModel::AddEmptyMesh(){
+void MeshManagerModel::AddEmptyMesh()
+{
        AddEmptyMesh_();
+       Save();
        RefreshOutputs(true);
 }
 
@@ -199,26 +324,27 @@ void MeshManagerModel::InsertMeshesAtCurrent_(std::vector<vtkPolyData*> meshList
 {
        if(!meshList.empty())
        {
-               std::vector<MeshModel*> tmpVect;
+               std::vector<std::shared_ptr<MeshModel>> tmpVect;
                MeshModel *meshModel;
                for(int i = 0; i < meshList.size(); i++){
-                       meshModel = new MeshModel(meshList[i], meshId);
-                       tmpVect.push_back(meshModel);
+                       tmpVect.push_back(std::make_shared<MeshModel>(meshList[i], meshId));
                        meshId++;
                }
                _meshes.insert(_meshes.begin() + currentMesh, tmpVect.begin(), tmpVect.end());
        }else{
-               printf("PG MeshManagerModel::AddMeshes Empty list of meshes \n");
+               printf("PG MeshManagerModel::InsertMeshesAtCurrent Empty list of meshes \n");
        }
 }
 
 void MeshManagerModel::InsertMeshesAtCurrent(std::vector<vtkPolyData*> meshList)
 {
        InsertMeshesAtCurrent_(meshList);
+       Save();
        RefreshOutputs(true);
 }
 
-void MeshManagerModel::SelectMesh(int i) {
+void MeshManagerModel::SelectMesh(int i) 
+{
        if(i >= 0 && i < _meshes.size()){
                int prevCurrent = currentMesh;
                currentMesh = i;
@@ -231,7 +357,8 @@ void MeshManagerModel::SelectMesh(int i) {
        }
 }
 
-void MeshManagerModel::SelectMeshByName(std::string meshName) {
+void MeshManagerModel::SelectMeshByName(std::string meshName) 
+{
        if(!_meshes.empty()){
                bool found = false;
                for(int i = 0; i < _meshes.size() && !found; i++){
@@ -243,60 +370,77 @@ void MeshManagerModel::SelectMeshByName(std::string meshName) {
        }
 }
 
-void MeshManagerModel::DeleteMesh(int position){
+void MeshManagerModel::DeleteMesh_(int position)
+{
        if(position >= 0 && position < _meshes.size()){
-               delete _meshes.at(position);
                _meshes.erase(_meshes.begin() + position);
                currentMesh = currentMesh + (position <= currentMesh?-1:0);
                if(currentMesh < 0){
                        currentMesh = 0;
                }
-               RefreshOutputs(true);
        }
 }
 
-void MeshManagerModel::DeleteMeshByName(std::string meshName){
+void MeshManagerModel::DeleteMesh(int position)
+{
+       DeleteMesh_(position);
+       Save();
+       RefreshOutputs(true);
+}
+
+void MeshManagerModel::DeleteMeshByName(std::string meshName)
+{
        if(!_meshes.empty()){
                bool found = false;
                for(int i = 0; i < _meshes.size() && !found; i++){
                        if(_meshes.at(i)->GetName() == meshName){
                                found = true;
-                               delete _meshes.at(i);
-                               _meshes.erase(_meshes.begin() + i);
-                               currentMesh = currentMesh + (i <= currentMesh?-1:0);
-                               if(currentMesh < 0){
-                                       currentMesh = 0;
-                               }
+                               DeleteMesh_(i);
                                RefreshOutputs(true);
                        }
                }
        }
 }
 
-void MeshManagerModel::DeleteCurrentMesh(){
-       if(!_meshes.empty()){   
-               delete _meshes.at(currentMesh);
-               _meshes.erase(_meshes.begin() + currentMesh);
-               currentMesh--;
-               if(currentMesh < 0){
-                       currentMesh = 0;
-               }
+void MeshManagerModel::DeleteCurrentMesh()
+{
+       if(!_meshes.empty()){
+               DeleteMesh_(currentMesh);
+               Save();
                RefreshOutputs(true);
        }
 }
 
-void MeshManagerModel::DeleteAll(){
-       if(!_meshes.empty()){   
-               for (MeshModel* element : _meshes)
-               {
-                       delete element;
-               } 
+void MeshManagerModel::ReplaceMesh(std::vector<vtkPolyData*> meshList)
+{
+       if(GetNumberOfMeshes() >= 1)
+       {
+               DeleteMesh_(currentMesh);
+       }
+       InsertMeshesAtCurrent_(meshList);
+       Save();
+       RefreshOutputs(true);
+}
+
+void MeshManagerModel::DeleteAll_()
+{
+       if(!_meshes.empty())
+       {
+               currentMesh = 0;
                _meshes.clear();
                RefreshOutputs(true);
        }
 }
 
-void MeshManagerModel::NextMesh(){
+void MeshManagerModel::DeleteAll()
+{
+       DeleteAll_();
+       Save();
+       RefreshOutputs(true);
+}
+
+void MeshManagerModel::NextMesh()
+{
        currentMesh++;
        if(currentMesh >= _meshes.size()){
                currentMesh = _meshes.size()-1;
@@ -304,7 +448,8 @@ void MeshManagerModel::NextMesh(){
        RefreshOutputs(true);
 }
 
-void MeshManagerModel::PreviousMesh(){
+void MeshManagerModel::PreviousMesh()
+{
        currentMesh--;
        if(currentMesh < 0){
                currentMesh = 0;
@@ -312,7 +457,8 @@ void MeshManagerModel::PreviousMesh(){
        RefreshOutputs(true);
 }
 
-MeshModel* MeshManagerModel::GetMeshModel(){
+std::shared_ptr<MeshModel> MeshManagerModel::GetMeshModel()
+{
        return _meshes.at(currentMesh);
 }
 
@@ -339,44 +485,47 @@ vtkPolyData*  MeshManagerModel::GetMeshTemp()
 void MeshManagerModel::SetMeshBase(vtkPolyData* mesh)
 {
     if(!_meshes.empty()){
+        _meshes.at(currentMesh) = std::make_shared<MeshModel>(_meshes.at(currentMesh).get());
         _meshes.at(currentMesh)->SetMeshBase(mesh);
+        Save();
         RefreshOutputs(true);
     }else{
-       printf("PG MeshManagerModel::SetMeshMemoryMode Mesh vector is empty \n");
+       printf("PG MeshManagerModel::SetMeshBase Mesh vector is empty \n");
     }
 }
 
 void MeshManagerModel::SetMeshMemoryMode(vtkPolyData* mesh)
 {
-       //if(!_meshes.empty()){
-       if(_meshes.size() > 1){
-               DeleteAll();
+       if(_meshes.size() > 1)
+       {
+               DeleteAll_();
        }
-       if(_meshes.size() == 0){
+       if(_meshes.size() == 0)
+       {
                AddEmptyMesh_();
        }
        _meshes.at(currentMesh)->SetMeshMemoryMode(mesh);
        RefreshOutputs(true);
-    //}else{
-    // printf("PG MeshManagerModel::SetMeshMemoryMode Mesh vector is empty \n");
-    //}
 }
 
 void MeshManagerModel::ResetMeshTemp()
 {
-       if(!_meshes.empty()){
+       if(!_meshes.empty())
+       {
        _meshes.at(currentMesh)->ResetMeshTemp();
        RefreshOutputs(true);
     }else{
        printf("PG MeshManagerModel::ResetMeshTemp Mesh vector is empty \n");
     }
-    
 }
 
 void MeshManagerModel::CopySetMeshBase(vtkPolyData* mesh)
 {
-       if(!_meshes.empty()){
+       if(!_meshes.empty())
+       {
+               _meshes.at(currentMesh) = std::make_shared<MeshModel>(_meshes.at(currentMesh).get());
                _meshes.at(currentMesh)->CopySetMeshBase(mesh);
+               Save();
                RefreshOutputs(true);
        }
        else{
@@ -392,3 +541,59 @@ std::vector<std::string> MeshManagerModel::GetMeshNames()
        }
        return names;
 }
+
+void MeshManagerModel::Undo()
+{      
+       RestoreState(history->Undo());
+       RefreshOutputs(true);
+}
+
+void MeshManagerModel::Redo()
+{
+       RestoreState(history->Redo());
+       RefreshOutputs(true);
+}
+
+void MeshManagerModel::Save()
+{
+       history->Save(new ManagerState(_meshes, meshId, currentMesh));
+}
+
+void MeshManagerModel::RestoreState(ManagerState* state)
+{
+       if(state != NULL)
+       {       
+               _meshes = state->GetMeshes();
+               meshId = state->GetMeshId();
+               currentMesh = state->GetCurrentMesh();
+       }
+}
+
+//
+//Manager State
+//
+MeshManagerModel::ManagerState::ManagerState(std::vector<std::shared_ptr<MeshModel>> meshesToSave, int meshId, int currentMesh)
+{
+       savedMeshes = meshesToSave;
+       savedId = meshId;
+       savedCurrentMesh = currentMesh;
+}
+
+MeshManagerModel::ManagerState::~ManagerState()
+{
+}
+
+std::vector<std::shared_ptr<MeshModel>>& MeshManagerModel::ManagerState::GetMeshes()
+{
+       return savedMeshes;
+}
+
+int MeshManagerModel::ManagerState::GetMeshId()
+{
+       return savedId;
+}
+int MeshManagerModel::ManagerState::GetCurrentMesh()
+{
+       return savedCurrentMesh;
+}
+