print.h

// ***********************************************************************
// Assembly         : task_scheduler
// Author           : viknash
// ***********************************************************************
// <copyright file="print.h" >
//     Copyright (c) viknash. All rights reserved.
// </copyright>
// <summary></summary>
// ***********************************************************************
#pragma once

#include <atomic>
#include <cstdint>
#include <iostream>
#include <mutex>
#include <sstream>
#include <thread>

#include "platform.h"
#include "globals.h"


namespace task_scheduler
{

#if defined(TASK_SCHEDULER_DEBUG)
    struct out_t
    {
        template < typename T > out_t &operator<<(T &&x)
        {
            buffer << x;
            return *this;
        };

        out_t() { buffer << thread_name << "> "; }

        ~out_t()
        {
            std::lock_guard< std::mutex > lk(get<globals>()->io_mutex);
            ts_windows_only(SetConsoleTextAttribute(get<globals>()->console_handle, get_thread_number()););
            tcout << buffer.str() << std::endl;
        }

        tostringstream buffer;
    };
#else  // defined(TASK_SCHEDULER_DEBUG)
    struct out_t
    {
        template < typename T > out_t &operator<<(T &&x) { return *this; };
    };
#endif // defined(TASK_SCHEDULER_DEBUG)

#define ts_print_internal(arg)                                                                                         \
    out_t arg;                                                                                                         \
    arg

#define ts_print(x)                                                                                                    \
    {                                                                                                                  \
        ts_print_internal(ts_unique_variable(printout)) << x;                                                          \
    }
};