wb-log.h

Go to the documentation of this file.

// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
// Copyright 2014-2015 Tsinghua University
// Author: wb.th08@gmail.com (Bin Wang), ozj@tsinghua.edu.cn (Zhijian Ou)
//
// All h, cpp, cc, and script files (e.g. bat, sh, pl, py) should include the above
// license declaration. Different coding language may use different comment styles.


#pragma once
#include <iostream>
#include <fstream>
#include <iomanip>
#include <ctime>
#include <cstring>
using namespace std;
#include "wb-linux.h"

namespace wb
{
#define wbLog_LevelSpace "\t"

#define wbLog_OutputSpace(io) \
    for (int i=0; i<m_nLevel; i++) io<<wbLog_LevelSpace;
#define wbLog_Output(x) \
    if (m_bOutputLog) { {wbLog_OutputSpace(m_fileLog)} m_fileLog<<x; } \
    if (m_bOutputCmd) { {wbLog_OutputSpace(cout)} cout<<x; }\
    return *this;

    class ProgressBar
    {
    private:
        long long m_num; 
        long long m_total; 
        char m_symbol[3];   
        int m_lastprec; 
        const int m_barmaxlen = 50; 

        string m_strhead; 
    public:
        ProgressBar() {
            m_total = 100;
            memcpy(m_symbol, "[=]", sizeof(3));
            m_lastprec = -1;
            m_strhead = "";
        }
        ProgressBar(long long total, const char *head = "", const char* sym = "[=]"){
            Reset(total, head, sym);
        }
        void Reset(long long total = 100, const char *head = "", const char* sym = "[=]");
        void Reset(long long n, long long total = 100, const char *head = "", const char* sym = "[=]");
        void Update(long long n = -1);
    };

    class Log
    {
    protected:
        ofstream m_fileLog; 
        bool m_bOutputLog; 
        bool m_bOutputCmd; 

        short m_nLevel;  

        ProgressBar m_bar; 
    public:
        Log();
        ~Log();

        void ReFile(const char *path, bool bNew = true);
        bool &bOutputCmd();
        bool &bOutputLog();
        Log &operator << (ostream& (*op) (ostream&));
        //template <typename _Arg>
        //Log &operator << (const _Smanip<_Arg>& _Manip) { cout << _Manip; return *this; }
        Log &operator << (int x);
        Log &operator << (short x);
        Log &operator << (long x);
        Log &operator << (long long x);
        Log &operator << (unsigned int x);
        Log &operator << (unsigned short x);
        Log &operator << (unsigned long x);
        Log &operator << (float x);
        Log &operator << (double x);
        Log &operator << (char x);
        Log &operator << (const char* x);
        Log &operator << (const void* x);
        Log &operator << (bool x);
        Log &operator << (string &x);
        template <typename T>
        Log &output(T *pArray, int n, const char *pgap=" ");
        void LevelDown();
        void LevelUp();
        void Progress(long long n = -1, bool bInit = false, long long total = 100, const char* head = "");
        void Progress(FILE *fp, bool bInit = false, const char* head="");
    };

    extern Log lout; 

    template <typename T>
    Log &Log::output(T *pArray, int n, const char *pgap/*=" "*/)
    {
        for (int i = 0; i < n; i++) {
            *this << pArray[i] << pgap;
        }
        return *this;
    }


#define lout_variable(x) {wb::lout<<#x" = "<<x<<endl;}
#define lout_variable_precent(x,y) {wb::lout<<#x" = "<<x<<" ("<<100.0*(x)/(y)<<"%)  /"#y<<"="<<y<<endl;}
#define lout_variable_rate(x,y) {wb::lout<<100.0*(x)/(y)<<"%("<<x<<"/"<<y<<")";}
#define lout_array(x, n) {wb::lout<<#x"=[ "; for(int i=0; i<n; i++) wb::lout<<x[i]<<" "; wb::lout<<"]"<<endl; }
#define lout_error(x) {wb::lout.bOutputCmd()=true; wb::lout<<"[ERROR] "<<x<<endl; exit(0);}
#define lout_warning(x) {wb::lout<<"[WARNING] "<<x<<endl;}
#define lout_assert(p) {if(!(p)) lout_error("! ("#p")"); }

#define precent(x, n) (x)<<"("<<100.0*(x)/(n)<<"%)"





}