X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FPartSet%2FPartSet_Tools.cpp;h=916a31a71f2a4e958e3ddbf706eb0ce35df99548;hb=94ba553e7b92f11a936e027b49bbd1d501eeee44;hp=02f216cfcb7deb9520354995fb1e000df8e0952f;hpb=b3eb2ec99597987ae45f7b0f8edafb03a6cae129;p=modules%2Fshaper.git diff --git a/src/PartSet/PartSet_Tools.cpp b/src/PartSet/PartSet_Tools.cpp index 02f216cfc..916a31a71 100644 --- a/src/PartSet/PartSet_Tools.cpp +++ b/src/PartSet/PartSet_Tools.cpp @@ -6,11 +6,16 @@ #include #include +#include #include +#include #include #include #include +#include +#include +#include #include #include @@ -18,23 +23,26 @@ #include #include #include +#include +#include +#include +#include #include +#include +#include +#include -#include -#include -#include -#include - -#include - -#include +#include #include #include +#include #include #include #include #include +#include +#include #ifdef _DEBUG #include @@ -63,12 +71,14 @@ gp_Pnt PartSet_Tools::convertClickToPoint(QPoint thePoint, Handle(V3d_View) theV gp_Pnt ConvertedPoint(X, Y, Z); gp_Pnt2d ConvertedPointOnPlane = ProjLib::Project(PlaneOfTheView, ConvertedPoint); - gp_Pnt ResultPoint = ElSLib::Value(ConvertedPointOnPlane.X(), ConvertedPointOnPlane.Y(), PlaneOfTheView); + gp_Pnt ResultPoint = ElSLib::Value(ConvertedPointOnPlane.X(), ConvertedPointOnPlane.Y(), + PlaneOfTheView); return ResultPoint; } void PartSet_Tools::convertTo2D(const gp_Pnt& thePoint, FeaturePtr theSketch, - Handle(V3d_View) theView, double& theX, double& theY) +Handle(V3d_View) theView, + double& theX, double& theY) { if (!theSketch) return; @@ -76,19 +86,18 @@ void PartSet_Tools::convertTo2D(const gp_Pnt& thePoint, FeaturePtr theSketch, AttributeDoublePtr anAttr; boost::shared_ptr aData = theSketch->data(); - boost::shared_ptr anOrigin = - boost::dynamic_pointer_cast(aData->attribute(SKETCH_ATTR_ORIGIN)); + boost::shared_ptr anOrigin = boost::dynamic_pointer_cast( + aData->attribute(SketchPlugin_Sketch::ORIGIN_ID())); - boost::shared_ptr aX = - boost::dynamic_pointer_cast(aData->attribute(SKETCH_ATTR_DIRX)); - boost::shared_ptr anY = - boost::dynamic_pointer_cast(aData->attribute(SKETCH_ATTR_DIRY)); + boost::shared_ptr aX = boost::dynamic_pointer_cast( + aData->attribute(SketchPlugin_Sketch::DIRX_ID())); + boost::shared_ptr anY = boost::dynamic_pointer_cast( + aData->attribute(SketchPlugin_Sketch::DIRY_ID())); gp_Pnt anOriginPnt(anOrigin->x(), anOrigin->y(), anOrigin->z()); gp_Vec aVec(anOriginPnt, thePoint); - if (!theView.IsNull()) - { + if (!theView.IsNull()) { V3d_Coordinate XEye, YEye, ZEye, XAt, YAt, ZAt; theView->Eye(XEye, YEye, ZEye); @@ -99,22 +108,21 @@ void PartSet_Tools::convertTo2D(const gp_Pnt& thePoint, FeaturePtr theSketch, gp_Vec anEyeVec(EyePoint, AtPoint); anEyeVec.Normalize(); - boost::shared_ptr aNormal = - boost::dynamic_pointer_cast(aData->attribute(SKETCH_ATTR_NORM)); + boost::shared_ptr aNormal = boost::dynamic_pointer_cast( + aData->attribute(SketchPlugin_Sketch::NORM_ID())); gp_Vec aNormalVec(aNormal->x(), aNormal->y(), aNormal->z()); double aDen = anEyeVec * aNormalVec; double aLVec = aDen != 0 ? aVec * aNormalVec / aDen : DBL_MAX; - gp_Vec aDeltaVec = anEyeVec*aLVec; + gp_Vec aDeltaVec = anEyeVec * aLVec; aVec = aVec - aDeltaVec; } theX = aVec.X() * aX->x() + aVec.Y() * aX->y() + aVec.Z() * aX->z(); theY = aVec.X() * anY->x() + aVec.Y() * anY->y() + aVec.Z() * anY->z(); } -void PartSet_Tools::convertTo3D(const double theX, const double theY, - FeaturePtr theSketch, +void PartSet_Tools::convertTo3D(const double theX, const double theY, FeaturePtr theSketch, gp_Pnt& thePoint) { if (!theSketch) @@ -122,113 +130,72 @@ void PartSet_Tools::convertTo3D(const double theX, const double theY, boost::shared_ptr aData = theSketch->data(); - boost::shared_ptr aC = - boost::dynamic_pointer_cast(aData->attribute(SKETCH_ATTR_ORIGIN)); - boost::shared_ptr aX = - boost::dynamic_pointer_cast(aData->attribute(SKETCH_ATTR_DIRX)); - boost::shared_ptr aY = - boost::dynamic_pointer_cast(aData->attribute(SKETCH_ATTR_DIRY)); + boost::shared_ptr aC = boost::dynamic_pointer_cast( + aData->attribute(SketchPlugin_Sketch::ORIGIN_ID())); + boost::shared_ptr aX = boost::dynamic_pointer_cast( + aData->attribute(SketchPlugin_Sketch::DIRX_ID())); + boost::shared_ptr aY = boost::dynamic_pointer_cast( + aData->attribute(SketchPlugin_Sketch::DIRY_ID())); - boost::shared_ptr aSum = aC->pnt()->xyz()->added( - aX->dir()->xyz()->multiplied(theX))->added(aY->dir()->xyz()->multiplied(theY)); + boost::shared_ptr aSum = aC->pnt()->xyz()->added(aX->dir()->xyz()->multiplied(theX)) + ->added(aY->dir()->xyz()->multiplied(theY)); boost::shared_ptr aPoint = boost::shared_ptr(new GeomAPI_Pnt(aSum)); thePoint = gp_Pnt(aPoint->x(), aPoint->y(), aPoint->z()); } -boost::shared_ptr PartSet_Tools::createFeaturePrs(const std::string& theKind, - FeaturePtr theSketch, - FeaturePtr theFeature) -{ - boost::shared_ptr aFeaturePrs; - - if (theKind == PartSet_FeaturePointPrs::getKind()) { - aFeaturePrs = boost::shared_ptr(new PartSet_FeaturePointPrs(theSketch)); - } - else if (theKind == PartSet_FeatureLinePrs::getKind()) { - aFeaturePrs = boost::shared_ptr(new PartSet_FeatureLinePrs(theSketch)); - } - else if (theKind == PartSet_FeatureCirclePrs::getKind()) { - aFeaturePrs = boost::shared_ptr(new PartSet_FeatureCirclePrs(theSketch)); - } - else if (theKind == PartSet_FeatureArcPrs::getKind()) { - aFeaturePrs = boost::shared_ptr(new PartSet_FeatureArcPrs(theSketch)); - } - else if (theKind == PartSet_FeatureLengthPrs::getKind()) { - aFeaturePrs = boost::shared_ptr(new PartSet_FeatureLengthPrs(theSketch)); - } - - if (theFeature && aFeaturePrs) - aFeaturePrs->init(theFeature, FeaturePtr()); - - return aFeaturePrs; -} - FeaturePtr PartSet_Tools::nearestFeature(QPoint thePoint, Handle_V3d_View theView, FeaturePtr theSketch, - const std::list& theFeatures) + const QList& theFeatures) { double aX, anY; gp_Pnt aPoint = PartSet_Tools::convertClickToPoint(thePoint, theView); PartSet_Tools::convertTo2D(aPoint, theSketch, theView, aX, anY); FeaturePtr aFeature; - std::list::const_iterator anIt = theFeatures.begin(), aLast = theFeatures.end(); - - FeaturePtr aDeltaFeature; + FeaturePtr aDeltaFeature; double aMinDelta = -1; - XGUI_ViewerPrs aPrs; - for (; anIt != aLast; anIt++) { - aPrs = *anIt; - if (!aPrs.feature()) + ModuleBase_ViewerPrs aPrs; + foreach (ModuleBase_ViewerPrs aPrs, theFeatures) { + if (!aPrs.object()) continue; - double aDelta = distanceToPoint(aPrs.feature(), aX, anY); + boost::shared_ptr aSketchFeature = boost::dynamic_pointer_cast< + SketchPlugin_Feature>(aPrs.object()); + if (!aSketchFeature) + continue; + double aDelta = aSketchFeature->distanceToPoint( + boost::shared_ptr(new GeomAPI_Pnt2d(aX, anY))); if (aMinDelta < 0 || aMinDelta > aDelta) { aMinDelta = aDelta; - aDeltaFeature = aPrs.feature(); + // TODO aDeltaFeature = aPrs.result(); } } return aDeltaFeature; } -double PartSet_Tools::distanceToPoint(FeaturePtr theFeature, - double theX, double theY) -{ - boost::shared_ptr aFeaturePrs = PartSet_Tools::createFeaturePrs( - theFeature->getKind(), FeaturePtr(), theFeature); - double aDelta = 0; - if (aFeaturePrs) - aDelta = aFeaturePrs->distanceToPoint(theFeature, theX, theY); - - return aDelta; -} - -void PartSet_Tools::moveFeature(FeaturePtr theFeature, double theDeltaX, double theDeltaY) -{ - if (!theFeature) - return; - - boost::shared_ptr aFeaturePrs = PartSet_Tools::createFeaturePrs( - theFeature->getKind(), FeaturePtr(), theFeature); - if (aFeaturePrs) - aFeaturePrs->move(theDeltaX, theDeltaY); -} - boost::shared_ptr PartSet_Tools::document() { - return ModelAPI_PluginManager::get()->rootDocument(); + return ModelAPI_Session::get()->moduleDocument(); } -void PartSet_Tools::setFeaturePoint(FeaturePtr theFeature, double theX, double theY, - const std::string& theAttribute) +boost::shared_ptr PartSet_Tools::getFeaturePoint(FeaturePtr theFeature, + double theX, double theY) { - if (!theFeature) - return; - boost::shared_ptr aData = theFeature->data(); - boost::shared_ptr aPoint = - boost::dynamic_pointer_cast(aData->attribute(theAttribute)); - if (aPoint) - aPoint->setValue(theX, theY); + boost::shared_ptr aClickedPoint = boost::shared_ptr( + new GeomAPI_Pnt2d(theX, theY)); + std::list > anAttiributes = + theFeature->data()->attributes(GeomDataAPI_Point2D::type()); + std::list >::const_iterator anIt = anAttiributes.begin(), + aLast = anAttiributes.end(); + boost::shared_ptr aFPoint; + for (; anIt != aLast && !aFPoint; anIt++) { + boost::shared_ptr aCurPoint = boost::dynamic_pointer_cast< + GeomDataAPI_Point2D>(*anIt); + if (aCurPoint && aCurPoint->pnt()->distance(aClickedPoint) < Precision::Confusion()) + aFPoint = aCurPoint; + } + + return aFPoint; } void PartSet_Tools::setFeatureValue(FeaturePtr theFeature, double theValue, @@ -237,64 +204,227 @@ void PartSet_Tools::setFeatureValue(FeaturePtr theFeature, double theValue, if (!theFeature) return; boost::shared_ptr aData = theFeature->data(); - AttributeDoublePtr anAttribute = - boost::dynamic_pointer_cast(aData->attribute(theAttribute)); + AttributeDoublePtr anAttribute = boost::dynamic_pointer_cast( + aData->attribute(theAttribute)); if (anAttribute) anAttribute->setValue(theValue); } -bool PartSet_Tools::featureValue(FeaturePtr theFeature, const std::string& theAttribute, - double& theValue) +double PartSet_Tools::featureValue(FeaturePtr theFeature, const std::string& theAttribute, + bool& isValid) +{ + isValid = false; + double aValue; + if (theFeature) { + boost::shared_ptr aData = theFeature->data(); + AttributeDoublePtr anAttribute = boost::dynamic_pointer_cast( + aData->attribute(theAttribute)); + if (anAttribute) { + aValue = anAttribute->value(); + isValid = true; + } + } + return aValue; +} + +FeaturePtr PartSet_Tools::feature(FeaturePtr theFeature, const std::string& theAttribute, + const std::string& theKind) { - bool aResult = false; + FeaturePtr aFeature; if (!theFeature) - return aResult; + return aFeature; + boost::shared_ptr aData = theFeature->data(); - AttributeDoublePtr anAttribute = - boost::dynamic_pointer_cast(aData->attribute(theAttribute)); - if (anAttribute) { - theValue = anAttribute->value(); - aResult = true; + boost::shared_ptr anAttr = boost::dynamic_pointer_cast< + ModelAPI_AttributeRefAttr>(aData->attribute(theAttribute)); + if (anAttr) { + aFeature = boost::dynamic_pointer_cast(anAttr->object()); + if (!theKind.empty() && aFeature && aFeature->getKind() != theKind) { + aFeature = FeaturePtr(); + } } - return aResult; + return aFeature; } -void PartSet_Tools::createConstraint(FeaturePtr theSketch, +void PartSet_Tools::createConstraint(CompositeFeaturePtr theSketch, boost::shared_ptr thePoint1, boost::shared_ptr thePoint2) { - boost::shared_ptr aDoc = document(); - FeaturePtr aFeature = aDoc->addFeature(SKETCH_CONSTRAINT_COINCIDENCE_KIND); - + FeaturePtr aFeature; if (theSketch) { - boost::shared_ptr aSketch = - boost::dynamic_pointer_cast(theSketch); - aSketch->addSub(aFeature); + aFeature = theSketch->addFeature(SketchPlugin_ConstraintCoincidence::ID()); + } else { + boost::shared_ptr aDoc = document(); + aFeature = aDoc->addFeature(SketchPlugin_ConstraintCoincidence::ID()); } boost::shared_ptr aData = aFeature->data(); - boost::shared_ptr aRef1 = - boost::dynamic_pointer_cast(aData->attribute(CONSTRAINT_ATTR_ENTITY_A)); + boost::shared_ptr aRef1 = boost::dynamic_pointer_cast< + ModelAPI_AttributeRefAttr>(aData->attribute(SketchPlugin_Constraint::ENTITY_A())); aRef1->setAttr(thePoint1); - boost::shared_ptr aRef2 = - boost::dynamic_pointer_cast(aData->attribute(CONSTRAINT_ATTR_ENTITY_B)); + boost::shared_ptr aRef2 = boost::dynamic_pointer_cast< + ModelAPI_AttributeRefAttr>(aData->attribute(SketchPlugin_Constraint::ENTITY_B())); aRef2->setAttr(thePoint2); - if (aFeature) // TODO: generate an error if feature was not created + if (aFeature) // TODO: generate an error if feature was not created aFeature->execute(); } -boost::shared_ptr PartSet_Tools::findPoint(FeaturePtr theFeature, - double theX, double theY) +void PartSet_Tools::setConstraints(CompositeFeaturePtr theSketch, FeaturePtr theFeature, + const std::string& theAttribute, double theClickedX, + double theClickedY) +{ + // find a feature point by the selection mode + //boost::shared_ptr aPoint = featurePoint(theMode); + boost::shared_ptr aFeaturePoint = boost::dynamic_pointer_cast< + GeomDataAPI_Point2D>(theFeature->data()->attribute(theAttribute)); + if (!aFeaturePoint) + return; + + // get all sketch features. If the point with the given coordinates belong to any sketch feature, + // the constraint is created between the feature point and the found sketch point + boost::shared_ptr aData = theSketch->data(); + boost::shared_ptr aRefList = boost::dynamic_pointer_cast< + ModelAPI_AttributeRefList>(aData->attribute(SketchPlugin_Sketch::FEATURES_ID())); + + std::list aFeatures = aRefList->list(); + std::list::const_iterator anIt = aFeatures.begin(), aLast = aFeatures.end(); + std::list > anAttiributes; + boost::shared_ptr aClickedPoint = boost::shared_ptr( + new GeomAPI_Pnt2d(theClickedX, theClickedY)); + for (; anIt != aLast; anIt++) { + FeaturePtr aFeature = boost::dynamic_pointer_cast(*anIt); + // find the given point in the feature attributes + anAttiributes = aFeature->data()->attributes(GeomDataAPI_Point2D::type()); + std::list >::const_iterator anIt = anAttiributes.begin(), + aLast = anAttiributes.end(); + boost::shared_ptr aFPoint; + for (; anIt != aLast && !aFPoint; anIt++) { + boost::shared_ptr aCurPoint = boost::dynamic_pointer_cast< + GeomDataAPI_Point2D>(*anIt); + if (aCurPoint && aCurPoint->pnt()->distance(aClickedPoint) < Precision::Confusion()) + aFPoint = aCurPoint; + } + if (aFPoint) + PartSet_Tools::createConstraint(theSketch, aFPoint, aFeaturePoint); + } +} + +boost::shared_ptr PartSet_Tools::sketchPlane(CompositeFeaturePtr theSketch) { - boost::shared_ptr aFeaturePrs = PartSet_Tools::createFeaturePrs( - theFeature->getKind(), FeaturePtr(), theFeature); + boost::shared_ptr aPlane; + double aA, aB, aC, aD; - boost::shared_ptr aPoint2D; - if (aFeaturePrs) - aPoint2D = aFeaturePrs->findPoint(theFeature, theX, theY); + boost::shared_ptr aData = theSketch->data(); + boost::shared_ptr anOrigin = boost::dynamic_pointer_cast( + aData->attribute(SketchPlugin_Sketch::ORIGIN_ID())); + boost::shared_ptr aNormal = boost::dynamic_pointer_cast( + aData->attribute(SketchPlugin_Sketch::NORM_ID())); + aA = aNormal->x(); + aB = aNormal->y(); + aC = aNormal->z(); + aD = 0; + + aPlane = boost::shared_ptr(new GeomAPI_Pln(aA, aB, aC, aD)); + return aPlane; +} + +boost::shared_ptr PartSet_Tools::point3D(boost::shared_ptr thePoint2D, + CompositeFeaturePtr theSketch) +{ + boost::shared_ptr aPoint; + if (!theSketch || !thePoint2D) + return aPoint; + + boost::shared_ptr aC = boost::dynamic_pointer_cast( + theSketch->data()->attribute(SketchPlugin_Sketch::ORIGIN_ID())); + boost::shared_ptr aX = boost::dynamic_pointer_cast( + theSketch->data()->attribute(SketchPlugin_Sketch::DIRX_ID())); + boost::shared_ptr aY = boost::dynamic_pointer_cast( + theSketch->data()->attribute(SketchPlugin_Sketch::DIRY_ID())); + + return thePoint2D->to3D(aC->pnt(), aX->dir(), aY->dir()); +} - return aPoint2D; +bool PartSet_Tools::isConstraintFeature(const std::string& theKind) +{ + return theKind == SketchPlugin_ConstraintDistance::ID() + || theKind == SketchPlugin_ConstraintLength::ID() + || theKind == SketchPlugin_ConstraintRadius::ID() + || theKind == SketchPlugin_ConstraintRigid::ID(); +} + +ResultPtr PartSet_Tools::createFixedObjectByEdge(const ModuleBase_ViewerPrs& thePrs, CompositeFeaturePtr theSketch) +{ + TopoDS_Shape aShape = thePrs.shape(); + if (aShape.ShapeType() != TopAbs_EDGE) + return ResultPtr(); + + Standard_Real aStart, aEnd; + Handle(V3d_View) aNullView; + FeaturePtr aMyFeature; + + Handle(Geom_Curve) aCurve = BRep_Tool::Curve(TopoDS::Edge(aShape), aStart, aEnd); + GeomAdaptor_Curve aAdaptor(aCurve); + if (aAdaptor.GetType() == GeomAbs_Line) { + // Create line + aMyFeature = theSketch->addFeature(SketchPlugin_Line::ID()); + + //DataPtr aData = myFeature->data(); + //boost::shared_ptr anEndAttr = + // boost::dynamic_pointer_cast(aData->attribute(SketchPlugin_Line::END_ID())); + + //double aX, aY; + //gp_Pnt Pnt1 = aAdaptor.Value(aStart); + //convertTo2D(Pnt1, theSketch, aNullView, aX, aY); + //setFeaturePoint(myFeature, aX, aY, SketchPlugin_Line::START_ID()); + + //gp_Pnt Pnt2 = aAdaptor.Value(aEnd); + //convertTo2D(Pnt2, theSketch, aNullView, aX, aY); + //setFeaturePoint(myFeature, aX, aY, SketchPlugin_Line::END_ID()); + } else if (aAdaptor.GetType() == GeomAbs_Circle) { + if (aAdaptor.IsClosed()) { + // Create circle + aMyFeature = theSketch->addFeature(SketchPlugin_Circle::ID()); + //gp_Circ aCirc = aAdaptor.Circle(); + //gp_Pnt aCenter = aCirc.Location(); + + //double aX, aY; + //convertTo2D(aCenter, theSketch, aNullView, aX, aY); + //setFeaturePoint(myFeature, aX, aY, SketchPlugin_Circle::CENTER_ID()); + //setFeatureValue(myFeature, aCirc.Radius(), SketchPlugin_Circle::RADIUS_ID()); + } else { + // Create arc + aMyFeature = theSketch->addFeature(SketchPlugin_Arc::ID()); + } + } + if (aMyFeature) { + DataPtr aData = aMyFeature->data(); + AttributeSelectionPtr anAttr = + boost::dynamic_pointer_cast + (aData->attribute(SketchPlugin_Feature::EXTERNAL_ID())); + + ResultPtr aRes = boost::dynamic_pointer_cast(thePrs.object()); + if (anAttr && aRes) { + boost::shared_ptr anEdge(new GeomAPI_Shape); + anEdge->setImpl(new TopoDS_Shape(aShape)); + + anAttr->setValue(aRes, anEdge); + aMyFeature->execute(); + return aMyFeature->lastResult(); + } + } + return ResultPtr(); +} + +bool PartSet_Tools::isContainPresentation(const QList& theSelected, + const ModuleBase_ViewerPrs& thePrs) +{ + foreach (ModuleBase_ViewerPrs aPrs, theSelected) { + if (aPrs.object() == thePrs.object()) + return true; + } + return false; }