X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FCollectionPlugin%2FCollectionPlugin_WidgetField.cpp;h=52a54c022b656b3cf0bc8fd46d414673b00b3121;hb=97917d3698f5a2f7fc9596e7c755ff8f6751e373;hp=d0500355ed05d30e5d20ad3dd500f935f787f1ff;hpb=266f5a5c2132719049c303265adee362d0e702b4;p=modules%2Fshaper.git diff --git a/src/CollectionPlugin/CollectionPlugin_WidgetField.cpp b/src/CollectionPlugin/CollectionPlugin_WidgetField.cpp index d0500355e..52a54c022 100644 --- a/src/CollectionPlugin/CollectionPlugin_WidgetField.cpp +++ b/src/CollectionPlugin/CollectionPlugin_WidgetField.cpp @@ -1,10 +1,35 @@ -// Copyright (C) 2014-20xx CEA/DEN, EDF R&D - -// File: CollectionPlugin_WidgetField.cpp -// Created: 16 Nov 2016 -// Author: Vitaly SMETANNIKOV +// Copyright (C) 2014-2019 CEA/DEN, EDF R&D +// +// This library is free software; you can redistribute it and/or +// modify it under the terms of the GNU Lesser General Public +// License as published by the Free Software Foundation; either +// version 2.1 of the License, or (at your option) any later version. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// Lesser General Public License for more details. +// +// You should have received a copy of the GNU Lesser General Public +// License along with this library; if not, write to the Free Software +// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA +// +// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com +// #include "CollectionPlugin_WidgetField.h" +#include "CollectionPlugin_Field.h" + +#include +#include +#include +#include +#include + +#include +#include +#include +#include #include #include @@ -16,10 +41,93 @@ #include #include #include +#include +#include +#include +#include +#include +#include +#include + +const char* MYFirstCol = "Shape"; +const char* MYTrue = "True"; +const char* MYFalse = "False"; + +DataTableItemDelegate::DataTableItemDelegate(ModelAPI_AttributeTables::ValueType theType) + : QStyledItemDelegate(), myType(theType) +{ +} + + +QWidget* DataTableItemDelegate::createEditor(QWidget* theParent, + const QStyleOptionViewItem & theOption, + const QModelIndex& theIndex ) const +{ + QWidget* aEditor = 0; + if ((theIndex.column() == 0) && (theIndex.row() > 0)) { + QWidget* aWgt = QStyledItemDelegate::createEditor(theParent, theOption, theIndex); + QLineEdit* aEdt = static_cast(aWgt); + aEdt->setReadOnly(true); + aEditor = aEdt; + } else { + QLineEdit* aLineEdt = 0; + switch (myType) { + case ModelAPI_AttributeTables::DOUBLE: + aLineEdt = dynamic_cast(QStyledItemDelegate::createEditor(theParent, + theOption, + theIndex)); + if (aLineEdt) { + aLineEdt->setValidator(new QDoubleValidator(aLineEdt)); + aEditor = aLineEdt; + } + break; + case ModelAPI_AttributeTables::INTEGER: + aLineEdt = dynamic_cast(QStyledItemDelegate::createEditor(theParent, + theOption, + theIndex)); + if (aLineEdt) { + aLineEdt->setValidator(new QIntValidator(aLineEdt)); + aEditor = aLineEdt; + } + break; + case ModelAPI_AttributeTables::BOOLEAN: + { + QComboBox* aBox = new QComboBox(theParent); + aBox->addItem(MYFalse); + aBox->addItem(MYTrue); + aEditor = aBox; + } + break; + default: + aEditor = QStyledItemDelegate::createEditor(theParent, theOption, theIndex); + } + } + if (myType == ModelAPI_AttributeTables::BOOLEAN) + connect(aEditor, SIGNAL(currentTextChanged(const QString&)), + SLOT(onEditItem(const QString&))); + else + connect(aEditor, SIGNAL(textEdited(const QString&)), + SLOT(onEditItem(const QString&))); + return aEditor; +} +void DataTableItemDelegate::onEditItem(const QString& theText) +{ + QWidget* aWgt = dynamic_cast(sender()); + commitData(aWgt); +} + + + +//********************************************************************************** +//********************************************************************************** +//********************************************************************************** CollectionPlugin_WidgetField:: - CollectionPlugin_WidgetField(QWidget* theParent, const Config_WidgetAPI* theData): -ModuleBase_ModelWidget(theParent, theData) + CollectionPlugin_WidgetField(QWidget* theParent, + ModuleBase_IWorkshop* theWorkshop, + const Config_WidgetAPI* theData): +ModuleBase_WidgetSelector(theParent, theWorkshop, theData), myHeaderEditor(0), + myIsTabEdit(false), myActivation(false) { QVBoxLayout* aMainLayout = new QVBoxLayout(this); @@ -32,17 +140,18 @@ ModuleBase_ModelWidget(theParent, theData) // Type of shapes myShapeTypeCombo = new QComboBox(aTypesWgt); QStringList aShapeTypes; - aShapeTypes << tr("Vertices") << tr("Edges") << tr("Faces") - << tr("Solids") << tr("Results") << tr("Parts"); + aShapeTypes << tr("Vertices") << tr("Edges") << tr("Faces") + << tr("Solids") << tr("Objects") << tr("Parts"); myShapeTypeCombo->addItems(aShapeTypes); aTypesLayout->addRow(tr("Type of shapes"), myShapeTypeCombo); // Type of field myFieldTypeCombo = new QComboBox(aTypesWgt); QStringList aFieldTypes; - aFieldTypes << tr("Double") << tr("Integer") << tr("Boolean") + aFieldTypes << tr("Boolean") << tr("Integer") << tr("Double") << tr("String"); myFieldTypeCombo->addItems(aFieldTypes); + myFieldTypeCombo->setCurrentIndex(2); aTypesLayout->addRow(tr("Type of field"), myFieldTypeCombo); // Number of components @@ -83,36 +192,15 @@ ModuleBase_ModelWidget(theParent, theData) aSliderLayout->addWidget(myMaxLbl); // Stamp value - aStepLayout->addWidget(new QLabel(tr("Stamp"), aStepFrame), 2, 0); - myStampSpn = new QSpinBox(aStepFrame); - aStepLayout->addWidget(myStampSpn, 2, 1); - - // Data table - myDataTbl = new QTableWidget(2, 2, aStepFrame); - myDataTbl->verticalHeader()->hide(); - myDataTbl->horizontalHeader()->hide(); - myDataTbl->setRowHeight(0, 25); - myDataTbl->setRowHeight(1, 25); + myCompNamesList << "Comp 1"; + myStepWgt = new QStackedWidget(aStepFrame); + aStepLayout->addWidget(myStepWgt, 2, 0, 1, 2); - QTableWidgetItem* aItem = new QTableWidgetItem("Shape"); - aItem->setBackgroundColor(Qt::lightGray); - aItem->setFlags(Qt::NoItemFlags | Qt::ItemIsEnabled); - myDataTbl->setItem(0, 0, aItem); + myDelegate = + new DataTableItemDelegate((ModelAPI_AttributeTables::ValueType) + myFieldTypeCombo->currentIndex()); - aItem = new QTableWidgetItem("Comp 1"); - aItem->setBackgroundColor(Qt::lightGray); - myDataTbl->setItem(0, 1, aItem); - - aItem = new QTableWidgetItem("Default value"); - aItem->setBackgroundColor(Qt::lightGray); - aItem->setFlags(Qt::NoItemFlags | Qt::ItemIsEnabled); - myDataTbl->setItem(1, 0, aItem); - - aItem = new QTableWidgetItem("0"); - aItem->setBackgroundColor(Qt::lightGray); - myDataTbl->setItem(1, 1, aItem); - - aStepLayout->addWidget(myDataTbl, 3, 0, 1, 2); + appendStepControls(); // Buttons below QWidget* aBtnWgt = new QWidget(this); @@ -125,100 +213,841 @@ ModuleBase_ModelWidget(theParent, theData) aBtnLayout->addStretch(1); - QPushButton* aRemoveBtn = new QPushButton(tr("Remove step"), aBtnWgt); - aBtnLayout->addWidget(aRemoveBtn); + myRemoveBtn = new QPushButton(tr("Remove step"), aBtnWgt); + aBtnLayout->addWidget(myRemoveBtn); + myRemoveBtn->setEnabled(false); connect(myNbComponentsSpn, SIGNAL(valueChanged(int)), SLOT(onNbCompChanged(int))); connect(aAddBtn, SIGNAL(clicked(bool)), SLOT(onAddStep())); - connect(aRemoveBtn, SIGNAL(clicked(bool)), SLOT(onRemoveStep())); + connect(myRemoveBtn, SIGNAL(clicked(bool)), SLOT(onRemoveStep())); connect(myStepSlider, SIGNAL(valueChanged(int)), SLOT(onStepMove(int))); + connect(myStepSlider, SIGNAL(rangeChanged(int, int)), SLOT(onRangeChanged(int, int))); + connect(myFieldTypeCombo, SIGNAL(currentIndexChanged(int)), SLOT(onFieldTypeChanged(int))); + connect(myShapeTypeCombo, SIGNAL(currentIndexChanged(int)), SLOT(onShapeTypeChanged(int))); + connect(qApp, SIGNAL(focusChanged(QWidget*, QWidget*)), SLOT(onFocusChanged(QWidget*, QWidget*))); } +//********************************************************************************** +void CollectionPlugin_WidgetField::appendStepControls() +{ + QWidget* aWidget = new QWidget(myStepWgt); + QGridLayout* aStepLayout = new QGridLayout(aWidget); + aStepLayout->setContentsMargins(0, 0, 0, 0); + + aStepLayout->addWidget(new QLabel(tr("Stamp"), aWidget), 0, 0); + + QSpinBox* aStampSpn = new QSpinBox(aWidget); + aStepLayout->addWidget(aStampSpn, 0, 1); + + myStampSpnList.append(aStampSpn); + + // Data table + QTableWidget* aDataTbl = new QTableWidget(1, myCompNamesList.count() + 1, aWidget); + aDataTbl->installEventFilter(this); + aDataTbl->setItemDelegate(myDelegate); + + QIntList aColWidth; + if (!myDataTblList.isEmpty()) { + QTableWidget* aFirstTable = myDataTblList.first(); + for (int i = 0; i < aFirstTable->columnCount(); i++) + aColWidth.append(aFirstTable->columnWidth(i)); + } + myDataTblList.append(aDataTbl); + + aDataTbl->verticalHeader()->hide(); + aDataTbl->setRowHeight(0, 25); + aDataTbl->setHorizontalScrollMode(QAbstractItemView::ScrollPerPixel); + + connect(aDataTbl->horizontalHeader(), SIGNAL(sectionResized(int, int, int)), + SLOT(onColumnResize(int, int, int))); + + updateHeaders(aDataTbl); + + QTableWidgetItem* aItem = new QTableWidgetItem("Default value"); + aItem->setBackgroundColor(Qt::lightGray); + aItem->setFlags(Qt::NoItemFlags | Qt::ItemIsEnabled); + aDataTbl->setItem(0, 0, aItem); + + // Set default value item + for (int i = 0; i < myCompNamesList.count(); i++) { + aItem = createDefaultItem(); + aItem->setBackgroundColor(Qt::lightGray); + aDataTbl->setItem(0, i + 1, aItem); + } + + if (aColWidth.length() > 0) { + for (int i = 0; i < aDataTbl->columnCount(); i++) { + if (i < aColWidth.size()) + aDataTbl->setColumnWidth(i, aColWidth.at(i)); + } + } + aStepLayout->addWidget(aDataTbl, 1, 0, 1, 2); + connect(aDataTbl, SIGNAL(cellChanged(int, int)), SLOT(onTableEdited(int, int))); + + myStepWgt->addWidget(aWidget); + aDataTbl->horizontalHeader()->viewport()->installEventFilter(this); +} + +//********************************************************************************** +void CollectionPlugin_WidgetField::deactivate() +{ + ModuleBase_WidgetSelector::deactivate(); + storeValueCustom(); +} + +//********************************************************************************** +//void CollectionPlugin_WidgetField::showEvent(QShowEvent* theEvent) +//{ +// myShapeTypeCombo->setEnabled(!isEditingMode()); +// myFieldTypeCombo->setEnabled(!isEditingMode()); +// myNbComponentsSpn->setEnabled(!isEditingMode()); +//} + +//********************************************************************************** +bool CollectionPlugin_WidgetField::eventFilter(QObject* theObject, QEvent* theEvent) +{ + QObject* aObject = 0; + foreach(QTableWidget* aTable, myDataTblList) { + if (aTable->horizontalHeader()->viewport() == theObject) { + aObject = theObject; + break; + } + } + if (aObject) { + if (theEvent->type() == QEvent::MouseButtonDblClick) { + if (myHeaderEditor) { //delete previous editor + myHeaderEditor->deleteLater(); + myHeaderEditor = 0; + } + QMouseEvent* aMouseEvent = static_cast(theEvent); + QHeaderView* aHeader = static_cast(aObject->parent()); + QTableWidget* aTable = static_cast(aHeader->parentWidget()); + + int aShift = aTable->horizontalScrollBar()->value(); + int aPos = aMouseEvent->x(); + int aIndex = aHeader->logicalIndex(aHeader->visualIndexAt(aPos)); + if (aIndex > 0) { + QRect aRect; + aRect.setLeft(aHeader->sectionPosition(aIndex)); + aRect.setWidth(aHeader->sectionSize(aIndex)); + aRect.setTop(0); + aRect.setHeight(aHeader->height()); + aRect.adjust(1, 1, -1, -1); + aRect.translate(-aShift, 0); + + myHeaderEditor = new QLineEdit(aHeader->viewport()); + myHeaderEditor->move(aRect.topLeft()); + myHeaderEditor->resize(aRect.size()); + myHeaderEditor->setFrame(false); + QString aText = aHeader->model()-> + headerData(aIndex, aHeader->orientation()).toString(); + myHeaderEditor->setText(aText); + myHeaderEditor->setFocus(); + myEditIndex = aIndex; //save for future use + myHeaderEditor->installEventFilter(this); //catch focus out event + //if user presses Enter it should close editor + connect(myHeaderEditor, SIGNAL(returnPressed()), aTable, SLOT(setFocus())); + myHeaderEditor->show(); + return true; + } + } + } else if ((theObject == myHeaderEditor) && (theEvent->type() == QEvent::FocusOut)) { + QString aNewTitle = myHeaderEditor->text(); + //save item text + myCompNamesList.replace(myEditIndex - 1, aNewTitle); + myHeaderEditor->deleteLater(); //safely delete editor + myHeaderEditor = 0; + // Store into data model + AttributeStringArrayPtr aStringsAttr = + myFeature->data()->stringArray(CollectionPlugin_Field::COMPONENTS_NAMES_ID()); + aStringsAttr->setValue(myEditIndex - 1, aNewTitle.toStdString()); + foreach(QTableWidget* aTable, myDataTblList) { + updateHeaders(aTable); + } + } else if (theEvent->type() == QEvent::FocusIn) { + QTableWidget* aTable = dynamic_cast(theObject); + if (aTable) { + ModuleBase_IPropertyPanel* aPanel = myWorkshop->propertyPanel(); + if (aPanel->activeWidget() != this) { + myActivation = true; + aPanel->activateWidget(this, false); + } + } + } + return ModuleBase_WidgetSelector::eventFilter(theObject, theEvent); +} + +//********************************************************************************** +QTableWidgetItem* CollectionPlugin_WidgetField::createDefaultItem() const +{ + QTableWidgetItem* aItem = new QTableWidgetItem(); + switch (myFieldTypeCombo->currentIndex()) { + case ModelAPI_AttributeTables::DOUBLE: + case ModelAPI_AttributeTables::INTEGER: + aItem->setText("0"); + break; + case ModelAPI_AttributeTables::BOOLEAN: + aItem->setText(MYFalse); + break; + case ModelAPI_AttributeTables::STRING: + aItem->setText(""); + break; + } + return aItem; +} + +//********************************************************************************** +QTableWidgetItem* CollectionPlugin_WidgetField:: + createValueItem(ModelAPI_AttributeTables::Value& theVal) const +{ + QTableWidgetItem* aItem = new QTableWidgetItem(); + aItem->setText(getValueText(theVal)); + return aItem; +} + +//********************************************************************************** +QString CollectionPlugin_WidgetField::getValueText(ModelAPI_AttributeTables::Value& theVal) const +{ + switch (myFieldTypeCombo->currentIndex()) { + case ModelAPI_AttributeTables::DOUBLE: + return QString::number(theVal.myDouble); + case ModelAPI_AttributeTables::INTEGER: + return QString::number(theVal.myInt); + case ModelAPI_AttributeTables::BOOLEAN: + return theVal.myBool? MYTrue : MYFalse; + case ModelAPI_AttributeTables::STRING: + return theVal.myStr.c_str(); + } + return ""; +} + + +//********************************************************************************** +void CollectionPlugin_WidgetField::updateHeaders(QTableWidget* theDataTbl) const +{ + QStringList aHeaders; + aHeaders << tr(MYFirstCol); + aHeaders << myCompNamesList; + theDataTbl->setHorizontalHeaderLabels(aHeaders); +} + +//********************************************************************************** +void CollectionPlugin_WidgetField::removeStepControls() +{ + int aCurWgtId = myStepWgt->currentIndex(); + QWidget* aWgt = myStepWgt->currentWidget(); + myStepWgt->removeWidget(aWgt); + + myStampSpnList.removeAt(aCurWgtId); + myDataTblList.removeAt(aCurWgtId); + aWgt->deleteLater(); +} + +//********************************************************************************** QList CollectionPlugin_WidgetField::getControls() const { QList aControls; // this control will accept focus and will be highlighted in the Property Panel - //aControls.push_back(myComboBox); + aControls.append(myShapeTypeCombo); + aControls.append(myFieldTypeCombo); + aControls.append(myNbComponentsSpn); + if (myStampSpnList.size() > 0) + aControls.append(myStampSpnList.first()); + if (myDataTblList.size() > 0) + aControls.append(myDataTblList.first()); + return aControls; } +//********************************************************************************** bool CollectionPlugin_WidgetField::storeValueCustom() { - //AttributePtr anAttribute = myFeature->attribute(SamplePanelPlugin_Feature::VALUE_ID()); - //AttributeIntegerPtr aValueAttribute = - // std::dynamic_pointer_cast(anAttribute); - //aValueAttribute->setValue(myComboBox->currentIndex()); - //updateObject(myFeature); + DataPtr aData = myFeature->data(); + // Store number of components + AttributeStringArrayPtr aStringsAttr = + aData->stringArray(CollectionPlugin_Field::COMPONENTS_NAMES_ID()); + int aNbComps = myCompNamesList.size(); + aStringsAttr->setSize(aNbComps); + for ( int i = 0; i < aNbComps; i++) + aStringsAttr->setValue(i, myCompNamesList.at(i).toStdString()); + + AttributeTablesPtr aTablesAttr = aData->tables(CollectionPlugin_Field::VALUES_ID()); + // Store number of steps + int aNbSteps = myDataTblList.size(); + + // Store Type of the field values + int aFldType = myFieldTypeCombo->currentIndex(); + + AttributeIntArrayPtr aStampsAttr = aData->intArray(CollectionPlugin_Field::STAMPS_ID()); + aStampsAttr->setSize(aNbSteps); + // Store data + QTableWidget* aTable = myDataTblList.first(); + int aRows = aTable->rowCount(); + // first column contains selected names which should not be stored + int aColumns = aTable->columnCount() - 1; + + aTablesAttr->setSize(aRows, aColumns, aNbSteps); + aTablesAttr->setType((ModelAPI_AttributeTables::ValueType)aFldType); + for (int i = 0; i < aNbSteps; i++) { + aStampsAttr->setValue(i, myStampSpnList.at(i)->value()); + aTable = myDataTblList.at(i); + for (int j = 0; j < aColumns; j++) { + for (int k = 0; k < aRows; k++) { + QString aTblVal = aTable->item(k, j + 1)->text(); + aTablesAttr->setValue(getValue(aTblVal), k, j, i); + } + } + } return true; } +//********************************************************************************** bool CollectionPlugin_WidgetField::restoreValueCustom() { - //AttributePtr anAttribute = myFeature->attribute(SamplePanelPlugin_Feature::VALUE_ID()); - //AttributeIntegerPtr aValueAttribute = - // std::dynamic_pointer_cast(anAttribute); + bool isBlocked; + DataPtr aData = myFeature->data(); - //bool isBlocked = myComboBox->blockSignals(true); - //myComboBox->setCurrentIndex(aValueAttribute->value()); - //myComboBox->blockSignals(isBlocked); + AttributeSelectionListPtr aSelList = aData->selectionList(CollectionPlugin_Field::SELECTED_ID()); + std::string aTypeStr = aSelList->selectionType(); + if (aTypeStr == "") + return false; // The attribute is not initialized + isBlocked = myShapeTypeCombo->blockSignals(true); + myShapeTypeCombo->setCurrentIndex(getSelectionType(aTypeStr)); + myShapeTypeCombo->blockSignals(isBlocked); + // Get number of components + AttributeStringArrayPtr aStringsAttr = + aData->stringArray(CollectionPlugin_Field::COMPONENTS_NAMES_ID()); + + myCompNamesList.clear(); + for (int i = 0; i < aStringsAttr->size(); i++) { + myCompNamesList.append(aStringsAttr->value(i).c_str()); + } + isBlocked = myNbComponentsSpn->blockSignals(true); + myNbComponentsSpn->setValue(myCompNamesList.size()); + myNbComponentsSpn->blockSignals(isBlocked); + + AttributeTablesPtr aTablesAttr = aData->tables(CollectionPlugin_Field::VALUES_ID()); + // Get number of steps + int aNbSteps = aTablesAttr->tables(); + myStepSlider->setMaximum(aNbSteps); + //myStepSlider->setValue(1); + // Clear old tables + while (myDataTblList.count() > aNbSteps) { + QWidget* aWgt = myStepWgt->widget(myStepWgt->count() - 1); + myStepWgt->removeWidget(aWgt); + aWgt->deleteLater(); + + myStampSpnList.removeLast(); + myDataTblList.removeLast(); + } + while (myDataTblList.count() < aNbSteps) + appendStepControls(); + //myStepWgt->setCurrentIndex(myStepSlider->value() - 1); + clearData(); + + // Get Type of the field values + isBlocked = myFieldTypeCombo->blockSignals(true); + myFieldTypeCombo->setCurrentIndex(aTablesAttr->type()); + myFieldTypeCombo->blockSignals(isBlocked); + myDelegate->setDataType(aTablesAttr->type()); + + AttributeIntArrayPtr aStampsAttr = aData->intArray(CollectionPlugin_Field::STAMPS_ID()); + // Fill data table + int aRows = aTablesAttr->rows(); + int aCols = aTablesAttr->columns(); + + // Get width of columns + QIntList aColWidth; + QTableWidget* aFirstTable = myDataTblList.first(); + for (int i = 0; i < aFirstTable->columnCount(); i++) + aColWidth.append(aFirstTable->columnWidth(i)); + + QTableWidgetItem* aItem = 0; + for (int i = 0; i < aNbSteps; i++) { + myStampSpnList.at(i)->setValue(aStampsAttr->value(i)); + QTableWidget* aTable = myDataTblList.at(i); + isBlocked = aTable->blockSignals(true); + aTable->setRowCount(aRows); + aTable->setColumnCount(aCols + 1); + updateHeaders(aTable); + for (int j = 0; j < aCols + 1; j++) { + for (int k = 0; k < aRows; k++) { + aItem = aTable->item(k, j); + if ((j == 0) && (k > 0)) { + // Add selection names + AttributeSelectionPtr aAttr = aSelList->value(k - 1); + if (aItem) { + aItem->setText(aAttr->namingName().c_str()); + } else { + aItem = new QTableWidgetItem(aAttr->namingName().c_str()); + aTable->setItem(k, j, aItem); + } + } else if (j > 0) { + // Add Values + ModelAPI_AttributeTables::Value aVal = aTablesAttr->value(k, j - 1, i); + if (aItem) { + aItem->setText(getValueText(aVal)); + } else { + aItem = createValueItem(aVal); + if (k == 0) + aItem->setBackgroundColor(Qt::lightGray); + aTable->setItem(k, j, aItem); + } + } + } + } + // Restore columns width + for (int i = 0; i < aTable->columnCount(); i++) { + if (i < aColWidth.size()) + aTable->setColumnWidth(i, aColWidth.at(i)); + } + + aTable->blockSignals(isBlocked); + } return true; } +//********************************************************************************** +int CollectionPlugin_WidgetField::getSelectionType(const std::string& theStr) const +{ + QString aType(theStr.c_str()); + aType = aType.toLower(); + if (aType == "vertex") + return 0; + else if (aType == "edge") + return 1; + else if (aType == "face") + return 2; + else if (aType == "solid") + return 3; + else if (aType == "object") + return 4; + else if (aType == "part") + return 5; + return -1; +} + + +//********************************************************************************** +std::string CollectionPlugin_WidgetField::getSelectionType(int theType) const +{ + switch (theType) { + case 0: //"Vertices" + return "vertex"; + case 1: // "Edges" + return "edge"; + case 2: // "Faces" + return "face"; + case 3: // "Solids" + return "solid"; + case 4: // "Results" + return "object"; + case 5: // "Parts" + return "part"; + } + return ""; +} + +//********************************************************************************** +QIntList CollectionPlugin_WidgetField::shapeTypes() const +{ + QIntList aRes; + switch (myShapeTypeCombo->currentIndex()) { + case 0: //"Vertices" + aRes.append(ModuleBase_Tools::shapeType("vertex")); + break; + case 1: // "Edges" + aRes.append(ModuleBase_Tools::shapeType("edge")); + break; + case 2: // "Faces" + aRes.append(ModuleBase_Tools::shapeType("face")); + break; + case 3: // "Solids" + aRes.append(ModuleBase_Tools::shapeType("solid")); + break; + case 4: // "Results" + aRes.append(ModuleBase_Tools::shapeType("object")); + break; + case 5: // "Parts" + // TODO: Selection mode for Parts + break; + } + return aRes; +} + +//********************************************************************************** +ModelAPI_AttributeTables::Value CollectionPlugin_WidgetField::getValue(QString theStrVal) const +{ + ModelAPI_AttributeTables::Value aVal; + switch (myFieldTypeCombo->currentIndex()) { + case ModelAPI_AttributeTables::BOOLEAN: + aVal.myBool = (theStrVal == MYTrue)? true : false; + break; + case ModelAPI_AttributeTables::DOUBLE: + aVal.myDouble = theStrVal.toDouble(); + break; + case ModelAPI_AttributeTables::INTEGER: + aVal.myInt = theStrVal.toInt(); + break; + case ModelAPI_AttributeTables::STRING: + aVal.myStr = theStrVal.toStdString(); + } + return aVal; +} + + +//********************************************************************************** void CollectionPlugin_WidgetField::onNbCompChanged(int theVal) { - int aOldCol = myDataTbl->columnCount() - 1; - int aNbRows = myDataTbl->rowCount(); - myDataTbl->setColumnCount(theVal + 1); + int aOldCol = myCompNamesList.count(); + int aNbRows = myDataTblList.first()->rowCount(); int aDif = theVal - aOldCol; QTableWidgetItem* aItem = 0; - for (int i = 0; i < aDif; i++) { - for (int j = 0; j < aNbRows; j++) { - aItem = new QTableWidgetItem(); - if (j == 0) - aItem->setText(QString("Comp %1").arg(i + aOldCol + 1)); - else - aItem->setText("0"); - if (j < 3) - aItem->setBackgroundColor(Qt::lightGray); - myDataTbl->setItem(j, i + aOldCol + 1, aItem); + + while (myCompNamesList.count() != theVal) { + if (aDif > 0) + myCompNamesList.append(QString("Comp %1").arg(myCompNamesList.count() + 1)); + else + myCompNamesList.removeLast(); + } + + AttributeTablesPtr aTablesAttr = myFeature->data()->tables(CollectionPlugin_Field::VALUES_ID()); + aTablesAttr->setSize(aNbRows, myCompNamesList.size(), myDataTblList.size()); + + foreach(QTableWidget* aDataTbl, myDataTblList) { + aDataTbl->setColumnCount(theVal + 1); + updateHeaders(aDataTbl); + for (int i = aOldCol; i < myCompNamesList.count(); i++) { + for (int j = 0; j < aNbRows; j++) { + aItem = aDataTbl->item(j, i + 1); + if (!aItem) { + aItem = createDefaultItem(); + if (j == 0) + aItem->setBackgroundColor(Qt::lightGray); + aDataTbl->setItem(j, i + 1, aItem); + } + } } } + emit valuesChanged(); } +//********************************************************************************** void CollectionPlugin_WidgetField::onAddStep() { int aMax = myStepSlider->maximum(); aMax++; myStepSlider->setMaximum(aMax); myMaxLbl->setText(QString::number(aMax)); - clearData(); + appendStepControls(); myStepSlider->setValue(aMax); + + AttributeTablesPtr aTablesAttr = myFeature->data()->tables(CollectionPlugin_Field::VALUES_ID()); + aTablesAttr->setSize(aTablesAttr->rows(), aTablesAttr->columns(), myDataTblList.size()); + + + AttributeSelectionListPtr aSelList = + myFeature->data()->selectionList(CollectionPlugin_Field::SELECTED_ID()); + if (!aSelList->isInitialized()) + return; + int aSelNb = aSelList->size(); + if (aSelNb == 0) + return; + + int aColumns = myNbComponentsSpn->value() + 1; + int aRows = aSelNb + 1; + QTableWidget* aTable = myDataTblList.last(); + aTable->setRowCount(aRows); + QTableWidgetItem* aItem = 0; + for(int i = 0; i < aColumns; i++) { + if (i == 0) { + for(int j = 1; j < aRows; j++) { + aItem = aTable->item(j, i); + if (!aItem) { + aItem = new QTableWidgetItem(); + aTable->setItem(j, i, aItem); + } + AttributeSelectionPtr aAttr = aSelList->value(j - 1); + aItem->setText(aAttr->namingName().c_str()); + aItem->setToolTip(aAttr->namingName().c_str()); + } + } else { + QString aDefVal = aTable->item(0, i)->text(); + for(int j = 1; j < aRows; j++) { + aItem = aTable->item(j, i); + if (!aItem) { + aItem = new QTableWidgetItem(); + aTable->setItem(j, i, aItem); + } + aItem->setText(aDefVal); + } + } + } + emit valuesChanged(); } +//********************************************************************************** void CollectionPlugin_WidgetField::onRemoveStep() { int aMax = myStepSlider->maximum(); aMax--; - myStepSlider->setMaximum(aMax); myMaxLbl->setText(QString::number(aMax)); + removeStepControls(); + myStepSlider->setMaximum(aMax); + + //AttributeTablesPtr aTablesAttr = myFeature->data()->tables(CollectionPlugin_Field::VALUES_ID()); + //aTablesAttr->setSize(aTablesAttr->rows(), aTablesAttr->columns(), myDataTblList.size()); + emit valuesChanged(); } +//********************************************************************************** void CollectionPlugin_WidgetField::clearData() { - int aNbRows = myDataTbl->rowCount(); - int aNbCol = myDataTbl->columnCount(); - - QString aDefValue; - for (int i = 1; i < aNbCol; i++) { - aDefValue = myDataTbl->item(1, i)->text(); - for (int j = 2; j < aNbRows; j++) { - myDataTbl->item(j, i)->setText(aDefValue); - } + foreach(QTableWidget* aDataTbl, myDataTblList) { + aDataTbl->setRowCount(1); } } +//********************************************************************************** void CollectionPlugin_WidgetField::onStepMove(int theStep) { myCurStepLbl->setText(QString::number(theStep)); + myStepWgt->setCurrentIndex(theStep - 1); +} + +//********************************************************************************** +bool CollectionPlugin_WidgetField:: + isValidSelectionCustom(const std::shared_ptr& thePrs) +{ + return (myShapeTypeCombo->currentIndex() == 5)? false : true; +} + +//********************************************************************************** +bool CollectionPlugin_WidgetField:: + setSelection(QList>& theValues, const bool theToValidate) +{ + if (myActivation) { + myActivation = false; + return false; + } + // Ignore selection for Parts mode + if (myShapeTypeCombo->currentIndex() == 5) + return false; + + QList aSelected; + QList::const_iterator anIt; + for (anIt = theValues.begin(); anIt != theValues.end(); anIt++) { + ModuleBase_ViewerPrsPtr aValue = *anIt; + ResultPtr aRes = std::dynamic_pointer_cast(aValue->object()); + if (theToValidate && aRes.get()) { + if (myShapeTypeCombo->currentIndex() > 3) + aSelected.append(aValue); + else if (acceptSubShape(aValue->shape(), aRes)) + aSelected.append(aValue); + } + } + AttributeSelectionListPtr aSelList = + myFeature->data()->selectionList(CollectionPlugin_Field::SELECTED_ID()); + aSelList->setSelectionType(getSelectionType(myShapeTypeCombo->currentIndex())); + aSelList->clear(); + + ResultPtr aResult; + GeomShapePtr aShape; + int aNbData = 0; + foreach(ModuleBase_ViewerPrsPtr aPrs, aSelected) { + aResult = std::dynamic_pointer_cast(aPrs->object()); + aShape = aPrs->shape(); + if (!aResult.get() && !aShape.get()) + continue; + if (!aSelList->isInList(aResult, aShape)) { + aSelList->append(aResult, aShape); + aNbData++; + } + } + int aColumns = myDataTblList.first()->columnCount(); + int aRows = myDataTblList.first()->rowCount(); + int aNewRows = aNbData + 1; + AttributeTablesPtr aTablesAttr = myFeature->data()->tables(CollectionPlugin_Field::VALUES_ID()); + aTablesAttr->setSize(aNewRows, aColumns - 1, myDataTblList.size()); + + QTableWidgetItem* aItem = 0; + foreach(QTableWidget* aTable, myDataTblList) { + aTable->setRowCount(aNewRows); + if (aNewRows > aRows) { + // Add new data + for(int i = 0; i < aColumns; i++) { + if (i == 0) { + for(int j = 1; j < aNewRows; j++) { + aItem = aTable->item(j, i); + if (!aItem) { + aItem = new QTableWidgetItem(); + aTable->setItem(j, i, aItem); + } + AttributeSelectionPtr aAttr = aSelList->value(j - 1); + aItem->setText(aAttr->namingName().c_str()); + aItem->setToolTip(aAttr->namingName().c_str()); + } + } else { + QString aDefVal = aTable->item(0, i)->text(); + for(int j = aRows; j < aNewRows; j++) { + aItem = aTable->item(j, i); + if (!aItem) { + aItem = new QTableWidgetItem(); + aTable->setItem(j, i, aItem); + } + aItem->setText(aDefVal); + } + } + } + } else { + // Update only selection name + for(int j = 1; j < aNewRows - 1; j++) { + AttributeSelectionPtr aAttr = aSelList->value(j); + aTable->item(j, 0)->setText(aAttr->namingName().c_str()); + aTable->item(j, 0)->setToolTip(aAttr->namingName().c_str()); + } + } + } + return true; +} + +//********************************************************************************** +void CollectionPlugin_WidgetField::onFieldTypeChanged(int theIdx) +{ + ModelAPI_AttributeTables::ValueType aOldType = myDelegate->dataType(); + if (aOldType != theIdx) { + myDelegate->setDataType((ModelAPI_AttributeTables::ValueType)theIdx); + int aColumns = myDataTblList.first()->columnCount(); + int aRows = myDataTblList.first()->rowCount(); + foreach(QTableWidget* aTable, myDataTblList) { + for(int i = 1; i < aColumns; i++) { + for(int j = 0; j < aRows; j++) { + switch (theIdx) { + case ModelAPI_AttributeTables::DOUBLE: + case ModelAPI_AttributeTables::INTEGER: + aTable->item(j, i)->setText("0"); + break; + case ModelAPI_AttributeTables::BOOLEAN: + aTable->item(j, i)->setText(MYFalse); + break; + case ModelAPI_AttributeTables::STRING: + aTable->item(j, i)->setText(""); + break; + } + } + } + } + emit valuesChanged(); + } +} + +//********************************************************************************** +void CollectionPlugin_WidgetField::onTableEdited(int theRow, int theCol) +{ + // Do not store here column of names + if (theCol == 0) + return; + if (!myFeature.get()) + return; + QTableWidget* aTable = static_cast(sender()); + int aNb = myDataTblList.indexOf(aTable); + if (aNb == -1) + return; + ModelAPI_AttributeTables::Value aVal = getValue(aTable->item(theRow, theCol)->text()); + + AttributeTablesPtr aTablesAttr = myFeature->data()->tables(CollectionPlugin_Field::VALUES_ID()); + if (aTablesAttr->isInitialized()) + aTablesAttr->setValue(aVal,theRow, theCol - 1, aNb); + else + emit valuesChanged(); +} + +//********************************************************************************** +void CollectionPlugin_WidgetField::onShapeTypeChanged(int theType) +{ + updateSelectionModesAndFilters(theType == 5? false:true); + + AttributeSelectionListPtr aSelList = + myFeature->data()->selectionList(CollectionPlugin_Field::SELECTED_ID()); + + std::string aTypeName = getSelectionType(theType); + if (aTypeName == aSelList->selectionType()) + return; + aSelList->setSelectionType(aTypeName); + + //Clear old selection + clearData(); + aSelList->clear(); + AttributeTablesPtr aTablesAttr = myFeature->data()->tables(CollectionPlugin_Field::VALUES_ID()); + aTablesAttr->setSize(1, myNbComponentsSpn->value(), myDataTblList.size()); + updateObject(myFeature); + emit valuesChanged(); +} + +//********************************************************************************** +bool CollectionPlugin_WidgetField::processEnter() +{ + if (myIsTabEdit) { + myIsTabEdit = false; + return true; + } + QWidget* aCurrWgt = qApp->focusWidget(); + int aCurWgtId = myStepWgt->currentIndex(); + if ((aCurrWgt == myShapeTypeCombo) || + (aCurrWgt == myFieldTypeCombo) || + (aCurrWgt == myNbComponentsSpn) || + (aCurrWgt == myStampSpnList[aCurWgtId]) || + (aCurrWgt == myDataTblList[aCurWgtId])) { + setFocus(); + return true; + } + return false; +} + +//********************************************************************************** +void CollectionPlugin_WidgetField::onFocusChanged(QWidget* theOld, QWidget* theNew) +{ + if (theNew && (!myIsTabEdit)) + myIsTabEdit = dynamic_cast(theNew); +} + +//********************************************************************************** +void CollectionPlugin_WidgetField::onRangeChanged(int theMin, int theMax) +{ + myMaxLbl->setText(QString::number(theMax)); + myRemoveBtn->setEnabled(theMax > 1); +} + +//********************************************************************************** +void CollectionPlugin_WidgetField::onColumnResize(int theIndex, int theOld, int theNew) +{ + if (myDataTblList.count() < 2) + return; + QObject* aSender = sender(); + foreach(QTableWidget* aTable, myDataTblList) { + if (aTable->horizontalHeader() != aSender) + aTable->setColumnWidth(theIndex, theNew); + } } + +//********************************************************************************** +QList> + CollectionPlugin_WidgetField::getAttributeSelection() const +{ + QList> aList; + if(myFeature) { + DataPtr aData = myFeature->data(); + AttributeSelectionListPtr aSelList = + aData->selectionList(CollectionPlugin_Field::SELECTED_ID()); + AttributeSelectionPtr aAttr; + ObjectPtr anObject; + for (int i = 0; i < aSelList->size(); i++) { + aAttr = aSelList->value(i); + ModuleBase_ViewerPrsPtr + aPrs(new ModuleBase_ViewerPrs(aAttr->context(), aAttr->value(), NULL)); + aList.append(aPrs); + } + } + return aList; +} \ No newline at end of file