src/CollectionPlugin/CollectionPlugin_WidgetField.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 email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
  18 //
  19
  20 #include "CollectionPlugin_WidgetField.h"
  21 #include "CollectionPlugin_Field.h"
  22
  23 #include <ModuleBase_Tools.h>
  24 #include <ModuleBase_IWorkshop.h>
  25 #include <ModuleBase_IModule.h>
  26 #include <ModuleBase_ISelection.h>
  27 #include <ModuleBase_IPropertyPanel.h>
  28
  29 #include <ModelAPI_AttributeSelectionList.h>
  30 #include <ModelAPI_AttributeStringArray.h>
  31 #include <ModelAPI_AttributeInteger.h>
  32 #include <ModelAPI_AttributeIntArray.h>
  33
  34 #include <QLayout>
  35 #include <QWidget>
  36 #include <QFormLayout>
  37 #include <QComboBox>
  38 #include <QSpinBox>
  39 #include <QLabel>
  40 #include <QSlider>
  41 #include <QTableWidget>
  42 #include <QPushButton>
  43 #include <QHeaderView>
  44 #include <QStackedWidget>
  45 #include <QValidator>
  46 #include <QLineEdit>
  47 #include <QEvent>
  48 #include <QMouseEvent>
  49 #include <QScrollBar>
  50 #include <QApplication>
  51
  52 const char* MYFirstCol = "Shape";
  53 const char* MYTrue = "True";
  54 const char* MYFalse = "False";
  55
  56 DataTableItemDelegate::DataTableItemDelegate(ModelAPI_AttributeTables::ValueType theType)
  57   : QStyledItemDelegate(), myType(theType)
  58 {
  59 }
  60
  61
  62 QWidget* DataTableItemDelegate::createEditor(QWidget* theParent,
  63                                              const QStyleOptionViewItem & theOption,
  64                                              const QModelIndex& theIndex ) const
  65 {
  66   QWidget* aEditor = 0;
  67   if ((theIndex.column() == 0) && (theIndex.row() > 0)) {
  68     QWidget* aWgt = QStyledItemDelegate::createEditor(theParent, theOption, theIndex);
  69     QLineEdit* aEdt = static_cast<QLineEdit*>(aWgt);
  70     aEdt->setReadOnly(true);
  71     aEditor = aEdt;
  72   } else {
  73     QLineEdit* aLineEdt = 0;
  74     switch (myType) {
  75     case ModelAPI_AttributeTables::DOUBLE:
  76       aLineEdt = dynamic_cast<QLineEdit*>(QStyledItemDelegate::createEditor(theParent,
  77                                                                             theOption,
  78                                                                             theIndex));
  79       if (aLineEdt) {
  80         aLineEdt->setValidator(new QDoubleValidator(aLineEdt));
  81         aEditor = aLineEdt;
  82       }
  83       break;
  84     case ModelAPI_AttributeTables::INTEGER:
  85       aLineEdt = dynamic_cast<QLineEdit*>(QStyledItemDelegate::createEditor(theParent,
  86                                                                             theOption,
  87                                                                             theIndex));
  88       if (aLineEdt) {
  89         aLineEdt->setValidator(new QIntValidator(aLineEdt));
  90         aEditor = aLineEdt;
  91       }
  92       break;
  93     case ModelAPI_AttributeTables::BOOLEAN:
  94       {
  95         QComboBox* aBox = new QComboBox(theParent);
  96         aBox->addItem(MYFalse);
  97         aBox->addItem(MYTrue);
  98         aEditor = aBox;
  99       }
 100       break;
 101     default:
 102       aEditor = QStyledItemDelegate::createEditor(theParent, theOption, theIndex);
 103     }
 104   }
 105   if (myType == ModelAPI_AttributeTables::BOOLEAN)
 106     connect(aEditor, SIGNAL(currentTextChanged(const QString&)),
 107       SLOT(onEditItem(const QString&)));
 108   else
 109     connect(aEditor, SIGNAL(textEdited(const QString&)),
 110       SLOT(onEditItem(const QString&)));
 111   return aEditor;
 112 }
 113
 114 void DataTableItemDelegate::onEditItem(const QString& theText)
 115 {
 116   QWidget* aWgt = dynamic_cast<QWidget*>(sender());
 117   commitData(aWgt);
 118 }
 119
 120
 121
 122 //**********************************************************************************
 123 //**********************************************************************************
 124 //**********************************************************************************
 125 CollectionPlugin_WidgetField::
 126   CollectionPlugin_WidgetField(QWidget* theParent,
 127                                ModuleBase_IWorkshop* theWorkshop,
 128                                const Config_WidgetAPI* theData):
 129 ModuleBase_WidgetSelector(theParent, theWorkshop, theData), myHeaderEditor(0),
 130   myIsTabEdit(false), myActivation(false)
 131 {
 132   QVBoxLayout* aMainLayout = new QVBoxLayout(this);
 133
 134   // Types definition controls
 135   QWidget* aTypesWgt = new QWidget(this);
 136   QFormLayout* aTypesLayout = new QFormLayout(aTypesWgt);
 137   aTypesLayout->setContentsMargins(0, 0, 0, 0);
 138   aMainLayout->addWidget(aTypesWgt);
 139
 140   // Type of shapes
 141   myShapeTypeCombo = new QComboBox(aTypesWgt);
 142   QStringList aShapeTypes;
 143   aShapeTypes << tr("Vertices") << tr("Edges") << tr("Faces")
 144     << tr("Solids") << tr("Objects") << tr("Parts");
 145   myShapeTypeCombo->addItems(aShapeTypes);
 146   aTypesLayout->addRow(tr("Type of shapes"), myShapeTypeCombo);
 147
 148   // Type of field
 149   myFieldTypeCombo = new QComboBox(aTypesWgt);
 150   QStringList aFieldTypes;
 151   aFieldTypes << tr("Boolean") << tr("Integer") << tr("Double")
 152     << tr("String");
 153   myFieldTypeCombo->addItems(aFieldTypes);
 154   myFieldTypeCombo->setCurrentIndex(2);
 155   aTypesLayout->addRow(tr("Type of field"), myFieldTypeCombo);
 156
 157   // Number of components
 158   myNbComponentsSpn = new QSpinBox(aTypesWgt);
 159   myNbComponentsSpn->setMinimum(1);
 160   aTypesLayout->addRow(tr("Nb. of components"), myNbComponentsSpn);
 161
 162   // Steps controls
 163   QFrame* aStepFrame = new QFrame(this);
 164   aStepFrame->setFrameShape(QFrame::Box);
 165   aStepFrame->setFrameStyle(QFrame::StyledPanel);
 166   QGridLayout* aStepLayout = new QGridLayout(aStepFrame);
 167   aMainLayout->addWidget(aStepFrame);
 168
 169   // Current step label
 170   aStepLayout->addWidget(new QLabel(tr("Current step"), aStepFrame), 0, 0);
 171   myCurStepLbl = new QLabel("1", aStepFrame);
 172   QFont aFont = myCurStepLbl->font();
 173   aFont.setBold(true);
 174   myCurStepLbl->setFont(aFont);
 175   aStepLayout->addWidget(myCurStepLbl, 0, 1);
 176
 177   // Steps slider
 178   QWidget* aSliderWidget = new QWidget(aStepFrame);
 179   aStepLayout->addWidget(aSliderWidget, 1, 0, 1, 2);
 180   QHBoxLayout* aSliderLayout = new QHBoxLayout(aSliderWidget);
 181   aSliderLayout->setContentsMargins(0, 0, 0, 0);
 182
 183   aSliderLayout->addWidget(new QLabel("1", aSliderWidget));
 184
 185   myStepSlider = new QSlider(Qt::Horizontal, aSliderWidget);
 186   myStepSlider->setTickPosition(QSlider::TicksBelow);
 187   myStepSlider->setRange(1, 1);
 188   myStepSlider->setPageStep(myStepSlider->singleStep());
 189   aSliderLayout->addWidget(myStepSlider, 1);
 190
 191   myMaxLbl = new QLabel("1", aSliderWidget);
 192   aSliderLayout->addWidget(myMaxLbl);
 193
 194   // Stamp value
 195   myCompNamesList << "Comp 1";
 196   myStepWgt = new QStackedWidget(aStepFrame);
 197   aStepLayout->addWidget(myStepWgt, 2, 0, 1, 2);
 198
 199   myDelegate =
 200     new DataTableItemDelegate((ModelAPI_AttributeTables::ValueType)
 201     myFieldTypeCombo->currentIndex());
 202
 203   appendStepControls();
 204
 205   // Buttons below
 206   QWidget* aBtnWgt = new QWidget(this);
 207   aMainLayout->addWidget(aBtnWgt);
 208   QHBoxLayout* aBtnLayout = new QHBoxLayout(aBtnWgt);
 209   aBtnLayout->setContentsMargins(0, 0, 0, 0);
 210
 211   QPushButton* aAddBtn = new QPushButton(tr("Add step"), aBtnWgt);
 212   aBtnLayout->addWidget(aAddBtn);
 213
 214   aBtnLayout->addStretch(1);
 215
 216   myRemoveBtn = new QPushButton(tr("Remove step"), aBtnWgt);
 217   aBtnLayout->addWidget(myRemoveBtn);
 218   myRemoveBtn->setEnabled(false);
 219
 220   connect(myNbComponentsSpn, SIGNAL(valueChanged(int)), SLOT(onNbCompChanged(int)));
 221   connect(aAddBtn, SIGNAL(clicked(bool)), SLOT(onAddStep()));
 222   connect(myRemoveBtn, SIGNAL(clicked(bool)), SLOT(onRemoveStep()));
 223   connect(myStepSlider, SIGNAL(valueChanged(int)), SLOT(onStepMove(int)));
 224   connect(myStepSlider, SIGNAL(rangeChanged(int, int)), SLOT(onRangeChanged(int, int)));
 225   connect(myFieldTypeCombo, SIGNAL(currentIndexChanged(int)), SLOT(onFieldTypeChanged(int)));
 226   connect(myShapeTypeCombo, SIGNAL(currentIndexChanged(int)), SLOT(onShapeTypeChanged(int)));
 227   connect(qApp, SIGNAL(focusChanged(QWidget*, QWidget*)), SLOT(onFocusChanged(QWidget*, QWidget*)));
 228 }
 229
 230 //**********************************************************************************
 231 void CollectionPlugin_WidgetField::appendStepControls()
 232 {
 233   QWidget* aWidget = new QWidget(myStepWgt);
 234   QGridLayout* aStepLayout = new QGridLayout(aWidget);
 235   aStepLayout->setContentsMargins(0, 0, 0, 0);
 236
 237   aStepLayout->addWidget(new QLabel(tr("Stamp"), aWidget), 0, 0);
 238
 239   QSpinBox* aStampSpn = new QSpinBox(aWidget);
 240   aStepLayout->addWidget(aStampSpn, 0, 1);
 241
 242   myStampSpnList.append(aStampSpn);
 243
 244   // Data table
 245   QTableWidget* aDataTbl = new QTableWidget(1, myCompNamesList.count() + 1, aWidget);
 246   aDataTbl->installEventFilter(this);
 247   aDataTbl->setItemDelegate(myDelegate);
 248
 249   QIntList aColWidth;
 250   if (!myDataTblList.isEmpty()) {
 251     QTableWidget* aFirstTable = myDataTblList.first();
 252     for (int i = 0; i < aFirstTable->columnCount(); i++)
 253       aColWidth.append(aFirstTable->columnWidth(i));
 254   }
 255   myDataTblList.append(aDataTbl);
 256
 257   aDataTbl->verticalHeader()->hide();
 258   aDataTbl->setRowHeight(0, 25);
 259   aDataTbl->setHorizontalScrollMode(QAbstractItemView::ScrollPerPixel);
 260
 261   connect(aDataTbl->horizontalHeader(), SIGNAL(sectionResized(int, int, int)),
 262           SLOT(onColumnResize(int, int, int)));
 263
 264   updateHeaders(aDataTbl);
 265
 266   QTableWidgetItem* aItem = new QTableWidgetItem("Default value");
 267   aItem->setBackgroundColor(Qt::lightGray);
 268   aItem->setFlags(Qt::NoItemFlags | Qt::ItemIsEnabled);
 269   aDataTbl->setItem(0, 0, aItem);
 270
 271   // Set default value item
 272   for (int i = 0; i < myCompNamesList.count(); i++) {
 273     aItem = createDefaultItem();
 274     aItem->setBackgroundColor(Qt::lightGray);
 275     aDataTbl->setItem(0, i + 1, aItem);
 276   }
 277
 278   if (aColWidth.length() > 0) {
 279     for (int i = 0; i < aDataTbl->columnCount(); i++) {
 280       if (i < aColWidth.size())
 281       aDataTbl->setColumnWidth(i, aColWidth.at(i));
 282     }
 283   }
 284   aStepLayout->addWidget(aDataTbl, 1, 0, 1, 2);
 285   connect(aDataTbl, SIGNAL(cellChanged(int, int)), SLOT(onTableEdited(int, int)));
 286
 287   myStepWgt->addWidget(aWidget);
 288   aDataTbl->horizontalHeader()->viewport()->installEventFilter(this);
 289 }
 290
 291 //**********************************************************************************
 292 void CollectionPlugin_WidgetField::deactivate()
 293 {
 294   ModuleBase_WidgetSelector::deactivate();
 295   storeValueCustom();
 296 }
 297
 298 //**********************************************************************************
 299 //void CollectionPlugin_WidgetField::showEvent(QShowEvent* theEvent)
 300 //{
 301 //  myShapeTypeCombo->setEnabled(!isEditingMode());
 302 //  myFieldTypeCombo->setEnabled(!isEditingMode());
 303 //  myNbComponentsSpn->setEnabled(!isEditingMode());
 304 //}
 305
 306 //**********************************************************************************
 307 bool CollectionPlugin_WidgetField::eventFilter(QObject* theObject, QEvent* theEvent)
 308 {
 309   QObject* aObject = 0;
 310   foreach(QTableWidget* aTable, myDataTblList) {
 311     if (aTable->horizontalHeader()->viewport() == theObject) {
 312       aObject = theObject;
 313       break;
 314     }
 315   }
 316   if (aObject) {
 317     if (theEvent->type() == QEvent::MouseButtonDblClick) {
 318       if (myHeaderEditor) { //delete previous editor
 319         myHeaderEditor->deleteLater();
 320         myHeaderEditor = 0;
 321       }
 322       QMouseEvent* aMouseEvent = static_cast<QMouseEvent*>(theEvent);
 323       QHeaderView* aHeader = static_cast<QHeaderView*>(aObject->parent());
 324       QTableWidget* aTable = static_cast<QTableWidget*>(aHeader->parentWidget());
 325
 326       int aShift = aTable->horizontalScrollBar()->value();
 327       int aPos = aMouseEvent->x();
 328       int aIndex = aHeader->logicalIndex(aHeader->visualIndexAt(aPos));
 329       if (aIndex > 0) {
 330         QRect aRect;
 331         aRect.setLeft(aHeader->sectionPosition(aIndex));
 332         aRect.setWidth(aHeader->sectionSize(aIndex));
 333         aRect.setTop(0);
 334         aRect.setHeight(aHeader->height());
 335         aRect.adjust(1, 1, -1, -1);
 336         aRect.translate(-aShift, 0);
 337
 338         myHeaderEditor = new QLineEdit(aHeader->viewport());
 339         myHeaderEditor->move(aRect.topLeft());
 340         myHeaderEditor->resize(aRect.size());
 341         myHeaderEditor->setFrame(false);
 342         QString aText = aHeader->model()->
 343           headerData(aIndex, aHeader->orientation()).toString();
 344         myHeaderEditor->setText(aText);
 345         myHeaderEditor->setFocus();
 346         myEditIndex = aIndex; //save for future use
 347         myHeaderEditor->installEventFilter(this); //catch focus out event
 348         //if user presses Enter it should close editor
 349         connect(myHeaderEditor, SIGNAL(returnPressed()), aTable, SLOT(setFocus()));
 350         myHeaderEditor->show();
 351         return true;
 352       }
 353     }
 354   } else if ((theObject == myHeaderEditor) && (theEvent->type() == QEvent::FocusOut)) {
 355     QString aNewTitle = myHeaderEditor->text();
 356     //save item text
 357     myCompNamesList.replace(myEditIndex - 1, aNewTitle);
 358     myHeaderEditor->deleteLater(); //safely delete editor
 359     myHeaderEditor = 0;
 360     // Store into data model
 361     AttributeStringArrayPtr aStringsAttr =
 362       myFeature->data()->stringArray(CollectionPlugin_Field::COMPONENTS_NAMES_ID());
 363     aStringsAttr->setValue(myEditIndex - 1, aNewTitle.toStdString());
 364     foreach(QTableWidget* aTable, myDataTblList) {
 365       updateHeaders(aTable);
 366     }
 367   } else if (theEvent->type() == QEvent::FocusIn) {
 368     QTableWidget* aTable = dynamic_cast<QTableWidget*>(theObject);
 369     if (aTable) {
 370       ModuleBase_IPropertyPanel* aPanel = myWorkshop->propertyPanel();
 371       if (aPanel->activeWidget() != this) {
 372         myActivation = true;
 373         aPanel->activateWidget(this, false);
 374       }
 375     }
 376   }
 377   return ModuleBase_WidgetSelector::eventFilter(theObject, theEvent);
 378 }
 379
 380 //**********************************************************************************
 381 QTableWidgetItem* CollectionPlugin_WidgetField::createDefaultItem() const
 382 {
 383   QTableWidgetItem* aItem = new QTableWidgetItem();
 384   switch (myFieldTypeCombo->currentIndex()) {
 385   case ModelAPI_AttributeTables::DOUBLE:
 386   case ModelAPI_AttributeTables::INTEGER:
 387     aItem->setText("0");
 388     break;
 389   case ModelAPI_AttributeTables::BOOLEAN:
 390     aItem->setText(MYFalse);
 391     break;
 392   case ModelAPI_AttributeTables::STRING:
 393     aItem->setText("");
 394     break;
 395   }
 396   return aItem;
 397 }
 398
 399 //**********************************************************************************
 400 QTableWidgetItem* CollectionPlugin_WidgetField::
 401   createValueItem(ModelAPI_AttributeTables::Value& theVal) const
 402 {
 403   QTableWidgetItem* aItem = new QTableWidgetItem();
 404   aItem->setText(getValueText(theVal));
 405   return aItem;
 406 }
 407
 408 //**********************************************************************************
 409 QString CollectionPlugin_WidgetField::getValueText(ModelAPI_AttributeTables::Value& theVal) const
 410 {
 411   switch (myFieldTypeCombo->currentIndex()) {
 412   case ModelAPI_AttributeTables::DOUBLE:
 413     return QString::number(theVal.myDouble);
 414   case ModelAPI_AttributeTables::INTEGER:
 415     return QString::number(theVal.myInt);
 416   case ModelAPI_AttributeTables::BOOLEAN:
 417     return theVal.myBool? MYTrue : MYFalse;
 418   case ModelAPI_AttributeTables::STRING:
 419     return theVal.myStr.c_str();
 420   }
 421   return "";
 422 }
 423
 424
 425 //**********************************************************************************
 426 void CollectionPlugin_WidgetField::updateHeaders(QTableWidget* theDataTbl) const
 427 {
 428   QStringList aHeaders;
 429   aHeaders << tr(MYFirstCol);
 430   aHeaders << myCompNamesList;
 431   theDataTbl->setHorizontalHeaderLabels(aHeaders);
 432 }
 433
 434 //**********************************************************************************
 435 void CollectionPlugin_WidgetField::removeStepControls()
 436 {
 437   int aCurWgtId = myStepWgt->currentIndex();
 438   QWidget* aWgt = myStepWgt->currentWidget();
 439   myStepWgt->removeWidget(aWgt);
 440
 441   myStampSpnList.removeAt(aCurWgtId);
 442   myDataTblList.removeAt(aCurWgtId);
 443   aWgt->deleteLater();
 444 }
 445
 446 //**********************************************************************************
 447 QList<QWidget*> CollectionPlugin_WidgetField::getControls() const
 448 {
 449   QList<QWidget*> aControls;
 450   // this control will accept focus and will be highlighted in the Property Panel
 451   aControls.append(myShapeTypeCombo);
 452   aControls.append(myFieldTypeCombo);
 453   aControls.append(myNbComponentsSpn);
 454   if (myStampSpnList.size() > 0)
 455     aControls.append(myStampSpnList.first());
 456   if (myDataTblList.size() > 0)
 457     aControls.append(myDataTblList.first());
 458
 459   return aControls;
 460 }
 461
 462 //**********************************************************************************
 463 bool CollectionPlugin_WidgetField::storeValueCustom()
 464 {
 465   DataPtr aData = myFeature->data();
 466   // Store number of components
 467   AttributeStringArrayPtr aStringsAttr =
 468     aData->stringArray(CollectionPlugin_Field::COMPONENTS_NAMES_ID());
 469   int aNbComps = myCompNamesList.size();
 470   aStringsAttr->setSize(aNbComps);
 471   for ( int i = 0; i < aNbComps; i++)
 472     aStringsAttr->setValue(i, myCompNamesList.at(i).toStdString());
 473
 474   AttributeTablesPtr aTablesAttr = aData->tables(CollectionPlugin_Field::VALUES_ID());
 475   // Store number of steps
 476   int aNbSteps =  myDataTblList.size();
 477
 478   // Store Type of the field values
 479   int aFldType = myFieldTypeCombo->currentIndex();
 480
 481   AttributeIntArrayPtr aStampsAttr = aData->intArray(CollectionPlugin_Field::STAMPS_ID());
 482   aStampsAttr->setSize(aNbSteps);
 483   // Store data
 484   QTableWidget* aTable = myDataTblList.first();
 485   int aRows = aTable->rowCount();
 486   // first column contains selected names which should not be stored
 487   int aColumns = aTable->columnCount() - 1;
 488
 489   aTablesAttr->setSize(aRows, aColumns, aNbSteps);
 490   aTablesAttr->setType((ModelAPI_AttributeTables::ValueType)aFldType);
 491   for (int i = 0; i < aNbSteps; i++) {
 492     aStampsAttr->setValue(i, myStampSpnList.at(i)->value());
 493     aTable = myDataTblList.at(i);
 494     for (int j = 0; j < aColumns; j++) {
 495       for (int k = 0; k < aRows; k++) {
 496         QString aTblVal = aTable->item(k, j + 1)->text();
 497         aTablesAttr->setValue(getValue(aTblVal), k, j, i);
 498       }
 499     }
 500   }
 501   updateObject(myFeature);
 502   return true;
 503 }
 504
 505 //**********************************************************************************
 506 bool CollectionPlugin_WidgetField::restoreValueCustom()
 507 {
 508   bool isBlocked;
 509   DataPtr aData = myFeature->data();
 510
 511   AttributeSelectionListPtr aSelList = aData->selectionList(CollectionPlugin_Field::SELECTED_ID());
 512   std::string aTypeStr = aSelList->selectionType();
 513   if (aTypeStr == "")
 514     return false; // The attribute is not initialized
 515   isBlocked = myShapeTypeCombo->blockSignals(true);
 516   myShapeTypeCombo->setCurrentIndex(getSelectionType(aTypeStr));
 517   myShapeTypeCombo->blockSignals(isBlocked);
 518
 519   // Get number of components
 520   AttributeStringArrayPtr aStringsAttr =
 521   aData->stringArray(CollectionPlugin_Field::COMPONENTS_NAMES_ID());
 522
 523   myCompNamesList.clear();
 524   for (int i = 0; i < aStringsAttr->size(); i++) {
 525     myCompNamesList.append(aStringsAttr->value(i).c_str());
 526   }
 527   isBlocked = myNbComponentsSpn->blockSignals(true);
 528   myNbComponentsSpn->setValue(myCompNamesList.size());
 529   myNbComponentsSpn->blockSignals(isBlocked);
 530
 531   AttributeTablesPtr aTablesAttr = aData->tables(CollectionPlugin_Field::VALUES_ID());
 532   // Get number of steps
 533   int aNbSteps = aTablesAttr->tables();
 534   myStepSlider->setMaximum(aNbSteps);
 535   //myStepSlider->setValue(1);
 536   // Clear old tables
 537   while (myDataTblList.count() > aNbSteps) {
 538     QWidget* aWgt = myStepWgt->widget(myStepWgt->count() - 1);
 539     myStepWgt->removeWidget(aWgt);
 540     aWgt->deleteLater();
 541
 542     myStampSpnList.removeLast();
 543     myDataTblList.removeLast();
 544   }
 545   while (myDataTblList.count() < aNbSteps)
 546     appendStepControls();
 547   //myStepWgt->setCurrentIndex(myStepSlider->value() - 1);
 548   clearData();
 549
 550   // Get Type of the field values
 551   isBlocked = myFieldTypeCombo->blockSignals(true);
 552   myFieldTypeCombo->setCurrentIndex(aTablesAttr->type());
 553   myFieldTypeCombo->blockSignals(isBlocked);
 554   myDelegate->setDataType(aTablesAttr->type());
 555
 556   AttributeIntArrayPtr aStampsAttr = aData->intArray(CollectionPlugin_Field::STAMPS_ID());
 557   // Fill data table
 558   int aRows = aTablesAttr->rows();
 559   int aCols = aTablesAttr->columns();
 560
 561   // Get width of columns
 562   QIntList aColWidth;
 563   QTableWidget* aFirstTable = myDataTblList.first();
 564   for (int i = 0; i < aFirstTable->columnCount(); i++)
 565     aColWidth.append(aFirstTable->columnWidth(i));
 566
 567   QTableWidgetItem* aItem = 0;
 568   for (int i = 0; i < aNbSteps; i++) {
 569     myStampSpnList.at(i)->setValue(aStampsAttr->value(i));
 570     QTableWidget* aTable = myDataTblList.at(i);
 571     isBlocked = aTable->blockSignals(true);
 572     aTable->setRowCount(aRows);
 573     aTable->setColumnCount(aCols + 1);
 574     updateHeaders(aTable);
 575     for (int j = 0; j < aCols + 1; j++) {
 576       for (int k = 0; k < aRows; k++) {
 577         aItem = aTable->item(k, j);
 578         if ((j == 0) && (k > 0)) {
 579           // Add selection names
 580           AttributeSelectionPtr aAttr = aSelList->value(k - 1);
 581           if (aItem) {
 582             aItem->setText(aAttr->namingName().c_str());
 583           } else {
 584             aItem = new QTableWidgetItem(aAttr->namingName().c_str());
 585             aTable->setItem(k, j, aItem);
 586           }
 587         } else if (j > 0) {
 588           // Add Values
 589           ModelAPI_AttributeTables::Value aVal = aTablesAttr->value(k, j - 1, i);
 590           if (aItem) {
 591             aItem->setText(getValueText(aVal));
 592           } else {
 593             aItem = createValueItem(aVal);
 594             if (k == 0)
 595               aItem->setBackgroundColor(Qt::lightGray);
 596             aTable->setItem(k, j, aItem);
 597           }
 598         }
 599       }
 600     }
 601     // Restore columns width
 602     for (int i = 0; i < aTable->columnCount(); i++) {
 603       if (i < aColWidth.size())
 604         aTable->setColumnWidth(i, aColWidth.at(i));
 605     }
 606
 607     aTable->blockSignals(isBlocked);
 608   }
 609   return true;
 610 }
 611
 612 //**********************************************************************************
 613 int CollectionPlugin_WidgetField::getSelectionType(const std::string& theStr) const
 614 {
 615   QString aType(theStr.c_str());
 616   aType = aType.toLower();
 617   if (aType == "vertex")
 618     return 0;
 619   else if (aType == "edge")
 620     return 1;
 621   else if (aType == "face")
 622     return 2;
 623   else if (aType == "solid")
 624     return 3;
 625   else if (aType == "object")
 626     return 4;
 627   else if (aType == "part")
 628     return 5;
 629   return -1;
 630 }
 631
 632
 633 //**********************************************************************************
 634 std::string CollectionPlugin_WidgetField::getSelectionType(int theType) const
 635 {
 636   switch (theType) {
 637   case 0: //"Vertices"
 638     return "vertex";
 639   case 1: // "Edges"
 640     return "edge";
 641   case 2: // "Faces"
 642     return "face";
 643   case 3: // "Solids"
 644     return "solid";
 645   case 4: // "Results"
 646     return "object";
 647   case 5: // "Parts"
 648     return "part";
 649   }
 650   return "";
 651 }
 652
 653 //**********************************************************************************
 654 QIntList CollectionPlugin_WidgetField::shapeTypes() const
 655 {
 656   QIntList aRes;
 657   switch (myShapeTypeCombo->currentIndex()) {
 658   case 0: //"Vertices"
 659     aRes.append(ModuleBase_Tools::shapeType("vertex"));
 660     break;
 661   case 1: // "Edges"
 662     aRes.append(ModuleBase_Tools::shapeType("edge"));
 663     break;
 664   case 2: // "Faces"
 665     aRes.append(ModuleBase_Tools::shapeType("face"));
 666     break;
 667   case 3: // "Solids"
 668     aRes.append(ModuleBase_Tools::shapeType("solid"));
 669     break;
 670   case 4: // "Results"
 671     aRes.append(ModuleBase_Tools::shapeType("object"));
 672     break;
 673   case 5: // "Parts"
 674     // TODO: Selection mode for Parts
 675     break;
 676   }
 677   return aRes;
 678 }
 679
 680 //**********************************************************************************
 681 ModelAPI_AttributeTables::Value CollectionPlugin_WidgetField::getValue(QString theStrVal) const
 682 {
 683   ModelAPI_AttributeTables::Value aVal;
 684   switch (myFieldTypeCombo->currentIndex()) {
 685   case ModelAPI_AttributeTables::BOOLEAN:
 686     aVal.myBool = (theStrVal == MYTrue)? true : false;
 687     break;
 688   case ModelAPI_AttributeTables::DOUBLE:
 689     aVal.myDouble = theStrVal.toDouble();
 690     break;
 691   case ModelAPI_AttributeTables::INTEGER:
 692     aVal.myInt = theStrVal.toInt();
 693     break;
 694   case ModelAPI_AttributeTables::STRING:
 695     aVal.myStr = theStrVal.toStdString();
 696   }
 697   return aVal;
 698 }
 699
 700
 701 //**********************************************************************************
 702 void CollectionPlugin_WidgetField::onNbCompChanged(int theVal)
 703 {
 704   int aOldCol = myCompNamesList.count();
 705   int aNbRows = myDataTblList.first()->rowCount();
 706   int aDif = theVal - aOldCol;
 707   QTableWidgetItem* aItem = 0;
 708
 709   while (myCompNamesList.count() != theVal) {
 710     if (aDif > 0)
 711       myCompNamesList.append(QString("Comp %1").arg(myCompNamesList.count() + 1));
 712     else
 713       myCompNamesList.removeLast();
 714   }
 715
 716   AttributeTablesPtr aTablesAttr = myFeature->data()->tables(CollectionPlugin_Field::VALUES_ID());
 717   aTablesAttr->setSize(aNbRows, myCompNamesList.size(), myDataTblList.size());
 718
 719   foreach(QTableWidget* aDataTbl, myDataTblList) {
 720     aDataTbl->setColumnCount(theVal + 1);
 721     updateHeaders(aDataTbl);
 722     for (int i = aOldCol; i < myCompNamesList.count(); i++) {
 723       for (int j = 0; j < aNbRows; j++) {
 724         aItem = aDataTbl->item(j, i + 1);
 725         if (!aItem) {
 726           aItem = createDefaultItem();
 727           if (j == 0)
 728             aItem->setBackgroundColor(Qt::lightGray);
 729           aDataTbl->setItem(j, i + 1, aItem);
 730         }
 731       }
 732     }
 733   }
 734   emit valuesChanged();
 735 }
 736
 737 //**********************************************************************************
 738 void CollectionPlugin_WidgetField::onAddStep()
 739 {
 740   int aMax = myStepSlider->maximum();
 741   aMax++;
 742   myStepSlider->setMaximum(aMax);
 743   myMaxLbl->setText(QString::number(aMax));
 744   appendStepControls();
 745   myStepSlider->setValue(aMax);
 746
 747   AttributeTablesPtr aTablesAttr = myFeature->data()->tables(CollectionPlugin_Field::VALUES_ID());
 748   aTablesAttr->setSize(aTablesAttr->rows(), aTablesAttr->columns(), myDataTblList.size());
 749
 750
 751   AttributeSelectionListPtr aSelList =
 752     myFeature->data()->selectionList(CollectionPlugin_Field::SELECTED_ID());
 753   if (!aSelList->isInitialized())
 754     return;
 755   int aSelNb = aSelList->size();
 756   if (aSelNb == 0)
 757     return;
 758
 759   int aColumns = myNbComponentsSpn->value() + 1;
 760   int aRows = aSelNb + 1;
 761   QTableWidget* aTable = myDataTblList.last();
 762   aTable->setRowCount(aRows);
 763   QTableWidgetItem* aItem = 0;
 764   for(int i = 0; i < aColumns; i++) {
 765     if (i == 0) {
 766       for(int j = 1; j < aRows; j++) {
 767         aItem = aTable->item(j, i);
 768         if (!aItem) {
 769           aItem = new QTableWidgetItem();
 770           aTable->setItem(j, i, aItem);
 771         }
 772         AttributeSelectionPtr aAttr = aSelList->value(j - 1);
 773         aItem->setText(aAttr->namingName().c_str());
 774         aItem->setToolTip(aAttr->namingName().c_str());
 775       }
 776     } else {
 777       QString aDefVal = aTable->item(0, i)->text();
 778       for(int j = 1; j < aRows; j++) {
 779         aItem = aTable->item(j, i);
 780         if (!aItem) {
 781           aItem = new QTableWidgetItem();
 782           aTable->setItem(j, i, aItem);
 783         }
 784         aItem->setText(aDefVal);
 785       }
 786     }
 787   }
 788   emit valuesChanged();
 789 }
 790
 791 //**********************************************************************************
 792 void CollectionPlugin_WidgetField::onRemoveStep()
 793 {
 794   int aMax = myStepSlider->maximum();
 795   aMax--;
 796   myMaxLbl->setText(QString::number(aMax));
 797   removeStepControls();
 798   myStepSlider->setMaximum(aMax);
 799
 800   //AttributeTablesPtr aTablesAttr = myFeature->data()->tables(CollectionPlugin_Field::VALUES_ID());
 801   //aTablesAttr->setSize(aTablesAttr->rows(), aTablesAttr->columns(), myDataTblList.size());
 802   emit valuesChanged();
 803 }
 804
 805 //**********************************************************************************
 806 void CollectionPlugin_WidgetField::clearData()
 807 {
 808   foreach(QTableWidget* aDataTbl, myDataTblList) {
 809     aDataTbl->setRowCount(1);
 810   }
 811 }
 812
 813 //**********************************************************************************
 814 void CollectionPlugin_WidgetField::onStepMove(int theStep)
 815 {
 816   myCurStepLbl->setText(QString::number(theStep));
 817   myStepWgt->setCurrentIndex(theStep - 1);
 818 }
 819
 820 //**********************************************************************************
 821 bool CollectionPlugin_WidgetField::
 822   isValidSelectionCustom(const std::shared_ptr<ModuleBase_ViewerPrs>& thePrs)
 823 {
 824   return (myShapeTypeCombo->currentIndex() == 5)? false : true;
 825 }
 826
 827 //**********************************************************************************
 828 bool CollectionPlugin_WidgetField::
 829   setSelection(QList<std::shared_ptr<ModuleBase_ViewerPrs>>& theValues, const bool theToValidate)
 830 {
 831   if (myActivation) {
 832     myActivation = false;
 833     return false;
 834   }
 835   // Ignore selection for Parts mode
 836   if (myShapeTypeCombo->currentIndex() == 5)
 837     return false;
 838
 839   QList<ModuleBase_ViewerPrsPtr> aSelected;
 840   QList<ModuleBase_ViewerPrsPtr>::const_iterator anIt;
 841   for (anIt = theValues.begin(); anIt != theValues.end(); anIt++) {
 842     ModuleBase_ViewerPrsPtr aValue = *anIt;
 843     ResultPtr aRes = std::dynamic_pointer_cast<ModelAPI_Result>(aValue->object());
 844     if (theToValidate && aRes.get()) {
 845       if (myShapeTypeCombo->currentIndex() > 3)
 846         aSelected.append(aValue);
 847       else if (acceptSubShape(aValue->shape(), aRes))
 848         aSelected.append(aValue);
 849     }
 850   }
 851   AttributeSelectionListPtr aSelList =
 852     myFeature->data()->selectionList(CollectionPlugin_Field::SELECTED_ID());
 853   aSelList->setSelectionType(getSelectionType(myShapeTypeCombo->currentIndex()));
 854   aSelList->clear();
 855
 856   ResultPtr aResult;
 857   GeomShapePtr aShape;
 858   int aNbData = 0;
 859   foreach(ModuleBase_ViewerPrsPtr aPrs, aSelected) {
 860     aResult = std::dynamic_pointer_cast<ModelAPI_Result>(aPrs->object());
 861     aShape = aPrs->shape();
 862     if (!aResult.get() && !aShape.get())
 863       continue;
 864     if (!aSelList->isInList(aResult, aShape)) {
 865       aSelList->append(aResult, aShape);
 866       aNbData++;
 867     }
 868   }
 869   int aColumns = myDataTblList.first()->columnCount();
 870   int aRows = myDataTblList.first()->rowCount();
 871   int aNewRows = aNbData + 1;
 872   AttributeTablesPtr aTablesAttr = myFeature->data()->tables(CollectionPlugin_Field::VALUES_ID());
 873   aTablesAttr->setSize(aNewRows, aColumns - 1, myDataTblList.size());
 874
 875   QTableWidgetItem* aItem = 0;
 876   foreach(QTableWidget* aTable, myDataTblList) {
 877     aTable->setRowCount(aNewRows);
 878     if (aNewRows > aRows) {
 879       // Add new data
 880       for(int i = 0; i < aColumns; i++) {
 881         if (i == 0) {
 882           for(int j = 1; j < aNewRows; j++) {
 883             aItem = aTable->item(j, i);
 884             if (!aItem) {
 885               aItem = new QTableWidgetItem();
 886               aTable->setItem(j, i, aItem);
 887             }
 888             AttributeSelectionPtr aAttr = aSelList->value(j - 1);
 889             aItem->setText(aAttr->namingName().c_str());
 890             aItem->setToolTip(aAttr->namingName().c_str());
 891           }
 892         } else {
 893           QString aDefVal = aTable->item(0, i)->text();
 894           for(int j = aRows; j < aNewRows; j++) {
 895             aItem = aTable->item(j, i);
 896             if (!aItem) {
 897               aItem = new QTableWidgetItem();
 898               aTable->setItem(j, i, aItem);
 899             }
 900             aItem->setText(aDefVal);
 901           }
 902         }
 903       }
 904     } else {
 905       // Update only selection name
 906       for(int j = 1; j < aNewRows - 1; j++) {
 907         AttributeSelectionPtr aAttr = aSelList->value(j);
 908         aTable->item(j, 0)->setText(aAttr->namingName().c_str());
 909         aTable->item(j, 0)->setToolTip(aAttr->namingName().c_str());
 910       }
 911     }
 912   }
 913   return true;
 914 }
 915
 916 //**********************************************************************************
 917 void CollectionPlugin_WidgetField::onFieldTypeChanged(int theIdx)
 918 {
 919   ModelAPI_AttributeTables::ValueType aOldType = myDelegate->dataType();
 920   if (aOldType != theIdx) {
 921     myDelegate->setDataType((ModelAPI_AttributeTables::ValueType)theIdx);
 922     int aColumns = myDataTblList.first()->columnCount();
 923     int aRows = myDataTblList.first()->rowCount();
 924     foreach(QTableWidget* aTable, myDataTblList) {
 925       for(int i = 1; i < aColumns; i++) {
 926         for(int j = 0; j < aRows; j++) {
 927           switch (theIdx) {
 928           case ModelAPI_AttributeTables::DOUBLE:
 929           case ModelAPI_AttributeTables::INTEGER:
 930             aTable->item(j, i)->setText("0");
 931             break;
 932           case ModelAPI_AttributeTables::BOOLEAN:
 933             aTable->item(j, i)->setText(MYFalse);
 934             break;
 935           case ModelAPI_AttributeTables::STRING:
 936             aTable->item(j, i)->setText("");
 937             break;
 938           }
 939         }
 940       }
 941     }
 942     emit valuesChanged();
 943   }
 944 }
 945
 946 //**********************************************************************************
 947 void CollectionPlugin_WidgetField::onTableEdited(int theRow, int theCol)
 948 {
 949   // Do not store here column of names
 950   if (theCol == 0)
 951     return;
 952   if (!myFeature.get())
 953     return;
 954   QTableWidget* aTable = static_cast<QTableWidget*>(sender());
 955   int aNb = myDataTblList.indexOf(aTable);
 956   if (aNb == -1)
 957     return;
 958   ModelAPI_AttributeTables::Value aVal = getValue(aTable->item(theRow, theCol)->text());
 959
 960   AttributeTablesPtr aTablesAttr = myFeature->data()->tables(CollectionPlugin_Field::VALUES_ID());
 961   if (aTablesAttr->isInitialized())
 962     aTablesAttr->setValue(aVal,theRow, theCol - 1, aNb);
 963   else
 964     emit valuesChanged();
 965 }
 966
 967 //**********************************************************************************
 968 void CollectionPlugin_WidgetField::onShapeTypeChanged(int theType)
 969 {
 970   activateSelectionAndFilters(theType == 5? false:true);
 971
 972   AttributeSelectionListPtr aSelList =
 973     myFeature->data()->selectionList(CollectionPlugin_Field::SELECTED_ID());
 974
 975   std::string aTypeName = getSelectionType(theType);
 976   if (aTypeName == aSelList->selectionType())
 977     return;
 978   aSelList->setSelectionType(aTypeName);
 979
 980   //Clear old selection
 981   clearData();
 982   aSelList->clear();
 983   AttributeTablesPtr aTablesAttr = myFeature->data()->tables(CollectionPlugin_Field::VALUES_ID());
 984   aTablesAttr->setSize(1, myNbComponentsSpn->value(), myDataTblList.size());
 985   emit valuesChanged();
 986 }
 987
 988 //**********************************************************************************
 989 bool CollectionPlugin_WidgetField::processEnter()
 990 {
 991   if (myIsTabEdit) {
 992     myIsTabEdit = false;
 993     return true;
 994   }
 995   QWidget* aCurrWgt = qApp->focusWidget();
 996   int aCurWgtId = myStepWgt->currentIndex();
 997   if ((aCurrWgt == myShapeTypeCombo) ||
 998       (aCurrWgt == myFieldTypeCombo) ||
 999       (aCurrWgt == myNbComponentsSpn) ||
1000       (aCurrWgt == myStampSpnList[aCurWgtId]) ||
1001       (aCurrWgt == myDataTblList[aCurWgtId])) {
1002     setFocus();
1003     return true;
1004   }
1005   return false;
1006 }
1007
1008 //**********************************************************************************
1009 void CollectionPlugin_WidgetField::onFocusChanged(QWidget* theOld, QWidget* theNew)
1010 {
1011   if (theNew && (!myIsTabEdit))
1012     myIsTabEdit = dynamic_cast<QLineEdit*>(theNew);
1013 }
1014
1015 //**********************************************************************************
1016 void CollectionPlugin_WidgetField::onRangeChanged(int theMin, int theMax)
1017 {
1018   myMaxLbl->setText(QString::number(theMax));
1019   myRemoveBtn->setEnabled(theMax > 1);
1020 }
1021
1022 //**********************************************************************************
1023 void CollectionPlugin_WidgetField::onColumnResize(int theIndex, int theOld, int theNew)
1024 {
1025   if (myDataTblList.count() < 2)
1026     return;
1027   QObject* aSender = sender();
1028   foreach(QTableWidget* aTable, myDataTblList) {
1029     if (aTable->horizontalHeader() != aSender)
1030       aTable->setColumnWidth(theIndex, theNew);
1031   }
1032 }
1033
1034 //**********************************************************************************
1035 QList<std::shared_ptr<ModuleBase_ViewerPrs>>
1036   CollectionPlugin_WidgetField::getAttributeSelection() const
1037 {
1038   QList<std::shared_ptr<ModuleBase_ViewerPrs>> aList;
1039   if(myFeature) {
1040     DataPtr aData = myFeature->data();
1041     AttributeSelectionListPtr aSelList =
1042       aData->selectionList(CollectionPlugin_Field::SELECTED_ID());
1043     AttributeSelectionPtr aAttr;
1044     ObjectPtr anObject;
1045     for (int i = 0; i < aSelList->size(); i++) {
1046       aAttr = aSelList->value(i);
1047       ModuleBase_ViewerPrsPtr
1048         aPrs(new ModuleBase_ViewerPrs(aAttr->context(), aAttr->value(), NULL));
1049       aList.append(aPrs);
1050     }
1051   }
1052   return aList;
1053 }