src/ModuleBase/ModuleBase_WidgetConcealedObjects.cpp

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