Salome HOME
Bug #334: Different behavior in Edit calculation case input panel.
[modules/hydro.git] / src / HYDROData / HYDROData_Tool.cxx
index cd6090370b15f1941099fa91fea84db1bba7bd78..333205a0e371be975b631e52c75e5e0991e767f0 100644 (file)
@@ -1,13 +1,40 @@
 
 #include "HYDROData_Tool.h"
 
+#include "HYDROData_ArtificialObject.h"
 #include "HYDROData_Image.h"
 #include "HYDROData_Iterator.h"
+#include "HYDROData_NaturalObject.h"
+
+#include <TopTools_SequenceOfShape.hxx>
+#include <TopExp_Explorer.hxx>
 
 #include <QFile>
 #include <QStringList>
 #include <QTextStream>
 
+#include <limits>
+#define CLASS2D 1
+#ifdef CLASS2D
+#include <gp_Pnt.hxx>
+#include <gp_Vec.hxx>
+#include <gp_Pln.hxx>
+#include <TopExp.hxx>
+#include <Geom_Plane.hxx>
+#include <BRep_Tool.hxx>
+#include <BRepBuilderAPI_FindPlane.hxx>
+#include <TopoDS.hxx>
+#include <TopoDS_Vertex.hxx>
+#include <TopoDS_Wire.hxx>
+#include <TopoDS_Face.hxx>
+#include <TopoDS_Iterator.hxx>
+#include <TopTools_MapOfShape.hxx>
+#include <TColgp_SequenceOfVec.hxx>
+#include <TopTools_ShapeMapHasher.hxx>
+#undef _NCollection_MapHasher
+#endif
+static int aMaxNameId = std::numeric_limits<int>::max();
+
 void HYDROData_Tool::WriteStringsToFile( QFile&             theFile,
                                          const QStringList& theStrings,
                                          const QString&     theSep )
@@ -23,42 +50,33 @@ void HYDROData_Tool::WriteStringsToFile( QFile&             theFile,
   anOutStream << aWriteStr << theSep << theSep;
 }
 
