Salome HOME
[bos #26534] EDF 24265 - Problem of recover in TUI
[modules/shaper.git] / src / FeaturesAPI / FeaturesAPI_Recover.cpp
1 // Copyright (C) 2014-2021  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 "FeaturesAPI_Recover.h"
21
22 #include <ModelHighAPI_Dumper.h>
23 #include <ModelHighAPI_Reference.h>
24 #include <ModelHighAPI_Tools.h>
25
26 //=================================================================================================
27 FeaturesAPI_Recover::FeaturesAPI_Recover(const std::shared_ptr<ModelAPI_Feature>& theFeature)
28 : ModelHighAPI_Interface(theFeature)
29 {
30   initialize();
31 }
32
33 //=================================================================================================
34 FeaturesAPI_Recover::FeaturesAPI_Recover(const std::shared_ptr<ModelAPI_Feature>& theFeature,
35   const ModelHighAPI_Reference& theBaseFeature,
36   const std::list<ModelHighAPI_Selection>& theRecoveredList,
37   const bool theRecoverCompound)
38 : ModelHighAPI_Interface(theFeature)
39 {
40   if (initialize()) {
41     std::string aMethod = theRecoverCompound ? FeaturesPlugin_Recover::METHOD_COMPOUND()
42                                              : FeaturesPlugin_Recover::METHOD_DEFAULT();
43     fillAttribute(aMethod, theFeature->string(FeaturesPlugin_Recover::METHOD()));
44
45     fillAttribute(theBaseFeature.feature(), mybaseFeature);
46     setRecoveredList(theRecoveredList);
47   }
48 }
49
50 //=================================================================================================
51 FeaturesAPI_Recover::~FeaturesAPI_Recover()
52 {}
53
54 //=================================================================================================
55 void FeaturesAPI_Recover::setBaseFeature(const ModelHighAPI_Reference& theBaseFeature)
56 {
57
58   fillAttribute(theBaseFeature.feature(), mybaseFeature);
59   execute();
60 }
61
62 //=================================================================================================
63 void FeaturesAPI_Recover::setRecoveredList(const std::list<ModelHighAPI_Selection>& theRecoverList)
64 {
65   fillAttribute(theRecoverList, myrecoveredList);
66   execute();
67 }
68
69 //==================================================================================================
70 void FeaturesAPI_Recover::dump(ModelHighAPI_Dumper& theDumper) const
71 {
72   FeaturePtr aBase = feature();
73   const std::string& aDocName = theDumper.name(aBase->document());
74
75   AttributeReferencePtr anAttrBaseFeature =
76     aBase->reference(FeaturesPlugin_Recover::BASE_FEATURE());
77   AttributeRefListPtr anAttrRecoveredEntities =
78     aBase->reflist(FeaturesPlugin_Recover::RECOVERED_ENTITIES());
79
80   FeaturePtr aFeature = ModelAPI_Feature::feature(anAttrBaseFeature->value());
81
82   theDumper << aBase << " = model.addRecover(" << aDocName << ", "
83             << aFeature << ", " << anAttrRecoveredEntities;
84
85   AttributeStringPtr aMethod = aBase->string(FeaturesPlugin_Recover::METHOD());
86   if (aMethod && aMethod->isInitialized() &&
87       aMethod->value() != FeaturesPlugin_Recover::METHOD_DEFAULT())
88     theDumper << ", " << true;
89
90   theDumper << ")" << std::endl;
91 }
92
93 //=================================================================================================
94 RecoverPtr addRecover(const std::shared_ptr<ModelAPI_Document>& thePart,
95   const ModelHighAPI_Reference& theBaseFeature,
96   const std::list<ModelHighAPI_Selection>& theRecoveredList, const bool theRecoverCompound)
97 {
98   std::shared_ptr<ModelAPI_Feature> aFeature = thePart->addFeature(FeaturesAPI_Recover::ID());
99   return RecoverPtr(new FeaturesAPI_Recover(
100     aFeature, theBaseFeature, theRecoveredList, theRecoverCompound));
101 }