src/SketchAPI/SketchAPI_Rotation.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 "SketchAPI_Rotation.h"
  22 #include <SketchAPI_SketchEntity.h>
  23 //--------------------------------------------------------------------------------------
  24 #include <ModelHighAPI_Dumper.h>
  25 #include <ModelHighAPI_Tools.h>
  26
  27 #include <SketchPlugin_SketchEntity.h>
  28 //--------------------------------------------------------------------------------------
  29 SketchAPI_Rotation::SketchAPI_Rotation(
  30     const std::shared_ptr<ModelAPI_Feature> & theFeature)
  31 : ModelHighAPI_Interface(theFeature)
  32 {
  33   initialize();
  34 }
  35
  36 SketchAPI_Rotation::SketchAPI_Rotation(
  37     const std::shared_ptr<ModelAPI_Feature> & theFeature,
  38     const std::list<std::shared_ptr<ModelAPI_Object> > & theObjects,
  39     const ModelHighAPI_RefAttr & theCenter,
  40     const ModelHighAPI_Double & theAngle,
  41     const ModelHighAPI_Integer & theNumberOfObjects,
  42     bool theFullValue)
  43 : ModelHighAPI_Interface(theFeature)
  44 {
  45   if (initialize()) {
  46     fillAttribute(theObjects, rotationList());
  47     fillAttribute(theCenter, center());
  48     fillAttribute(theAngle, angle());
  49     fillAttribute(theNumberOfObjects, numberOfObjects());
  50     fillAttribute(theFullValue ? "FullAngle" : "SingleAngle", valueType());
  51
  52     execute(true);
  53   }
  54 }
  55
  56 SketchAPI_Rotation::~SketchAPI_Rotation()
  57 {
  58
  59 }
  60
  61 std::list<std::shared_ptr<SketchAPI_SketchEntity> > SketchAPI_Rotation::rotated() const
  62 {
  63   std::list<ObjectPtr> aList = rotatedObjects()->list();
  64   // remove all initial features
  65   std::list<FeaturePtr> anIntermediate;
  66   std::list<ObjectPtr>::const_iterator anIt = aList.begin();
  67   for (; anIt != aList.end(); ++anIt) {
  68     FeaturePtr aFeature = ModelAPI_Feature::feature(*anIt);
  69     AttributeBooleanPtr isCopy = aFeature->boolean(SketchPlugin_SketchEntity::COPY_ID());
  70     if (isCopy.get() && isCopy->value())
  71       anIntermediate.push_back(aFeature);
  72   }
  73   return SketchAPI_SketchEntity::wrap(anIntermediate);
  74 }
  75
  76 //--------------------------------------------------------------------------------------
  77
  78 void SketchAPI_Rotation::dump(ModelHighAPI_Dumper& theDumper) const
  79 {
  80   FeaturePtr aBase = feature();
  81   const std::string& aSketchName = theDumper.parentName(aBase);
  82
  83   AttributeRefListPtr aRotObjects = rotationList();
  84   AttributeRefAttrPtr aCenter = center();
  85   AttributeDoublePtr anAngle = angle();
  86   AttributeIntegerPtr aNbCopies = numberOfObjects();
  87   bool isFullValue = valueType()->value() != "SingleAngle";
  88
  89   // Check all attributes are already dumped. If not, store the constraint as postponed.
  90   if (!theDumper.isDumped(aCenter) || !theDumper.isDumped(aRotObjects)) {
  91     theDumper.postpone(aBase);
  92     return;
  93   }
  94
  95   theDumper << aBase << " = " << aSketchName << ".addRotation("
  96             << aRotObjects << ", " << aCenter << ", " << anAngle << ", " << aNbCopies;
  97   if (isFullValue)
  98     theDumper << ", " << isFullValue;
  99   theDumper << ")" << std::endl;
 100
 101   // Dump variables for a list of rotated features
 102   theDumper << "[";
 103   std::list<std::shared_ptr<SketchAPI_SketchEntity> > aList = rotated();
 104   std::list<std::shared_ptr<SketchAPI_SketchEntity> >::const_iterator anIt = aList.begin();
 105   for (; anIt != aList.end(); ++anIt) {
 106     if (anIt != aList.begin())
 107       theDumper << ", ";
 108     theDumper << (*anIt)->feature();
 109   }
 110   theDumper << "] = " << theDumper.name(aBase) << ".rotated()" << std::endl;
 111
 112   // Set necessary "auxiliary" flag for rotated features
 113   // (flag is set if it differs to base entity)
 114   std::list<ObjectPtr> aRotList = aRotObjects->list();
 115   std::list<ObjectPtr>::const_iterator aRIt = aRotList.begin();
 116   anIt = aList.begin();
 117   for (; aRIt != aRotList.end(); ++aRIt) {
 118     FeaturePtr aFeature = ModelAPI_Feature::feature(*aRIt);
 119     if (!aFeature)
 120       continue;
 121     bool aBaseAux = aFeature->boolean(SketchPlugin_SketchEntity::AUXILIARY_ID())->value();
 122
 123     for (int i = 1; i < aNbCopies->value(); ++i, ++anIt) {
 124       aFeature = (*anIt)->feature();
 125       bool aFeatAux = aFeature->boolean(SketchPlugin_SketchEntity::AUXILIARY_ID())->value();
 126       if (aFeatAux != aBaseAux)
 127         theDumper << theDumper.name((*anIt)->feature(), false)
 128                   << ".setAuxiliary(" << aFeatAux << ")" <<std::endl;
 129     }
 130   }
 131 }