simuStack.cpp

#include <iostream>
#include <sstream>
#include <string>
#include <fstream>
#include <cstdlib>
#include <ctime>

#include <unistd.h>

#define STACK_RUNNING 		0
#define STACK_FILE_CHANGED 	1
#define STACK_FILE_DELETED 	2
#define STACK_NOT_RUNNING 	3

#define SIMULIST_OK		0
#define WRONG_PATH 		1
#define NO_TEMPLATE		3
#define NO_INPUT_FILE	4
#define NO_OUTPUT_FILE	5
#define UPDATE_DONE		6

using namespace std;

void openStream(fstream& str, const char* file, ios_base::openmode m){
	
	str.open(file, m);

	if (!str.is_open()){
		cout << endl << "Error while opening " << file << " file" << endl;
		exit(1);
	}
	
	return;
}

int main(int argc, char** argv){

	unsigned int index = 1;
	time_t timestamp;
	int stackID = (int) getpid();
	string line;
	int exitCode;
	char cmd[20] = "";

	static const char pid[] = "stackPid";
	static const char simus[] = "simuList";
	static const char done[] = "doneSimuList";
	static const char started[] = "startedSimuList";
	static const char log[] = "logSimuStack";
	
	fstream startedList;
	fstream pidfile;
	fstream infile;
	fstream donelist;
	fstream logfile;
	
	sprintf(cmd,"./Scripts/checkPid %d",stackID);
	exitCode = system(cmd);
	if(	(WEXITSTATUS(exitCode) == STACK_RUNNING) 		|| 	\
		(WEXITSTATUS(exitCode) == STACK_FILE_CHANGED)	||	\
		(WEXITSTATUS(exitCode) == STACK_FILE_DELETED)	){
		
		openStream(pidfile, pid, fstream::in);
		pidfile >> stackID;
		pidfile.close();
		
		cout << endl << "Stack already running with pid " << stackID << endl;
		exit(1);
	}

	openStream(pidfile, pid, fstream::out);
	pidfile << stackID << endl;
	pidfile.close();
	
	//TODO: VERIFICAR SI HABIA ALGO INICIADO Y REEMPLAZAR EL INPUT POR EL INPUT RESTART
	//TODO: REEMPLAZAR EN EL INPUT RESTART EL VALOR DE RESTART MAYOR ENCONTRADO EN LA CARPETA
	//TODO: NORMALIZAR LOS COMANDOS, i.e: PRIMERO UN CD (PARA UBICARSE)
	cout << endl << "Cleaning simulation list..." << endl;
	system("./Scripts/cleanStackList");

	cout << endl << "Looking for stopped simulations..." << endl;
	exitCode=system("./Scripts/updateRestart");
	if( WEXITSTATUS(exitCode) == WRONG_PATH ){
		cout << "Found wrong path. Verify and launch again." << endl;
		exit(1);
	} else if( WEXITSTATUS(exitCode) == NO_TEMPLATE ){
		cout << "No input restart template found! Verify and launch again." << endl;
		exit(1);
	} else if( WEXITSTATUS(exitCode) == NO_INPUT_FILE ){
		cout << "No input file found! Verify and launch again." << endl;
		exit(1);
	} else if( WEXITSTATUS(exitCode) == NO_OUTPUT_FILE ){
		cout << "No output file found! Verify and launch again." << endl;
	} else if( 	(WEXITSTATUS(exitCode) == SIMULIST_OK)	|| \
				(WEXITSTATUS(exitCode) == UPDATE_DONE)	){
		cout << "Update done. OK" << endl;
	}		
	
	cout << "Stack ready to run." << endl;
	
	openStream(infile, simus, fstream::in);
		
	timestamp = time(0);	
	
	openStream(logfile, log, fstream::out | fstream::app);
	logfile << endl;
	logfile << "#################################################" << endl;
	logfile << "Lammps Stack Running..." << endl;
	logfile << ctime(&timestamp);
	logfile << "#################################################" << endl << endl;
	logfile.close();
	
	while(1){
		while (getline(infile, line,'#')){
			if(line != ""){
				
				timestamp = time(0);
				
				openStream(logfile, log, fstream::out | fstream::app);			
				logfile << "#################################################" << endl;
				logfile << "COMMAND: " << endl << line << endl;
				logfile << "Timestamp (ASCII): " << ctime(&timestamp);
				logfile << "Timestamp: " << timestamp << endl;
				logfile << "#################################################" << endl;
				logfile.close();
				
				openStream(startedList, started, fstream::out | fstream::app);
				startedList << index << " " << timestamp << endl; 
				startedList.close();
				
				int err = system(line.c_str());
				
				openStream(logfile, log, fstream::out | fstream::app);											
				logfile << "(Up) Command completed with error code: " << WEXITSTATUS(err) << endl;
				logfile << "#################################################" << endl << endl;
				logfile.close();
				
				openStream(donelist, done, fstream::out | fstream::app);
				donelist << index++ << " " << timestamp << endl;
				donelist.close();
			}
		}
		
		sleep(5);
		
		/*	Borra el eofbit para poder leer las simulaciones 
		 * 	que se agreguen mientras se ejecutaba la anterior
		 */
		infile.clear();
	}

	exit(0);
}