MainWindow.cpp

#include <QtDebug>

#include "MainWindow.h"
#include "TreeModel.h"
#include <qtoolbutton.h>
#include <qpushbutton.h>

#include <stdio.h>    //FILE, fopen(), fclose()
#include <stdlib.h>
#include <sys/stat.h> //stat, stat()
#include <string>     //string
#include <fstream>    //fstream 
#include <iostream>   //cout

MainWindow::MainWindow() {

	// stuff for QSettings
	QCoreApplication::setOrganizationName("MINC");
	QCoreApplication::setApplicationName("brain-view");
	
	QSettings settings;
//	qDebug() << "Debug. [ MainWindow() ] Settings path 1: " << settings.fileName();
	QSettings set2(QSettings::IniFormat, QSettings::UserScope, "MINC", "brain-view");
//	qDebug() << "Debug. [ MainWindow() ] Settings path 2: " << set2.fileName();
//	qDebug() << "Debug. [ MainWindow() ] Application Pid: " << QCoreApplication::applicationPid();
	
	createActions();
	createMenus();
	initColourbars();

	viewer = new BrainQuarter();
	//colourBar = new ColourBarForm(this);
	//colourBarDock = new QDockWidget(tr("Colour Bar"), this);
	//addDockWidget(Qt::LeftDockWidgetArea, colourBarDock);
	//colourBarDock->setWidget(colourBar);

	resourceForm = new ResourceForm(this, viewer);
	resourceDock = new QDockWidget(tr("Resources"), this);
	addDockWidget(Qt::RightDockWidgetArea, resourceDock);
	resourceDock->setWidget(resourceForm);
	
// 	tagForm = new TagForm(this, viewer);
// 	tagDock = new QDockWidget(tr("Tags"), this);
// 	addDockWidget(Qt::BottomDockWidgetArea, tagDock);
// 	tagDock->setWidget(tagForm);
// 	resourceDock->addWidget(tagForm);
	
	
	setCentralWidget(viewer);


}


void MainWindow::initColourbars() {

    // get the path from the settings
    QSettings settings(QSettings::IniFormat, QSettings::UserScope, "MINC", "brain-view");
    QString defaultDir = QString(QFileInfo(settings.fileName()).path()).append("/textures/");
    QString textureDir = settings.value("textureDir", defaultDir).toString();
    settings.setValue("textureDir", textureDir);
  //  qDebug() << "Debug. [ initColourbars() ] TextureDir: " << textureDir;

    // create the texture dir if it does not already exist
    QDir td(textureDir);
    if (! td.exists()) {
        td.mkpath(textureDir);
    }

    // for every colourbar in the resource file check to see if it
    // already exists. If not, write it out to file
	QDirIterator it(":/resources/colourbars");
	while ( it.hasNext() ) {
		QString fileNameFull = it.next();
    //    qDebug() << "Debug. [ initColourbars() ] Texture file full path: " << fileNameFull;
		QFileInfo fileInfo = it.fileInfo();
        QString fileName = fileInfo.baseName();
      //  qDebug() << "Debug. [ initColourbars() ] Texture file basename: " << fileName;
        fileName.append(".png");
        fileName.prepend("/");
        fileName.prepend(textureDir);
        if (! QFile(fileName).exists()) {
        //    qDebug() << "Debug. [ initColourbars() ] Adding texture to local texture dir: " << fileInfo.filePath();
            QImage i(fileInfo.filePath());
            bool a = i.isNull();
            bool b = i.save(fileName);
          //  qDebug() << "Debug. [ initColourbars() ] Created colourbar: " << fileName << " " << a << " " << b;
        }
	}

}


MainWindow::~MainWindow() {
}


bool MainWindow::openFile(QFile &filename) {

    QFileInfo fileInfo(filename);	
    QString ext = fileInfo.suffix();

	bool status = false;
    if (ext == "obj" || ext == "h5") {
		status = resourceForm->insertGeometry(filename);
	}
    else if (ext == "txt" || ext == "vertstats") {
        status = resourceForm->insertVertstats(filename);
    }
    else if (ext == "tag") {
		status = resourceForm->insertTagfile(filename);
    }
    else if (ext == "config") {
		status = resourceForm->insertLabelLUfile(filename);
    }

	Q_ASSERT_X( status == true, "MainWindow::openFile", "Failed insert into Resource form" );

	return true;
}

void MainWindow::openFileFromDialog() {
	QString name = QFileDialog::getOpenFileName(this,
                tr("Open File"), QString(), tr("Files (*.obj *.txt *.vertstats *.tag *.h5 *.config)"));
	if (! name.isNull()) {
		QFile filename(name);
		openFile(filename);
	}
}


/*void MainWindow::setTagOptionFromDialog(int tagopt){
        qDebug() << "Tag option : " << tagopt;
	//viewer->setTagOpt(tagopt);
}
*/

