source: code/branches/wiimote/src/external/wiicpp/log/dataset.cpp @ 9780

#include "dataset.h"
#include "wiic_internal.h"

/** 
 * Dataset destructor, which calls clear.
 */
Dataset::~Dataset()
{
        clear();
}

/** 
 * Add a new training to the dataset.
 * 
 * @param training Add a training to the dataset.
 */
void Dataset::addTraining(Training* t)
{
        if(t)
                trainings.push_back(t);
                
        loaded = true;
}

/**
 * Load a dataset stored in a file.
 *
 * @param filename Dataset filename
 */
bool Dataset::loadDataset(const string& nomefile)
{
        int version; 
        string line, dummy;
        ifstream infile(nomefile.c_str());

        if(!infile.is_open()) {
                cout << "[Error] Unable to open the dataset file" << endl;
                return false;
        }
        else {
                trainings.clear();
                
                // Version
                getline(infile,line); // First line is WiiC log version
                if(line.find("WiiC") == string::npos) {
                        cout << "[Error] Bad log format." << endl;
                        return false;
                }
                istringstream iline(line);
                iline >> dummy >> version;
                if(version > WIIC_LOG_VERSION) {
                        cout << "[Error] Unsupported WiiC log version." << endl;
                        return false;
                }
                getline(infile,line); // Date (we should load this as well...)
                
                // For each training
                int counter = 0;
                while(!infile.eof()) {
                        getline(infile,line);
                        istringstream iline(line);
                        string cmd; 
                        unsigned long ts;
                        iline >> cmd >> ts;
                        if(cmd == "START") {
                                // Each training is inserted in the training vector     
                                trainings.push_back(new Training());
                                trainings[counter]->setTimestampFromMidnight(ts);
                                if(!(trainings[counter++]->loadTraining(infile))) {
                                        cout << "[Error] Unable to load a training in the dataset" << endl ;
                                        return false;
                                }
                        }
                }
        }
        infile.close();
        loaded = true;
        
        return loaded;
}

/**
 * Delete all trainings and clear the buffer. This method will take 
 * care of freeing the memory of each training in the dataset, 
 * hence you don't need to free them in your code.
 */
void Dataset::clear()
{
        for(unsigned int i = 0 ; i < trainings.size() ; i++) {
                if(trainings[i]) {
                        delete trainings[i];
                        trainings[i] = 0;
                }
        }
        trainings.clear();
        loaded = false;
}

/**
 * Save the dataset into a file for training and recognition.
 *
 * @param filename Pathname of the destination file
 * @param addr Wii device mac address
 */
bool Dataset::save(const char* file, const char* addr) const
{
        // Open file
        ofstream out(file, ios::trunc);
        if(!out.is_open()) { // File does not exist, hence I create it
                cout << "[Error] Unable to open " << file << endl;
                return false;
        }
        
        // Save header
        saveHeader(out, addr);
        
        // For each training
        for(int i = 0 ; i < size() ; i++)       {
                const Training* training = trainingAt(i);
                if(training)
                        training->save(out);
        }
        
        out.close();

        return true;
}

/**
 * Save the dataset header into a file.
 *
 * @param out Output file stream
 * @param addr Wii device MAC address
 */     
void Dataset::saveHeader(ofstream& out, const char* addr) const
{
        // Log Version
        out << "WiiC " << WIIC_LOG_VERSION << endl;

        // Date
    time_t tim=time(NULL);
    char *s=ctime(&tim);
    s[strlen(s)-1]=0;        // remove \n
        out << s << endl;
        
        // Mac Address
        out << addr << endl;
}