X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FSketchPlugin%2FSketchPlugin_ConstraintRadius.cpp;h=7916442e33e04df9c98698c3d6f104df6e7b1156;hb=026f4997e7531acff5d6677877d4e948088c46ac;hp=36ec6c384b0f9d7e9231dddf90d408ac268575a9;hpb=a5ffc4b734b71d9a78da1e18909ff3526cf2a188;p=modules%2Fshaper.git diff --git a/src/SketchPlugin/SketchPlugin_ConstraintRadius.cpp b/src/SketchPlugin/SketchPlugin_ConstraintRadius.cpp index 36ec6c384..7916442e3 100644 --- a/src/SketchPlugin/SketchPlugin_ConstraintRadius.cpp +++ b/src/SketchPlugin/SketchPlugin_ConstraintRadius.cpp @@ -1,164 +1,214 @@ -// File: SketchPlugin_ConstraintRadius.cpp -// Created: 26 May 2014 -// Author: Artem ZHIDKOV +// Copyright (C) 2014-2021 CEA/DEN, EDF R&D +// +// This library is free software; you can redistribute it and/or +// modify it under the terms of the GNU Lesser General Public +// License as published by the Free Software Foundation; either +// version 2.1 of the License, or (at your option) any later version. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// Lesser General Public License for more details. +// +// You should have received a copy of the GNU Lesser General Public +// License along with this library; if not, write to the Free Software +// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA +// +// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com +// #include "SketchPlugin_ConstraintRadius.h" #include #include #include +#include + +#include +#include #include +#include #include +#include +#include #include #include #include #include +#include #include #include #include #include +const double tolerance = 1.e-7; + SketchPlugin_ConstraintRadius::SketchPlugin_ConstraintRadius() { + myFlyoutUpdate = false; } void SketchPlugin_ConstraintRadius::initAttributes() { - data()->addAttribute(SketchPlugin_Constraint::VALUE(), ModelAPI_AttributeDouble::type()); - data()->addAttribute(SketchPlugin_Constraint::ENTITY_A(), ModelAPI_AttributeRefAttr::type()); - data()->addAttribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT(), GeomDataAPI_Point2D::type()); + data()->addAttribute(SketchPlugin_Constraint::VALUE(), ModelAPI_AttributeDouble::typeId()); + data()->addAttribute(SketchPlugin_Constraint::ENTITY_A(), ModelAPI_AttributeRefAttr::typeId()); + data()->addAttribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT(), GeomDataAPI_Point2D::typeId()); + + data()->addAttribute(SketchPlugin_ConstraintRadius::LOCATION_TYPE_ID(), + ModelAPI_AttributeInteger::typeId()); + ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), LOCATION_TYPE_ID()); +} + +void SketchPlugin_ConstraintRadius::colorConfigInfo(std::string& theSection, std::string& theName, + std::string& theDefault) +{ + theSection = "Visualization"; + theName = "sketch_dimension_color"; + theDefault = SKETCH_DIMENSION_COLOR; } void SketchPlugin_ConstraintRadius::execute() { - boost::shared_ptr aRef = boost::dynamic_pointer_cast< + std::shared_ptr aRef = std::dynamic_pointer_cast< ModelAPI_AttributeRefAttr>(data()->attribute(SketchPlugin_Constraint::ENTITY_A())); FeaturePtr aFeature = ModelAPI_Feature::feature(aRef->object()); if (aFeature) { - double aRadius = 0; - boost::shared_ptr aData = aFeature->data(); - if (aFeature->getKind() == SketchPlugin_Circle::ID()) { - AttributeDoublePtr anAttribute = boost::dynamic_pointer_cast( - aData->attribute(SketchPlugin_Circle::RADIUS_ID())); - if (anAttribute) - aRadius = anAttribute->value(); - } else if (aFeature->getKind() == SketchPlugin_Arc::ID()) { - boost::shared_ptr aCenterAttr = boost::dynamic_pointer_cast< - GeomDataAPI_Point2D>(aData->attribute(SketchPlugin_Arc::CENTER_ID())); - boost::shared_ptr aStartAttr = boost::dynamic_pointer_cast< - GeomDataAPI_Point2D>(aData->attribute(SketchPlugin_Arc::START_ID())); - if (aCenterAttr && aStartAttr) - aRadius = aCenterAttr->pnt()->distance(aStartAttr->pnt()); - } - boost::shared_ptr aValueAttr = boost::dynamic_pointer_cast< - ModelAPI_AttributeDouble>(data()->attribute(SketchPlugin_Constraint::VALUE())); - if(!aValueAttr->isInitialized()) { - aValueAttr->setValue(aRadius); - } + // the value should to be computed here, + // not in the getAISObject in order to change the model value + // inside the object transaction. This is important for creating a constraint by preselection. + // The display of the presentation in this case happens after the transaction commit + std::shared_ptr aFlyoutAttr = std::dynamic_pointer_cast< + GeomDataAPI_Point2D>(data()->attribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT())); + if (!aFlyoutAttr->isInitialized()) + compute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT()); } } -AISObjectPtr SketchPlugin_ConstraintRadius::getAISObject(AISObjectPtr thePrevious) +bool SketchPlugin_ConstraintRadius::compute(const std::string& theAttributeId) { + if (theAttributeId != SketchPlugin_Constraint::FLYOUT_VALUE_PNT()) + return false; + + std::shared_ptr aCyrcFeature; + double aRadius = circleRadius(aCyrcFeature); + if (aRadius < 0) + return false; + + // Flyout point + std::shared_ptr aFlyoutAttr = std::dynamic_pointer_cast< + GeomDataAPI_Point2D>(data()->attribute(theAttributeId)); + // Prepare a circle + if (aCyrcFeature->getKind() == SketchPlugin_Circle::ID()) { // circle + std::shared_ptr aCenterAttr = + std::dynamic_pointer_cast( + aCyrcFeature->data()->attribute(SketchPlugin_Circle::CENTER_ID())); + double aShift = aRadius * 1.1; + std::shared_ptr aPnt = aCenterAttr->pnt(); + std::shared_ptr aFPnt = + std::shared_ptr(new GeomAPI_Pnt2d(aPnt->x() + aShift, aPnt->y() + aShift)); + aFlyoutAttr->setValue(aFPnt); + } else { // arc + std::shared_ptr aStartAttr = std::dynamic_pointer_cast< + GeomDataAPI_Point2D>(aCyrcFeature->data()->attribute(SketchPlugin_Arc::START_ID())); + aFlyoutAttr->setValue(aStartAttr->pnt()); + } + return true; +} + +double SketchPlugin_ConstraintRadius::circleRadius(std::shared_ptr& theCirc) +{ + static const double kErrorResult = -1.; if (!sketch()) - return thePrevious; + return kErrorResult; - boost::shared_ptr aData = data(); - boost::shared_ptr anAttr = boost::dynamic_pointer_cast< + std::shared_ptr aData = data(); + std::shared_ptr anAttr = std::dynamic_pointer_cast< ModelAPI_AttributeRefAttr>(aData->attribute(SketchPlugin_Constraint::ENTITY_A())); if (!anAttr) - return thePrevious; - FeaturePtr aFeature = ModelAPI_Feature::feature(anAttr->object()); - std::string aKind = aFeature ? aFeature->getKind() : ""; + return kErrorResult; + theCirc = ModelAPI_Feature::feature(anAttr->object()); + std::string aKind = theCirc ? theCirc->getKind() : ""; if (aKind != SketchPlugin_Circle::ID() && aKind != SketchPlugin_Arc::ID()) - return thePrevious; - - // Flyout point - boost::shared_ptr aFlyoutAttr = boost::dynamic_pointer_cast< - GeomDataAPI_Point2D>(aData->attribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT())); - if (!aFlyoutAttr->isInitialized()) - return thePrevious; - boost::shared_ptr aFlyoutPnt = sketch()->to3D(aFlyoutAttr->x(), aFlyoutAttr->y()); + return kErrorResult; - // Prepare a circle - aData = aFeature->data(); - boost::shared_ptr aCenterAttr; - double aRadius; + DataPtr aCircData = theCirc->data(); + std::shared_ptr aCenterAttr; if (aKind == SketchPlugin_Circle::ID()) { - aCenterAttr = boost::dynamic_pointer_cast( - aData->attribute(SketchPlugin_Circle::CENTER_ID())); - AttributeDoublePtr aCircRadius = boost::dynamic_pointer_cast( - aData->attribute(SketchPlugin_Circle::RADIUS_ID())); - aRadius = aCircRadius->value(); - } else if (aKind == SketchPlugin_Arc::ID()) { - aCenterAttr = boost::dynamic_pointer_cast( - aData->attribute(SketchPlugin_Arc::CENTER_ID())); - boost::shared_ptr aStartAttr = boost::dynamic_pointer_cast< - GeomDataAPI_Point2D>(aData->attribute(SketchPlugin_Arc::START_ID())); - aRadius = aCenterAttr->pnt()->distance(aStartAttr->pnt()); + AttributeDoublePtr aCircRadius = std::dynamic_pointer_cast( + aCircData->attribute(SketchPlugin_Circle::RADIUS_ID())); + return aCircRadius->value(); + } else { + aCenterAttr = std::dynamic_pointer_cast( + aCircData->attribute(SketchPlugin_Arc::CENTER_ID())); + std::shared_ptr aStartAttr = std::dynamic_pointer_cast< + GeomDataAPI_Point2D>(aCircData->attribute(SketchPlugin_Arc::START_ID())); + return aCenterAttr->pnt()->distance(aStartAttr->pnt()); } - - boost::shared_ptr aCenter = sketch()->to3D(aCenterAttr->x(), aCenterAttr->y()); - boost::shared_ptr aNDir = boost::dynamic_pointer_cast( - sketch()->data()->attribute(SketchPlugin_Sketch::NORM_ID())); - boost::shared_ptr aNormal = aNDir->dir(); - boost::shared_ptr aCircle(new GeomAPI_Circ(aCenter, aNormal, aRadius)); - - // Value - boost::shared_ptr aValueAttr = boost::dynamic_pointer_cast< - ModelAPI_AttributeDouble>(aData->attribute(SketchPlugin_Constraint::VALUE())); - double aValue = aRadius; - if (aValueAttr && aValueAttr->isInitialized()) - aValue = aValueAttr->value(); - - AISObjectPtr anAIS = thePrevious; - if (!anAIS) - anAIS = AISObjectPtr(new GeomAPI_AISObject); - anAIS->createRadius(aCircle, aFlyoutPnt, aValue); - - // Set color from preferences - std::vector aRGB = Config_PropManager::color("Visualization", "radius_color", RADIUS_COLOR); - anAIS->setColor(aRGB[0], aRGB[1], aRGB[2]); - return anAIS; + return kErrorResult; } -void SketchPlugin_ConstraintRadius::move(double theDeltaX, double theDeltaY) +AISObjectPtr SketchPlugin_ConstraintRadius::getAISObject(AISObjectPtr thePrevious) { - boost::shared_ptr aData = data(); - if (!aData->isValid()) - return; + if (!sketch()) + return thePrevious; - boost::shared_ptr aRef = boost::dynamic_pointer_cast< - ModelAPI_AttributeRefAttr>(data()->attribute(SketchPlugin_Constraint::ENTITY_A())); - FeaturePtr aFeature = ModelAPI_Feature::feature(aRef->object()); - if (!aFeature) - return; - std::string aCenterAttrName; - if (aFeature->getKind() == SketchPlugin_Circle::ID()) - aCenterAttrName = SketchPlugin_Circle::CENTER_ID(); - else if (aFeature->getKind() == SketchPlugin_Arc::ID()) - aCenterAttrName = SketchPlugin_Arc::CENTER_ID(); - boost::shared_ptr aCenterAttr = boost::dynamic_pointer_cast< - GeomDataAPI_Point2D>(aFeature->data()->attribute(aCenterAttrName)); - boost::shared_ptr aCenter = aCenterAttr->pnt(); - - // The specified delta applied on the circle curve, - // so it will be scaled due to distance between flyout and center points - boost::shared_ptr aFlyoutAttr = boost::dynamic_pointer_cast< - GeomDataAPI_Point2D>(aData->attribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT())); - boost::shared_ptr aFlyout = aFlyoutAttr->pnt(); - - boost::shared_ptr aRadius = boost::dynamic_pointer_cast< - ModelAPI_AttributeDouble>(aData->attribute(SketchPlugin_Constraint::VALUE())); - double aScale = aFlyout->distance(aCenter) / aRadius->value(); - - boost::shared_ptr aCircle(new GeomAPI_Circ2d(aCenter, aFlyout)); - aFlyout->setX(aFlyout->x() + aScale * theDeltaX); - aFlyout->setY(aFlyout->y() + aScale * theDeltaY); - aFlyout = aCircle->project(aFlyout); - - aFlyoutAttr->setValue(aFlyout->x(), aFlyout->y()); + AISObjectPtr anAIS = SketcherPrs_Factory::radiusConstraint(this, sketch(), + thePrevious); + if (anAIS.get() && !thePrevious.get()) + SketchPlugin_Tools::setDimensionColor(anAIS); + return anAIS; +} + +void SketchPlugin_ConstraintRadius::attributeChanged(const std::string& theID) { + if (theID == SketchPlugin_Constraint::ENTITY_A()) { + std::shared_ptr aValueAttr = std::dynamic_pointer_cast< + ModelAPI_AttributeDouble>(data()->attribute(SketchPlugin_Constraint::VALUE())); + if (!aValueAttr->isInitialized()) { + // only if it is not initialized, try to compute the current value + std::shared_ptr aCyrcFeature; + double aRadius = circleRadius(aCyrcFeature); + if (aRadius > 0) { // set as value the radius of updated reference to another circle + aValueAttr->setValue(aRadius); + } + } + } else if (theID == SketchPlugin_Constraint::FLYOUT_VALUE_PNT() && !myFlyoutUpdate) { + // Recalculate flyout point in local coordinates of the circle (or arc): + // coordinates are calculated according to center of the shape + std::shared_ptr aFlyoutAttr = + std::dynamic_pointer_cast( + attribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT())); + + AttributeRefAttrPtr aRefAttr = std::dynamic_pointer_cast( + attribute(SketchPlugin_Constraint::ENTITY_A())); + if (!aRefAttr || !aRefAttr->isObject()) + return; + FeaturePtr aFeature = ModelAPI_Feature::feature(aRefAttr->object()); + if (!aFeature || (aFeature->getKind() != SketchPlugin_Arc::ID() && + aFeature->getKind() != SketchPlugin_Circle::ID())) + return; + + std::string aCenterAttrName; + if (aFeature->getKind() == SketchPlugin_Circle::ID()) + aCenterAttrName = SketchPlugin_Circle::CENTER_ID(); + else if (aFeature->getKind() == SketchPlugin_Arc::ID()) + aCenterAttrName = SketchPlugin_Arc::CENTER_ID(); + std::shared_ptr aCenterAttr = std::dynamic_pointer_cast< + GeomDataAPI_Point2D>(aFeature->data()->attribute(aCenterAttrName)); + std::shared_ptr aCenter = aCenterAttr->pnt(); + std::shared_ptr aRadius = std::dynamic_pointer_cast< + ModelAPI_AttributeDouble>(attribute(SketchPlugin_Constraint::VALUE())); + + std::shared_ptr aFlyoutPnt = aFlyoutAttr->pnt(); + double aDist = aFlyoutPnt->distance(aCenter); + if (aDist < tolerance) + return; + + myFlyoutUpdate = true; + std::shared_ptr aFlyoutDir = aFlyoutPnt->xy()->decreased(aCenter->xy()); + aFlyoutAttr->setValue(aFlyoutDir->x(), aFlyoutDir->y()); + myFlyoutUpdate = false; + } }