]> SALOME platform Git repositories - modules/shaper.git/blob - src/Model/Model_ResultBody.cpp
Salome HOME
Merge branch 'master' of newgeom:newgeom
[modules/shaper.git] / src / Model / Model_ResultBody.cpp
1 // File:        Model_ResultBody.cpp
2 // Created:     08 Jul 2014
3 // Author:      Mikhail PONIKAROV
4
5 #include <Model_ResultBody.h>
6 #include <Model_Data.h>
7 #include <TNaming_Builder.hxx>
8 #include <TNaming_NamedShape.hxx>
9 #include <TopoDS_Shape.hxx>
10 #include <TDF_ChildIterator.hxx>
11 #include <TopTools_MapOfShape.hxx>
12 #include <TopExp_Explorer.hxx>
13 #include <GeomAPI_Shape.h>
14 #include <GeomAlgoAPI_MakeShape.h>
15 // DEB
16 //#include <TCollection_AsciiString.hxx>
17 //#include <TDF_Tool.hxx>
18 Model_ResultBody::Model_ResultBody()
19 {
20   setIsConcealed(false);
21 }
22
23 void Model_ResultBody::store(const boost::shared_ptr<GeomAPI_Shape>& theShape)
24 {
25   boost::shared_ptr<Model_Data> aData = boost::dynamic_pointer_cast<Model_Data>(data());
26   if (aData) {
27     TDF_Label& aShapeLab = aData->shapeLab();
28     // clean builders
29     clean();   
30     // store the new shape as primitive
31     TNaming_Builder aBuilder(aShapeLab);
32     if (!theShape)
33       return;  // bad shape
34     TopoDS_Shape aShape = theShape->impl<TopoDS_Shape>();
35     if (aShape.IsNull())
36       return;  // null shape inside
37
38     aBuilder.Generated(aShape);
39   }
40 }
41
42 void Model_ResultBody::storeGenerated(const boost::shared_ptr<GeomAPI_Shape>& theFromShape,
43                                           const boost::shared_ptr<GeomAPI_Shape>& theToShape)
44 {
45   boost::shared_ptr<Model_Data> aData = boost::dynamic_pointer_cast<Model_Data>(data());
46   if (aData) {
47     TDF_Label& aShapeLab = aData->shapeLab();
48     // clean builders
49     clean();   
50     // store the new shape as primitive
51     TNaming_Builder aBuilder(aShapeLab);
52     if (!theFromShape || !theToShape)
53       return;  // bad shape
54     TopoDS_Shape aShapeBasis = theFromShape->impl<TopoDS_Shape>();
55     if (aShapeBasis.IsNull())
56       return;  // null shape inside
57         TopoDS_Shape aShapeNew = theToShape->impl<TopoDS_Shape>();
58     if (aShapeNew.IsNull())
59       return;  // null shape inside
60     aBuilder.Generated(aShapeBasis, aShapeNew);
61   }
62 }
63
64 void Model_ResultBody::storeModified(const boost::shared_ptr<GeomAPI_Shape>& theOldShape,
65                                           const boost::shared_ptr<GeomAPI_Shape>& theNewShape)
66 {
67   boost::shared_ptr<Model_Data> aData = boost::dynamic_pointer_cast<Model_Data>(data());
68   if (aData) {
69     TDF_Label& aShapeLab = aData->shapeLab();
70     // clean builders
71     clean();   
72     // store the new shape as primitive
73     TNaming_Builder aBuilder(aShapeLab);
74     if (!theOldShape || !theNewShape)
75       return;  // bad shape
76     TopoDS_Shape aShapeOld = theOldShape->impl<TopoDS_Shape>();
77     if (aShapeOld.IsNull())
78       return;  // null shape inside
79         TopoDS_Shape aShapeNew = theNewShape->impl<TopoDS_Shape>();
80     if (aShapeNew.IsNull())
81       return;  // null shape inside
82     aBuilder.Generated(aShapeOld, aShapeNew);
83   }
84 }
85
86 boost::shared_ptr<GeomAPI_Shape> Model_ResultBody::shape()
87 {
88   boost::shared_ptr<Model_Data> aData = boost::dynamic_pointer_cast<Model_Data>(data());
89   if (aData) {
90     TDF_Label& aShapeLab = aData->shapeLab();
91     Handle(TNaming_NamedShape) aName;
92     if (aShapeLab.FindAttribute(TNaming_NamedShape::GetID(), aName)) {
93       TopoDS_Shape aShape = aName->Get();
94       if (!aShape.IsNull()) {
95         boost::shared_ptr<GeomAPI_Shape> aRes(new GeomAPI_Shape);
96         aRes->setImpl(new TopoDS_Shape(aShape));
97         return aRes;
98       }
99     }
100   }
101   return boost::shared_ptr<GeomAPI_Shape>();
102 }
103
104 void Model_ResultBody::clean()
105 {
106   std::vector<TNaming_Builder*>::iterator aBuilder = myBuilders.begin();
107   for(; aBuilder != myBuilders.end(); aBuilder++)
108     delete *aBuilder;
109   myBuilders.clear();
110 }
111
112 Model_ResultBody::~Model_ResultBody()
113 {
114   clean();
115 }
116
117 TNaming_Builder* Model_ResultBody::builder(const int theTag)
118 {
119   if (myBuilders.size() <= (unsigned int)theTag) {
120     myBuilders.insert(myBuilders.end(), theTag - myBuilders.size() + 1, NULL);
121   }
122   if (!myBuilders[theTag]) {
123     boost::shared_ptr<Model_Data> aData = boost::dynamic_pointer_cast<Model_Data>(data());
124     myBuilders[theTag] = new TNaming_Builder(aData->shapeLab().FindChild(theTag));
125         //TCollection_AsciiString entry;//
126         //TDF_Tool::Entry(aData->shapeLab().FindChild(theTag), entry);
127         //cout << "Label = " <<entry.ToCString() <<endl;
128   }
129   return myBuilders[theTag];
130 }
131
132 void Model_ResultBody::generated(
133   const boost::shared_ptr<GeomAPI_Shape>& theNewShape, const int theTag)
134 {
135   TopoDS_Shape aShape = theNewShape->impl<TopoDS_Shape>();
136   builder(theTag)->Generated(aShape);
137 }
138
139 void Model_ResultBody::generated(const boost::shared_ptr<GeomAPI_Shape>& theOldShape,
140     const boost::shared_ptr<GeomAPI_Shape>& theNewShape, const int theTag)
141 {
142   TopoDS_Shape anOldShape = theOldShape->impl<TopoDS_Shape>();
143   TopoDS_Shape aNewShape = theNewShape->impl<TopoDS_Shape>();
144   builder(theTag)->Generated(anOldShape, aNewShape);
145 }
146
147
148 void Model_ResultBody::modified(const boost::shared_ptr<GeomAPI_Shape>& theOldShape,
149     const boost::shared_ptr<GeomAPI_Shape>& theNewShape, const int theTag)
150 {
151   TopoDS_Shape anOldShape = theOldShape->impl<TopoDS_Shape>();
152   TopoDS_Shape aNewShape = theNewShape->impl<TopoDS_Shape>();
153   builder(theTag)->Modify(anOldShape, aNewShape);
154 }
155
156 void Model_ResultBody::deleted(const boost::shared_ptr<GeomAPI_Shape>& theOldShape,
157     const int theTag)
158 {
159   TopoDS_Shape aShape = theOldShape->impl<TopoDS_Shape>();
160   builder(theTag)->Delete(aShape);
161 }
162
163 void Model_ResultBody::loadDeletedShapes (GeomAlgoAPI_MakeShape* theMS,
164                                                boost::shared_ptr<GeomAPI_Shape>  theShapeIn,
165                                                const int  theKindOfShape,
166                                                const int  theTag)
167 {
168   TopoDS_Shape aShapeIn = theShapeIn->impl<TopoDS_Shape>();
169   TopTools_MapOfShape aView;
170   TopExp_Explorer ShapeExplorer (aShapeIn, (TopAbs_ShapeEnum)theKindOfShape);
171   for (; ShapeExplorer.More(); ShapeExplorer.Next ()) {
172     const TopoDS_Shape& aRoot = ShapeExplorer.Current ();
173     if (!aView.Add(aRoot)) continue;
174         boost::shared_ptr<GeomAPI_Shape> aRShape(new GeomAPI_Shape());
175         aRShape->setImpl((new TopoDS_Shape(aRoot)));
176     if (theMS->isDeleted (aRShape)) {
177                 builder(theTag)->Delete(aRoot);
178     }
179   }
180 }
181
182 void Model_ResultBody::loadAndOrientModifiedShapes (
183                                                    GeomAlgoAPI_MakeShape* theMS,
184                                                boost::shared_ptr<GeomAPI_Shape>  theShapeIn,
185                                                const int  theKindOfShape,
186                                                const int  theTag,
187                                                GeomAPI_DataMapOfShapeShape& theSubShapes)
188 {
189   TopoDS_Shape aShapeIn = theShapeIn->impl<TopoDS_Shape>();
190   TopTools_MapOfShape aView;
191   TopExp_Explorer aShapeExplorer (aShapeIn, (TopAbs_ShapeEnum)theKindOfShape);
192   for (; aShapeExplorer.More(); aShapeExplorer.Next ()) {
193     const TopoDS_Shape& aRoot = aShapeExplorer.Current ();
194     if (!aView.Add(aRoot)) continue;
195         ListOfShape aList;
196         boost::shared_ptr<GeomAPI_Shape> aRShape(new GeomAPI_Shape());
197         aRShape->setImpl((new TopoDS_Shape(aRoot)));
198         theMS->generated(aRShape, aList);
199         std::list<boost::shared_ptr<GeomAPI_Shape> >::const_iterator anIt = aList.begin(), aLast = aList.end();
200     for (; anIt != aLast; anIt++) {
201       TopoDS_Shape aNewShape = (*anIt)->impl<TopoDS_Shape>();     
202           if (theSubShapes.isBound(*anIt)) {
203                 boost::shared_ptr<GeomAPI_Shape> aMapShape(theSubShapes.find(*anIt));
204                 aNewShape.Orientation(aMapShape->impl<TopoDS_Shape>().Orientation());
205       }
206       if (!aRoot.IsSame (aNewShape)) 
207                   builder(theTag)->Modify(aRoot,aNewShape);
208     }
209   }
210 }
211
212 void Model_ResultBody::loadAndOrientGeneratedShapes (
213                                                    GeomAlgoAPI_MakeShape* theMS,
214                                                boost::shared_ptr<GeomAPI_Shape>  theShapeIn,
215                                                const int  theKindOfShape,
216                                                const int  theTag,
217                                                GeomAPI_DataMapOfShapeShape& theSubShapes)
218 {
219   TopoDS_Shape aShapeIn = theShapeIn->impl<TopoDS_Shape>();
220   TopTools_MapOfShape aView;
221   TopExp_Explorer aShapeExplorer (aShapeIn, (TopAbs_ShapeEnum)theKindOfShape);
222   for (; aShapeExplorer.More(); aShapeExplorer.Next ()) {
223     const TopoDS_Shape& aRoot = aShapeExplorer.Current ();
224     if (!aView.Add(aRoot)) continue;
225         ListOfShape aList;
226         boost::shared_ptr<GeomAPI_Shape> aRShape(new GeomAPI_Shape());
227         aRShape->setImpl((new TopoDS_Shape(aRoot)));
228         theMS->generated(aRShape, aList);
229         std::list<boost::shared_ptr<GeomAPI_Shape> >::const_iterator anIt = aList.begin(), aLast = aList.end();
230     for (; anIt != aLast; anIt++) {
231       TopoDS_Shape aNewShape = (*anIt)->impl<TopoDS_Shape>();     
232           if (theSubShapes.isBound(*anIt)) {
233                 boost::shared_ptr<GeomAPI_Shape> aMapShape(theSubShapes.find(*anIt));
234                 aNewShape.Orientation(aMapShape->impl<TopoDS_Shape>().Orientation());
235       }
236       if (!aRoot.IsSame (aNewShape)) 
237                   builder(theTag)->Generated(aRoot,aNewShape);
238     }
239   }
240 }