src/ModuleBase/ModuleBase_WidgetConcealedObjects.cpp

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D\r
   2 \r
   3 // File:        ModuleBase_WidgetConcealedObjects.cpp\r
   4 // Created:     29 Jul 2016\r
   5 // Author:      Natalia ERMOLAEVA\r
   6 \r
   7 #include <ModuleBase_WidgetConcealedObjects.h>\r
   8 #include <ModuleBase_Tools.h>\r
   9 \r
  10 #include <ModelAPI_Result.h>\r
  11 #include <ModelAPI_AttributeReference.h>\r
  12 #include <ModelAPI_AttributeRefList.h>\r
  13 #include <ModelAPI_Session.h>\r
  14 #include <ModelAPI_Validator.h>\r
  15 #include <ModelAPI_Tools.h>\r
  16 \r
  17 #include <Config_WidgetAPI.h>\r
  18 \r
  19 #include <QGridLayout>\r
  20 \r
  21 #include <QWidget>\r
  22 #include <QTableWidget>\r
  23 #include <QHeaderView>\r
  24 #include <QToolButton>\r
  25 \r
  26 ModuleBase_WidgetConcealedObjects::ModuleBase_WidgetConcealedObjects(QWidget* theParent,\r
  27                                                      const Config_WidgetAPI* theData)\r
  28 : ModuleBase_ModelWidget(theParent, theData)\r
  29 {\r
  30   myBaseShapeAttribute = theData->getProperty("base_shape_attribute");\r
  31   QGridLayout* aMainLay = new QGridLayout(this);\r
  32   ModuleBase_Tools::adjustMargins(aMainLay);\r
  33 \r
  34   myView = new QTableWidget(this);\r
  35   aMainLay->addWidget(myView);\r
  36 \r
  37   myView->setColumnCount(2);\r
  38   myView->horizontalHeader()->setVisible(false);\r
  39   myView->verticalHeader()->setVisible(false);\r
  40 }\r
  41 \r
  42 ModuleBase_WidgetConcealedObjects::~ModuleBase_WidgetConcealedObjects()\r
  43 {\r
  44 }\r
  45 \r
  46 bool ModuleBase_WidgetConcealedObjects::storeValueCustom()\r
  47 {\r
  48   if(!myFeature)\r
  49     return false;\r
  50   DataPtr aData = myFeature->data();\r
  51   AttributeRefListPtr anAttributeList = aData->reflist(attributeID());\r
  52   anAttributeList->clear();\r
  53   int aSize1 = anAttributeList->size(false);\r
  54   for (int i = 0, aSize = myView->rowCount(); i < aSize; i++) {\r
  55     QToolButton* aButton = dynamic_cast<QToolButton*>(myView->cellWidget(i, 0));;\r
  56     if (aButton->isChecked())\r
  57       anAttributeList->append(myConcealedResults[i]);\r
  58   }\r
  59   int aSize = anAttributeList->size(false);\r
  60   return true;\r
  61 }\r
  62 \r
  63 bool ModuleBase_WidgetConcealedObjects::restoreValueCustom()\r
  64 {\r
  65   FeaturePtr aBaseFeature;\r
  66   ObjectPtr anObject;\r
  67   if (myFeature) {\r
  68     anObject = ModuleBase_Tools::getObject(myFeature->attribute(myBaseShapeAttribute));\r
  69     if (anObject.get() != NULL)\r
  70       aBaseFeature = ModelAPI_Feature::feature(anObject);\r
  71   }\r
  72   if (myBaseFeature != aBaseFeature) {\r
  73     myView->setRowCount(0);\r
  74     myConcealedResults.clear();\r
  75     myBaseFeature = aBaseFeature;\r
  76     if (myBaseFeature.get()) {\r
  77       std::list<std::shared_ptr<ModelAPI_Result> > aResults;\r
  78       ModelAPI_Tools::getConcealedResults(myBaseFeature, aResults);\r
  79       std::list<std::shared_ptr<ModelAPI_Result> >::const_iterator anIt = aResults.begin(),\r
  80                                                                    aLast = aResults.end();\r
  81       for (; anIt != aLast; anIt++) {\r
  82         ResultPtr aResult = *anIt;\r
  83 \r
  84         int aRowId = myView->rowCount();\r
  85         addViewRow(aResult);\r
  86         myConcealedResults[aRowId] = aResult;\r
  87       }\r
  88     }\r
  89   }\r
  90   \r
  91   DataPtr aData = myFeature->data();\r
  92   AttributeRefListPtr anAttributeList = aData->reflist(attributeID());\r
  93   int aSize = anAttributeList->size();\r
  94   for (int i = 0, aSize = myView->rowCount(); i < aSize; i++) {\r
  95     ResultPtr aResult = myConcealedResults[i];\r
  96     QToolButton* aButton = dynamic_cast<QToolButton*>(myView->cellWidget(i, 0));\r
  97     bool isChecked = anAttributeList->isInList(aResult);\r
  98 \r
  99     bool aBlocked = aButton->blockSignals(true);\r
 100     aButton->setChecked(isChecked);\r
 101     this->updateItemIcon(aButton);\r
 102     aButton->blockSignals(aBlocked);\r
 103   }\r
 104   return true;\r
 105 }\r
 106 \r
 107 QList<QWidget*> ModuleBase_WidgetConcealedObjects::getControls() const\r
 108 {\r
 109   QList<QWidget*> result;\r
 110   result << myView;\r
 111   return result;\r
 112 }\r
 113 \r
 114 void ModuleBase_WidgetConcealedObjects::addViewRow(const std::shared_ptr<ModelAPI_Result>& theResult)\r
 115 {\r
 116   int anId = myView->rowCount();\r
 117   myView->setRowCount(anId+1);\r
 118 \r
 119   QToolButton* aVisibilityBtn = new QToolButton(this);\r
 120   connect(aVisibilityBtn, SIGNAL(toggled(bool)), this, SLOT(onItemToggled(bool)));\r
 121   aVisibilityBtn->setCheckable(true);\r
 122   aVisibilityBtn->setChecked(false);\r
 123   updateItemIcon(aVisibilityBtn);\r
 124 \r
 125   myView->setCellWidget(anId, 0, aVisibilityBtn);\r
 126   myView->setItem(anId, 1, new QTableWidgetItem(theResult->data()->name().c_str()));\r
 127 }\r
 128 \r
 129 void ModuleBase_WidgetConcealedObjects::onItemToggled(bool theState)\r
 130 {\r
 131   emit valuesChanged();\r
 132   updateObject(myFeature);\r
 133 }\r
 134 \r
 135 void ModuleBase_WidgetConcealedObjects::updateItemIcon(QToolButton* theButton)\r
 136 {\r
 137   bool isChecked = theButton->isChecked();\r
 138   theButton->setIcon(isChecked ? QIcon(":icons/concealed_on.png")\r
 139                                : QIcon(":icons/concealed_off.png"));\r
 140   theButton->setSizePolicy(QSizePolicy::Fixed, QSizePolicy::Fixed);\r
 141 }\r