Subversion Repositories svn1

#include "qmdialogloadexternalteams.h"
#include "ui_qmdialogloadexternalteams.h"
#include "QFile"
#include "QTextStream"
#include "mainwindow.h"
#include <QRegExp>
#include <QtGui/QHBoxLayout>
#include <QtGui/QVBoxLayout>
#include <QtGui/QWidget>

#include    "consts.h"
#include    "structs.h"
#include    "proto.h"

QmDialogLoadExternalTeams::QmDialogLoadExternalTeams(const QString &efile,QWidget *parent) :
    QDialog(parent)
    {

    //
    // Create Windows
    //
    resize(550, 500);
    setSizeGripEnabled(true);
    setWindowTitle("Load External Team Data");

    QVBoxLayout *verticalLayout;
    verticalLayout = new QVBoxLayout(this);
    verticalLayout->setContentsMargins(0, 0, 0, 0);

    QVBoxLayout *verticalLayout2;
    verticalLayout2 = new QVBoxLayout(this);
    verticalLayout2->setContentsMargins(5, 5, 5, 5);

    QHBoxLayout *horizontalLayout;
    horizontalLayout = new QHBoxLayout();

    groupBox = new QGroupBox(this);
    groupBox->setTitle("Data");
    //groupBox->setGeometry(QRect(10, 0, 531, 451));
     verticalLayout->addWidget(groupBox);
     //verticalLayout->addLayout(verticalLayout2);
     groupBox->setLayout(verticalLayout2);

     tableWidget = new QTableWidget(groupBox);
     tableWidget->setSizePolicy(QSizePolicy::Expanding,QSizePolicy::Expanding);
     tableWidget->setGeometry(QRect(15, 21, 501, 421));
     tableWidget->setCornerButtonEnabled(false);
     tableWidget->verticalHeader()->setVisible(false);
     tableWidget->verticalHeader()->setDefaultSectionSize(20);
     tableWidget->verticalHeader()->setHighlightSections(true);
     verticalLayout2->addWidget(tableWidget);

    fileName = new QLineEdit(this);
    fileName->setObjectName(QString::fromUtf8("fileName"));
    fileName->setGeometry(QRect(20, 470, 331, 20));
    fileName->setReadOnly(true);
    fileName->setText(efile);
    horizontalLayout->addWidget(fileName);



    load = new QPushButton(this);
    load->setObjectName(QString::fromUtf8("load"));
    load->setGeometry(QRect(370, 470, 75, 23));
    load->setText("Load");
    horizontalLayout->addWidget(load);

    cancel = new QPushButton(this);
    cancel->setObjectName(QString::fromUtf8("cancel"));
    cancel->setGeometry(QRect(460, 470, 75, 23));
    cancel->setAutoDefault(false);
    cancel->setText("Cancel");
    horizontalLayout->addWidget(cancel);

    verticalLayout->addLayout(horizontalLayout);


    // Open the users file

    QFile file(efile);
    if ( ! file.open(QIODevice::ReadOnly | QIODevice::Text) )
    {
        MainWindow::showMessage("Cannot open external data file");
        return;
    }
    MainWindow::showMessage("Loading External Data");

    // Insert column headers
    tableWidget->setColumnCount(3 + ( 2 * config.num_legs) );
    QStringList labels;
    labels << "Team" << "Team Name" << "Cat";
    for (int ii = 1; ii <= config.num_legs; ii++ )
    {
        labels += QString("Leg:%1").arg(QString::number(ii));
        labels += QString("Age:%1").arg(QString::number(ii));
    }
    tableWidget->setHorizontalHeaderLabels(labels);

    // Process Each line of the file
    QTextStream in(&file);
    int ii = 0;
    int bad_cat = 0;
    QRegExp csv_split("\"?,\"?");
    while (!in.atEnd())
    {
         QString line = in.readLine();
         line = line.trimmed();             // Remove leading and training White Space
         line.remove(0xA0);                 // M$ special uglyness

         QStringList parts = splitCsvData(line);
         QString first = parts.value(0);
         bool ok;
         if ( first.isEmpty() )
             continue;
         int team = first.toInt(&ok);
         if ( ! ok || team <= 0 )
             continue;
         tableWidget->setRowCount(1+ii);

         // Insert Team number
         tableWidget->setItem(ii, 0, new QTableWidgetItem(first));
         parts.removeFirst();

         // Extract Team Name
         tableWidget->setItem(ii, 1, new QTableWidgetItem( parts.value(0) ) );
         parts.removeFirst();

         // Extract Team Cat
         QTableWidgetItem *item = new QTableWidgetItem( parts.value(0)) ;
         tableWidget->setItem(ii, 2, item );

        if (config.lookup_class(qPrintable(parts.value(0)) ) <= 0 )
        {
            item->setBackgroundColor(QColor(200,0,0,50));
            bad_cat++;
        }
        parts.removeFirst();

         int yy = 0;
         while ( parts.count() > 0)
         {
             // Name of competitor
             tableWidget->setItem(ii, 3+yy, new QTableWidgetItem( parts.value(0)));
             parts.removeFirst();

             // Posible age - if its a number
             int age = parts.value(0).toInt(&ok);
             if ( ok )
             {
                 if ( age > 0 )
                 {
                     tableWidget->setItem(ii, 4+yy, new QTableWidgetItem( parts.value(0)));
                 }
                 parts.removeFirst();
             }
             yy += 2;
         }
         ii++;
     }
    tableWidget->resizeColumnsToContents();

    // Report errors
    if (bad_cat)
    {
        MainWindow::showMessage("Invalid Categories in data");
    }

    // Connect up buttons

    connect (load, SIGNAL(clicked()), this, SLOT(loadData()));
    connect(cancel, SIGNAL(clicked()), this, SLOT(close()));
}

