#include "HYDROGUI_Module.h"
#include "HYDROGUI_Tool.h"
#include "HYDROGUI_LineEditDoubleValidator.h"
+#include "HYDROData_StricklerTable.h"
#include <LightApp_Application.h>
#include <QLabel>
#include <QTableWidget>
#include <QHeaderView>
+#include <QColorDialog>
+
+HYDROGUI_ColorDelegate::HYDROGUI_ColorDelegate( QWidget* theParent )
+{
+}
+
+HYDROGUI_ColorDelegate::~HYDROGUI_ColorDelegate()
+{
+}
+
+void HYDROGUI_ColorDelegate::paint( QPainter* thePainter, const QStyleOptionViewItem& theOption,
+ const QModelIndex& theIndex ) const
+{
+ QColor aColor = qVariantValue<QColor>( theIndex.data( Qt::BackgroundColorRole ) );
+ thePainter->fillRect( theOption.rect, aColor );
+}
+
+QWidget* HYDROGUI_ColorDelegate::createEditor( QWidget* theParent,
+ const QStyleOptionViewItem& theOption,
+ const QModelIndex& theIndex ) const
+{
+ QColor aColor = qVariantValue<QColor>( theIndex.data( Qt::BackgroundColorRole ) );
+ QColor aNewColor = QColorDialog::getColor( aColor );
+ if( aNewColor.isValid() )
+ {
+ QAbstractItemModel* aModel = const_cast<QAbstractItemModel*>( theIndex.model() );
+ QVariant aValue = qVariantFromValue( aNewColor );
+ aModel->setData( theIndex, aValue, Qt::BackgroundColorRole );
+ }
+ return 0;
+}
+
+void HYDROGUI_ColorDelegate::setEditorData( QWidget* theEditor, const QModelIndex& theIndex ) const
+{
+}
+
+void HYDROGUI_ColorDelegate::setModelData( QWidget* theEditor, QAbstractItemModel* theModel,
+ const QModelIndex& theIndex ) const
+{
+}
+
+QSize HYDROGUI_ColorDelegate::sizeHint( const QStyleOptionViewItem& theOption, const QModelIndex& theIndex ) const
+{
+ return theOption.rect.size();
+}
+
+
+
HYDROGUI_StricklerTableDlg::HYDROGUI_StricklerTableDlg( HYDROGUI_Module* theModule, const QString& theTitle, int theType )
: HYDROGUI_InputPanel( theModule, theTitle ),
QLabel* anImageNameLabel = new QLabel( tr( "NAME" ), aNameGroup );
myName = new QLineEdit( aNameGroup );
+ QGroupBox* aAttrNameGroup = new QGroupBox( tr( "STRICKLER_TABLE_ATTR_NAME" ), this );
+ QLabel* aAttrNameLabel = new QLabel( tr( "ATTR_NAME" ), aAttrNameGroup );
+ myAttrName = new QLineEdit( aAttrNameGroup );
+
QBoxLayout* anImageNameLayout = new QHBoxLayout( aNameGroup );
anImageNameLayout->setMargin( 5 );
anImageNameLayout->setSpacing( 5 );
anImageNameLayout->addWidget( anImageNameLabel );
anImageNameLayout->addWidget( myName );
+ QBoxLayout* anAttrNameLayout = new QHBoxLayout( aAttrNameGroup );
+ anAttrNameLayout->setMargin( 5 );
+ anAttrNameLayout->setSpacing( 5 );
+ anAttrNameLayout->addWidget( aAttrNameLabel );
+ anAttrNameLayout->addWidget( myAttrName );
+
// Strickler table
QGroupBox* aTableGroup = new QGroupBox( tr( "STRICKLER_TABLE_TABLE" ), this );
aTableGroup->setVisible( theType == Edit );
// Table
myTable = new QTableWidget( mainFrame() );
- myTable->setItemDelegate( new HYDROGUI_LineEditDoubleValidator( this ) );
+ myTable->setItemDelegateForColumn( 1, new HYDROGUI_LineEditDoubleValidator( this ) );
+ myTable->setItemDelegateForColumn( 3, new HYDROGUI_ColorDelegate( this ) );
myTable->setEditTriggers( QAbstractItemView::DoubleClicked |
QAbstractItemView::SelectedClicked |
QAbstractItemView::EditKeyPressed );
- myTable->setColumnCount( 2 );
+ myTable->setColumnCount( 4 );
QStringList aColumnNames;
- aColumnNames << tr( "STRICKLER_TYPE" ) << tr( "STRICKLER_COEFFICIENT" );
+ aColumnNames << tr( "STRICKLER_TYPE" ) << tr( "STRICKLER_COEFFICIENT" ) << tr( "ATTR_VALUE" ) << tr( "COLOR" );
myTable->setHorizontalHeaderLabels( aColumnNames );
- myTable->horizontalHeader()->setStretchLastSection( false);
- myTable->horizontalHeader()->setResizeMode( 0, QHeaderView::Stretch );
+ myTable->horizontalHeader()->setStretchLastSection( false );
+ myTable->horizontalHeader()->setResizeMode( 0, QHeaderView::ResizeToContents );
myTable->horizontalHeader()->setResizeMode( 1, QHeaderView::ResizeToContents );
+ myTable->horizontalHeader()->setResizeMode( 2, QHeaderView::ResizeToContents );
+ myTable->horizontalHeader()->setResizeMode( 3, QHeaderView::Stretch );
myTable->verticalHeader()->setResizeMode( QHeaderView::ResizeToContents );
// Common
addWidget( aFileNameGroup );
addWidget( aNameGroup );
+ addWidget( aAttrNameGroup );
addWidget( aTableGroup );
// Update controls
connect( myTable->selectionModel(), SIGNAL( selectionChanged( QItemSelection, QItemSelection ) ),
this, SLOT( onSelectionChanged() ) );
- setMinimumWidth( 350 );
+ setMinimumWidth( 450 );
if ( theType != Edit )
addStretch();
myName->setReadOnly( on );
}
-HYDROGUI_StricklerTableDlg::StricklerCoefficientList HYDROGUI_StricklerTableDlg::getData() const
+void HYDROGUI_StricklerTableDlg::getGuiData( Handle_HYDROData_StricklerTable& theTable ) const
{
- StricklerCoefficientList aRes;
- for ( int i = 0; i < myTable->rowCount(); i++ )
- {
- QTableWidgetItem* typeItem = myTable->item( i, 0 );
- QTableWidgetItem* coeffItem = myTable->item( i, 1 );
- aRes.append( StricklerCoefficient( typeItem->text(), coeffItem->text().toDouble() ) );
- }
- return aRes;
+ if( theTable.IsNull() )
+ return;
+
+ theTable->SetAttrName( myAttrName->text() );
+ theTable->Clear();
+ for ( int i = 0; i < myTable->rowCount(); i++ )
+ {
+ QTableWidgetItem* typeItem = myTable->item( i, 0 );
+ QString aType = typeItem->data( Qt::DisplayRole ).toString();
+
+ QTableWidgetItem* coeffItem = myTable->item( i, 1 );
+ QString aCoeffStr = coeffItem->data( Qt::DisplayRole ).toString();
+ double aCoeff = aCoeffStr.toDouble();
+
+ QTableWidgetItem* attrValueItem = myTable->item( i, 2 );
+ QString anAttrValue = attrValueItem->data( Qt::DisplayRole ).toString();
+
+ QTableWidgetItem* colorItem = myTable->item( i, 3 );
+ QColor aColor = colorItem->backgroundColor();
+
+ theTable->Set( aType, aCoeff );
+ theTable->SetAttrValue( aType, anAttrValue );
+ theTable->SetColor( aType, aColor );
+ }
}
-void HYDROGUI_StricklerTableDlg::setData(const StricklerCoefficientList& theData)
+void HYDROGUI_StricklerTableDlg::setGuiData( const Handle_HYDROData_StricklerTable& theTable )
{
- myTable->setRowCount( 0 );
+ myAttrName->setText( theTable->GetAttrName() );
- foreach ( const StricklerCoefficient& aData, theData ) {
- // Check the current Strickler type
- if ( aData.myType.isEmpty() ) {
- continue;
- }
+ if( theTable.IsNull() )
+ {
+ myTable->setRowCount( 0 );
+ return;
+ }
+
+ QStringList aTypes = theTable->GetTypes();
+ myTable->setRowCount( 0 );
+ foreach( QString aType, aTypes )
+ {
+ // Check the current Strickler type
+ if( aType.isEmpty() )
+ continue;
+
+ // Get Strickler data for the current Strickler type
+ QString aCoefficient = HYDROGUI_Tool::GetCoordinateString( theTable->Get( aType, 0.0 ), false );
+ QString anAttrValue = theTable->GetAttrValue( aType );
+ QColor aColor = theTable->GetColor( aType );
+
+ // Insert row with the data
+ int aRow = myTable->rowCount();
+ myTable->insertRow( aRow );
- // Get Strickler coefficient value for the current Strickler type
- QString aCoefficient = HYDROGUI_Tool::GetCoordinateString( aData.myCoefficient, false );
+ // "Type" column
+ QTableWidgetItem* aTypeItem = new QTableWidgetItem( aType );
+ myTable->setItem( aRow, 0, aTypeItem );
- // Insert row with the data
- int aRow = myTable->rowCount();
- myTable->insertRow( aRow );
+ // "Coefficient" column
+ myTable->setItem( aRow, 1, new QTableWidgetItem( aCoefficient ) );
- // "Type" column
- QTableWidgetItem* aTypeItem = new QTableWidgetItem( aData.myType );
- myTable->setItem( aRow, 0, aTypeItem );
+ // "Attribute value" column
+ myTable->setItem( aRow, 2, new QTableWidgetItem( anAttrValue ) );
- // "Coefficient" column
- myTable->setItem( aRow, 1, new QTableWidgetItem( aCoefficient ) );
- }
+ // "Color" column
+ QTableWidgetItem* anItem = new QTableWidgetItem();
+ anItem->setBackgroundColor( aColor );
+ myTable->setItem( aRow, 3, anItem );
+ }
- myTable->resizeColumnToContents( 0 );
- myTable->resizeRowsToContents();
+ myTable->resizeColumnToContents( 0 );
+ myTable->resizeRowsToContents();
- updateControls();
+ updateControls();
}
void HYDROGUI_StricklerTableDlg::updateControls()
