0026628: Button Erase (Delete where erase algorythm is used) works incorrect

[occt.git] / samples / mfc / standard / 08_HLR / src / HLRDoc.cpp

diff --git a/samples/mfc/standard/08_HLR/src/HLRDoc.cpp b/samples/mfc/standard/08_HLR/src/HLRDoc.cpp
index 59d83d6..3544bee 100755 (executable)
--- a/samples/mfc/standard/08_HLR/src/HLRDoc.cpp
+++ b/samples/mfc/standard/08_HLR/src/HLRDoc.cpp
@@ -165,6 +165,7 @@ void CHLRDoc::OnBUTTONHLRDialog()
     myCSelectionDialogIsCreated = true;
   }
   myCSelectionDialog->ShowWindow(SW_RESTORE);
+  myCSelectionDialog->UpdateWindow();
 }
 
 void CHLRDoc::OnFileImportBrep() 
@@ -187,14 +188,17 @@ void CHLRDoc::Fit()
 void CHLRDoc::OnObjectErase()
 {
   Standard_Boolean toUpdateDisplayable = Standard_False;
-  for (myAISContext->InitCurrent(); myAISContext->MoreCurrent(); myAISContext->NextCurrent())
+  myAISContext->InitCurrent();
+  while (myAISContext->MoreCurrent())
   {
-    myAISContext->Erase (myAISContext->Current(), Standard_True);
     if (myAISContext->Current()->Type() == AIS_KOI_Shape && myCSelectionDialogIsCreated)
     {
       myCSelectionDialog->DiplayableShape()->Remove (Handle(AIS_Shape)::DownCast (myAISContext->Current())->Shape());
       toUpdateDisplayable = Standard_True;
     }
+
+    myAISContext->Erase (myAISContext->Current(), Standard_True);
+    myAISContext->InitCurrent();
   }
 
   myAISContext->ClearCurrents();