src/ModelHighAPI/ModelHighAPI_Selection.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 "ModelHighAPI_Selection.h"
  22
  23 #include <ModelAPI_AttributeDouble.h>
  24 #include <ModelAPI_AttributeIntArray.h>
  25 #include <ModelAPI_AttributeSelection.h>
  26 #include <ModelAPI_AttributeSelectionList.h>
  27 #include <ModelAPI_Feature.h>
  28 #include <ModelAPI_ResultCompSolid.h>
  29 //--------------------------------------------------------------------------------------
  30
  31 //--------------------------------------------------------------------------------------
  32 ModelHighAPI_Selection::ModelHighAPI_Selection()
  33 : myVariantType(VT_Empty)
  34 {
  35 }
  36
  37 ModelHighAPI_Selection::ModelHighAPI_Selection(const std::shared_ptr<ModelAPI_Result>& theContext,
  38                                                const std::shared_ptr<GeomAPI_Shape>& theSubShape)
  39 : myVariantType(VT_ResultSubShapePair)
  40 , myResultSubShapePair(theContext, theSubShape)
  41 {
  42 }
  43
  44 ModelHighAPI_Selection::ModelHighAPI_Selection(const std::string& theType,
  45                                                const std::string& theSubShapeName)
  46 : myVariantType(VT_TypeSubShapeNamePair)
  47 , myTypeSubShapeNamePair(theType, theSubShapeName)
  48 {
  49 }
  50
  51 ModelHighAPI_Selection::~ModelHighAPI_Selection()
  52 {
  53 }
  54
  55 //--------------------------------------------------------------------------------------
  56 void ModelHighAPI_Selection::fillAttribute(
  57     const std::shared_ptr<ModelAPI_AttributeSelection> & theAttribute) const
  58 {
  59   switch(myVariantType) {
  60     case VT_Empty: return;
  61     case VT_ResultSubShapePair:
  62       theAttribute->setValue(myResultSubShapePair.first, myResultSubShapePair.second);
  63       return;
  64     case VT_TypeSubShapeNamePair:
  65       theAttribute->selectSubShape(myTypeSubShapeNamePair.first, myTypeSubShapeNamePair.second);
  66       if(theAttribute->isInvalid()) {
  67         FeaturePtr aFeature = ModelAPI_Feature::feature(theAttribute->owner());
  68         aFeature->setError(
  69           std::string("Error: attribute \"") + theAttribute->id() + std::string("\" is invalid."));
  70       }
  71       return;
  72   }
  73 }
  74
  75 //--------------------------------------------------------------------------------------
  76 void ModelHighAPI_Selection::appendToList(
  77     const std::shared_ptr<ModelAPI_AttributeSelectionList> & theAttribute) const
  78 {
  79   switch(myVariantType) {
  80     case VT_Empty: return;
  81     case VT_ResultSubShapePair:
  82       theAttribute->append(myResultSubShapePair.first, myResultSubShapePair.second);
  83       return;
  84     case VT_TypeSubShapeNamePair:
  85       // Note: the reverse order (first - type, second - sub-shape name)
  86       theAttribute->append(myTypeSubShapeNamePair.second, myTypeSubShapeNamePair.first);
  87       return;
  88   }
  89 }
  90
  91 //==================================================================================================
  92 ModelHighAPI_Selection::VariantType ModelHighAPI_Selection::variantType() const
  93 {
  94   return myVariantType;
  95 }
  96
  97 //==================================================================================================
  98 ResultSubShapePair ModelHighAPI_Selection::resultSubShapePair() const
  99 {
 100   return myResultSubShapePair;
 101 }
 102
 103 //==================================================================================================
 104 TypeSubShapeNamePair ModelHighAPI_Selection::typeSubShapeNamePair() const
 105 {
 106   return myTypeSubShapeNamePair;
 107 }
 108
 109 //==================================================================================================
 110 std::string ModelHighAPI_Selection::shapeType() const
 111 {
 112   switch(myVariantType) {
 113   case VT_ResultSubShapePair:
 114     return myResultSubShapePair.second.get() ? myResultSubShapePair.second->shapeTypeStr() :
 115                                                myResultSubShapePair.first->shape()->shapeTypeStr();
 116   case VT_TypeSubShapeNamePair: return myTypeSubShapeNamePair.first;
 117   }
 118
 119   return "SHAPE";
 120 }
 121
 122 //==================================================================================================
 123 void ModelHighAPI_Selection::setName(const std::string& theName)
 124 {
 125   if (myVariantType == VT_ResultSubShapePair) {
 126     std::shared_ptr<ModelAPI_Result> aResult = myResultSubShapePair.first;
 127     if(!aResult.get()) {
 128       return;
 129     }
 130     aResult->data()->setName(theName);
 131   }
 132 }
 133
 134 void ModelHighAPI_Selection::setColor(int theRed, int theGreen, int theBlue)
 135 {
 136   if (myVariantType != VT_ResultSubShapePair || !myResultSubShapePair.first.get())
 137     return;
 138
 139   AttributeIntArrayPtr aColor =
 140       myResultSubShapePair.first->data()->intArray(ModelAPI_Result::COLOR_ID());
 141   aColor->setSize(3);
 142   aColor->setValue(0, theRed);
 143   aColor->setValue(1, theGreen);
 144   aColor->setValue(2, theBlue);
 145 }
 146
 147 void ModelHighAPI_Selection::setDeflection(double theValue)
 148 {
 149   if (myVariantType != VT_ResultSubShapePair)
 150     return;
 151
 152   AttributeDoublePtr aDeflectionAttr =
 153     myResultSubShapePair.first->data()->real(ModelAPI_Result::DEFLECTION_ID());
 154
 155   aDeflectionAttr->setValue(theValue);
 156 }
 157
 158 int ModelHighAPI_Selection::numberOfSubs() const
 159 {
 160   if (myVariantType != VT_ResultSubShapePair)
 161     return 0;
 162
 163   ResultCompSolidPtr aCompSolid =
 164       std::dynamic_pointer_cast<ModelAPI_ResultCompSolid>(myResultSubShapePair.first);
 165   if (!aCompSolid)
 166     return 0;
 167
 168   return aCompSolid->numberOfSubs();
 169 }
 170
 171 ModelHighAPI_Selection ModelHighAPI_Selection::subResult(int theIndex) const
 172 {
 173   if (myVariantType != VT_ResultSubShapePair)
 174     return ModelHighAPI_Selection();
 175
 176   ResultCompSolidPtr aCompSolid =
 177       std::dynamic_pointer_cast<ModelAPI_ResultCompSolid>(myResultSubShapePair.first);
 178   if (!aCompSolid)
 179     return ModelHighAPI_Selection();
 180
 181   ResultBodyPtr aResult = aCompSolid->subResult(theIndex);
 182   return ModelHighAPI_Selection(aResult, aResult->shape());
 183 }