2 #include "HYDROData_SplitToZonesTool.h"
4 #include "HYDROData_PolylineXY.h"
5 #include "HYDROData_ShapesGroup.h"
7 #include <BRepAlgoAPI_Common.hxx>
8 #include <BRepAlgoAPI_Cut.hxx>
9 #include <BRepBuilderAPI_MakeFace.hxx>
11 #include <TopExp_Explorer.hxx>
13 #include <TopoDS_Edge.hxx>
14 #include <TopoDS_Iterator.hxx>
15 #include <TopoDS_Wire.hxx>
16 #include <TopoDS_Iterator.hxx>
17 #include <BRepCheck_Analyzer.hxx>
18 #include <NCollection_IncAllocator.hxx>
19 #include <BOPAlgo_Builder.hxx>
20 #include <BOPAlgo_PaveFiller.hxx>
21 #include <NCollection_DataMap.hxx>
22 #include <TopTools_ListOfShape.hxx>
23 #include <TopTools_ListIteratorOfListOfShape.hxx>
24 #include <TopTools_ShapeMapHasher.hxx>
25 typedef NCollection_DataMap<TopoDS_Shape, TopTools_ListOfShape, TopTools_ShapeMapHasher> HYDROData_DataMapOfShapeListOfShape;
26 typedef HYDROData_DataMapOfShapeListOfShape::Iterator HYDROData_DataMapIteratorOfDataMapOfShapeListOfShape;
27 typedef NCollection_DataMap<TopoDS_Shape, QStringList, TopTools_ShapeMapHasher> HYDROData_DataMapOfShapeListOfString;
28 typedef HYDROData_DataMapOfShapeListOfString::Iterator HYDROData_DataMapIteratorOfDataMapOfShapeListOfString;
29 #undef _NCollection_MapHasher
31 #define DEB_SPLIT_TO_ZONES 1
32 #ifdef DEB_SPLIT_TO_ZONES
33 #include <BRepTools.hxx>
35 TopoDS_Face HYDROData_SplitToZonesTool::SplitData::Face() const
41 if ( Shape.ShapeType() == TopAbs_FACE )
43 aResFace = TopoDS::Face( Shape );
45 else if ( Shape.ShapeType() == TopAbs_WIRE )
47 BRepBuilderAPI_MakeFace aMakeFace( TopoDS::Wire( Shape ), Standard_True );
49 if( aMakeFace.IsDone() )
50 aResFace = aMakeFace.Face();
57 HYDROData_SplitToZonesTool::SplitDataList
58 HYDROData_SplitToZonesTool::Split( const HYDROData_SequenceOfObjects& theObjectList,
59 const HYDROData_SequenceOfObjects& theGroupsList,
60 const Handle(HYDROData_PolylineXY)& thePolyline )
62 SplitDataList anOutputSplitDataList;
63 if(theObjectList.IsEmpty()) return anOutputSplitDataList;
65 // Collect the object shapes to split. InputDataList will contain elements which will hold shape & name_of_shape.
66 SplitDataList anInputSplitDataList;
67 //TCollection_AsciiString aNam("Faces_");
68 for( int anIndex = 1, aLength = theObjectList.Length(); anIndex <= aLength; anIndex++ )
70 Handle(HYDROData_Object) aGeomObj =
71 Handle(HYDROData_Object)::DownCast( theObjectList.Value( anIndex ) );
72 if( aGeomObj.IsNull() )
75 TopoDS_Shape aShape = aGeomObj->GetTopShape();
76 if ( aShape.IsNull() )
78 //TCollection_AsciiString aName = aNam + anIndex + ".brep";
79 //BRepTools::Write(aShape, aName.ToCString());
81 if ( aShape.ShapeType() == TopAbs_COMPOUND ) {
82 // Create split data for each face contained in the compound
83 TopExp_Explorer anExp( aShape, TopAbs_FACE );
84 for ( ; anExp.More(); anExp.Next() ) {
85 const TopoDS_Face& aFace = TopoDS::Face( anExp.Current() );
86 if ( !aFace.IsNull() ) {
87 SplitData aSplitData( SplitData::Data_Zone, aFace, aGeomObj->GetName() );
88 anInputSplitDataList.append( aSplitData );
92 SplitData aSplitData( SplitData::Data_Zone, aShape, aGeomObj->GetName() );
93 anInputSplitDataList.append( aSplitData );
97 SplitDataList anInputGroupList;
98 for( int anIndex = 1; anIndex <= theGroupsList.Length(); anIndex++ )
100 Handle(HYDROData_ShapesGroup) aGeomGroup =
101 Handle(HYDROData_ShapesGroup)::DownCast( theGroupsList.Value( anIndex ) );
102 if( aGeomGroup.IsNull() )
105 TopTools_SequenceOfShape aGroupShapes;
106 aGeomGroup->GetShapes( aGroupShapes );
107 for( int i = 1, aNbShapes = aGroupShapes.Length(); i <= aNbShapes; i++ ) {
108 const TopoDS_Shape& aGroupShape = aGroupShapes.Value( i );
109 if ( aGroupShape.IsNull() )
112 if ( aGroupShape.ShapeType() == TopAbs_COMPOUND ) {
113 TopExp_Explorer anExp( aGroupShape, TopAbs_EDGE );
114 for ( ; anExp.More(); anExp.Next() ) {
115 const TopoDS_Edge& anEdge = TopoDS::Edge( anExp.Current() );
116 if ( !anEdge.IsNull() ) {
117 SplitData aSplitData( SplitData::Data_Edge, anEdge, aGeomGroup->GetName() );
118 anInputGroupList.append( aSplitData );
122 SplitData aSplitData( SplitData::Data_Edge, aGroupShape, aGeomGroup->GetName() );
123 anInputGroupList.append( aSplitData );
127 // If only one shape is given we don't split it
128 // algorithm just returns the unpacked input data
129 if(theObjectList.Size() == 1) {
130 anOutputSplitDataList.append(anInputSplitDataList);
131 if(!theGroupsList.IsEmpty() )
132 anOutputSplitDataList.append(anInputGroupList);
133 return anOutputSplitDataList;
136 HYDROData_DataMapOfShapeListOfString aDM3;
137 if(!anInputGroupList.isEmpty()) {// Old edge ==> List_Of_Names
138 QStringList aListOfNames;
139 for (int i=0;i < anInputGroupList.size() ;i++) {
140 const TopoDS_Shape& aSh = anInputGroupList.at(i).Shape;
141 aDM3.Bind(aSh, anInputGroupList.at(i).ObjectNames);
144 // Step 1. Prepare Partition structures.
145 Handle(NCollection_BaseAllocator) pA1 = new NCollection_IncAllocator, pA2 = new NCollection_IncAllocator;
146 BOPAlgo_PaveFiller* aPaveFiller = new BOPAlgo_PaveFiller(pA1);
147 BOPAlgo_Builder* aBuilder = new BOPAlgo_Builder(pA2);
148 BOPCol_ListOfShape aLS;
149 QStringList aListOfNames;
150 for (int i=0;i < anInputSplitDataList.size() ;i++) {
151 const TopoDS_Shape& aSh = anInputSplitDataList.at(i).Shape;
152 aDM3.Bind(aSh, anInputSplitDataList.at(i).ObjectNames);
155 aPaveFiller->SetArguments(aLS);
156 aPaveFiller->Perform();
157 Standard_Integer anErr = aPaveFiller->ErrorStatus();
159 return anOutputSplitDataList;
160 BOPDS_PDS pDS = aPaveFiller->PDS();
162 return anOutputSplitDataList;
165 // Step 2. Split faces
166 BOPCol_ListIteratorOfListOfShape anIt(aPaveFiller->Arguments());
167 for (; anIt.More(); anIt.Next()) {
168 const TopoDS_Shape& aS = anIt.Value();
169 aBuilder->AddArgument(aS);
171 aBuilder->PerformWithFiller(*aPaveFiller);
172 anErr = aBuilder->ErrorStatus();
174 return anOutputSplitDataList;
175 const TopoDS_Shape& aResult = aBuilder->Shape();
176 if (aResult.IsNull())
177 return anOutputSplitDataList;
178 BRepCheck_Analyzer aCheck (aResult);
179 if(!aCheck.IsValid()) {
180 #ifdef DEB_SPLIT_TO_ZONES
181 cout << "result is not valid" <<endl;
182 BRepTools::Write(aResult, "SplitFacesNV.brep");
184 return anOutputSplitDataList;
186 #ifdef DEB_SPLIT_TO_ZONES
187 //BRepTools::Write(aResult, "SplitFacesV.brep");
190 // Step 3. Collect history
191 HYDROData_DataMapOfShapeListOfShape aDM1;
193 //TCollection_AsciiString aNamM ("EdgM_");
194 //TCollection_AsciiString aNamG ("EdgG_");
195 for (int i =1;anIt.More();anIt.Next(),i++) {
196 Standard_Boolean foundF(Standard_False);
197 const TopTools_ListOfShape& aListOfNew = aBuilder->Modified(anIt.Value());
198 if(!aListOfNew.IsEmpty())
199 foundF = Standard_True;
200 TopTools_ListOfShape aList;
201 TopTools_ListIteratorOfListOfShape it(aListOfNew);
202 for(;it.More();it.Next())
203 aList.Append(it.Value());
204 // Bug in History: partition should give only modified entities! => temporary solution is used
205 const TopTools_ListOfShape& aListOfGen = aBuilder->Generated(anIt.Value());
206 if(!aListOfGen.IsEmpty())
207 foundF = Standard_True;
208 it.Initialize(aListOfGen);
209 for(;it.More();it.Next())
210 aList.Append(it.Value());
211 if(!foundF) // face is not modified
212 aList.Append (anIt.Value());
213 aDM1.Bind(anIt.Value(), aList);
214 //TCollection_AsciiString aName;
215 if(!anInputGroupList.isEmpty() ) { // 1
217 TopExp_Explorer exp (anIt.Value(), TopAbs_EDGE);
218 for (int j =1;exp.More();exp.Next(),j++) {
219 Standard_Boolean foundE(Standard_False);
220 const TopTools_ListOfShape& aListM = aBuilder->Modified(exp.Current());
221 //cout << "NB_EDGE_M = " << aListM.Extent() <<endl;
222 if(aListM.Extent()) foundE = Standard_True;
223 it.Initialize(aListM);
224 for(int k=1;it.More();it.Next(),k++) {
225 aList.Append(it.Value());
226 //aName = aNamM + i + j +k +".brep";
227 //BRepTools::Write(it.Value(),aName.ToCString());
229 const TopTools_ListOfShape& aListG = aBuilder->Generated(exp.Current());
230 if(aListG.Extent()) foundE = Standard_True;
231 it.Initialize(aListG);
232 for(int k=1;it.More();it.Next(),k++) {
233 aList.Append(it.Value());
234 //aName = aNamG + i + j +k +".brep";
235 //BRepTools::Write(it.Value(),aName.ToCString());
237 //cout << "NB_EDGE = " << aList.Extent() <<endl;
239 aList.Append (exp.Current());
240 //aName = aNamG + i + j +".brep";
241 //BRepTools::Write(exp.Current(),aName.ToCString());
243 aDM1.Bind(exp.Current(), aList);
248 // aDM2: NewShape ==> ListOfOldShapes
249 HYDROData_DataMapOfShapeListOfShape aDM2;
250 HYDROData_DataMapIteratorOfDataMapOfShapeListOfShape aMIt(aDM1);
251 for(;aMIt.More();aMIt.Next()) {
252 const TopoDS_Shape& aKey = aMIt.Key();
253 TopTools_ListOfShape aList;
255 const TopTools_ListOfShape& aListOfNew = aMIt.Value();
256 TopTools_ListIteratorOfListOfShape it(aListOfNew);
257 for(;it.More();it.Next()) {
258 if(!aDM2.IsBound(it.Value()))
259 aDM2.Bind(it.Value(), aList);
261 TopTools_ListOfShape& aList = aDM2.ChangeFind(it.Value());
266 //cout << "DM2 Ext = " <<aDM2.Extent() <<endl;
267 // Step 4. Fill output structure.
268 aMIt.Initialize(aDM2);
269 for(int i =1;aMIt.More();aMIt.Next(),i++) {
270 SplitData aDestSplitData;
271 const TopoDS_Shape& aKey = aMIt.Key(); //new
272 aDestSplitData.Shape = aKey;
273 if(aKey.ShapeType() == TopAbs_FACE)
274 aDestSplitData.Type = SplitData::Data_Zone;
276 aDestSplitData.Type = SplitData::Data_Edge;
278 QStringList aListOfNames; // names processing
279 const TopTools_ListOfShape& aListOfOld = aMIt.Value();
280 TopTools_ListIteratorOfListOfShape it(aListOfOld);
281 for(;it.More();it.Next()) {
282 const TopoDS_Shape& aSh = it.Value(); //old
283 if(aDM3.IsBound(aSh)) {
284 const QStringList& ObjectNames = aDM3.Find(aSh);
285 aListOfNames.append(ObjectNames);
289 aDestSplitData.ObjectNames = aListOfNames;
290 anOutputSplitDataList.append(aDestSplitData);
292 //cout << "anOutputSplitDataList = " <<anOutputSplitDataList.size() <<endl;
293 return anOutputSplitDataList;
296 // Step 3. Extract the separate regions.
297 SplitDataList anExtractedSplitDataList;
298 SplitDataListIterator anOutputIter( anOutputSplitDataList );
299 while( anOutputIter.hasNext() )
301 const SplitData& anOutputSplitData = anOutputIter.next();
302 anExtractedSplitDataList.append( ExtractSeparateData( anOutputSplitData ) );
305 // return anExtractedSplitDataList;
308 bool HYDROData_SplitToZonesTool::SplitTwoData( const SplitData& theData1,
309 const SplitData& theData2,
310 SplitData& theData1Subtracted,
311 SplitData& theData2Subtracted,
312 SplitData& theDataIntersected )
314 const TopoDS_Shape& aShape1 = theData1.Shape;
315 const TopoDS_Shape& aShape2 = theData2.Shape;
317 const QStringList& anObjectNames1 = theData1.ObjectNames;
318 const QStringList& anObjectNames2 = theData2.ObjectNames;
320 BRepAlgoAPI_Common aCommon( aShape1, aShape2 );
321 TopoDS_Shape aCommonShape = aCommon.Shape();
322 if( aCommonShape.IsNull() )
324 theData1Subtracted = theData1;
325 theData2Subtracted = theData2;
329 BRepAlgoAPI_Cut aCut1( aShape1, aShape2 );
330 TopoDS_Shape aCut1Shape = aCut1.Shape();
332 BRepAlgoAPI_Cut aCut2( aShape2, aShape1 );
333 TopoDS_Shape aCut2Shape = aCut2.Shape();
335 theData1Subtracted = SplitData( SplitData::Data_Zone, aCut1Shape, anObjectNames1 );
336 theData2Subtracted = SplitData( SplitData::Data_Zone, aCut2Shape, anObjectNames2 );
337 theDataIntersected = SplitData( SplitData::Data_Zone, aCommonShape, anObjectNames1 + anObjectNames2 );
342 HYDROData_SplitToZonesTool::SplitDataList
343 HYDROData_SplitToZonesTool::ExtractSeparateData( const SplitData& theData )
345 SplitDataList aSplitDataList;
346 TopExp_Explorer anExp( theData.Shape, TopAbs_FACE );
347 for( ; anExp.More(); anExp.Next() )
349 TopoDS_Shape aShape = anExp.Current();
350 if( aShape.ShapeType() == TopAbs_FACE )
352 TopoDS_Face aFace = TopoDS::Face( aShape );
353 if( !aFace.IsNull() )
355 SplitData aSplitData( SplitData::Data_Zone, aFace, theData.ObjectNames );
356 aSplitDataList.append( aSplitData );
360 return aSplitDataList;