source: code/branches/wiimote/src/external/wiicpp/log/training.h @ 10256

#ifndef TRAINING_H
#define TRAINING_H

#include <fstream>
#include <iostream>
#include <vector>
#include <sstream>
#include "sample.h"
#include "accsample.h"
#include "gyrosample.h"

using namespace std;

/**
 * Class Training to save the individual training of a gesture  
 */
class Training
{
public:
        /**
         * Default constructor.
         */
        Training() { }
        ~Training();

        bool loadTraining(ifstream&);
        void save(ofstream&) const;
        void addSample(Sample*);
        void clear();

        /** 
         * Returns the i-th sample of the training as a constant pointer. 
         * 
         * @param i Sample index in the training set
         */
        inline const Sample* sampleAt(unsigned int i) const { 
                if(i < samples.size())
                        return samples[i]; 
                else {
                        cout << "[Error]: requested out of array bound index in training." << endl;
                        return 0;
                }
        }

        /** 
         * Returns the i-th sample of the training as a pointer. 
         * 
         * @param i Sample index in the training set
         */
        inline Sample* sampleAt(unsigned int i) { 
                if(i < samples.size())
                        return samples[i]; 
                else {
                        cout << "[Error]: requested out of array bound index in training." << endl;
                        return 0;
                }
        }
        
        /** 
         * Number of samples in the current training. 
         */
        inline unsigned int size() const { return samples.size(); }
        
        inline void setTimestampFromMidnight(unsigned long ts) { timestamp = ts; }

private:
        vector<Sample*> samples;
        unsigned long timestamp; // Training timestamp (beginning of the gesture)
};

#endif