logger_options.cc

Go to the documentation of this file.

/*!
 *
 * Copyright (C) 2015 Technical University of Liberec.  All rights reserved.
 *
 * This program is free software; you can redistribute it and/or modify it under
 * the terms of the GNU General Public License version 3 as published by the
 * Free Software Foundation. (http://www.gnu.org/licenses/gpl-3.0.en.html)
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more details.
 *
 *
 * @file    logger_options.cc
 * @brief
 */
 
 
#include "system/logger_options.hh"
#include "system/logger.hh"
#include "system/global_defs.h"
#include "system/time_point.hh"
#include <time.h>
#include <random>
 
using namespace std;
 
/*******************************************************************
 * implementation of LoggerOptions
 */
 
LoggerOptions& LoggerOptions::get_instance() {
    return *instance_;
}
 
 
 
std::string LoggerOptions::format_hh_mm_ss() {
    TimePoint t = TimePoint();
    double seconds = t-LoggerOptions::start_time;
    ASSERT(seconds > -numeric_limits<double>::epsilon())(seconds).error("Formating of negative time.");
 
    unsigned int h,m,s,ms;
    unsigned int full_time = (int)(seconds * 1000); // in first step in miliseconds
 
    ms = full_time % 1000;
    full_time /= 1000;
    s = full_time % 60;
    full_time /= 60;
    m = full_time % 60;
    h = full_time / 60;
 
    stringstream ss;
    if (h<10) ss << "0";
    ss << h << ":";
    if (m<10) ss << "0";
    ss << m << ":";
    if (s<10) ss << "0";
    ss << s << ".";
    if (ms<100) ss << "0";
    if (ms<10) ss << "0";
    ss << ms;
 
    return ss.str();
}
 
 
TimePoint LoggerOptions::start_time = TimePoint();
 
 
LoggerOptions* LoggerOptions::instance_ = new LoggerOptions();
 
 
LoggerOptions::LoggerOptions()
: mpi_rank_(-1), init_flag_(InitFlag::uninitialize) {}
 
 
LoggerOptions::~LoggerOptions() {
    if (file_stream_.is_open()) {
        file_stream_ << std::flush;
        file_stream_.close();
    }
}
 
 
void LoggerOptions::set_mpi_rank(int mpi_rank) {
    ASSERT(init_flag_ == InitFlag::uninitialize).error("Setup MPI must be performed before setting logger file.");
 
    this->mpi_rank_ = mpi_rank;
}
 
 
std::string LoggerOptions::log_file_name(std::string log_file_base) {
    ASSERT(init_flag_ == InitFlag::uninitialize).error("Recurrent initialization of logger file stream.");
 
    if ( (log_file_base.size()) == 0 || (log_file_base == "//") ) {
        init_flag_ = InitFlag::no_log;
        return "";
    }
 
    int mpi_rank = this->mpi_rank_;
    if (mpi_rank == -1) { // MPI is not set, random value is used
        std::random_device rd;
        std::mt19937 gen(rd());
        std::uniform_int_distribution<int> dis(0, 999999);
        mpi_rank = dis(gen);
        WarningOut() << "Unset MPI rank, random value '" << mpi_rank << "' of rank will be used.\n";
    }
    std::stringstream file_name;
    file_name << log_file_base << "." << mpi_rank << ".log";
    return file_name.str();
}
 
 
void LoggerOptions::set_stream(std::string abs_path) {
    ASSERT(init_flag_ == InitFlag::uninitialize).error("Recurrent initialization of logger file stream.");
 
    file_stream_.open(abs_path, ios_base::out);
    if (! file_stream_.is_open())
        THROW( ExcMessage() << EI_Message("Can not open logger file: '" + abs_path + "'!") );
 
    init_flag_ = InitFlag::initialize;
}
 
 
void LoggerOptions::reset() {
    mpi_rank_ = -1;
    init_flag_ = InitFlag::uninitialize;
    if (file_stream_.is_open()) {
        file_stream_ << std::flush;
        file_stream_.close();
    }
}