Bug.h

#ifndef BUG_H_
#define BUG_H_

#include <fstream>

#ifndef DEBUG
    //#define DEBUG
#endif

//#define TIMEONLY

struct Bug;

struct TimeOnly {
	TimeOnly(Bug & bug_) : bug(bug_) {}

	Bug & bug;
};

/*
    struct for debugging - this is gross but can be used pretty much like an ofstream,
                           except the debug messages are stripped while compiling if
                           DEBUG is not defined.
    example:
        Bug bug;
        bug.open("./debug.txt");
        bug << state << endl;
        bug << "testing" << 2.0 << '%' << endl;
        bug.close();
*/
struct Bug
{
    std::ofstream file;
    TimeOnly time;

    Bug() : time(*this)
    {

    };

    //opens the specified file
    inline void open(const std::string &filename)
    {
        #if defined(DEBUG) || defined(TIMEONLY)
            file.open(filename.c_str());
        #endif
    };

    //closes the ofstream
    inline void close()
    {
        #if defined(DEBUG) || defined(TIMEONLY)
            file.close();
        #endif
    };
};

//output function for endl
inline Bug& operator<<(Bug &bug, std::ostream& (*manipulator)(std::ostream&))
{
    #ifdef DEBUG
        bug.file << manipulator;
    #endif

    return bug;
};

//output function
template <class T>
inline Bug& operator<<(Bug &bug, const T &t)
{
    #ifdef DEBUG
        bug.file << t;
    #endif

    return bug;
};

//output function for endl
inline TimeOnly& operator<<(TimeOnly &bug, std::ostream& (*manipulator)(std::ostream&))
{
#if defined(DEBUG) || defined(TIMEONLY)
	bug.bug.file << manipulator;
#endif

	return bug;
};

//output function
template <class T>
inline TimeOnly& operator<<(TimeOnly &bug, const T &t)
{
#if defined(DEBUG) || defined(TIMEONLY)
	bug.bug.file << t;
#endif

	return bug;
};

#endif //BUG_H_