#include <QLabel>

#include <QListWidget>

#include <QListWidgetItem>

// left column

    // declare start button, caption and description

    btnStart = formattedButton(":/res/Trainings.png", true);

    btnStart->setToolTip(QPushButton::tr("Go!"));

    lblCaption = new QLabel(this);

    lblDescription = new QLabel(this);

    lblDescription->setWordWrap(true);

    // add start button, caption and description to 3 different rows

    pageLayout->addWidget(btnStart, 0, 0);

    pageLayout->addWidget(lblCaption, 1, 0);

    pageLayout->addWidget(lblDescription, 2, 0);

    // make first and last row stretch vertically

    pageLayout->setRowStretch(1, 0);

    // make both columns equal width

    pageLayout->setColumnStretch(0, 1);

    pageLayout->setColumnStretch(1, 1);

    // center widgets within their grid cells

    pageLayout->setAlignment(btnStart, Qt::AlignHCenter | Qt::AlignVCenter);

    pageLayout->setAlignment(lblCaption, Qt::AlignHCenter | Qt::AlignVCenter);

    pageLayout->setAlignment(lblDescription, Qt::AlignHCenter | Qt::AlignVCenter);

// right column

    lstMissions = new QListWidget(this);

    pageLayout->addWidget(lstMissions, 0, 1, 3, 1); // spans over 3 rows

    connect(lstMissions, SIGNAL(itemSelectionChanged()), this, SLOT(updateInfo()));

    connect(lstMissions, SIGNAL(itemDoubleClicked(QListWidgetItem *)), this, SLOT(startSelected()));

    connect(btnStart, SIGNAL(clicked()), this, SLOT(startSelected()));

    QStringList missionList = scriptList(tmpdir);

    missionList.sort();

    missionList.replaceInStrings(QRegExp("$")," *");

    QStringList defaultList = scriptList(tmpdir);

    defaultList.sort();

    missionList << defaultList;

    foreach (const QString & mission, missionList)

        QListWidgetItem * item = new QListWidgetItem(mission);

        // replace underscores in mission name with spaces

        item->setText(item->text().replace("_", " "));

        // store original name in data

        item->setData(Qt::UserRole, mission);

        lstMissions->addItem(item);

    updateInfo();

void PageTraining::startSelected()

{

    emit startMission(lstMissions->currentItem()->data(Qt::UserRole).toString());

}

void PageTraining::updateInfo()

{

    if (lstMissions->currentItem())

    {

        QString thumbFile = datadir->absolutePath() + "/Graphics/Missions/Training/" + lstMissions->currentItem()->data(Qt::UserRole).toString() + ".png";

        if (QFile::exists(thumbFile))

            btnStart->setIcon(QIcon(thumbFile));

        else

            btnStart->setIcon(QIcon(":/res/Trainings.png"));

        lblCaption->setText(lstMissions->currentItem()->text());

        // TODO load mission description from file

        lblDescription->setText("< Imagine\nMission\nDescription\nhere >\n\nThank you.");

    }

    else

    {

        btnStart->setIcon(QIcon(":/res/Trainings.png"));

        lblCaption->setText(tr("Select a mission on the right -->"));

        // TODO better text and tr()

        lblDescription->setText("Welcome to the Training screen.\n\n\n...\nWHAT?\nIt's not finished yet...");

    }

}