#include <HYDROData_ShapeFile.h>
#include <HYDROData_Document.h>
#include <HYDROData_StricklerTable.h>
+#include <HYDROData_ShapesTool.h>
#include <BOPAlgo_BOP.hxx>
#include <BOPAlgo_Builder.hxx>
#include <TopExp_Explorer.hxx>
#include <ShapeUpgrade_UnifySameDomain.hxx>
#include <TopExp.hxx>
-#include <TDataStd_Real.hxx>
#include <TopTools_IndexedMapOfShape.hxx>
#include <ShapeBuild_ReShape.hxx>
#include <ShapeFix_Shape.hxx>
#include <BRepCheck_Shell.hxx>
#include <BRepCheck_ListOfStatus.hxx>
#include <TopTools_SequenceOfShape.hxx>
+#include <Handle_Geom_Curve.hxx>
+#include <Handle_Geom_Line.hxx>
+#include <Handle_Geom_TrimmedCurve.hxx>
+#include <Geom_TrimmedCurve.hxx>
+#include <TopTools_DataMapOfShapeListOfShape.hxx>
+#include <NCollection_DoubleMap.hxx>
+#include <HYDROData_LCM_FaceClassifier.h>
+#include <QDir>
+#include <stdexcept>
#include <QFile>
#include <QString>
#include <QTextStream>
+#include <QFileInfo>
+
+#define _DEVDEBUG_
+#include "HYDRO_trace.hxx"
const char TELEMAC_FORMAT = 'f';
const int TELEMAC_PRECISION = 3;
*/
bool HYDROData_LandCoverMap::ExportTelemac( const QString& theFileName,
double theDeflection,
- const Handle(HYDROData_StricklerTable)& theTable ) const
+ const Handle(HYDROData_StricklerTable)& theTable,
+ QString& statMessage) const
{
TopoDS_Shape aLandCoverMapShape = GetShape();
TopTools_ListOfShape aListOfFaces;
aTypesMap.insert( anIt.Face().TShape(), anIt.StricklerType() );
}
- TopoDS_Shape aShape = MergeFaces( aListOfFaces, false );
+ TopoDS_Shape aShape = MergeFaces( aListOfFaces, false, NULL );
NCollection_IndexedMap<gp_Pnt> aVerticesMap;
NCollection_IndexedDataMap< TopoDS_Edge, QList<int> > anEdgesMap;
QFile aFile( theFileName );
if( !aFile.open( QFile::WriteOnly | QFile::Text ) )
- return false;
+ {
+ QString homeFilePath = QDir::home().absoluteFilePath( theFileName );
+ aFile.setFileName(homeFilePath);
+ if (aFile.open( QFile::WriteOnly | QFile::Text ) )
+ statMessage = "Telemac file have been exported to the home directory: " + homeFilePath;
+ else
+ return false;
+ }
+ else
+ {
+ QString absFilePath = QDir::current().absoluteFilePath( theFileName );
+ statMessage = "Telemac file have been exported to the current directory: " + absFilePath;
+ }
QTextStream aStream( &aFile );
aStream << "# nodes\n";
return false;
TopoDS_Shape aShape = theObject->GetTopShape();
- if( aShape.ShapeType()!=TopAbs_FACE )
- return false;
+ TopoDS_Face aFace;
+
+ if( aShape.ShapeType() ==TopAbs_FACE )
+ {
+ aFace = TopoDS::Face(aShape);
+ }
+ else if ( aShape.ShapeType() ==TopAbs_COMPOUND )
+ {
+ TopoDS_Iterator It(aShape);
+ for (; It.More(); It.Next())
+ if (It.Value().ShapeType() == TopAbs_FACE)
+ {
+ aFace = TopoDS::Face(It.Value());
+ break;
+ }
+ }
- TopoDS_Face aFace = TopoDS::Face( aShape );
+ if (aFace.IsNull())
+ return false;
return LocalPartition( aFace, theType );
}
*/
bool HYDROData_LandCoverMap::Split( const TopoDS_Shape& theShape )
{
- return LocalPartition( theShape, "" );
+ int aNbCL = GetLCCount();
+ bool aResult = LocalPartition( theShape, "" );
+ return aResult && aNbCL != GetLCCount();
}
bool HYDROData_LandCoverMap::Merge( const TopTools_ListOfShape& theFaces, const QString& theType )
{
// 1. to fuse the faces into the new face
- TopoDS_Shape aMergedFace = MergeFaces( theFaces, true );
- if( !aMergedFace.IsNull() && aMergedFace.ShapeType()==TopAbs_FACE )
- {
+ TopoDS_Shape aMergedFace = MergeFaces( theFaces, true, NULL );
+ bool aStat = true;
+ if( !aMergedFace.IsNull() )
+ {
// 2. to remove the merged faces from the current map
Remove( theFaces );
-
- // 3. to add the face into the map
- return LocalPartition( TopoDS::Face( aMergedFace ), theType );
+ TopExp_Explorer Exp(aMergedFace, TopAbs_FACE);
+ for( ; Exp.More(); Exp.Next() )
+ {
+ const TopoDS_Face& aCF = TopoDS::Face(Exp.Current());
+ // 3. to add the face into the map
+ aStat = aStat && LocalPartition( aCF, theType );
+ }
}
- return false;
+ else
+ aStat = false;
+ return aStat;
}
/**
@param theTolerance the operation's tolerance
@return result shape (face or shell)
*/
+
TopoDS_Shape HYDROData_LandCoverMap::MergeFaces( const TopTools_ListOfShape& theFaces,
- bool IsToUnify, double theTolerance )
+ bool IsToUnify,
+ TopTools_IndexedDataMapOfShapeListOfShape* theShHistory,
+ double theTolerance)
{
+ //DEBTRACE("MergeFaces");
int anError;
TopTools_ListIteratorOfListOfShape anIt;
BOPCol_ListOfShape aLC;
const TopoDS_Shape& aMergedShape = anAlgo.Shape();
+ // retrieve history of modifications
+ if (theShHistory)
+ {
+ theShHistory->Clear();
+ anIt.Initialize(theFaces);
+ for( ; anIt.More(); anIt.Next() )
+ {
+ const TopTools_ListOfShape aMLS = anAlgo.Modified( anIt.Value() );
+ theShHistory->Add(anIt.Value(), aMLS);
+ }
+ }
+ //
+
BRep_Builder aBuilder;
TopoDS_Shell aShell;
aBuilder.MakeShell( aShell );
return aResult;
}
-
/**
Change Strickler type for the list of faces to the given one
@param theFaces the faces to change type
aShapesList.Append( anIt.Face() );
aShapesList.Append( theNewShape );
+ //DEBTRACE("theNewType " << theNewType);
if( aShapesList.Size()==1 && theNewShape.ShapeType()==TopAbs_FACE )
{
aNewFaces.Add( TopoDS::Face( theNewShape ), theNewType );
for( ; aMIt.More(); aMIt.Next() )
{
//std::cout << " " << aMIt.Value() << std::endl;
+ //DEBTRACE(aMIt.Value());
int aKey = (int)(uintptr_t)aMIt.Value().TShape().operator->();
aShapesFromNewFace.Add( aKey );
}
for( ; anIt.More(); anIt.Next() )
{
QString aSType = anIt.StricklerType();
+ //DEBTRACE(anIt.StricklerType() << " " << anIt.Face());
//std::cout << "from " << anIt.Face() << ": " << anIt.StricklerType() << std::endl;
TopTools_ListOfShape aModified = aBuilder.Modified( anIt.Face() );
+ //
if( aModified.Extent() == 0 )
aModified.Append( anIt.Face() );
+ //DEBTRACE(anIt.StricklerType() << " " << anIt.Face());
TopTools_ListIteratorOfListOfShape aMIt( aModified );
for( ; aMIt.More(); aMIt.Next() )
if( isFace && !isAlsoFromNew )
aNewFaces.Add( TopoDS::Face( aShape ), aSType );
}
+ //DEBTRACE(anIt.StricklerType() << " " << anIt.Face());
}
-
// c. add the new shape if it is face with its type
if( theNewShape.ShapeType()==TopAbs_FACE )
aNewFaces.Add( TopoDS::Face( theNewShape ), theNewType );
+ //DEBTRACE(theNewShape << " " << theNewType);
// convert map of shape to type to compound and list of types
StoreLandCovers( aNewFaces );
+
+// anIt.Init( *this );
+// for( ; anIt.More(); anIt.Next() )
+// {
+// DEBTRACE(anIt.StricklerType() << " " << anIt.Face());
+// }
return true;
}
{
TopTools_ListOfShape aListOfFaces;
- int n = theMap.Size();
-
- Handle( TDataStd_ExtStringArray ) aTypes =
- TDataStd_ExtStringArray::Set( myLab.FindChild( DataTag_Types ), 0, n-1, Standard_True );
-
- HYDROData_MapOfFaceToStricklerType::Iterator aNFIt( theMap );
- for( int i=0; aNFIt.More(); aNFIt.Next(), i++ )
+ //DEBTRACE("theMap.Extent() " << theMap.Extent());
+ for( int i = 1; i <= theMap.Extent(); i++ )
{
- TopoDS_Face aFace = aNFIt.Key();
+ TopoDS_Face aFace = theMap.FindKey(i);
if( aFace.IsNull() )
continue;
- QString aType = aNFIt.Value();
aListOfFaces.Append(aFace);
- aTypes->SetValue( i, HYDROData_Tool::toExtString( aType ) );
}
+ TopTools_IndexedDataMapOfShapeListOfShape ShHistory;
+ ShHistory.Clear();
+
TopoDS_Shape aResult;
if( aListOfFaces.Extent() == 1 )
aResult = aListOfFaces.First();
else if( aListOfFaces.Extent() > 1 )
- aResult = MergeFaces( aListOfFaces, false );
+ aResult = MergeFaces( aListOfFaces, false, &ShHistory );
//remove internal edges
- aResult = RemoveInternal(aResult);
+ //if nothing changes => the result shape should be the same
+ //hence the map will be empty
+
+ NCollection_IndexedDataMap<TopoDS_Face, TopoDS_Face> ShF2FHistory;
+ RemoveInternal(aResult, &ShF2FHistory);
+
+ //one face => mark as unchanged
+ if( aListOfFaces.Extent() == 1 )
+ ShHistory.Add(aResult, TopTools_ListOfShape());
+
+ NCollection_IndexedDataMap<TopoDS_Face, QString> aChF2ST;
+ QStringList aSTypes;
+ //
+ for( int i = 1; i <= theMap.Extent(); i++ )
+ {
+ TopoDS_Face aFF = theMap.FindKey(i);
+ //DEBTRACE(" --- " << aFF);
+ if( aFF.IsNull() )
+ continue;
+ //DEBTRACE(ShHistory.IsEmpty());
+ //DEBTRACE(aFF.Checked());
+ TopTools_ListOfShape aLS;
+ try
+ {
+ aLS = ShHistory.FindFromKey(aFF); //TODO: bug to fix. Observed on an incomplete split of a face
+ }
+ catch (...)
+ {
+ DEBTRACE("TODO: bug to fix. Observed on an incomplete split of a face");
+ //continue; // No, keep aLS empty and propagate the type of the original face
+ }
+ if (aLS.IsEmpty())
+ {
+ //DEBTRACE("--- aLS.IsEmpty()");
+ QString aSType = theMap.FindFromKey(aFF);
+ //DEBTRACE(" --- " << aSType.toStdString());
+ if (ShF2FHistory.Contains(aFF))
+ {
+ //DEBTRACE("ShF2FHistory.FindFromKey(aFF) " << ShF2FHistory.FindFromKey(aFF));
+ aChF2ST.Add(ShF2FHistory.FindFromKey(aFF), aSType);
+ }
+ else
+ {
+ //DEBTRACE("aFF " << aFF);
+ aChF2ST.Add(aFF, aSType);
+ }
+ }
+ else
+ {
+ //DEBTRACE("--- !aLS.IsEmpty()");
+ TopTools_ListIteratorOfListOfShape anIt(aLS);
+ for (; anIt.More(); anIt.Next())
+ {
+ QString aSType = theMap.FindFromKey(aFF);
+ //DEBTRACE(" --- " << aSType.toStdString());
+ const TopoDS_Face& aMF = TopoDS::Face(anIt.Value());
+ //if (ShF2FHistory.Contains(aFF))
+ if (ShF2FHistory.Contains(aMF))
+ {
+ //DEBTRACE("ShF2FHistory.FindFromKey(aMF) " << ShF2FHistory.FindFromKey(aFF));
+ aChF2ST.Add(ShF2FHistory.FindFromKey(aMF), aSType);
+ }
+ else
+ {
+ //DEBTRACE("aMF " << aMF);
+ aChF2ST.Add(aMF, aSType);
+ }
+ }
+ }
+ }
+ //
SetShape( aResult );
+ //
+ //Explorer Exp(*this);
+ TopExp_Explorer FExp(aResult, TopAbs_FACE);
+ for( ; FExp.More(); FExp.Next() )
+ {
+ TopoDS_Face aFace = TopoDS::Face(FExp.Current());
+ QString aST = "";
+ if (aChF2ST.Contains(aFace))
+ aST = aChF2ST.FindFromKey(aFace);
+ //DEBTRACE("aFace " << aFace << " aST " << aST.toStdString());
+ aSTypes << aST;
+ }
+
+ Handle( TDataStd_ExtStringArray ) aTypes = TDataStd_ExtStringArray::Set( myLab.FindChild( DataTag_Types ), 0, aSTypes.size() - 1, Standard_True );
+ int k = 0;
+ foreach (QString aST, aSTypes)
+ {
+ //DEBTRACE("aST " << aST.toStdString());
+ aTypes->SetValue( k, HYDROData_Tool::toExtString( aST ) );
+ k++;
+ }
+
}
/**
QString aDbfFileName = thePyScriptPath;
aDbfFileName.replace( ".py", ".dbf" );
- ExportSHP( aShpFileName, true, 1 );
+ ExportSHP( aShpFileName, true, 0.1 );
QString anAttr = "CODE_06"; //TODO: some custom choice
QStringList anAttrValues, aTypes;
ExportDBF( aDbfFileName, anAttr, anAttrValues, aTypes );
aResList << QString( "%1.ImportSHP( '%2' )" ).
- arg( aName ).arg( aShpFileName );
+ arg( aName ).arg( QFileInfo( aShpFileName ).fileName() );
Dump( "attr_values", anAttrValues, aResList );
Dump( "types", aTypes, aResList );
aResList << QString( "%1.ImportDBF( '%2', '%3', attr_values, types )" ).
- arg( aName ).arg( aDbfFileName ).arg( anAttr );
+ arg( aName ).arg( QFileInfo( aDbfFileName ).fileName() ).arg( anAttr );
return aResList;
}
-TopoDS_Shape HYDROData_LandCoverMap::RemoveInternal(const TopoDS_Shape& InSh)
+void HYDROData_LandCoverMap::RemoveInternal(TopoDS_Shape& ShToRebuild, NCollection_IndexedDataMap<TopoDS_Face, TopoDS_Face>* aF2FReplace)
{
+ //DEBTRACE("RemoveInternal");
//Shape must be topologically correct
- TopExp_Explorer anExp(InSh, TopAbs_EDGE);
- TopTools_ListOfShape anEdgesToRemove;
-
- for(; anExp.More(); anExp.Next() )
+ TopExp_Explorer anExpF(ShToRebuild, TopAbs_FACE);
+ //
+ for(; anExpF.More(); anExpF.Next() )
{
- TopoDS_Edge CurEdge = TopoDS::Edge(anExp.Current());
- if (CurEdge.Orientation() == TopAbs_INTERNAL)
- anEdgesToRemove.Append(CurEdge);
+ TopoDS_Face CurFace = TopoDS::Face(anExpF.Current());
+ //
+ TopExp_Explorer anExp(CurFace, TopAbs_EDGE);
+ TopTools_ListOfShape anEdgesToRemove;
+ //
+ for(; anExp.More(); anExp.Next() )
+ {
+ TopoDS_Edge CurEdge = TopoDS::Edge(anExp.Current());
+ if (CurEdge.Orientation() == TopAbs_INTERNAL)
+ anEdgesToRemove.Append(CurEdge);
+ }
+ //
+ if (!anEdgesToRemove.IsEmpty())
+ {
+ Handle_ShapeBuild_ReShape aReshape = new ShapeBuild_ReShape();
+ TopoDS_Shape OutF = aReshape->Apply(CurFace);
+ TopTools_ListIteratorOfListOfShape aIt(anEdgesToRemove);
+ for (; aIt.More(); aIt.Next())
+ aReshape->Remove(aIt.Value());
+ OutF = aReshape->Apply(CurFace);
+
+ Handle(ShapeFix_Shape) sfs = new ShapeFix_Shape;
+ sfs->Init(OutF);
+ sfs->Perform();
+ OutF = sfs->Shape();
+ aF2FReplace->Add(CurFace, TopoDS::Face(OutF));
+ }
}
-
- Handle_ShapeBuild_ReShape aReshape = new ShapeBuild_ReShape();
- TopoDS_Shape OutSh = aReshape->Apply(InSh);
- TopTools_ListIteratorOfListOfShape aIt(anEdgesToRemove);
- for (; aIt.More(); aIt.Next())
- aReshape->Remove(aIt.Value());
- OutSh = aReshape->Apply(InSh);
-
- Handle(ShapeFix_Shape) sfs = new ShapeFix_Shape;
- sfs->Init(OutSh);
- sfs->Perform();
- return sfs->Shape();
+ //
+ Handle_ShapeBuild_ReShape anExtReshape = new ShapeBuild_ReShape();
+ for (int i = 1; i <= aF2FReplace->Extent(); i++)
+ {
+ TopoDS_Face aFK = aF2FReplace->FindKey(i);
+ TopoDS_Face aFV = aF2FReplace->FindFromIndex(i);
+ anExtReshape->Replace(aFK, aFV);
+ ShToRebuild = anExtReshape->Apply(ShToRebuild);
+ }
+// for (int i = 1; i <= aF2FReplace->Extent(); i++)
+// {
+// DEBTRACE("aF2FReplace key,value " << aF2FReplace->FindKey(i) << " " << aF2FReplace->FindFromIndex(i));
+// }
}
void HYDROData_LandCoverMap::SetTransparency( double theTransparency )
{
- Handle(TDataStd_Real) anAttr;
- TDF_Label aLabel = myLab.FindChild( DataTag_Transparency );
- if( !aLabel.FindAttribute( TDataStd_Real::GetID(), anAttr ) )
- aLabel.AddAttribute( anAttr = new TDataStd_Real() );
- anAttr->Set( theTransparency );
+ SetDouble( DataTag_Transparency, theTransparency );
}
double HYDROData_LandCoverMap::GetTransparency() const
{
- Handle(TDataStd_Real) anAttr;
- TDF_Label aLabel = myLab.FindChild( DataTag_Transparency );
- if( !aLabel.FindAttribute( TDataStd_Real::GetID(), anAttr ) )
- return 0.5;
-
- return anAttr->Get();
-
+ return GetDouble( DataTag_Transparency, 0.5 );
}
bool HYDROData_LandCoverMap::ImportSHP( const QString& theSHPFileName,
QStringList aPolyList;
TopTools_SequenceOfShape aFaces;
int aSHapeType = -1;
- int Stat = anImporter.ImportPolygons(theSHPFileName, aPolyList, aFaces, aSHapeType);
+ int Stat = anImporter.ImportPolygons(HYDROData_Document::Document(1), theSHPFileName, aPolyList, aFaces, aSHapeType);
//
if (Stat != 1)
return false;
{
HYDROData_ShapeFile anExporter;
QStringList aList;
- anExporter.Export(theSHPFileName, this, aList, bUseDiscr, theDefl );
+ anExporter.Export(HYDROData_Document::Document(1), theSHPFileName, this, aList, bUseDiscr, theDefl );
if (aList.empty())
return true;
else
return false;
}
+bool HYDROData_LandCoverMap::CheckLinear()
+{
+ TopoDS_Shape InpShape = GetShape();
+ TopExp_Explorer anEdgeEx(InpShape, TopAbs_EDGE);
+ for (; anEdgeEx.More(); anEdgeEx.Next())
+ {
+ TopoDS_Edge E = TopoDS::Edge(anEdgeEx.Current());
+ double aFP, aLP;
+ Handle_Geom_Curve aCur = BRep_Tool::Curve(E, aFP, aLP);
+ Handle(Geom_Line) aLine = Handle(Geom_Line)::DownCast(aCur);
+ if (aLine.IsNull())
+ {
+ Handle(Geom_TrimmedCurve) aTC = Handle(Geom_TrimmedCurve)::DownCast(aCur);
+ if (!aTC.IsNull())
+ {
+ Handle(Geom_Line) aLine = Handle(Geom_Line)::DownCast(aTC->BasisCurve());
+ if (aLine.IsNull())
+ return false;
+ }
+ else
+ return false;
+ }
+ }
+ return true;
+}
+
+void HYDROData_LandCoverMap::UpdateLocalCS( double theDx, double theDy )
+{
+ TopoDS_Shape aShape = GetShape();
+ TopoDS_Shape aLocatedShape = HYDROData_ShapesTool::Translated( aShape, theDx, theDy, 0 );
+ SetShape( aLocatedShape );
+}
+
+void HYDROData_LandCoverMap::ClassifyPoints( const std::vector<gp_XY>& thePoints, std::vector<std::set <QString> >& theTypes ) const
+{
+ HYDROData_LCM_FaceClassifier FC(this);
+ FC.Classify(thePoints, theTypes, NULL);
+}
+
+void HYDROData_LandCoverMap::ClassifyPoints( const std::vector<gp_XY>& thePoints,
+ Handle(HYDROData_StricklerTable) theTable,
+ std::vector<double>& theCoeffs, double DefValue, bool UseMax ) const
+{
+ std::vector<std::set <QString> > Types;
+ HYDROData_LCM_FaceClassifier FC(this);
+ FC.Classify(thePoints, Types, NULL);
+ theCoeffs.resize(thePoints.size());
+ for (size_t i = 0; i < Types.size(); i++)
+ {
+ const std::set<QString>& SStr = Types[i];
+ if (SStr.empty())
+ theCoeffs[i] = DefValue;
+ else
+ {
+ std::set<QString>::const_iterator it;
+ std::vector<double> C1(SStr.size());
+ for (it = SStr.begin(); it != SStr.end(); ++it)
+ C1.push_back(theTable->Get( *it, DefValue ));
+ double Val;
+ if (UseMax)
+ Val = *(std::max_element( C1.begin(), C1.end() ) );
+ else
+ Val = *(std::min_element( C1.begin(), C1.end() ) );
+ theCoeffs[i] = Val;
+ }
+ }
+}