source: trunk/examples/sql/overview/table3/main.cpp

/****************************************************************************
** $Id: main.cpp 2051 2007-02-21 10:04:20Z chehrlic $
**
** Copyright (C) 1992-2007 Trolltech ASA.  All rights reserved.
**
** This file is part of an example program for Qt.  This example
** program may be used, distributed and modified without limitation.
**
*****************************************************************************/

#include "main.h"
#include <qdatatable.h>

StatusPicker::StatusPicker( QWidget *parent, const char *name )
    : QComboBox( parent, name )
{
    QSqlCursor cur( "status" );
    cur.select( cur.index( "name" ) );

    int i = 0;
    while ( cur.next() ) {
        insertItem( cur.value( "name" ).toString(), i );
        index2id[i] = cur.value( "id" ).toInt();
        i++;
    }
}


int StatusPicker::statusId() const
{
    return index2id[ currentItem() ];
}


void StatusPicker::setStatusId( int statusid )
{
    QMap<int,int>::Iterator it;
    for ( it = index2id.begin(); it != index2id.end(); ++it ) {
        if ( it.data() == statusid ) {
            setCurrentItem( it.key() );
            break;
        }
    }
}



QWidget *CustomSqlEditorFactory::createEditor(
    QWidget *parent, const QSqlField *field )
{
    if ( field->name() == "statusid" ) {
        QWidget *editor = new StatusPicker( parent );
        return editor;
    }

    return QSqlEditorFactory::createEditor( parent, field );
}


int main( int argc, char *argv[] )
{
    QApplication app( argc, argv );

    if ( createConnections() ) {
        QSqlCursor staffCursor( "staff" );

        QDataTable              *staffTable     = new QDataTable( &staffCursor );
        QSqlPropertyMap         *propMap        = new QSqlPropertyMap();
        CustomSqlEditorFactory  *editorFactory  = new CustomSqlEditorFactory();
        propMap->insert( "StatusPicker", "statusid" );
        staffTable->installPropertyMap( propMap );
        staffTable->installEditorFactory( editorFactory );

        app.setMainWidget( staffTable );

        staffTable->addColumn( "forename", "Forename" );
        staffTable->addColumn( "surname",  "Surname" );
        staffTable->addColumn( "salary",   "Annual Salary" );
        staffTable->addColumn( "statusid", "Status" );

        QStringList order = QStringList() << "surname" << "forename";
        staffTable->setSort( order );

        staffTable->refresh();
        staffTable->show();

        return app.exec();
    }

    return 1;
}