QmDialogLoadExternalTeams::~QmDialogLoadExternalTeams()
{

}

void QmDialogLoadExternalTeams::loadData(void)
{
    qDebug ("LoadData");
    team_type team_buf;
    int bad_cat = 0;
    for ( int ii = 0; ii < tableWidget->rowCount(); ii++)
    {
        if ( tableWidget->isRowHidden(ii))
        {
            continue;
        }
        bool bad = false;
        QTableWidgetItem *item;
        item = tableWidget->item(ii,0);
        if (item)
        {
            int team = item->data(Qt::EditRole).toInt();
            if ( team > 0 && team <= config.max_team )
            {
                g_record( team, &team_buf );

                // Name
                item = tableWidget->item(ii,1);
                if (item)
                {
                    strncpy(team_buf.name , qPrintable(item->data(Qt::EditRole).toString()), sizeof(team_buf.name));
                }

                // Category
                item = tableWidget->item(ii,2);
                if (item)
                {
                    int category = config.lookup_class(qPrintable(item->data(Qt::EditRole).toString()) );
                    if (category)
                    {
                        team_buf.teamclass = category;
                    }
                    else
                    {
                        bad_cat++;
                        bad = true;
                    }
                }

                // Team member names and ages
                int member = 0;
                for (int yy = 3; yy < tableWidget->columnCount(); yy+= 2, member++)
                {
                    if (member > config.num_legs)
                    {
                        break;
                    }
                    item = tableWidget->item(ii,yy);
                    if (item)
                    {
                        strncpy(team_buf.members[member].name , qPrintable(item->data(Qt::EditRole).toString()), sizeof(team_buf.members[member].name));
                    }
                    item = tableWidget->item(ii,1+yy);
                    if (item)
                    {
                        int age = item->data(Qt::EditRole).toInt();
                        if (age)
                        {
                            team_buf.members[member].age = age;
                        }
                    }
                }

                put_team_record( team, &team_buf );
            }
            else
            {
                bad = true;
            }
        }
        else
        {
            bad = true;
        }
        if (!bad)
        {
            tableWidget->hideRow(ii);
        }
    }

    // Report errors
    if (bad_cat)
    {
        MainWindow::showMessage("Invalid Categories in data");
    }
}

/*========================================================================
 *
 *  Generate team name file
 *
 *  Purpose:
 *      This function is called to Generate team name file in the format
 *      that can be read by the load command
 *
 *      The file contains team number,Team name,Team class
 *      The operator is prompted to enter the name of the file
 *
 *  Parameters:
 *      None
 *
 *  Returns:
 *      Nothing
 *
 *========================================================================*/
