setcolorrange.cpp

#include "setcolorrange.h"
#include "ui_setcolorrange.h"
#include <QFile>
#include <QTextStream>
#include <QString>
#include <iostream>
#include <sstream>
#include <fstream>

using namespace std;

SetColorRange::SetColorRange(QWidget *parent) :
    QWidget(parent),
    ui(new Ui::SetColorRange)
{
    eye = new Vision;
    ui->setupUi(this);


    //setAttribute(Qt::WA_DeleteOnClose);
    connect(ui->horizontalSlider, SIGNAL(valueChanged(int)), this, SLOT(on_horizontalSlider_sliderMoved(int)));
    connect(ui->horizontalSlider_2, SIGNAL(valueChanged(int)), this, SLOT(on_horizontalSlider_2_sliderMoved(int)));
    connect(ui->horizontalSlider_3, SIGNAL(valueChanged(int)), this, SLOT(on_horizontalSlider_3_sliderMoved(int)));
    connect(ui->horizontalSlider_4, SIGNAL(valueChanged(int)), this, SLOT(on_horizontalSlider_4_sliderMoved(int)));
    connect(ui->horizontalSlider_5, SIGNAL(valueChanged(int)), this, SLOT(on_horizontalSlider_5_sliderMoved(int)));
    connect(ui->horizontalSlider_6, SIGNAL(valueChanged(int)), this, SLOT(on_horizontalSlider_6_sliderMoved(int)));
    connect(ui->pushButton_2, SIGNAL(clicked(bool)), this, SLOT(on_pushButton_2_clicked()));
    connect(eye, SIGNAL(processedImage(QImage)), this, SLOT(updateVisionUI(QImage)));
    eye->set_mode(1);
}

void SetColorRange::showEvent(QShowEvent *event){
    QWidget::showEvent(event);

    string file = "Config/" + this->subject;
    fstream input(file, fstream::in);
    int low[3], upper[3];

    if(!input){
        ui->horizontalSlider->setValue(0);
        ui->horizontalSlider_2->setValue(0);
        ui->horizontalSlider_3->setValue(0);
        ui->horizontalSlider_4->setValue(255);
        ui->horizontalSlider_5->setValue(255);
        ui->horizontalSlider_6->setValue(255);
        clog << "Values set to default." << endl;
        return;
    }

    input >> low[0] >> low[1] >> low[2];
    input >> upper[0] >> upper[1] >> upper[2];

    input.close();

    ui->horizontalSlider->setValue(low[0]);
    ui->horizontalSlider_2->setValue(low[1]);
    ui->horizontalSlider_3->setValue(low[2]);
    ui->horizontalSlider_4->setValue(upper[0]);
    ui->horizontalSlider_5->setValue(upper[1]);
    ui->horizontalSlider_6->setValue(upper[2]);

    if(eye == NULL){
        eye = new Vision;
        eye->set_mode(1);
    }

    on_pushButton_2_clicked();
}

void SetColorRange::closeEvent(QCloseEvent *event){
    QWidget::closeEvent(event);

    if(!eye->isStopped()){
        eye->Stop();
        eye->wait();
        eye->release_cam();
    }
}

void SetColorRange::updateVisionUI(QImage img)
{
    if(!img.isNull()){
        ui->label_7->setAlignment(Qt::AlignCenter);
        ui->label_7->setPixmap(QPixmap::fromImage(img).scaled(ui->label_7->size(), Qt::KeepAspectRatio, Qt::FastTransformation));
    }
}

void SetColorRange::set_vision(Vision *eye, int cam_id)
{
    if(!eye->is_open()){
        eye->open_camera(cam_id);
    }
    this->eye = eye;
}

void SetColorRange::on_horizontalSlider_sliderMoved(int position)
{
    vector<int> low = eye->get_low();
    low[0] = position;
    eye->set_low(low);
    ui->lcdNumber->display(position);
}

void SetColorRange::on_horizontalSlider_2_sliderMoved(int position)
{
    vector<int> low = eye->get_low();
    low[1] = position;
    eye->set_low(low);
    ui->lcdNumber_2->display(position);
}

void SetColorRange::on_horizontalSlider_3_sliderMoved(int position)
{
    vector<int> low = eye->get_low();
    low[2] = position;
    eye->set_low(low);
    ui->lcdNumber_3->display(position);
}

void SetColorRange::on_horizontalSlider_4_sliderMoved(int position)
{
    vector<int> upper = eye->get_upper();
    upper[0] = position;
    eye->set_upper(upper);
    ui->lcdNumber_4->display(position);
}

void SetColorRange::on_horizontalSlider_5_sliderMoved(int position)
{
    vector<int> upper = eye->get_upper();
    upper[1] = position;
    eye->set_upper(upper);
    ui->lcdNumber_5->display(position);
}

void SetColorRange::on_horizontalSlider_6_sliderMoved(int position)
{
    vector<int> upper = eye->get_upper();
    upper[2] = position;
    eye->set_upper(upper);
    ui->lcdNumber_6->display(position);
}

void SetColorRange::on_pushButton_2_clicked()
{
    std::cout << cam_id << std::endl;
    if(!eye->open_camera(cam_id)){
        QMessageBox msgBox;
        msgBox.setText("The camera could not be opened!");
        msgBox.exec();
    }
    eye->Play();
}

void SetColorRange::set_camid(int cam_id){
    this->cam_id = cam_id;
}

void SetColorRange::set_subject(string subject)
{
    this->subject = subject;
}

SetColorRange::~SetColorRange()
{
    delete eye;
}

void SetColorRange::on_pushButton_clicked()
{
    int i = 0;
    string line, path = "Config/" + subject;
    stringstream ss;
    fstream file;

    file.open(path.c_str(), fstream::in);

    if(!file){
        cout << "File not open (1)" << endl;
    }

    for(int i = 0; i < 3; ++i){
        ss << eye->get_low()[i] << " ";
    }

    ss << endl;

    for(int i = 0; i < 3; ++i){
        ss << eye->get_upper()[i] << " ";
    }

    ss << endl;

    while(getline(file, line)){
        if(i >= 2){
            ss << line << endl;
        }

        i++;
    }

    file.close();
    file.clear();

    file.open(path.c_str(), fstream::out);

    if(!file){
        cout << "File not open (2)" << endl;
    }

    file << ss.str();

    file.close();
}