1 // Copyright (C) 2014-2019 CEA/DEN, EDF R&D
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.
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.
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
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
20 #include "CollectionPlugin_Group.h"
22 #include <ModelAPI_Data.h>
23 #include <ModelAPI_Document.h>
24 #include <ModelAPI_AttributeInteger.h>
25 #include <ModelAPI_AttributeString.h>
26 #include <ModelAPI_AttributeSelectionList.h>
27 #include <ModelAPI_AttributeIntArray.h>
28 #include <ModelAPI_ResultGroup.h>
31 CollectionPlugin_Group::CollectionPlugin_Group()
35 void CollectionPlugin_Group::initAttributes()
37 AttributeSelectionListPtr aList = std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(
38 data()->addAttribute(LIST_ID(), ModelAPI_AttributeSelectionList::typeId()));
39 aList->setWholeResultAllowed(true); // allow to select the whole result
42 void CollectionPlugin_Group::execute()
44 if (results().empty() || firstResult()->isDisabled()) { // just create result if not exists
45 ResultPtr aGroup = document()->createGroup(data());
50 // returns name with suffix, not existing in the existing set
51 static std::string findName(
52 const std::string theOrigin, int& theSuffix, std::set<std::string>& theExisting)
56 std::ostringstream aName;
57 aName<<theOrigin<<"_"<<theSuffix;
60 } while(theExisting.count(aRes));
61 theExisting.insert(aRes);
66 bool CollectionPlugin_Group::customAction(const std::string& theActionId)
68 if (theActionId == "split") {
69 DocumentPtr aDoc = document();
70 // collect all existing names of features to give unique names
71 std::set<std::string> aFeatNames, aResNames;
72 std::list<FeaturePtr> allFeat = aDoc->allFeatures();
73 std::list<FeaturePtr>::iterator allFeatIter = allFeat.begin();
74 for(; allFeatIter != allFeat.end(); allFeatIter++) {
75 FeaturePtr aFeat = *allFeatIter;
76 if (aFeat->data().get() && aFeat->data()->isValid()) {
77 aFeatNames.insert(aFeat->name());
78 if (aFeat->getKind() == ID() && aFeat->data().get() && aFeat->data()->isValid()) {
79 std::list<ResultPtr>::const_iterator aRess = aFeat->results().cbegin();
80 for(; aRess != aFeat->results().cend(); aRess++) {
81 ResultPtr aRes = *aRess;
82 if (aRes->data().get() && aRes->data()->isValid()) {
83 aResNames.insert(aRes->data()->name());
90 AttributeSelectionListPtr aList = selectionList(LIST_ID());
91 std::set<int> aRemoved;
92 bool aStay = false; // to indicate that the good attribute found stays in the list
93 int anIndex = 1; // index of the name assigned to group-feature and result
94 // added in the order: 3 2 1 orig=0, so, keep the results to give names later
95 std::list<ObjectPtr> aResults;
96 for(int aNext = aList->size() - 1; aNext >= 0; aNext--) {
97 AttributeSelectionPtr anOldAttr = aList->value(aNext);
98 if (anOldAttr->isInvalid() || !anOldAttr->context().get()) {// remove invalids
99 aRemoved.insert(aNext);
106 aRemoved.insert(aNext);
107 FeaturePtr aNew = aDoc->addFeature(ID(), false);
108 AttributeSelectionListPtr aNewList = aNew->selectionList(LIST_ID());
109 aNewList->setSelectionType(aList->selectionType());
110 aNewList->append(anOldAttr->contextObject(), anOldAttr->value());
111 aResults.push_front(aNew); // to keep the order
113 aResults.push_back(data()->owner());
114 // remove all selections except the first
115 aList->remove(aRemoved);
117 if (aResults.size() > 1) { // rename if there are new groups appeared only
118 std::list<ObjectPtr>::iterator aResIter = aResults.begin();
119 for(int aSuffix = 1; aResIter != aResults.end(); aResIter++) {
120 FeaturePtr aFeat = std::dynamic_pointer_cast<ModelAPI_Feature>(*aResIter);
121 aFeat->data()->setName(findName(name(), aSuffix, aFeatNames));
122 if (!aFeat->results().empty() && !results().empty()) {
123 int aResSuf = aSuffix - 1;
124 std::string aResName = findName(firstResult()->data()->name(), aResSuf, aResNames);
125 aFeat->firstResult()->data()->setName(aResName);
126 // set the same color of result as in origin
127 if (firstResult()->data()->attribute(ModelAPI_Result::COLOR_ID()).get()) {
128 AttributeIntArrayPtr aSourceColor =
129 firstResult()->data()->intArray(ModelAPI_Result::COLOR_ID());
130 if (aSourceColor.get() && aSourceColor->size()) {
131 AttributeIntArrayPtr aDestColor =
132 aFeat->firstResult()->data()->intArray(ModelAPI_Result::COLOR_ID());
133 aDestColor->setSize(aSourceColor->size());
134 for(int a = 0; a < aSourceColor->size(); a++)
135 aDestColor->setValue(a, aSourceColor->value(a));
140 // remove also filters if split performed
141 FiltersFeaturePtr aFilters = aList->filters();
142 if (aFilters.get()) {
143 std::list<std::string> aFiltersList = aFilters->filters();
144 std::list<std::string>::iterator aFilterName = aFiltersList.begin();
145 for(; aFilterName != aFiltersList.end(); aFilterName++) {
146 aFilters->removeFilter(*aFilterName);