void QmDialogLoadExternalTeams::storeData(const QString &efile)
{
    QFile file(efile);
    if ( ! file.open(QIODevice::WriteOnly | QIODevice::Truncate | QIODevice::Text) )
    {
        MainWindow::showMessage("Cannot open external data file");
        return;
    }
    QTextStream out(&file);

    // Print headings
    out << toCsv("Team Number");
    out << "," << toCsv("Team Name");
    out << "," <<  toCsv("Class Abr");

    for( int j = 1; j <= config.num_legs; j++ )
    {
        out << "," << toCsv("Competitor Name");
        out << "," << toCsv("Age");
    }
    out << endl;

    /*
     * Put the data into the file
     */
    team_type   team_buf;
    for(int i = config.min_team; i <= config.max_team; i++ )
    {
        if( valid_field( i ) && g_record( i, &team_buf ) )
        {
            /*
            **  Basic information
            **      - Team number
            **      - Full team name
            */
            out << toCsv(team_buf.numb);
            out << "," << toCsv(team_buf.name);
            out << "," << toCsv(team_buf.teamclass == 0 ? "" : config.team_class[team_buf.teamclass - 1].abr);

            for(int j = 1; j <= config.num_legs; j++ )
            {
                out << "," << toCsv(team_buf.members[j-1].name);
                out << "," << toCsv(team_buf.members[j-1].age);
            }
            out <<endl;
        }
    }
}

QStringList QmDialogLoadExternalTeams::splitCsvData( const QString str)
{
    QStringList results;

    const QChar *data = str.constData();
    while (!data->isNull())
    {
        QString result;
        bool quoted = false;
        /*
        **  Extract the next record
        */
        while ( TRUE )
        {
            QChar uch = *data;

            /*
            **  End of the field
            */
            if ( uch == '\n' || uch == '\r' || uch == '\0' )
                break;

            data++;

            /*
            ** Ugly character from MS CSV files
            */
            if ( uch == (char) 0xA0 )
            {
                continue;
            }

            if ( !quoted && uch == ',' )
            {
                break;
            }

            /*
            **  An unquoted " will start scanning for a matching quote
            */
            if ( !quoted && uch == '"' )
            {
                quoted = true;
                continue;
            }

            /*
            **  A quoted " may be an embedded quote or the end of a quote
            */
            if ( quoted && uch == '"' )
            {
                if ( *data != '"' )
                {
                    quoted = FALSE;
                    continue;
                }

                /*
                **  Skip one " and pick up the next
                */
                ++data;
            }

            /*
            **  Save this character
            */
            result += uch;
        }

        /*
        **  Clean up the extracted string
        */
        results += result.trimmed();
     }
    return results;
}

/*========================================================================
 *
 *  Generate team name file
 *
 *  Purpose:
 *      This function is called to Generate team name file
 *
 *      The file contains team number,Team name,Team class
 *      The operator is prompted to enter the name of the file
 *
 *  Parameters:
 *      None
 *
 *  Returns:
 *      Nothing
 *
 *========================================================================*/

void QmDialogLoadExternalTeams::storeTeamInfo(const QString &efile)
{
    QFile file(efile);
    if ( ! file.open(QIODevice::WriteOnly | QIODevice::Truncate | QIODevice::Text) )
    {
        MainWindow::showMessage("Cannot open external team info file");
        return;
    }
    QTextStream out(&file);

     /*
     * Put the data into the file
     */
    team_type   team_buf;
    for(int i = config.min_team; i <= config.max_team; i++ )
    {
        if( valid_field( i ) && g_record( i, &team_buf ) )
        {
            /*
            **  Basic information
            **      - Team number
            **      - Full team name
            **      - Category
            */
            out.setFieldAlignment(QTextStream::AlignLeft);
            out.setFieldWidth(5);
            out << team_buf.numb;
            out.setFieldWidth(0);
            out << ",";
            out.setFieldWidth(MAX_TM_NAME+1);
            out << team_buf.name;
            out.setFieldWidth(0);
            out << ",";
            out << (team_buf.teamclass <= 0 ? "" : config.team_class[team_buf.teamclass - 1].abr);
            out <<endl;
        }
    }
}

QString QmDialogLoadExternalTeams::toCsv(const QString &str)
{
    QString result = QString(str);
    if ( result.contains("\"") || result.contains(",") )
    {
        result.replace("\"", "\"\"");
        result.prepend("\"");
        result.append("\"");
    }
    return result;
}

QString QmDialogLoadExternalTeams::toCsv(const int data)
{
    return QString::number(data);
}