#include "base/bdmbase.h"
#include "base/user_info.h"
#include "generator.h"
#include "additive_generator.h"
#include "size_generator.h"
#include "square_mat_point.h"
#include <iostream>
#include <iomanip>
#include <stdlib.h>

using namespace std;
using namespace bdm;
using namespace itpp;

UIREGISTER ( size_generator );
UIREGISTER ( additive_generator );
//UIREGISTER ( square_mat_point );

const char *generator_file_name = "generator.cfg";
const char *agenda_file_name = "agenda.cfg";
int agenda_length = 10;

int main ( int argc, char const *argv[] ) {
	RNG_randomize();

	bool unknown = false;
	int update_next = 0; // 1 generator file, 2 agenda file, 3 agenda length
	const char **param = argv + 1;
	while ( *param && !unknown ) {
		if ( update_next ) {
			if ( update_next == 1 ) {
				generator_file_name = *param;
			} else if ( update_next == 2 ) {
				agenda_file_name = *param;
			} else {
				int length = atoi ( *param );
				if ( length > 0 ) {
					agenda_length = length;
				} else {
					cerr << "invalid agenda length value ignored" << endl;
				}
			}

			update_next = 0;
		} else {
			if ( !strcmp ( *param, "-a" ) ) {
				update_next = 2;
			} else if ( !strcmp ( *param, "-g" ) ) {
				update_next = 1;
			} else if ( !strcmp ( *param, "-l" ) ) {
				update_next = 3;
			} else {
				unknown = true;
			}
		}

		++param;
	}

	if ( unknown || update_next ) {
		cerr << "usage: " << argv[0] << " [ -g generator.cfg ] [ -a agenda_output.cfg ] [ -l agenda_length ]" << endl;
	} else {
		Array<shared_ptr<square_mat_point> > mag ( agenda_length );

		UIFile gspec ( generator_file_name );
		shared_ptr<generator> gen ( UI::build<generator> ( gspec, "generator", UI::compulsory ) );
		for ( int i = 0; i < agenda_length; ++i ) {
			mat m = gen->next();
			square_mat_point *p = new square_mat_point();
			p->set_parameters ( m, randu ( m.rows() ), randu() );
			mag ( i ) = p;
		}

		UIFile fag;
		UI::save ( mag, fag, "agenda" );
		fag.save ( agenda_file_name );
	}
}