-void HYDROData_Tool::SetMustBeUpdatedImages(
+void HYDROData_Tool::SetMustBeUpdatedObjects(
   const Handle(HYDROData_Document)& theDoc  )
 {
-  bool aChanged = true;
+  bool anIsChanged = true;
 
-  // iterate until there is no changes because images on all level of dependency must be updated
-  while ( aChanged )
+  // iterate until there is no changes because objects on all level of dependency must be updated
+  while ( anIsChanged )
   {
-    aChanged = false;
+    anIsChanged = false;
 
-    HYDROData_Iterator anIter( theDoc, KIND_IMAGE );
+    HYDROData_Iterator anIter( theDoc );
     for ( ; anIter.More(); anIter.Next() )
     {
-      Handle(HYDROData_Image) anImage = 
-        Handle(HYDROData_Image)::DownCast( anIter.Current() );
-      if ( anImage.IsNull() || anImage->MustBeUpdated() )
+      Handle(HYDROData_Entity) anObject = anIter.Current();
+      if ( anObject.IsNull() || anObject->IsMustBeUpdated() )
         continue;
 
-      Handle(HYDROData_Image) aTrsfRefImage = anImage->GetTrsfReferenceImage();
-      if ( !aTrsfRefImage.IsNull() && aTrsfRefImage->MustBeUpdated() )
+      HYDROData_SequenceOfObjects aRefSeq = anObject->GetAllReferenceObjects();
+      for ( int i = 1, n = aRefSeq.Length(); i <= n; ++i )
       {
-        anImage->MustBeUpdated( true );
-        aChanged = true;
-        continue;
-      }
+        Handle(HYDROData_Entity) aRefObject = aRefSeq.Value( i );
+        if ( aRefObject.IsNull() || !aRefObject->IsMustBeUpdated() )
+          continue;
 
-      for ( int i = 0, aNBRefs = anImage->NbReferences(); i < aNBRefs; ++i )
-      {
-        Handle(HYDROData_Image) aRefImage =
-          Handle(HYDROData_Image)::DownCast( anImage->Reference( i ) );
-        if ( !aRefImage.IsNull() && aRefImage->MustBeUpdated() )
-        {
-           // image references to updated => also must be updated
-           anImage->MustBeUpdated(true);
-           aChanged = true;
-        }
+        anObject->SetToUpdate( true );
+        anIsChanged = true;
+        break;
       }
     }
   }
@@ -66,7 +84,8 @@ void HYDROData_Tool::SetMustBeUpdatedImages(
 
 QString HYDROData_Tool::GenerateObjectName( const Handle(HYDROData_Document)& theDoc,
                                             const QString&                    thePrefix,
-                                            const QStringList&                theUsedNames )
+                                            const QStringList&                theUsedNames,
+                                            const bool                        theIsTryToUsePurePrefix )
 {
   QStringList aNamesList( theUsedNames );
 
@@ -87,66 +106,134 @@ QString HYDROData_Tool::GenerateObjectName( const Handle(HYDROData_Document)& th
 
   QString aName;
 
-  int anId = 1;
-  while( anId < 1000 )
-  {
-    aName = QString( "%1_%2" ).arg( thePrefix ).arg( QString::number( anId++ ) );
+  if ( theIsTryToUsePurePrefix && !aNamesList.contains( thePrefix ) ) {
+    aName = thePrefix;
+  } else {
+    int anId = 1;
+    while( anId < aMaxNameId )
+    {
+      aName = QString( "%1_%2" ).arg( thePrefix ).arg( QString::number( anId++ ) );
 
-    // check that there are no other objects with the same name in the document
-    if ( !aNamesList.contains( aName ) )
-      break;
+      // check that there are no other objects with the same name in the document
+      if ( !aNamesList.contains( aName ) )
+        break;
+    }
   }
 
   return aName;
 }
 
-Handle(HYDROData_Entity) HYDROData_Tool::FindObjectByName( const Handle(HYDROData_Document)& theDoc,
-                                                           const QString&                    theName,
-                                                           const ObjectKind                  theObjectKind )
+bool HYDROData_Tool::IsGeometryObject( const Handle(HYDROData_Entity)& theObject )
 {
-  Handle(HYDROData_Entity) anObject;
-  if ( theName.isEmpty() || theDoc.IsNull() )
-    return anObject;
-
-  QStringList aNamesList;
-  aNamesList << theName;
-
-  HYDROData_SequenceOfObjects aSeqOfObjs = FindObjectsByNames( theDoc, aNamesList, theObjectKind );
-  if( aSeqOfObjs.IsEmpty() )
-    return anObject;
+  if ( theObject.IsNull() )
+    return false;
   
-  anObject = aSeqOfObjs.First();
-  return anObject;
+  return theObject->IsKind( STANDARD_TYPE(HYDROData_ArtificialObject) ) ||
+         theObject->IsKind( STANDARD_TYPE(HYDROData_NaturalObject) );
 }
 
-HYDROData_SequenceOfObjects HYDROData_Tool::FindObjectsByNames( const Handle(HYDROData_Document)& theDoc,
-                                                                const QStringList&                theNames,
-                                                                const ObjectKind                  theObjectKind )
+void HYDROData_Tool::UpdateChildObjectName( const QString&                  theOldStr,
+                                            const QString&                  theNewStr,
+                                            const Handle(HYDROData_Entity)& theObject )
 {
-  HYDROData_SequenceOfObjects aResSeq;
-  if( theDoc.IsNull() )
-    return aResSeq;
+  if ( theObject.IsNull() )
+    return;
 
-  QStringList aNamesList = theNames;
+  QString anObjName = theObject->GetName();
+  if ( theOldStr.isEmpty() )
+  {
+    while ( anObjName.startsWith( '_' ) )
+      anObjName.remove( 0, 1 );
 
-  HYDROData_Iterator anIter( theDoc, theObjectKind );
-  for( ; anIter.More(); anIter.Next() )
+    anObjName.prepend( theNewStr + "_" );
+  }
+  else if ( anObjName.startsWith( theOldStr ) )
   {
-    Handle(HYDROData_Entity) anObject = anIter.Current();
-    if( anObject.IsNull() )
-      continue;
+    anObjName.replace( 0, theOldStr.length(), theNewStr );
+  }
+  else
+    return;
 
-    QString anObjName = anObject->GetName();
-    if ( anObjName.isEmpty() || !aNamesList.contains( anObjName ) )
-      continue;
+  theObject->SetName( anObjName );
+}
+
+QString HYDROData_Tool::GenerateNameForPython( const MapOfTreatedObjects& theTreatedObjects,
+                                               const QString&             thePrefix )
+{
+  QString aName = thePrefix;
+  if ( !theTreatedObjects.contains( aName ) )
+    return aName;
 
-    aResSeq.Append( anObject );
+  int anId = 1;
+  while( anId < aMaxNameId )
+  {
+    aName = QString( "%1_%2" ).arg( thePrefix ).arg( QString::number( anId++ ) );
 
-    aNamesList.removeAll( anObjName );
-    if ( aNamesList.isEmpty() )
+    // check that there are no other objects with the same name
+    if ( !theTreatedObjects.contains( aName ) )
       break;
   }
 
-  return aResSeq;
+  return aName;
 }
-
+//======================================================================================================
+// the Face to be: planer, whitout holes
+#ifdef CLASS2D
+TopAbs_State HYDROData_Tool::ComputePointState( const gp_XY& thePnt2d, const TopoDS_Face& theFace )
+{
+  TopAbs_State aState(TopAbs_UNKNOWN);
+  if(theFace.IsNull()) return aState;  
+
+#ifdef CLASS2D    
+  TopoDS_Wire aWire; int nb(0);
+  TopoDS_Iterator it(theFace);
+  for(;it.More();it.Next()) {
+         aWire = TopoDS::Wire(it.Value());
+         nb++;
+  }
+  if(nb > 1 || aWire.IsNull()) return aState;
+  gp_Pln aPlane;
+  gp_Vec aNormal;
+  BRepBuilderAPI_FindPlane fndPlane (theFace, Precision::Confusion());
+  if(fndPlane.Found())
+    aPlane = fndPlane.Plane()->Pln();
+  else
+    return aState;
+  aNormal = gp_Vec(aPlane.Axis().Direction());
+  if(theFace.Orientation() == TopAbs_REVERSED) 
+    aNormal.Reverse();
+  gp_Pnt aPoint = gp_Pnt (thePnt2d.X(), thePnt2d.Y(), 0);
+  TColgp_SequenceOfVec aSeq;
+  TopTools_MapOfShape aMap;
+  it.Initialize(aWire);
+  for (;it.More(); it.Next()) {
+       const TopoDS_Vertex& aVx = TopExp::FirstVertex(TopoDS::Edge(it.Value()), Standard_True);
+       if(!aMap.Add(aVx))
+         continue;
+       const gp_Pnt& aCurPnt = BRep_Tool::Pnt(TopoDS::Vertex(aVx));
+       if(aPoint.IsEqual(aCurPnt,Precision::Confusion())) {
+      aState = TopAbs_ON;
+         return aState;
+       }
+       gp_Vec aVec (aPoint, aCurPnt);
+       aSeq.Append(aVec);
+  }
+  Standard_Real anAng(0.0);
+  for(int i = 1;i < aSeq.Length();i++) {
+    const gp_Vec& aV1 = aSeq.Value(i);
+       const gp_Vec& aV2 = aSeq.Value(i+1);
+        anAng += aV1.AngleWithRef(aV2, aNormal);
+  }
+  anAng += aSeq.Value(aSeq.Length()).AngleWithRef(aSeq.Value(1), aNormal);
+  if(abs(anAng) > Precision::Angular())
+    aState = TopAbs_IN;
+  else
+        aState = TopAbs_OUT;
+
+#else
+  BRepTopAdaptor_FClass2d aClassifier( theFace, Precision::Confusion() ); 
+  aState = aClassifier.Perform( gp_Pnt2d( thePoint ), Standard_False );
+#endif
+  return aState;
+}
+#endif
\ No newline at end of file