0027398: Integrate Qt Browser Widget to Open CASCADE Technology

[occt.git] / tools / View / View_Displayer.cxx

diff --git a/tools/View/View_Displayer.cxx b/tools/View/View_Displayer.cxx
new file mode 100644 (file)
index 0000000..cb90118
--- /dev/null
+++ b/tools/View/View_Displayer.cxx
@@ -0,0 +1,207 @@
+// Created on: 2017-06-16
+// Created by: Natalia ERMOLAEVA
+// Copyright (c) 2017 OPEN CASCADE SAS
+//
+// This file is part of Open CASCADE Technology software library.
+//
+// This library is free software; you can redistribute it and/or modify it under
+// the terms of the GNU Lesser General Public License version 2.1 as published
+// by the Free Software Foundation, with special exception defined in the file
+// OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
+// distribution for complete text of the license and disclaimer of any warranty.
+//
+// Alternatively, this file may be used under the terms of Open CASCADE
+// commercial license or contractual agreement. 
+
+#include <View_Displayer.hxx>
+
+#include <AIS_InteractiveContext.hxx>
+#include <AIS_InteractiveObject.hxx>
+#include <AIS_ListIteratorOfListOfInteractive.hxx>
+#include <V3d_View.hxx>
+#include <V3d_Viewer.hxx>
+#include <View_Viewer.hxx>
+#include <View_Widget.hxx>
+
+// =======================================================================
+// function : Constructor
+// purpose :
+// =======================================================================
+View_Displayer::View_Displayer()
+: myIsKeepPresentations (false), myDisplayMode (-1)
+{
+}
+
+// =======================================================================
+// function : SetContext
+// purpose :
+// =======================================================================
+void View_Displayer::SetContext (const Handle(AIS_InteractiveContext)& theContext)
+{
+  NCollection_DataMap<View_PresentationType, NCollection_Shared<AIS_ListOfInteractive> > aDisplayed = myDisplayed;
+  EraseAllPresentations (true);
+  myContext = theContext;
+
+  for (NCollection_DataMap<View_PresentationType, NCollection_Shared<AIS_ListOfInteractive> >::Iterator aDisplayedIt(aDisplayed);
+    aDisplayedIt.More(); aDisplayedIt.Next())
+  {
+    View_PresentationType aType = aDisplayedIt.Key();
+    for (AIS_ListIteratorOfListOfInteractive aPresentationsIt (aDisplayedIt.Value());
+         aPresentationsIt.More(); aPresentationsIt.Next())
+      DisplayPresentation (aPresentationsIt.Value(), aType, false);
+  }
+  UpdateViewer();
+}
+
+// =======================================================================
+// function : SetDisplayMode
+// purpose :
+// =======================================================================
+void View_Displayer::SetDisplayMode (const int theDisplayMode,
+                                     const View_PresentationType theType,
+                                     const bool theToUpdateViewer)
+{
+  myDisplayMode = theDisplayMode;
+  if (GetContext().IsNull())
+    return;
+
+  NCollection_Shared<AIS_ListOfInteractive> aDisplayed;
+  DisplayedPresentations (aDisplayed, theType);
+
+  for (AIS_ListIteratorOfListOfInteractive aDisplayedIt (aDisplayed); aDisplayedIt.More(); aDisplayedIt.Next())
+    GetContext()->SetDisplayMode (aDisplayedIt.Value(), theDisplayMode, Standard_False);
+
+  if (theToUpdateViewer)
+    UpdateViewer();
+}
+
+// =======================================================================
+// function : DisplayPresentation
+// purpose :
+// =======================================================================
+void View_Displayer::DisplayPresentation (const Handle(Standard_Transient)& thePresentation,
+                                          const View_PresentationType theType,
+                                          const bool theToUpdateViewer)
+{
+  if (GetContext().IsNull())
+    return;
+
+  NCollection_Shared<AIS_ListOfInteractive> aDisplayed;
+  DisplayedPresentations (aDisplayed, theType);
+  if (!myIsKeepPresentations)
+    ErasePresentations (theType, false);
+
+  Handle(AIS_InteractiveObject) aPresentation = Handle(AIS_InteractiveObject)::DownCast (thePresentation);
+  if (!aPresentation.IsNull() && aPresentation->GetContext().IsNull())
+  {
+    // one presentation can not be shown in several contexts
+    if (theType == View_PresentationType_Additional)
+    {
+      Quantity_Color aColor;
+      if (myColorAttributes.Find (View_PresentationType_Additional, aColor))
+        aPresentation->SetColor (aColor);
+    }
+    GetContext()->Display (aPresentation, false);
+    if (myDisplayMode != -1)
+      GetContext()->SetDisplayMode (aPresentation, myDisplayMode, false);
+    aDisplayed.Append (aPresentation);
+  }
+
+  if (!myIsKeepPresentations)
+  {
+    Handle(V3d_View) aView = GetView();
+    if (!aView.IsNull())
+    {
+      aView->FitAll();
+      aView->Redraw();
+    }
+  }
+  myDisplayed.Bind (theType, aDisplayed);
+
+  if (theToUpdateViewer)
+    UpdateViewer();
+}
+
+// =======================================================================
+// function : EraseAllPresentations
+// purpose :
+// =======================================================================
+void View_Displayer::EraseAllPresentations (const bool theToUpdateViewer)
+{
+  for (NCollection_DataMap<View_PresentationType, NCollection_Shared<AIS_ListOfInteractive> >::Iterator aDisplayedIt(myDisplayed);
+       aDisplayedIt.More(); aDisplayedIt.Next())
+    ErasePresentations (aDisplayedIt.Key(), false);
+
+  if (theToUpdateViewer)
+    UpdateViewer();
+}
+
+// =======================================================================
+// function : ErasePresentations
+// purpose :
+// =======================================================================
+void View_Displayer::ErasePresentations (const View_PresentationType theType, const bool theToUpdateViewer)
+{
+  if (GetContext().IsNull())
+    return;
+
+  NCollection_Shared<AIS_ListOfInteractive> aDisplayed;
+  DisplayedPresentations (aDisplayed, theType);
+
+  for (AIS_ListIteratorOfListOfInteractive aDisplayedIt (aDisplayed); aDisplayedIt.More(); aDisplayedIt.Next())
+    GetContext()->Remove (aDisplayedIt.Value(), Standard_False);
+
+  if (theToUpdateViewer)
+    UpdateViewer();
+}
+
+// =======================================================================
+// function : UpdateViewer
+// purpose :
+// =======================================================================
+void View_Displayer::UpdateViewer()
+{
+  if (GetContext().IsNull())
+    return;
+
+  GetContext()->UpdateCurrentViewer();
+}
+
+// =======================================================================
+// function : SetAttributeColor
+// purpose :
+// =======================================================================
+void View_Displayer::SetAttributeColor (const Quantity_Color& theColor, const View_PresentationType theType)
+{
+  myColorAttributes.Bind (theType, theColor);
+}
+
+// =======================================================================
+// function : DisplayedPresentations
+// purpose :
+// =======================================================================
+void View_Displayer::DisplayedPresentations (NCollection_Shared<AIS_ListOfInteractive>& thePresentations,
+                                             const View_PresentationType theType)
+{
+  myDisplayed.Find (theType, thePresentations);
+}
+
+// =======================================================================
+// function : getView
+// purpose :
+// =======================================================================
+Handle(V3d_View) View_Displayer::GetView() const
+{
+  Handle(V3d_View) aView;
+  if (GetContext().IsNull())
+    return aView;
+
+  const Handle(V3d_Viewer)& aViewer = GetContext()->CurrentViewer();
+  if (!aViewer.IsNull())
+  {
+    aViewer->InitActiveViews();
+    if (aViewer->MoreActiveViews())
+      aView = aViewer->ActiveView();
+  }
+  return aView;
+}