src/SketchPlugin/SketchPlugin_ConstraintCoincidence.cpp

   1 // Copyright (C) 2014-2017  CEA/DEN, EDF R&D
   2 //
   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.
   7 //
   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.
  12 //
  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
  16 //
  17 // See http://www.salome-platform.org/ or
  18 // email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
  19 //
  20
  21 #include "SketchPlugin_ConstraintCoincidence.h"
  22
  23 #include <SketchPlugin_Point.h>
  24 #include <SketchPlugin_Line.h>
  25 #include <SketchPlugin_Arc.h>
  26 #include <SketchPlugin_Circle.h>
  27
  28 #include <ModelAPI_Events.h>
  29 #include <ModelGeomAlgo_Point2D.h>
  30 #include <GeomDataAPI_Point2D.h>
  31
  32 #include <SketcherPrs_Factory.h>
  33
  34 #include <Events_Loop.h>
  35
  36 SketchPlugin_ConstraintCoincidence::SketchPlugin_ConstraintCoincidence()
  37 {
  38 }
  39
  40 void SketchPlugin_ConstraintCoincidence::initAttributes()
  41 {
  42   data()->addAttribute(SketchPlugin_Constraint::ENTITY_A(), ModelAPI_AttributeRefAttr::typeId());
  43   data()->addAttribute(SketchPlugin_Constraint::ENTITY_B(), ModelAPI_AttributeRefAttr::typeId());
  44 }
  45
  46 void SketchPlugin_ConstraintCoincidence::execute()
  47 {
  48 }
  49
  50 AISObjectPtr SketchPlugin_ConstraintCoincidence::getAISObject(AISObjectPtr thePrevious)
  51 {
  52   if (!sketch())
  53     return thePrevious;
  54
  55   AISObjectPtr anAIS = SketcherPrs_Factory::coincidentConstraint(this, sketch(), thePrevious);
  56   return anAIS;
  57 }
  58
  59 FeaturePtr SketchPlugin_ConstraintCoincidence::findCoincidenceFeature(const FeaturePtr& theFeature1,
  60                                                                       const FeaturePtr& theFeature2)
  61 {
  62   FeaturePtr aResultFeature;
  63
  64   std::list<AttributePtr> anAttrList;
  65   if (theFeature1->getKind() == SketchPlugin_Circle::ID() ||
  66       theFeature2->getKind() == SketchPlugin_Circle::ID())
  67     return aResultFeature;
  68
  69   if (theFeature2->getKind() == SketchPlugin_Line::ID()) {
  70     anAttrList.push_back(theFeature2->attribute(SketchPlugin_Line::START_ID()));
  71     anAttrList.push_back(theFeature2->attribute(SketchPlugin_Line::END_ID()));
  72   } else if (theFeature2->getKind() == SketchPlugin_Arc::ID()) {
  73     anAttrList.push_back(theFeature2->attribute(SketchPlugin_Arc::START_ID()));
  74     anAttrList.push_back(theFeature2->attribute(SketchPlugin_Arc::END_ID()));
  75   }
  76
  77   const std::set<AttributePtr>& aRefsList = theFeature1->data()->refsToMe();
  78   std::set<AttributePtr>::const_iterator aRefIt = aRefsList.begin();
  79   for (; aRefIt != aRefsList.end() && !aResultFeature.get(); ++aRefIt) {
  80     FeaturePtr aConstrFeature = std::dynamic_pointer_cast<ModelAPI_Feature>((*aRefIt)->owner());
  81     if (aConstrFeature->getKind() != SketchPlugin_ConstraintCoincidence::ID())
  82       continue;
  83     AttributeRefAttrPtr aRefAttr = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(*aRefIt);
  84     AttributePtr anAttr = aRefAttr->attr();
  85     if (anAttr->id() == SketchPlugin_Arc::CENTER_ID())
  86       continue;
  87
  88     anAttr = aConstrFeature->attribute(SketchPlugin_Constraint::ENTITY_A());
  89     if (anAttr == *aRefIt)
  90       anAttr = aConstrFeature->attribute(SketchPlugin_Constraint::ENTITY_B());
  91
  92     aRefAttr = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(anAttr);
  93     if (!aRefAttr)
  94       continue;
  95     anAttr = aRefAttr->attr();
  96     for (std::list<AttributePtr>::const_iterator anIt = anAttrList.begin();
  97          anIt != anAttrList.end() && !aResultFeature.get(); ++anIt)
  98       if (*anIt == anAttr)
  99         aResultFeature = aConstrFeature;
 100   }
 101   return aResultFeature;
 102 }
 103
 104 AttributePoint2DPtr SketchPlugin_ConstraintCoincidence::getPoint(const FeaturePtr& theFeature)
 105 {
 106   AttributePoint2DPtr aPoint = ModelGeomAlgo_Point2D::getPointOfRefAttr(theFeature.get(),
 107                                                          SketchPlugin_Constraint::ENTITY_A(),
 108                                  SketchPlugin_Point::ID(), SketchPlugin_Point::COORD_ID());
 109   if (!aPoint.get())
 110     aPoint = ModelGeomAlgo_Point2D::getPointOfRefAttr(theFeature.get(),
 111                                                       SketchPlugin_Constraint::ENTITY_B(),
 112                                  SketchPlugin_Point::ID(), SketchPlugin_Point::COORD_ID());
 113   return aPoint;
 114 }
 115
 116 void SketchPlugin_ConstraintCoincidence::createCoincidenceFeature(SketchPlugin_Sketch* theSketch,
 117                                            const std::shared_ptr<GeomDataAPI_Point2D>& thePoint1,
 118                                            const std::shared_ptr<GeomDataAPI_Point2D>& thePoint2)
 119 {
 120   FeaturePtr aFeature;
 121   if (theSketch) {
 122     aFeature = theSketch->addFeature(SketchPlugin_ConstraintCoincidence::ID());
 123   } else {
 124     std::shared_ptr<ModelAPI_Document> aDoc = theSketch->document();
 125     aFeature = aDoc->addFeature(SketchPlugin_ConstraintCoincidence::ID());
 126   }
 127
 128   std::shared_ptr<ModelAPI_Data> aData = aFeature->data();
 129
 130   std::shared_ptr<ModelAPI_AttributeRefAttr> aRef1 = std::dynamic_pointer_cast<
 131       ModelAPI_AttributeRefAttr>(aData->attribute(SketchPlugin_Constraint::ENTITY_A()));
 132   aRef1->setAttr(thePoint1);
 133
 134   std::shared_ptr<ModelAPI_AttributeRefAttr> aRef2 = std::dynamic_pointer_cast<
 135       ModelAPI_AttributeRefAttr>(aData->attribute(SketchPlugin_Constraint::ENTITY_B()));
 136   aRef2->setAttr(thePoint2);
 137
 138   // we need to flush created signal in order to coincidence is processed by solver
 139   Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_CREATED));
 140 }