src/SketchPlugin/SketchPlugin_MultiRotation.cpp

   1 // Copyright (C) 2014-2020  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 email : webmaster.salome@opencascade.com
  18 //
  19
  20 #include "SketchPlugin_MultiRotation.h"
  21 #include "SketchPlugin_Tools.h"
  22
  23 #include <GeomDataAPI_Point2D.h>
  24 #include <ModelAPI_AttributeRefAttr.h>
  25 #include <ModelAPI_AttributeBoolean.h>
  26 #include <ModelAPI_AttributeDouble.h>
  27 #include <ModelAPI_AttributeString.h>
  28 #include <ModelAPI_AttributeInteger.h>
  29 #include <ModelAPI_Data.h>
  30 #include <ModelAPI_ResultConstruction.h>
  31 #include <ModelAPI_AttributeRefList.h>
  32 #include <ModelAPI_Events.h>
  33 #include <ModelAPI_Session.h>
  34 #include <ModelAPI_Validator.h>
  35 #include <ModelAPI_Tools.h>
  36
  37 #include <SketchPlugin_SketchEntity.h>
  38
  39 #include <GeomAPI_Pnt2d.h>
  40 #include <GeomAPI_XY.h>
  41
  42 #include <SketcherPrs_Factory.h>
  43
  44 #include <cmath>
  45
  46 static const double PI = 3.1415926535897932;
  47 static const double PERIOD = 360.0;
  48 static const double ANGLETOL = 1.e-7;
  49
  50 SketchPlugin_MultiRotation::SketchPlugin_MultiRotation()
  51   : isUpdatingAngle(false)
  52 {
  53 }
  54
  55 void SketchPlugin_MultiRotation::initAttributes()
  56 {
  57   data()->addAttribute(CENTER_ID(), ModelAPI_AttributeRefAttr::typeId());
  58
  59   data()->addAttribute(ANGLE_TYPE(),   ModelAPI_AttributeString::typeId());
  60   data()->addAttribute(ANGLE_ID(), ModelAPI_AttributeDouble::typeId());
  61   data()->addAttribute(NUMBER_OF_OBJECTS_ID(), ModelAPI_AttributeInteger::typeId());
  62   data()->addAttribute(SketchPlugin_Constraint::ENTITY_A(), ModelAPI_AttributeRefList::typeId());
  63   data()->addAttribute(SketchPlugin_Constraint::ENTITY_B(), ModelAPI_AttributeRefList::typeId());
  64   data()->addAttribute(ROTATION_LIST_ID(), ModelAPI_AttributeRefList::typeId());
  65   data()->addAttribute(REVERSED_ID(), ModelAPI_AttributeBoolean::typeId());
  66
  67   ModelAPI_Session::get()->validators()->
  68     registerNotObligatory(getKind(), SketchPlugin_Constraint::ENTITY_A());
  69   ModelAPI_Session::get()->validators()->
  70     registerNotObligatory(getKind(), SketchPlugin_Constraint::ENTITY_B());
  71 }
  72
  73 void SketchPlugin_MultiRotation::execute()
  74 {
  75   if (!sketch()) {
  76     // it is possible, that this method is called before this feature has back reference to sketch
  77     // in this case, the execute is performed after this is done
  78     return;
  79   }
  80
  81   AttributeRefListPtr aRotationObjectRefs = reflist(ROTATION_LIST_ID());
  82   int aNbCopies = integer(NUMBER_OF_OBJECTS_ID())->value() - 1;
  83   if (aNbCopies <= 0)
  84     return;
  85
  86   // Obtain center and angle of rotation
  87   AttributeRefAttrPtr aCenter = data()->refattr(CENTER_ID());
  88   if (!aCenter || !aCenter->isInitialized())
  89     return;
  90
  91   //double anAngle = std::dynamic_pointer_cast<ModelAPI_AttributeDouble>(
  92   //                                                           attribute(ANGLE_ID()))->value();
  93
  94   // Convert angle to radians
  95   //anAngle *= PI / 180.0;
  96
  97   // Wait all objects being created, then send update events
  98   static Events_ID anUpdateEvent = Events_Loop::eventByName(EVENT_OBJECT_UPDATED);
  99   bool isUpdateFlushed = Events_Loop::loop()->isFlushed(anUpdateEvent);
 100   if (isUpdateFlushed)
 101     Events_Loop::loop()->setFlushed(anUpdateEvent, false);
 102
 103   std::shared_ptr<ModelAPI_Data> aData = data();
 104   AttributeRefListPtr aRefListOfShapes = std::dynamic_pointer_cast<ModelAPI_AttributeRefList>(
 105       aData->attribute(SketchPlugin_Constraint::ENTITY_A()));
 106   AttributeRefListPtr aRefListOfRotated = std::dynamic_pointer_cast<ModelAPI_AttributeRefList>(
 107       aData->attribute(SketchPlugin_Constraint::ENTITY_B()));
 108   int aCurrentNbCopies = aRefListOfShapes->size() ?
 109       aRefListOfRotated->size() / aRefListOfShapes->size() - 1 : 0;
 110   std::list<ObjectPtr> anInitialList = aRefListOfShapes->list();
 111   std::list<ObjectPtr> aTargetList = aRefListOfRotated->list();
 112   std::list<ObjectPtr> anAddition;
 113   std::vector<bool> isUsed(anInitialList.size(), false);
 114   // collect new items and check the items to remove
 115   for(int anInd = 0; anInd < aRotationObjectRefs->size(); anInd++) {
 116     ObjectPtr anObject = aRotationObjectRefs->object(anInd);
 117     std::list<ObjectPtr>::const_iterator anIt = anInitialList.begin();
 118     std::vector<bool>::iterator aUsedIt = isUsed.begin();
 119     for (; anIt != anInitialList.end(); anIt++, aUsedIt++)
 120       if (*anIt == anObject) {
 121         *aUsedIt = true;
 122         break;
 123       }
 124     if (anIt == anInitialList.end())
 125       anAddition.push_back(anObject);
 126   }
 127   // remove unused items
 128   std::list<ObjectPtr>::iterator anInitIter = anInitialList.begin();
 129   std::list<ObjectPtr>::iterator aTargetIter = aTargetList.begin();
 130   std::vector<bool>::iterator aUsedIter = isUsed.begin();
 131   std::set<FeaturePtr> aFeaturesToBeRemoved;
 132   for (; aUsedIter != isUsed.end(); aUsedIter++) {
 133     if (!(*aUsedIter)) {
 134       aRefListOfShapes->remove(*anInitIter);
 135       aRefListOfRotated->remove(*aTargetIter++);
 136       for (int i = 0; i < aCurrentNbCopies && aTargetIter != aTargetList.end();
 137            i++, aTargetIter++) {
 138         aRefListOfRotated->remove(*aTargetIter);
 139         // remove the corresponding feature from the sketch
 140         ResultConstructionPtr aRC =
 141             std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(*aTargetIter);
 142         DocumentPtr aDoc = aRC ? aRC->document() : DocumentPtr();
 143         FeaturePtr aFeature =  aDoc ? aDoc->feature(aRC) : FeaturePtr();
 144         if (aFeature)
 145           aFeaturesToBeRemoved.insert(aFeature);
 146       }
 147     } else {
 148       for (int i = 0; i <= aNbCopies && aTargetIter != aTargetList.end(); i++)
 149         aTargetIter++;
 150     }
 151     if (anInitIter != anInitialList.end())
 152       anInitIter++;
 153   }
 154   ModelAPI_Tools::removeFeaturesAndReferences(aFeaturesToBeRemoved);
 155   aFeaturesToBeRemoved.clear();
 156   // change number of copies
 157   if (aCurrentNbCopies != 0 && aNbCopies != aCurrentNbCopies) {
 158     bool isAdd = aNbCopies > aCurrentNbCopies;
 159     int aMinCopies = isAdd ? aCurrentNbCopies : aNbCopies;
 160     int aMaxCopies = isAdd ? aNbCopies : aCurrentNbCopies;
 161     int ind = 0;
 162
 163     aTargetList = aRefListOfRotated->list();
 164     aTargetIter = aTargetList.begin();
 165     ObjectPtr anObjToCopy = *aTargetIter;
 166     while (aTargetIter != aTargetList.end()) {
 167       aRefListOfRotated->remove(*aTargetIter);
 168       aTargetIter++;
 169       ind++;
 170       if (ind > aMinCopies && ind <=aMaxCopies) {
 171         while (ind <= aMaxCopies) {
 172           if (isAdd) {
 173             // Add new shifted item
 174             ObjectPtr anObject = copyFeature(anObjToCopy);
 175             aTargetList.insert(aTargetIter, anObject);
 176           } else {
 177             // remove object
 178             std::list<ObjectPtr>::iterator aRemoveIt = aTargetIter++;
 179             ObjectPtr anObject = *aRemoveIt;
 180             aTargetList.erase(aRemoveIt);
 181             aRefListOfRotated->remove(anObject);
 182             // remove the corresponding feature from the sketch
 183             ResultConstructionPtr aRC =
 184                 std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(anObject);
 185             DocumentPtr aDoc = aRC ? aRC->document() : DocumentPtr();
 186             FeaturePtr aFeature =  aDoc ? aDoc->feature(aRC) : FeaturePtr();
 187             if (aFeature)
 188               aFeaturesToBeRemoved.insert(aFeature);
 189           }
 190           ind++;
 191         }
 192         ind = 0;
 193         if (aTargetIter != aTargetList.end())
 194           anObjToCopy = *aTargetIter;
 195       }
 196     }
 197     ModelAPI_Tools::removeFeaturesAndReferences(aFeaturesToBeRemoved);
 198     aFeaturesToBeRemoved.clear();
 199
 200     for (aTargetIter = aTargetList.begin(); aTargetIter != aTargetList.end(); aTargetIter++)
 201       aRefListOfRotated->append(*aTargetIter);
 202   }
 203   // add new items
 204   std::list<ObjectPtr>::iterator anAddIter = anAddition.begin();
 205   for (; anAddIter != anAddition.end(); anAddIter++) {
 206     aRefListOfShapes->append(*anAddIter);
 207     aRefListOfRotated->append(*anAddIter);
 208     for (int i = 0; i < aNbCopies; i++) {
 209       ObjectPtr anObject = copyFeature(*anAddIter);
 210       aRefListOfRotated->append(anObject);
 211     }
 212   }
 213
 214 ////  if (fabs(anAngle) > 1.e-12) {
 215 ////    // Recalculate positions of features
 216 ////    aTargetList = aRefListOfRotated->list();
 217 ////    aTargetIter = aTargetList.begin();
 218 ////    while (aTargetIter != aTargetList.end()) {
 219 ////      ObjectPtr anInitialObject = *aTargetIter++;
 220 ////      for (int i = 0; i < aNbCopies && aTargetIter != aTargetList.end(); i++, aTargetIter++)
 221 //// rotateFeature(anInitialObject, *aTargetIter, aCenter->x(), aCenter->y(), anAngle * (i + 1));
 222 ////    }
 223 ////  }
 224
 225   // send events to update the sub-features by the solver
 226   if (isUpdateFlushed)
 227     Events_Loop::loop()->setFlushed(anUpdateEvent, true);
 228 }
 229
 230 AISObjectPtr SketchPlugin_MultiRotation::getAISObject(AISObjectPtr thePrevious)
 231 {
 232   if (!sketch())
 233     return thePrevious;
 234
 235   AISObjectPtr anAIS = SketcherPrs_Factory::rotateConstraint(this, sketch(),
 236                                                              thePrevious);
 237   return anAIS;
 238 }
 239
 240 void SketchPlugin_MultiRotation::erase()
 241 {
 242   static Events_Loop* aLoop = Events_Loop::loop();
 243   static Events_ID aRedispEvent = aLoop->eventByName(EVENT_OBJECT_TO_REDISPLAY);
 244
 245   // Set copy attribute to false on all copied features.
 246   AttributeRefListPtr aRefListOfShapes = std::dynamic_pointer_cast<ModelAPI_AttributeRefList>(
 247       data()->attribute(SketchPlugin_Constraint::ENTITY_A()));
 248   AttributeRefListPtr aRefListOfRotated = std::dynamic_pointer_cast<ModelAPI_AttributeRefList>(
 249       data()->attribute(SketchPlugin_Constraint::ENTITY_B()));
 250
 251   if(aRefListOfShapes.get() && aRefListOfRotated.get()) {
 252     for(int anIndex = 0; anIndex < aRefListOfRotated->size(); anIndex++) {
 253       ObjectPtr anObject = aRefListOfRotated->object(anIndex);
 254       if(aRefListOfShapes->isInList(anObject)) {
 255         // Don't modify attribute of original features, just skip.
 256         continue;
 257       }
 258       if(anObject.get()) {
 259         ResultPtr aRes = std::dynamic_pointer_cast<ModelAPI_Result>(anObject);
 260         if(aRes.get()) {
 261           FeaturePtr aFeature = aRes->document()->feature(aRes);
 262           if(aFeature.get()) {
 263             AttributeBooleanPtr aBooleanAttr =
 264               aFeature->boolean(SketchPlugin_SketchEntity::COPY_ID());
 265             if(aBooleanAttr.get()) {
 266               if (ModelAPI_Session::get()->isOperation()) // if this is not undo or redo
 267                 aBooleanAttr->setValue(false);
 268               // Redisplay object as it is not copy anymore.
 269               ModelAPI_EventCreator::get()->sendUpdated(aRes, aRedispEvent);
 270             }
 271           }
 272         }
 273       }
 274     }
 275   }
 276
 277   SketchPlugin_ConstraintBase::erase();
 278 }
 279
 280 ObjectPtr SketchPlugin_MultiRotation::copyFeature(ObjectPtr theObject)
 281 {
 282   ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(theObject);
 283   FeaturePtr aFeature = ModelAPI_Feature::feature(theObject);
 284   if (!aFeature || !aResult)
 285     return ObjectPtr();
 286
 287   FeaturePtr aNewFeature =
 288     SketchPlugin_Sketch::addUniqueNamedCopiedFeature(aFeature, sketch(), true);
 289   aNewFeature->execute();
 290
 291   static Events_ID aRedisplayEvent = Events_Loop::eventByName(EVENT_OBJECT_TO_REDISPLAY);
 292   ModelAPI_EventCreator::get()->sendUpdated(aNewFeature, aRedisplayEvent);
 293
 294   std::shared_ptr<GeomAPI_Shape> aShapeIn = aResult->shape();
 295   const std::list<ResultPtr>& aResults = aNewFeature->results();
 296   std::list<ResultPtr>::const_iterator anIt = aResults.begin();
 297   for (; anIt != aResults.end(); anIt++) {
 298     ResultConstructionPtr aRC = std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(*anIt);
 299     if (!aRC) continue;
 300     std::shared_ptr<GeomAPI_Shape> aShapeOut = aRC->shape();
 301     if ((aShapeIn->isVertex() && aShapeOut->isVertex()) ||
 302         (aShapeIn->isEdge() && aShapeOut->isEdge()) ||
 303         (aShapeIn->isFace() && aShapeOut->isFace()))
 304       return aRC;
 305   }
 306   return ObjectPtr();
 307 }
 308
 309 /*void SketchPlugin_MultiRotation::rotateFeature(
 310     ObjectPtr theInitial, ObjectPtr theTarget,
 311     double theCenterX, double theCenterY, double theAngle)
 312 {
 313   std::shared_ptr<GeomAPI_Pnt2d> aCenter(new GeomAPI_Pnt2d(theCenterX, theCenterY));
 314   double cosA = std::cos(theAngle);
 315   double sinA = std::sin(theAngle);
 316
 317   FeaturePtr anInitialFeature = ModelAPI_Feature::feature(theInitial);
 318   FeaturePtr aTargetFeature = ModelAPI_Feature::feature(theTarget);
 319
 320   // block feature update
 321   aTargetFeature->data()->blockSendAttributeUpdated(true);
 322
 323   std::list<AttributePtr> anInitAttrList =
 324       anInitialFeature->data()->attributes(GeomDataAPI_Point2D::typeId());
 325   std::list<AttributePtr> aTargetAttrList =
 326       aTargetFeature->data()->attributes(GeomDataAPI_Point2D::typeId());
 327   std::list<AttributePtr>::iterator anInitIt = anInitAttrList.begin();
 328   std::list<AttributePtr>::iterator aTargetIt = aTargetAttrList.begin();
 329   for (; anInitIt != anInitAttrList.end(); anInitIt++, aTargetIt++) {
 330     std::shared_ptr<GeomDataAPI_Point2D> aPointFrom =
 331         std::dynamic_pointer_cast<GeomDataAPI_Point2D>(*anInitIt);
 332     std::shared_ptr<GeomDataAPI_Point2D> aPointTo =
 333         std::dynamic_pointer_cast<GeomDataAPI_Point2D>(*aTargetIt);
 334     std::shared_ptr<GeomAPI_XY> aPnt = aPointFrom->pnt()->xy();
 335     if (aPnt->distance(aCenter->xy()) > 1.e-7) {
 336       std::shared_ptr<GeomAPI_XY> aDir = aPnt->decreased(aCenter->xy());
 337       double dx = cosA * aDir->x() - sinA * aDir->y();
 338       double dy = sinA * aDir->x() + cosA * aDir->y();
 339       aPnt->setX(aCenter->x() + dx);
 340       aPnt->setY(aCenter->y() + dy);
 341     }
 342     aPointTo->setValue(aPnt->x(), aPnt->y());
 343   }
 344
 345   // unblock feature update
 346   aTargetFeature->data()->blockSendAttributeUpdated(false);
 347 }*/
 348
 349
 350 void SketchPlugin_MultiRotation::attributeChanged(const std::string& theID)
 351 {
 352   if (theID == ROTATION_LIST_ID()) {
 353     AttributeRefListPtr aRotationObjectRefs = reflist(ROTATION_LIST_ID());
 354     if (aRotationObjectRefs->size() == 0) {
 355       int aNbCopies = integer(NUMBER_OF_OBJECTS_ID())->value()-1;
 356       if (aNbCopies <= 0)
 357         return;
 358
 359       // Clear list of objects
 360       AttributeRefListPtr aRefListOfRotated = reflist(SketchPlugin_Constraint::ENTITY_B());
 361       std::list<ObjectPtr> aTargetList = aRefListOfRotated->list();
 362       std::list<ObjectPtr>::iterator aTargetIter = aTargetList.begin();
 363       std::set<FeaturePtr> aFeaturesToBeRemoved;
 364       while (aTargetIter != aTargetList.end()) {
 365         aTargetIter++;
 366         for (int i = 0; i < aNbCopies && aTargetIter != aTargetList.end(); i++, aTargetIter++) {
 367           aRefListOfRotated->remove(*aTargetIter);
 368           // remove the corresponding feature from the sketch
 369           FeaturePtr aFeature = ModelAPI_Feature::feature(*aTargetIter);
 370           if (aFeature)
 371             aFeaturesToBeRemoved.insert(aFeature);
 372         }
 373       }
 374       ModelAPI_Tools::removeFeaturesAndReferences(aFeaturesToBeRemoved);
 375
 376       aRefListOfRotated->clear();
 377       reflist(SketchPlugin_Constraint::ENTITY_A())->clear();
 378     }
 379   }
 380   else if (!isUpdatingAngle && real(ANGLE_ID())->isInitialized())
 381   {
 382     isUpdatingAngle = true;
 383     AttributeDoublePtr anAngle = real(ANGLE_ID());
 384     if (theID == ANGLE_TYPE() && integer(NUMBER_OF_OBJECTS_ID())->isInitialized()) {
 385       if (string(ANGLE_TYPE())->value() != "SingleAngle")
 386         anAngle->setValue(anAngle->value() * (integer(NUMBER_OF_OBJECTS_ID())->value() - 1));
 387       else
 388       {
 389         int aNbSplits = integer(NUMBER_OF_OBJECTS_ID())->value();
 390         if (anAngle->value() < PERIOD - ANGLETOL)
 391           aNbSplits -= 1;
 392         anAngle->setValue(anAngle->value() / aNbSplits);
 393       }
 394     }
 395     else if (theID == ANGLE_ID()) {
 396       if (anAngle->value() > PERIOD + ANGLETOL || anAngle->value() < -ANGLETOL)
 397         anAngle->setValue(anAngle->value() + PERIOD * ceil(-anAngle->value() / PERIOD));
 398       if (fabs(anAngle->value() - PERIOD) < ANGLETOL)
 399         anAngle->setValue(PERIOD);
 400       else if (fabs(anAngle->value()) < ANGLETOL)
 401         anAngle->setValue(0.);
 402     }
 403     isUpdatingAngle = false;
 404   }
 405 }