void MainWindow::createActions() {

	openAction = new QAction(tr("&Open File"), this);
	openAction->setShortcuts(QKeySequence::Open);
	connect(openAction, SIGNAL(triggered()), this, SLOT(openFileFromDialog()));

	saveTagAction = new QAction(tr("&Save Tags to File"), this);
	saveTagAction->setShortcuts(QKeySequence::Save);
	connect(saveTagAction, SIGNAL(triggered()), this, SLOT(saveTagFromDialog()));
	
	quitAction = new QAction(tr("&Quit"), this);
	quitAction->setShortcut(tr("Ctrl+Q"));
	connect(quitAction, SIGNAL(triggered()), this, SLOT(quitApplication()));


	tagGroup = new QActionGroup(this);
	surfaceAction = new QAction(tr("&Top Surface"), this);
	tagGroup->addAction(surfaceAction);
	surfaceAction->setCheckable(TRUE);
	//surfaceAction->setChecked(TRUE);
	//surfaceAction->isExclusive(TRUE);
	connect(surfaceAction, SIGNAL(triggered()), this, SLOT(setTagOptionFromDialog0()));

	midpointAction = new QAction(tr("&MidPoint"), this);
	tagGroup->addAction(midpointAction);
	midpointAction->setCheckable(TRUE);
	midpointAction->setChecked(true);
	connect(midpointAction, SIGNAL(triggered()), this, SLOT(setTagOptionFromDialog1()));
	
	backsurfaceAction = new QAction(tr("&Back Surface"), this);
	tagGroup->addAction(backsurfaceAction);
	backsurfaceAction->setCheckable(TRUE);
	connect(backsurfaceAction, SIGNAL(triggered()), this, SLOT(setTagOptionFromDialog2()));

	
	
	verboseAction = new QAction(tr("&Verbose"), this);
	verboseAction->setShortcut(tr("Ctrl+V"));
	//verboseAction->isCheckable ();
	verboseAction->setCheckable(TRUE);
	verboseAction->setChecked(false);
	//connect(verboseAction, SIGNAL(triggered()), this, SLOT(verboseApplication()));

}


void MainWindow::createMenus() {

	fileMenu = menuBar()->addMenu(tr("&File"));
	fileMenu->addAction(openAction);
	fileMenu->addAction(saveTagAction);
	fileMenu->addAction(quitAction);

	tagMenu = menuBar()->addMenu(tr("&Tag"));
	tagMenu->addAction(surfaceAction);
	tagMenu->addAction(midpointAction);
	tagMenu->addAction(backsurfaceAction);

	outMenu = menuBar()->addMenu(tr("&Output"));
	outMenu->addAction(verboseAction);

}



void MainWindow::createTagFunc(){
	Tagpoint = (float*) malloc(sizeof(float) * 3);
	Tagpoint = resourceForm->addTagpoint();
	Tagsize = resourceForm->returnTagsize();
	if ( verboseAction->isChecked() )
		qDebug() << "Created tag : " << Tagpoint[0] <<" , " << Tagpoint[1] << " , " << Tagpoint[2] ;	
}

void MainWindow::addTagFunc(){
// 	Tagpoint = (float*) malloc(sizeof(float) * 3);
// 	Tagpoint = resourceForm->addTagpoint();
		
	//remove any duplication of tag points:
	bool tagfound=false;
	for (int i=0; i< (tagsVect.size())/4; i++){
		if (tagsVect[4*i]==Tagpoint[0] && tagsVect[4*i+1]==Tagpoint[1] && tagsVect[4*i+2]==Tagpoint[2] && tagsVect[4*i+3]==Tagsize){			
			tagfound=true;
			break;
		}
	}
	if (!tagfound){
		tagsVect.append(Tagpoint[0]);	//Tag(x)
		tagsVect.append(Tagpoint[1]);	//Tag(y)
		tagsVect.append(Tagpoint[2]);	//Tag(z)
		tagsVect.append(Tagsize);	//Tagsize
		if ( verboseAction->isChecked() )
			qDebug() << "Added tag : " << Tagpoint[0] <<" , " << Tagpoint[1] << " , " << Tagpoint[2] ;	
	}
	
}



void MainWindow::saveTagFromDialog() {
	QString sname = QFileDialog::getSaveFileName(this,
                tr("Save Tags to File"), QString(), tr("Files (*.tag)"));
	if (! sname.isNull()) {
		QFile sfilename(sname);
		if ( verboseAction->isChecked() )
			qDebug() << "construct a file " << sfilename.fileName().toLatin1().data();
		saveTagFileFunc(sfilename);

	}
	
}

void MainWindow::saveTagFileFunc(QFile &filename){
	//create an empty file filename and save all the created tags in the file in format of .tag for follow_tree.py
// 	filename.open(IO_WriteOnly, stderr );

	FILE *pFile;
	pFile = fopen (filename.fileName().toLatin1().data(), "w");
	if (pFile == NULL)
	{
		qDebug() <<"ERROR! No file for saving the tags!" ;
		return;
	}
	else
	{
		fprintf (pFile, "MNI Tag Point File\nVolumes = 1;\n\nPoints =\n");
		int tagnum= (tagsVect.size())/4;	//number of tags (x,y,z,size)
		
		if ( verboseAction->isChecked() )
			qDebug() << tagnum <<"tags created!";
		

		for (int i=0; i<(tagnum-1) ; i++){
			fprintf (pFile, "%f %f %f %f 1 1 \"Marker\"\n", tagsVect[4*i], tagsVect[4*i+1],tagsVect[4*i+2],tagsVect[4*i+3] );
		}
		fprintf (pFile, "%f %f %f %f 1 1 \"Marker\";", tagsVect[4*(tagnum-1)], tagsVect[4*(tagnum-1)+1],tagsVect[4*(tagnum-1)+2],tagsVect[4*(tagnum-1)+3]);
		fclose(pFile);
		//tagsVect.clear();
	}
}

void MainWindow::quitApplication() {
	// code to prompt for exit, save stuff, etc. could go here
	//qDebug() << "Debug. [ MainWindow() ] Closing ... " ;
	//root->unref(); //will do in BrainQuarter destructor   
	delete viewer;
	exit(0);
}

void MainWindow::verboseStatus() {
	resourceForm->setVerboseStatus(verboseAction->isChecked());
}