examples/Qt/facerecgapp/mainwindow.cpp

/****************************************************************************
**
** Copyright (C) 2011 Nokia Corporation and/or its subsidiary(-ies).
** All rights reserved.
** Contact: Nokia Corporation (qt-info@nokia.com)
**
** This file is part of the examples of the Qt Toolkit.
**
** $QT_BEGIN_LICENSE:BSD$
** You may use this file under the terms of the BSD license as follows:
**
** "Redistribution and use in source and binary forms, with or without
** modification, are permitted provided that the following conditions are
** met:
**   * Redistributions of source code must retain the above copyright
**     notice, this list of conditions and the following disclaimer.
**   * Redistributions in binary form must reproduce the above copyright
**     notice, this list of conditions and the following disclaimer in
**     the documentation and/or other materials provided with the
**     distribution.
**   * Neither the name of Nokia Corporation and its Subsidiary(-ies) nor
**     the names of its contributors may be used to endorse or promote
**     products derived from this software without specific prior written
**     permission.
**
** THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
** "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
** LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
** A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
** OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
** SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
** LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
** DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
** THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
** (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
** OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE."
** $QT_END_LICENSE$
**
****************************************************************************/

#include <QDateTime>
#include "mainwindow.h"

MainWindow::MainWindow()
{
    infoLabel = new QLabel(tr("<i>Choose a menu option</i>"));
    infoLabel->setAlignment(Qt::AlignCenter);

    imageLabel = new QLabel();
    
    cameraViewfinder = new QCameraViewfinder(this);
    layout = new QVBoxLayout;
    layout->addWidget(infoLabel);
    QWidget *widget = new QWidget;
    setCentralWidget(widget);
    widget->setLayout(layout);
    createActions();
    createMenus();
    setWindowTitle(tr("Face Recogniser"));
    // Initialize the face recogniser, sets up the camera and timer.
    faceRecgApp = new FaceRecgApp(this);
    setupCamera();
    timer = new QTimer(this);
    timer->setInterval(3000);
    // Defines the connection for recogniser and the timer.
    connect(timer, SIGNAL(timeout()), this, SLOT(timerTimeout()));
    connect(faceRecgApp,SIGNAL(resultCompared(bool)),this, SLOT(comparedResult(bool)));
}
void MainWindow::trainwImg()
{
    camera->stop();
    layout->removeWidget(cameraViewfinder);
    layout->addWidget(infoLabel);
    layout->addWidget(imageLabel);
    imageLabel->clear();
    infoLabel->setText(tr("Invoked <b>Training</b>"));
    openImage();
    if(!fileName.isEmpty())
        faceRecgApp->train(fileName);
}

void MainWindow::recogniseUser()
{
    imageLabel->clear();
    imageLabel->adjustSize();
    layout->removeWidget(imageLabel);
    layout->removeWidget(infoLabel);
    infoLabel->setText(tr("Invoked <b>Recognise User </b>"));
    layout->addWidget(cameraViewfinder);
    layout->addWidget(infoLabel);
    camera->start();//starting the camera
    timer->start();
    setEnabled(false);
}


void MainWindow::timerTimeout()
{
    QDateTime now = QDateTime::currentDateTime();
    QString name = now.toString("hhmmss");
    capImgName = "e:/capImgL2/";
    capImgName.append(name);
    capImgName.append(".jpg");          
    camera->searchAndLock();
    imageCapture->capture(capImgName);
}

void MainWindow::createActions()
{
    trainAct = new QAction(tr("&Train"), this);
    trainAct->setStatusTip(tr("Traing with a image"));
    connect(trainAct, SIGNAL(triggered()), this, SLOT(trainwImg()));

    recogniseAct= new QAction(tr("&Recognise"), this);
    recogniseAct->setStatusTip(tr("Recognise the User"));
    connect(recogniseAct, SIGNAL(triggered()), this, SLOT(recogniseUser()));

    resetAct= new QAction(tr("Re&set"), this);
    resetAct->setStatusTip(tr("Reset the recogniser"));
    connect(resetAct, SIGNAL(triggered()), this, SLOT(reset()));

    exitAct = new QAction(tr("E&xit"), this);
    exitAct->setShortcuts(QKeySequence::Quit);
    exitAct->setStatusTip(tr("Exit the application"));
    connect(exitAct, SIGNAL(triggered()), this, SLOT(close()));
}

void MainWindow::createMenus()
{
    fileMenu = menuBar()->addMenu(tr("Face Rec App"));
    fileMenu->addAction(trainAct);
    fileMenu->addAction(recogniseAct);
    fileMenu->addAction(resetAct);
    fileMenu->addAction(exitAct);
}

void MainWindow::openImage()
{
    QString selectedFilter;
    QString filePath = QFileDialog::getOpenFileName(this,
                                   tr("Select the image"),
                                   QDir::homePath(),
                                   tr("Images (*.png *.bmp *.jpg)"),
                                   &selectedFilter,
                                   QFileDialog::DontUseNativeDialog );
   if (!filePath.isEmpty())
       {
       QImageReader reader(filePath);
       reader.setScaledSize(QSize(imageLabel->width(),imageLabel->height()));
       QImage image = reader.read();
       if (image.isNull()) {
           QMessageBox::information(this, tr("Image Viewer"),
                                    tr("Cannot load %1.").arg(filePath));
           qWarning("bad argument, Try again");
           return;
       }
       fileName = filePath;
       imageLabel->setPixmap(QPixmap::fromImage(image,Qt::AutoColor));
       infoLabel->setText("Training the recogniser...");
   }
   else
       {
       fileName = "";
       }
}
void MainWindow::setupCamera()
{

    //Selecting the secondary camera by passing argument value as 1.
    cameraDevice = QCamera::availableDevices()[1];
    camera = new QCamera(cameraDevice,this);

    camera->setViewfinder(cameraViewfinder);
    imageCapture = new QCameraImageCapture(camera);
    camera->setCaptureMode(QCamera::CaptureStillImage);

    connect(imageCapture,SIGNAL(imageSaved(int,QString)), this, SLOT(imageSaved(int,QString)));    
    if (camera->state() == QCamera::ActiveState) {
        camera->stop();
    }
}
void MainWindow::imageSaved(int id, const QString imgName)
{
    infoLabel->setText("Face recognition in progress...");
    faceRecgApp->test(imgName);    
}

void MainWindow::reset()
{
    infoLabel->setText("Face recogniser is reset.");
    camera->stop();
    layout->removeWidget(cameraViewfinder);  
    layout->addWidget(infoLabel);
    imageLabel->clear();
    layout->removeWidget(imageLabel);
    faceRecgApp->reset();    
}

void MainWindow::comparedResult(bool status)
{
    QMessageBox msgBox;
    if(status)
        {
        camera->stop();
        timer->stop();
        faceRecgApp->reset(); // reset the reecogniser.
        msgBox.setText("Face is recognised as same as the one trained");
        setEnabled(true); // enable the mainwindow.
        msgBox.setModal(true);
        msgBox.exec();
        close();
        }
    else
        {
        timer->stop();
        msgBox.setText("Face is different from the trained face. Try Again!!!");
        msgBox.setModal(true);        
        msgBox.exec();
        timer->start();
        }
}

---