1 // Copyright (C) 2014-2019 CEA/DEN, EDF R&D
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Lesser General Public
5 // License as published by the Free Software Foundation; either
6 // version 2.1 of the License, or (at your option) any later version.
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11 // Lesser General Public License for more details.
13 // You should have received a copy of the GNU Lesser General Public
14 // License along with this library; if not, write to the Free Software
15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
20 #include <GeomAPI_DataMapOfShapeMapOfShapes.h>
22 #include <NCollection_DataMap.hxx>
23 #include <NCollection_Map.hxx>
24 #include <TopoDS_Shape.hxx>
26 typedef NCollection_DataMap<TopoDS_Shape, NCollection_Map<TopoDS_Shape> > MAP;
27 #define MY_MAP implPtr<MAP>()
29 //=================================================================================================
30 GeomAPI_DataMapOfShapeMapOfShapes::GeomAPI_DataMapOfShapeMapOfShapes()
31 : GeomAPI_Interface(new NCollection_DataMap<TopoDS_Shape, NCollection_Map<TopoDS_Shape> >)
34 //=================================================================================================
35 bool GeomAPI_DataMapOfShapeMapOfShapes::bind(const std::shared_ptr<GeomAPI_Shape> theKey,
36 const ListOfShape& theItems)
38 const TopoDS_Shape& aKey = theKey->impl<TopoDS_Shape>();
39 if(MY_MAP->IsBound(aKey)) {
40 MY_MAP->ChangeFind(aKey).Clear();
42 for(ListOfShape::const_iterator anIt = theItems.cbegin(); anIt != theItems.cend(); anIt++) {
43 const TopoDS_Shape& anItem = (*anIt)->impl<TopoDS_Shape>();
44 if(MY_MAP->IsBound(aKey)) {
45 MY_MAP->ChangeFind(aKey).Add(anItem);
47 NCollection_Map<TopoDS_Shape> anItems;
49 MY_MAP->Bind(aKey, anItems);
56 //=================================================================================================
57 bool GeomAPI_DataMapOfShapeMapOfShapes::add(const std::shared_ptr<GeomAPI_Shape> theKey,
58 const std::shared_ptr<GeomAPI_Shape> theItem)
60 const TopoDS_Shape& aKey = theKey->impl<TopoDS_Shape>();
61 const TopoDS_Shape& anItem = theItem->impl<TopoDS_Shape>();
62 if(MY_MAP->IsBound(aKey)) {
63 return MY_MAP->ChangeFind(aKey).Add(anItem) == Standard_True;
65 NCollection_Map<TopoDS_Shape> anItems;
67 return MY_MAP->Bind(aKey, anItems) == Standard_True;
71 //=================================================================================================
72 bool GeomAPI_DataMapOfShapeMapOfShapes::isBound(const std::shared_ptr<GeomAPI_Shape> theKey) const
74 const TopoDS_Shape& aKey = theKey->impl<TopoDS_Shape>();
75 return MY_MAP->IsBound(aKey) == Standard_True;
78 //=================================================================================================
79 bool GeomAPI_DataMapOfShapeMapOfShapes::find(const std::shared_ptr<GeomAPI_Shape> theKey,
80 ListOfShape& theItems) const
82 const TopoDS_Shape& aKey = theKey->impl<TopoDS_Shape>();
84 if(MY_MAP->IsBound(aKey) == Standard_False) {
88 const NCollection_Map<TopoDS_Shape>& aMap = MY_MAP->Find(aKey);
89 for(NCollection_Map<TopoDS_Shape>::Iterator anIt(aMap); anIt.More(); anIt.Next()) {
90 std::shared_ptr<GeomAPI_Shape> aShape(new GeomAPI_Shape);
91 aShape->setImpl(new TopoDS_Shape(anIt.Value()));
92 theItems.push_back(aShape);
98 //=================================================================================================
99 bool GeomAPI_DataMapOfShapeMapOfShapes::unBind(const std::shared_ptr<GeomAPI_Shape> theKey)
101 const TopoDS_Shape& aKey = theKey->impl<TopoDS_Shape>();
102 return MY_MAP->UnBind(aKey) == Standard_True;
105 //=================================================================================================
106 void GeomAPI_DataMapOfShapeMapOfShapes::clear()
108 return MY_MAP->Clear();
111 //=================================================================================================
112 int GeomAPI_DataMapOfShapeMapOfShapes::size() const
114 return MY_MAP->Size();
118 //=================================================================================================
119 // iterator implementation
120 //=================================================================================================
122 class IteratorImpl : public GeomAPI_DataMapOfShapeMapOfShapes::iterator
127 IteratorImpl(const MAP::Iterator& theIterator)
128 : myIterator(theIterator)
131 IteratorImpl(const std::shared_ptr<IteratorImpl>& theIterator)
133 mySelf = theIterator;
136 bool operator==(const std::shared_ptr<IteratorImpl>& theOther) const
139 return theOther->myIterator.IsEqual(myIterator);
141 // theOther is end iterator => check the current iterator is not at the end
142 return !myIterator.More();
145 virtual iterator& operator++()
150 virtual iterator operator++(int)
152 std::shared_ptr<IteratorImpl> aSelf = std::dynamic_pointer_cast<IteratorImpl>(mySelf);
153 std::shared_ptr<IteratorImpl> aCopy(new IteratorImpl(aSelf->myIterator));
155 return IteratorImpl(aCopy);
158 virtual key_type first() const
161 return iterator::first();
163 GeomShapePtr aShape(new GeomAPI_Shape);
164 aShape->setImpl(new TopoDS_Shape(myIterator.Key()));
168 virtual mapped_type second() const
171 return iterator::second();
174 const NCollection_Map<TopoDS_Shape>& aMap = myIterator.Value();
175 for(NCollection_Map<TopoDS_Shape>::Iterator anIt(aMap); anIt.More(); anIt.Next()) {
176 std::shared_ptr<GeomAPI_Shape> aShape(new GeomAPI_Shape);
177 aShape->setImpl(new TopoDS_Shape(anIt.Value()));
178 anItems.push_back(aShape);
184 MAP::Iterator myIterator;
188 GeomAPI_DataMapOfShapeMapOfShapes::iterator::iterator()
192 GeomAPI_DataMapOfShapeMapOfShapes::iterator::iterator(
193 const GeomAPI_DataMapOfShapeMapOfShapes::iterator& theOther)
194 : mySelf(theOther.mySelf)
198 const GeomAPI_DataMapOfShapeMapOfShapes::iterator&
199 GeomAPI_DataMapOfShapeMapOfShapes::iterator::operator=(
200 const GeomAPI_DataMapOfShapeMapOfShapes::iterator& theOther)
202 mySelf = theOther.mySelf;
206 bool GeomAPI_DataMapOfShapeMapOfShapes::iterator::operator==(const iterator& theOther) const
208 std::shared_ptr<IteratorImpl> aSelf = std::dynamic_pointer_cast<IteratorImpl>(mySelf);
209 std::shared_ptr<IteratorImpl> aSelfOther =
210 std::dynamic_pointer_cast<IteratorImpl>(theOther.mySelf);
212 return aSelf ? aSelf->operator==(aSelfOther)
213 : (aSelfOther ? aSelfOther->operator==(aSelf) : true);
216 bool GeomAPI_DataMapOfShapeMapOfShapes::iterator::operator!=(const iterator& theOther) const
218 return !operator==(theOther);
221 GeomAPI_DataMapOfShapeMapOfShapes::iterator&
222 GeomAPI_DataMapOfShapeMapOfShapes::iterator::operator++()
224 mySelf->operator++();
228 GeomAPI_DataMapOfShapeMapOfShapes::iterator
229 GeomAPI_DataMapOfShapeMapOfShapes::iterator::operator++(int)
234 GeomAPI_DataMapOfShapeMapOfShapes::iterator::key_type
235 GeomAPI_DataMapOfShapeMapOfShapes::iterator::first() const
237 return mySelf->first();
240 GeomAPI_DataMapOfShapeMapOfShapes::iterator::mapped_type
241 GeomAPI_DataMapOfShapeMapOfShapes::iterator::second() const
243 return mySelf->second();
248 GeomAPI_DataMapOfShapeMapOfShapes::iterator GeomAPI_DataMapOfShapeMapOfShapes::begin()
250 MAP::Iterator anIt(*MY_MAP);
251 std::shared_ptr<IteratorImpl> anIter(new IteratorImpl(anIt));
252 return IteratorImpl(anIter);
255 GeomAPI_DataMapOfShapeMapOfShapes::const_iterator GeomAPI_DataMapOfShapeMapOfShapes::begin() const
257 MAP::Iterator anIt(*MY_MAP);
258 std::shared_ptr<IteratorImpl> anIter(new IteratorImpl(anIt));
259 return IteratorImpl(anIter);
262 GeomAPI_DataMapOfShapeMapOfShapes::iterator GeomAPI_DataMapOfShapeMapOfShapes::end()
264 return IteratorImpl(std::shared_ptr<IteratorImpl>());
267 GeomAPI_DataMapOfShapeMapOfShapes::const_iterator GeomAPI_DataMapOfShapeMapOfShapes::end() const
269 return IteratorImpl(std::shared_ptr<IteratorImpl>());