mainwindow.cpp

/** ===========================================================
 * @file
 *
 * This file is a part of pcro project
 * <a href="https://github.com/maheshmhegade/pcro">https://github.com/maheshmhegade/pcro</a>
 *
 * @date    2010-03-03
 * @brief   pcro using pc as signal generator.
 * @section DESCRIPTION
 *
 * Using ALSA to output digital data as analogue sound which is standard a mathematical function.
 *
 * @author Copyright (C) 2012-2013 by Mahesh Hegde
 *         <a href="mailto:maheshmhegade at gmail dot com">maheshmhegade at gmail dot com</a>
 *
 * @section LICENSE
 *
 * This program is free software; you can redistribute it
 * and/or modify it under the terms of the GNU General
 * Public License as published by the Free Software Foundation;
 * either version 2, or (at your option)
 * any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * ============================================================ */

#include "mainwindow.h"

MainWindow::MainWindow(QWidget *parent) :
    QMainWindow(parent),
    ui(new Ui::MainWindow)
{
    liveVideoObject = new liveVideo();
    facemodeltostore = 0;
    allwaveObject = new alsaSoundcard();
    ui->setupUi(this);
    ui->FaceGV->hide();
    wave = new outputWave();

    //enable this option to start voice recognition on starting the application

    //QtConcurrent::run(this,&MainWindow::recognizeVoice);

    connect(this,SIGNAL(allValuesSet(VoiceRecognition *)),this,SLOT(setWaveValues(VoiceRecognition *)));
    connect(this,SIGNAL(plotAndPlayNow()),this,SLOT(plotAndPlay()));
}

MainWindow::~MainWindow()
{
    delete wave;
    delete allwaveObject;
    delete ui;
}

void MainWindow::on_applypushButton_clicked()
{
    generateWave();

    //run plotting and playing tasks in seperate threads to display waveform quickly
    //TODO:find a way to stop playing sound,whenever needed.
    QtConcurrent::run(this,&MainWindow::play_sound);
    QtConcurrent::run(this,&MainWindow::plotWave);
}

void MainWindow::plotWave()
{
    QVector<double> x(5000), y(5000);
    ui->widget->addGraph();
    // create graph and assign data to it:
    // give the axes some labels:
    ui->widget->xAxis->setLabel("x");
    ui->widget->yAxis->setLabel("y");
    // set axes ranges, so we see all data:
    ui->widget->xAxis->setRange(0, 5000);
    ui->widget->yAxis->setRange(-35000,35000);
    for (int i=0; i<5000; ++i)
    {
        x[i] = (double)i; // x goes from -1 to 1
        y[i] = wave->waveSamples[i];//*x[i]; // let's plot a quadratic function
    }
    // create graph and assign data to it:
    ui->widget->addGraph();
    ui->widget->graph(0)->setData(x,y);
    ui->widget->replot();
}

void MainWindow::play_sound()
{
    allwaveObject = new alsaSoundcard(); //= alsaSoundcard();
    allwaveObject->playBack(wave);
}

void MainWindow::setWaveValues(VoiceRecognition *voiceRecognizer)
{
    /*
    ui->wavecomboBox->setCurrentIndex(voiceRecognizer->waveType);
    ui->voltagelineEdit->setText(QString(QString::number(voiceRecognizer->waveVoltage)));
    ui->frequencylineEdit->setText(QString(QString::number(voiceRecognizer->waveFrequency)));
    ui->durationlineEdit->setText(QString(QString::number(voiceRecognizer->waveDuration)));
     */
    //transfer handle to plotAndPlay()
    emit plotAndPlayNow();
}

