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

#include "logger.h"
#include "wiicpp.h"
#include "wiic_internal.h"

void Logger::SetLogLevel(LogStatus status, int type, const string& file)
{               
        // Avoid repetitions
        if(status == logStatus && type == logType && file == logFile)
                return;
                
        if(status == WIIC_LOG_START) {
                ResetTimestamp();       
        
                if((file == "" && logFile != "") || (file == logFile)) { // Same file
                        if(logEnabled) 
                                out << "END" << endl;
                                
                        // New training
                        unsigned long trainingTs = (startTs.tv_sec % 86400) * 1000 + startTs.tv_usec / 1000;
                        out << "START " << trainingTs << endl;
                }
                else if(file != "") { // Open a new log file
                        if(out.is_open()) { // In case a previous log file is still open
                                if(logEnabled)
                                        out << "END" << endl;
                                out.close();
                        }
                                                
                        logFile = file;
                        
                        // Check if the file already exists, and whether it is a WiiC logfile
                        bool resume = CheckLogFile();
                        
                        out.open(logFile.c_str(),ios::app);
                        if(!out.is_open()) 
                                return;
                
                        // Sometimes we want to modify a log file, not to create a new one
                        if(!resume)
                                SaveLogHeader();

                        // New training                         
                        unsigned long trainingTs = (startTs.tv_sec % 86400) * 1000 + startTs.tv_usec / 1000;
                        out << "START " << trainingTs << endl;
                }
                else // How can I log if you don't provide me with a file???
                        return;
                                
                // Status init
                logType = type;
                logStatus = WIIC_LOG_START;
                logEnabled = true;
        }
        else if(status == WIIC_LOG_STOP) {
                out << "END" << endl;
                if(out.is_open())
                        out.close();
                        
                // Status reset
                logStatus = WIIC_LOG_STOP;
                logType = WIIC_LOG_NONE;
                logEnabled = false;
                logFile = "";
        }
        else {
                return;
        }       
}

bool Logger::CheckLogFile()
{
        ifstream in(logFile.c_str());
        if(!in.is_open())
                return false;
                
        string wiicToken;
        int logVersion;
        in >> wiicToken >> logVersion;
        in.close();
        
        if(wiicToken == "WiiC" && logVersion <= WIIC_LOG_VERSION)
                return true;
                
        return false;
}

/**
 * Save the dataset header into a file.
 *
 * @param out Output file stream
 * @param addr Wii device MAC address
 */     
void Logger::SaveLogHeader()
{
        // 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 << deviceAddr << endl;
}

void Logger::InitLog()
{
        // Compute the relative timestamp
        struct timeval t;
        gettimeofday(&t,0);
        relativeTs = ((t.tv_sec*1000000.0 + t.tv_usec) - (startTs.tv_sec*1000000.0 + startTs.tv_usec))/1000.0;
}

void Logger::LogAcc(float x, float y, float z)
{
        out << "ACC " << relativeTs << " " << x << " " << y << " " << z << endl;        
}

void Logger::LogGyro(float roll, float pitch, float yaw)
{
        out << "GYRO " << relativeTs << " " << roll << " " << pitch << " " << yaw << endl;
}