1 // Copyright (C) 2013 CEA/DEN, EDF R&D, OPEN CASCADE
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.
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 // File: CurveCreator_Curve.cxx
21 // Author: Sergey KHROMOV
23 #include "CurveCreator_Curve.hxx"
25 #include "CurveCreator.hxx"
26 #include "CurveCreator_Section.hxx"
27 #include "CurveCreator_Displayer.h"
29 #include <AIS_Point.hxx>
30 #include <AIS_Line.hxx>
31 #include <AIS_Shape.hxx>
32 #include <BRepBuilderAPI_MakeEdge.hxx>
33 #include <BRepBuilderAPI_MakeWire.hxx>
34 #include <Geom_CartesianPoint.hxx>
37 #include <TopoDS_Edge.hxx>
38 #include <TopoDS_Face.hxx>
39 #include <TopoDS_Wire.hxx>
40 #include <TColgp_HArray1OfPnt.hxx>
41 #include <GeomAPI_Interpolate.hxx>
45 //=======================================================================
46 // function: Constructor
48 //=======================================================================
49 CurveCreator_Curve::CurveCreator_Curve( const CurveCreator::Dimension theDimension )
51 myDimension (theDimension),
60 //=======================================================================
61 // function: Destructor
63 //=======================================================================
64 CurveCreator_Curve::~CurveCreator_Curve()
66 // Delete all allocated data.
70 //=======================================================================
71 // function: getDimension
73 //=======================================================================
74 CurveCreator::Dimension CurveCreator_Curve::getDimension() const
79 //=======================================================================
80 // function: getUniqSectionName
81 // purpose: return unique section name
82 //=======================================================================
83 std::string CurveCreator_Curve::getUniqSectionName() const
85 for( int i = 0 ; i < 1000000 ; i++ ){
87 sprintf( aBuffer, "Section_%d", i+1 );
88 std::string aName(aBuffer);
90 for( j = 0 ; j < mySections.size() ; j++ ){
91 if( mySections[j]->myName == aName )
94 if( j == mySections.size() )
100 //=======================================================================
101 // function: setDisplayer
102 // purpose: set curve changes Displayer
103 //=======================================================================
104 void CurveCreator_Curve::setDisplayer( CurveCreator_Displayer* theDisplayer )
106 myDisplayer = theDisplayer;
109 //=======================================================================
110 // function: removeDisplayer
111 // purpose: remove the attached Displayer
112 //=======================================================================
113 void CurveCreator_Curve::removeDisplayer()
118 //=======================================================================
121 //=======================================================================
122 bool CurveCreator_Curve::addEmptyDiff()
124 bool isEnabled = false;
126 if (myUndoDepth != 0) {
127 // Forget all Redos after the current one.
130 myListDiffs.erase(myCurrenPos, myListDiffs.end());
133 if (myUndoDepth == -1 || myNbUndos < myUndoDepth) {
134 // Increase the number of undos.
137 // If there are too many differences, remove the first one.
138 myListDiffs.pop_front();
141 // Add new difference.
142 myListDiffs.push_back(CurveCreator_Diff());
143 myCurrenPos = myListDiffs.end();
150 void CurveCreator_Curve::startOperation()
155 void CurveCreator_Curve::finishOperation()
160 //=======================================================================
161 // function: toICoord
163 //=======================================================================
164 int CurveCreator_Curve::toICoord(const int theIPnt) const
166 return theIPnt * myDimension;
169 std::vector<Handle_AIS_InteractiveObject> CurveCreator_Curve::constructSection( int theISection ) const
171 std::vector<Handle_AIS_InteractiveObject> aSectionRepresentation;
173 CurveCreator::SectionType aSectType = getSectionType( theISection );
174 int aSectSize = getNbPoints( theISection );
175 bool aSectIsClosed = isClosed( theISection );
177 if( aSectType == CurveCreator::Polyline )
180 for( ; iPoint < ( aSectSize - 1 ) ; iPoint++ ){
181 Handle_AIS_Point anAISPnt = getAISPoint(theISection, iPoint);
182 aSectionRepresentation.push_back( anAISPnt );
183 Handle_AIS_Line aLine = getAISLine( theISection, iPoint, iPoint+1 );
184 aSectionRepresentation.push_back( aLine );
186 if( aSectSize != 0 ){
187 Handle_AIS_Point anAISPnt = getAISPoint(theISection, iPoint);
188 aSectionRepresentation.push_back( anAISPnt );
189 if( isClosed(theISection) && ( aSectSize > 1 ) ){
190 Handle_AIS_Line aLine = getAISLine( theISection, iPoint, 0 );
191 aSectionRepresentation.push_back( aLine );
195 else if( aSectType == CurveCreator::Spline )
197 std::vector<double> aPoints;
198 for( int iPoint = 0; iPoint < aSectSize; iPoint++ )
200 Handle_AIS_Point anAISPnt = getAISPoint( theISection, iPoint );
201 aSectionRepresentation.push_back( anAISPnt );
203 CurveCreator::Coordinates aCoords = getPoint( theISection, iPoint );
204 double aX = aCoords[0];
205 double aY = aCoords[1];
206 double aZ = aCoords[2];
207 aPoints.push_back( aX );
208 aPoints.push_back( aY );
213 Handle(Geom_BSplineCurve) aBSplineCurve;
214 // fill array for algorithm by the received coordinates
215 int aLen = aPoints.size() / 2;
216 Handle(TColgp_HArray1OfPnt) aHCurvePoints = new TColgp_HArray1OfPnt (1, aLen);
217 std::vector<double>::const_iterator aListIter = aPoints.begin();
218 for (int ind = 1; ind <= aLen; ind++) {
219 gp_Pnt aPnt(gp::Origin());
220 aPnt.SetX(*aListIter);
222 aPnt.SetY(*aListIter);
225 aHCurvePoints->SetValue(ind, aPnt);
228 GeomAPI_Interpolate aGBC(aHCurvePoints, aSectIsClosed, gp::Resolution());
231 aBSplineCurve = aGBC.Curve();
233 TopoDS_Edge anEdge = BRepBuilderAPI_MakeEdge( aBSplineCurve ).Edge();
235 TopoDS_Wire aWire = BRepBuilderAPI_MakeWire( anEdge ).Wire();
237 Handle(AIS_Shape) aShape = new AIS_Shape( aWire );
238 aSectionRepresentation.push_back( aShape );
241 return aSectionRepresentation;
244 //=======================================================================
245 // function: setUndoDepth
247 //=======================================================================
248 void CurveCreator_Curve::setUndoDepth(const int theDepth)
251 // Reset all undo/redo data.
255 myCurrenPos = myListDiffs.end();
257 } else if (theDepth == -1) {
258 // There is nothing to do as the depth become unlimited.
260 } else if (theDepth > 0) {
261 // The new "real" depth is set.
262 if (theDepth < myNbRedos) {
263 // The new depth is less then number of redos. Remove the latest redos.
264 int aShift = (myNbRedos - theDepth);
265 ListDiff::iterator aFromPos = myListDiffs.end();
271 myListDiffs.erase(aFromPos, myListDiffs.end());
272 myNbRedos = theDepth;
275 if (theDepth < myNbUndos + myNbRedos) {
276 // The new depth is less then the total number of differences.
277 // Remove the first undos.
278 int aShift = (myNbUndos + myNbRedos - theDepth);
279 ListDiff::iterator aToPos = myListDiffs.begin();
285 myListDiffs.erase(myListDiffs.begin(), aToPos);
286 myNbUndos = theDepth - myNbRedos;
289 myUndoDepth = theDepth;
293 //=======================================================================
294 // function: getUndoDepth
296 //=======================================================================
297 int CurveCreator_Curve::getUndoDepth() const
302 Handle_AIS_Point CurveCreator_Curve::getAISPoint( int theISection, int theIPoint ) const
305 getCoordinates( theISection, theIPoint, anX, anY, aZ );
306 gp_Pnt aPoint( anX, anY, aZ);
308 AIS_Point* aPnt = new AIS_Point( new Geom_CartesianPoint(aPoint));
312 Handle_AIS_Line CurveCreator_Curve::getAISLine( int theISection, int theIPoint1, int theIPoint2 ) const
315 getCoordinates( theISection, theIPoint1, anX, anY, aZ );
316 gp_Pnt aPoint1( anX, anY, aZ);
318 double anX2, anY2, aZ2;
319 getCoordinates( theISection, theIPoint2, anX2, anY2, aZ2 );
320 //MTN to avoid crash during line construction
321 if( ( anX == anX2 ) && ( anY == anY2 ) && (aZ == aZ2 ) ){
325 gp_Pnt aPoint2( anX2, anY2, aZ2 );
327 AIS_Line* aLine = new AIS_Line( new Geom_CartesianPoint(aPoint1), new Geom_CartesianPoint(aPoint2) );
331 void CurveCreator_Curve::getCoordinates( int theISection, int theIPoint, double& theX, double& theY, double& theZ ) const
333 CurveCreator::Coordinates aCoords = getPoint( theISection, theIPoint );
337 if( getDimension() == CurveCreator::Dim3d ){
342 void CurveCreator_Curve::redisplayCurve()
345 myDisplayer->erase();
346 myDisplayer->display( constructWire() );
350 /************ Implementation of INTERFACE methods ************/
352 /***********************************************/
353 /*** Undo/Redo methods ***/
354 /***********************************************/
356 //! Get number of available undo operations
357 int CurveCreator_Curve::getNbUndo() const
362 //! Undo previous operation
363 bool CurveCreator_Curve::undo()
370 myCurrenPos->applyUndo(this);
376 //! Get number of available redo operations
377 int CurveCreator_Curve::getNbRedo() const
382 //! Redo last previously "undone" operation
383 bool CurveCreator_Curve::redo()
387 myCurrenPos->applyRedo(this);
396 /***********************************************/
397 /*** Section methods ***/
398 /***********************************************/
399 //=======================================================================
402 //=======================================================================
403 bool CurveCreator_Curve::clear()
405 // erase curve from the viewer
407 myDisplayer->erase();
408 // Delete all allocated data.
410 const int aNbSections = getNbSections();
412 for (; i < aNbSections; i++) {
413 delete mySections[i];
421 //! Join range of sections to one section (join all sections if -1 is passed in one of arguments)
422 bool CurveCreator_Curve::join( const int theISectionTo,
423 const int theISectionFrom )
425 if (theISectionTo != theISectionFrom) {
428 myListDiffs.back().init(this, CurveCreator_Operation::Join, theISectionTo, theISectionFrom);
430 CurveCreator_Section *aSection1 = mySections.at(theISectionTo);
431 CurveCreator_Section *aSection2 = mySections.at(theISectionFrom);
433 aSection1->myPoints.insert(aSection1->myPoints.end(),
434 aSection2->myPoints.begin(),
435 aSection2->myPoints.end());
437 removeSection(theISectionFrom);
446 //! Get number of sections
447 int CurveCreator_Curve::getNbSections() const
449 return mySections.size();
452 //! For internal use only! Undo/Redo is not used here.
453 int CurveCreator_Curve::addSectionInternal
454 (const std::string& theName, const CurveCreator::SectionType theType,
455 const bool theIsClosed, const CurveCreator::Coordinates &thePoints)
457 CurveCreator_Section *aSection = new CurveCreator_Section;
459 std::string aName = theName;
461 aName = getUniqSectionName();
463 aSection->myName = aName;
464 aSection->myType = theType;
465 aSection->myIsClosed = theIsClosed;
466 aSection->myPoints = thePoints;
467 mySections.push_back(aSection);
469 return mySections.size()-1;
472 //=======================================================================
473 // function: addSection
474 // purpose: adds an empty section
475 //=======================================================================
476 int CurveCreator_Curve::addSection
477 (const std::string& theName, const CurveCreator::SectionType theType,
478 const bool theIsClosed)
480 int resISection = -1;
481 // Set the difference.
483 CurveCreator::Coordinates aCoords; //empty list
484 if (addEmptyDiff()) {
485 myListDiffs.back().init(this, CurveCreator_Operation::AddSection,
486 theName, aCoords, theType, theIsClosed);
489 resISection = addSectionInternal(theName, theType, theIsClosed, aCoords);
494 //=======================================================================
495 // function: addSection
496 // purpose: adds a section with the given points
497 //=======================================================================
498 int CurveCreator_Curve::addSection
499 (const std::string& theName, const CurveCreator::SectionType theType,
500 const bool theIsClosed, const CurveCreator::Coordinates &thePoints)
502 int resISection = -1;
503 // Set the difference.
505 if (addEmptyDiff()) {
506 myListDiffs.back().init(this, CurveCreator_Operation::AddSection,
507 theName, thePoints, theType, theIsClosed);
510 resISection = addSectionInternal(theName, theType, theIsClosed, thePoints);
516 bool CurveCreator_Curve::removeSectionInternal( const int theISection )
518 if (theISection == -1) {
519 delete mySections.back();
520 mySections.pop_back();
522 CurveCreator::Sections::iterator anIterRm = mySections.begin() + theISection;
525 mySections.erase(anIterRm);
531 //! Removes the given sections.
532 bool CurveCreator_Curve::removeSection( const int theISection )
535 // Set the difference.
538 myListDiffs.back().init(this, CurveCreator_Operation::RemoveSection, theISection);
540 res = removeSectionInternal( theISection );
547 * Get number of points in specified section or (the total number of points
548 * in Curve if theISection is equal to -1).
550 int CurveCreator_Curve::getNbPoints( const int theISection ) const
554 if (theISection == -1) {
556 const int aNbSections = getNbSections();
558 for (; i < aNbSections; i++) {
559 aNbCoords += mySections[i]->myPoints.size();
562 aNbCoords = mySections.at(theISection)->myPoints.size();
565 return aNbCoords/myDimension;
568 //! Get "closed" flag of the specified section
569 bool CurveCreator_Curve::isClosed( const int theISection ) const
571 return mySections.at(theISection)->myIsClosed;
575 * Set "closed" flag of the specified section (all sections if
576 * \a theISection is -1).
578 bool CurveCreator_Curve::setClosed( const int theISection,
579 const bool theIsClosed )
581 if (theISection == -1) {
582 int aSize = mySections.size();
585 for (i = 0; i < aSize; i++) {
586 mySections[i]->myIsClosed = theIsClosed;
590 mySections.at(theISection)->myIsClosed = theIsClosed;
596 //! Returns specified section name
597 std::string CurveCreator_Curve::getSectionName( const int theISection ) const
599 if( ( theISection >= 0 ) && ( theISection < mySections.size() ))
600 return mySections.at(theISection)->myName;
604 /** Set name of the specified section */
605 bool CurveCreator_Curve::setSectionName
606 ( const int theISection, const std::string& theName )
608 if( ( theISection >= 0 ) && ( theISection < mySections.size() )){
609 mySections.at(theISection)->myName = theName;
615 //! Get type of the specified section
616 CurveCreator::SectionType CurveCreator_Curve::getSectionType
617 ( const int theISection ) const
619 return mySections.at(theISection)->myType;
623 * Set type of the specified section (or all sections
624 * if \a theISection is -1).
626 bool CurveCreator_Curve::setSectionType( const int theISection,
627 const CurveCreator::SectionType theType )
629 if (theISection == -1) {
631 const int aNbSections = getNbSections();
633 for (; i < aNbSections; i++) {
634 mySections[i]->myType = theType;
638 if( mySections.at(theISection)->myType != theType ){
639 mySections.at(theISection)->myType = theType;
647 /***********************************************/
648 /*** Point methods ***/
649 /***********************************************/
651 //! For internal use only! Undo/Redo is not used here.
652 bool CurveCreator_Curve::addPointsInternal( const CurveCreator::Coordinates& theCoords,
653 const std::vector<int> &theISections,
654 const std::vector<int> &theIPnts )
657 if( (theCoords.size()/getDimension()) == theISections.size() == theIPnts.size() ) {
658 for( int ind = 0; ind < theISections.size(); ind++ ) {
659 int anISection = theISections.at(ind);
660 CurveCreator_Section *aSection =
661 (anISection == -1 ? mySections.back() : mySections.at(anISection));
664 int anIPnt = theIPnts.at(ind);
665 CurveCreator::Coordinates::iterator anIterPosition;
667 anIterPosition = aSection->myPoints.end();
669 anIterPosition = aSection->myPoints.begin() + toICoord(anIPnt);
670 CurveCreator::Coordinates::const_iterator aFirstPosition =
671 theCoords.begin() + toICoord(ind);
672 aSection->myPoints.insert(anIterPosition,
674 aFirstPosition + getDimension());
685 * Add one point to the specified section starting from the given theIPnt index
686 * (or at the end of points if \a theIPnt is -1).
688 bool CurveCreator_Curve::addPoints( const CurveCreator::Coordinates& theCoords,
689 const int theISection,
693 CurveCreator::Coordinates aCoords = theCoords;
694 // Set the difference.
696 if (addEmptyDiff()) {
697 myListDiffs.back().init(this, CurveCreator_Operation::InsertPoints,
698 theCoords, theISection, theIPnt);
700 std::vector<int> anISections, anIPnts;
701 anISections.push_back( theISection );
702 anIPnts.push_back( theIPnt );
703 res = addPointsInternal( theCoords, anISections, anIPnts );
708 //! Set coordinates of specified point
709 bool CurveCreator_Curve::setPoint( const int theISection,
711 const CurveCreator::Coordinates& theNewCoords )
713 // Set the difference.
715 if (addEmptyDiff()) {
716 myListDiffs.back().init(this, CurveCreator_Operation::SetCoordinates,
717 theNewCoords, theISection, theIPnt);
723 if (theNewCoords.size() == myDimension) {
724 CurveCreator_Section *aSection = mySections.at(theISection);
726 for (i = 0; i < myDimension; i++) {
727 aSection->myPoints.at(toICoord(theIPnt) + i) = theNewCoords[i];
739 bool CurveCreator_Curve::removePointsInternal( const std::vector<int> &theISections,
740 const std::vector<int> &theIPnts )
743 if( theISections.size() == theIPnts.size() ) {
744 for( int ind = 0; ind < theISections.size(); ind++ ) {
745 int anISection = theISections.at(ind);
746 CurveCreator_Section *aSection = mySections.at(anISection);
748 int anIPnt = theIPnts.at(ind);
749 CurveCreator::Coordinates::iterator aFirstPosition;
751 aFirstPosition = aSection->myPoints.end();
753 aFirstPosition = aSection->myPoints.begin() + toICoord(anIPnt);
754 aSection->myPoints.erase( aFirstPosition, aFirstPosition + getDimension() );
764 //! Remove point with given id
765 bool CurveCreator_Curve::removePoint( const int theISection, const int theIPnt )
768 // Set the difference.
770 if (addEmptyDiff()) {
771 myListDiffs.back().init(this, CurveCreator_Operation::RemovePoints,
772 theISection, theIPnt);
774 std::vector<int> anISections, anIPnts;
775 anISections.push_back( theISection );
776 anIPnts.push_back( theIPnt );
777 res = removePointsInternal( anISections, anIPnts );
782 //=======================================================================
783 // function: getCoordinates
785 //=======================================================================
786 CurveCreator::Coordinates CurveCreator_Curve::getPoint( const int theISection,
787 const int theIPnt) const
789 CurveCreator_Section *aSection = mySections.at(theISection);
790 CurveCreator::Coordinates::const_iterator
791 anIter = aSection->myPoints.begin() + toICoord(theIPnt);
792 CurveCreator::Coordinates aResult(anIter, anIter + myDimension);
797 //=======================================================================
798 // function: getPoints
800 //=======================================================================
801 CurveCreator::Coordinates CurveCreator_Curve::getPoints( const int theISection ) const
803 return mySections.at(theISection)->myPoints;
807 /***********************************************/
808 /*** Presentation methods ***/
809 /***********************************************/
810 std::vector<Handle_AIS_InteractiveObject> CurveCreator_Curve::constructWire() const
812 std::vector<Handle_AIS_InteractiveObject> aCurveRepresentation;
813 std::vector<Handle_AIS_InteractiveObject> aSectionObjects;
814 for( int iSection = 0 ; iSection < getNbSections() ; iSection++ ){
815 aSectionObjects = constructSection( iSection );
816 for( int iObject = 0 ; iObject < aSectionObjects.size() ; iObject++ ){
817 aCurveRepresentation.push_back( aSectionObjects.at(iObject) );
820 return aCurveRepresentation;