void MainWindow::generateWave()
{
    wave->samplingFrequency = 44100;
    wave->waveDuration = ui->durationlineEdit->text().toInt();
    wave->waveFrequency = ui->frequencylineEdit->text().toInt();
    wave->waveAmplitude = ui->voltagelineEdit->text().toFloat();
    switch (ui->wavecomboBox->currentIndex())
    {
    case 0:
    {
        allwaveObject->generateSin(wave);
        break;
    }
    case 1:
    {
        allwaveObject->generateCos(wave);
        break;
    }
    case 2:
    {
        allwaveObject->generateTriangular(wave);
        break;
    }
    case 3:
    {
        allwaveObject->generateSquare(wave);
        break;
    }
    case 4:
    {
        allwaveObject->generateRamp(wave);
        break;
    }
    }
    QVector<double> x(5000), y(5000);
    ui->widget->addGraph();
    // create graph and assign data to it:
    // give the axes some labels:
    ui->widget->xAxis->setLabel("x");
    ui->widget->yAxis->setLabel("y");
    // set axes ranges, so we see all data:
    ui->widget->xAxis->setRange(0, 5000);
    ui->widget->yAxis->setRange(-35000,35000);
    for (int i=0; i<5000; ++i)
    {
        x[i] = (double)i; // x goes from -1 to 1
        y[i] = wave->waveSamples[i];//*x[i]; // let's plot a quadratic function
    }
    // create graph and assign data to it:
    ui->widget->addGraph();
    ui->widget->graph(0)->setData(x,y);
    ui->widget->replot();

    cout << "one" << endl;
    return;
}

void MainWindow::plotAndPlay()
{
    generateWave();

    //execute plotting and playing sound in seperate threads
    QtConcurrent::run(this,&MainWindow::play_sound);
    QtConcurrent::run(this,&MainWindow::plotWave);
}

void MainWindow::recognizeVoice(QString personNmae)
{
    VoiceRecognition *recognizer = new VoiceRecognition();

    //select acoustic model belonging to perticular person by supplying person name in QString form
    recognizer->SelectAcousticModel(personNmae);
    recognizer->startVoiceRecognition(ui);

    //tranfser handle to setWaveValues()
    emit allValuesSet(recognizer);
}

void MainWindow::on_SaveFacePB_clicked()
{
    Tlddatabase *tlddatabase = new Tlddatabase;
    facemodeltostore = faceData.first;
    facemodeltostore->Name = ui->FaceNameLE->text();
    cout << qPrintable(facemodeltostore->Name) << endl;
    tlddatabase->insertFaceModel(facemodeltostore);
    ui->FaceNameLE->clear();
    ui->FaceGV->hide();
    delete tlddatabase;
}

void MainWindow::on_DetectFacePB_clicked()
{
    Tldrecognition* const tmpTLD          = new Tldrecognition;
    faceData = tmpTLD->getModeltoStore();
    //facemodeltostore = faceData.first;
    ui->FaceGV->show();
    liveVideoObject->displayVideo(faceData.second,ui->FaceGV);
    delete tmpTLD;
}

void MainWindow::on_RecognizeFacePB_clicked()
{
    liveVideoObject->togglePlayStatus();

    if(liveVideoObject->playStatus == true)
    {
        Tlddatabase *tlddatabase = new Tlddatabase();

        QList<unitFaceModel*> comparemodels;

        for (int i = 1; i <= tlddatabase->queryNumfacesinDatabase();i++ )
        {
            comparemodels.push_back(tlddatabase->getFaceModel(i));
        }
        delete tlddatabase;

        Tldrecognition* const tmpTLD      = new Tldrecognition;
        std::pair<IplImage*,QString> faceName = tmpTLD->getRecognitionConfidence(comparemodels);
        ui->FaceGV->show();
        ui->FaceNameLE->clear();
        ui->FaceNameLE->setText(faceName.second);
        liveVideoObject->displayVideo(faceName.first,ui->FaceGV);
        qDebug() << "person "<< faceName.second << " recognized in front of the system";
        delete tmpTLD;
    }
    QtConcurrent::run(this,&MainWindow::recognizeVoice,(QString)"mylm");
}