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 //=======================================================================
453 // function: addSection
454 // purpose: adds an empty section
455 //=======================================================================
456 int CurveCreator_Curve::addSection
457 (const std::string& theName, const CurveCreator::SectionType theType,
458 const bool theIsClosed)
460 int resISection = -1;
461 // Set the difference.
463 if (addEmptyDiff()) {
464 CurveCreator::Coordinates aCoords; //empty list
465 myListDiffs.back().init(this, CurveCreator_Operation::AddSection,
466 theName, aCoords, theType, theIsClosed);
468 CurveCreator_Section *aSection = new CurveCreator_Section;
470 std::string aName = theName;
472 aName = getUniqSectionName();
474 aSection->myName = aName;
475 aSection->myType = theType;
476 aSection->myIsClosed = theIsClosed;
477 mySections.push_back(aSection);
481 return mySections.size()-1;
483 //=======================================================================
484 // function: addSection
485 // purpose: adds a section with the given points
486 //=======================================================================
487 int CurveCreator_Curve::addSection
488 (const std::string& theName, const CurveCreator::SectionType theType,
489 const bool theIsClosed, const CurveCreator::Coordinates &thePoints)
491 int resISection = -1;
492 //// Set the difference.
494 //if (addEmptyDiff()) {
495 // myListDiffs.back().init(this, CurveCreator_Operation::AddSection,
496 // theName, thePoints, theType, theIsClosed);
499 // create an empty section
500 resISection = addSection(theName, theType, theIsClosed);
501 if( resISection != -1 ) {
502 // attach the given points to created section
503 CurveCreator_Section *aSection = mySections.at(resISection);
504 aSection->myPoints = thePoints;
511 //! Removes the given sections.
512 bool CurveCreator_Curve::removeSection( const int theISection )
514 // Set the difference.
517 myListDiffs.back().init(this, CurveCreator_Operation::RemoveSection, theISection);
519 if (theISection == -1) {
520 delete mySections.back();
521 mySections.pop_back();
523 CurveCreator::Sections::iterator anIterRm = mySections.begin() + theISection;
526 mySections.erase(anIterRm);
535 * Get number of points in specified section or (the total number of points
536 * in Curve if theISection is equal to -1).
538 int CurveCreator_Curve::getNbPoints( const int theISection ) const
542 if (theISection == -1) {
544 const int aNbSections = getNbSections();
546 for (; i < aNbSections; i++) {
547 aNbCoords += mySections[i]->myPoints.size();
550 aNbCoords = mySections.at(theISection)->myPoints.size();
553 return aNbCoords/myDimension;
556 //! Get "closed" flag of the specified section
557 bool CurveCreator_Curve::isClosed( const int theISection ) const
559 return mySections.at(theISection)->myIsClosed;
563 * Set "closed" flag of the specified section (all sections if
564 * \a theISection is -1).
566 bool CurveCreator_Curve::setClosed( const int theISection,
567 const bool theIsClosed )
569 if (theISection == -1) {
570 int aSize = mySections.size();
573 for (i = 0; i < aSize; i++) {
574 mySections[i]->myIsClosed = theIsClosed;
578 mySections.at(theISection)->myIsClosed = theIsClosed;
584 //! Returns specifyed section name
585 std::string CurveCreator_Curve::getSectionName( const int theISection ) const
587 if( ( theISection >= 0 ) && ( theISection < mySections.size() ))
588 return mySections.at(theISection)->myName;
592 /** Set name of the specified section */
593 bool CurveCreator_Curve::setSectionName
594 ( const int theISection, const std::string& theName )
596 if( ( theISection >= 0 ) && ( theISection < mySections.size() )){
597 mySections.at(theISection)->myName = theName;
603 //! Get type of the specified section
604 CurveCreator::SectionType CurveCreator_Curve::getSectionType
605 ( const int theISection ) const
607 return mySections.at(theISection)->myType;
611 * Set type of the specified section (or all sections
612 * if \a theISection is -1).
614 bool CurveCreator_Curve::setSectionType( const int theISection,
615 const CurveCreator::SectionType theType )
617 if (theISection == -1) {
619 const int aNbSections = getNbSections();
621 for (; i < aNbSections; i++) {
622 mySections[i]->myType = theType;
626 if( mySections.at(theISection)->myType != theType ){
627 mySections.at(theISection)->myType = theType;
635 /***********************************************/
636 /*** Point methods ***/
637 /***********************************************/
640 * Add one point to the specified section starting from the given theIPnt index
641 * (or at the end of points if \a theIPnt is -1).
643 bool CurveCreator_Curve::addPoints( const CurveCreator::Coordinates& theCoords,
644 const int theISection,
648 CurveCreator::Coordinates aCoords = theCoords;
649 // Set the difference.
651 if (addEmptyDiff()) {
652 myListDiffs.back().init(this, CurveCreator_Operation::AddPoints,
653 theCoords, theISection);
655 CurveCreator_Section *aSection =
656 (theISection == -1 ? mySections.back() : mySections.at(theISection));
659 int anICoord = ( theIPnt == -1 ? 0 : toICoord(theIPnt) );
660 CurveCreator::Coordinates::iterator anIterPosition = aSection->myPoints.end();
662 anIterPosition = aSection->myPoints.begin() + toICoord(theIPnt);
663 aSection->myPoints.insert(anIterPosition,
664 theCoords.begin(), theCoords.end());
672 //! Set coordinates of specified point
673 bool CurveCreator_Curve::setPoint( const int theISection,
675 const CurveCreator::Coordinates& theNewCoords )
677 // Set the difference.
679 if (addEmptyDiff()) {
680 myListDiffs.back().init(this, CurveCreator_Operation::SetCoordinates,
681 theNewCoords, theISection, theIPnt);
687 if (theNewCoords.size() == myDimension) {
688 CurveCreator_Section *aSection = mySections.at(theISection);
690 for (i = 0; i < myDimension; i++) {
691 aSection->myPoints.at(toICoord(theIPnt) + i) = theNewCoords[i];
703 //! Remove point with given id
704 bool CurveCreator_Curve::removePoint( const int theISection, const int theIPnt )
707 // Set the difference.
709 if (addEmptyDiff()) {
710 myListDiffs.back().init(this, CurveCreator_Operation::RemovePoints,
711 theISection, theIPnt, 0);
713 CurveCreator_Section *aSection = mySections.at(theISection);
715 CurveCreator::Coordinates::iterator anIterPosition =
716 aSection->myPoints.begin() + toICoord(theIPnt);
717 aSection->myPoints.erase( anIterPosition );
725 //=======================================================================
726 // function: getCoordinates
728 //=======================================================================
729 CurveCreator::Coordinates CurveCreator_Curve::getPoint( const int theISection,
730 const int theIPnt) const
732 CurveCreator_Section *aSection = mySections.at(theISection);
733 CurveCreator::Coordinates::const_iterator
734 anIter = aSection->myPoints.begin() + toICoord(theIPnt);
735 CurveCreator::Coordinates aResult(anIter, anIter + myDimension);
740 //=======================================================================
741 // function: getPoints
743 //=======================================================================
744 CurveCreator::Coordinates CurveCreator_Curve::getPoints( const int theISection ) const
746 return mySections.at(theISection)->myPoints;
750 /***********************************************/
751 /*** Presentation methods ***/
752 /***********************************************/
753 std::vector<Handle_AIS_InteractiveObject> CurveCreator_Curve::constructWire() const
755 std::vector<Handle_AIS_InteractiveObject> aCurveRepresentation;
756 std::vector<Handle_AIS_InteractiveObject> aSectionObjects;
757 for( int iSection = 0 ; iSection < getNbSections() ; iSection++ ){
758 aSectionObjects = constructSection( iSection );
759 for( int iObject = 0 ; iObject < aSectionObjects.size() ; iObject++ ){
760 aCurveRepresentation.push_back( aSectionObjects.at(iObject) );
763 return aCurveRepresentation;