ftb/profiler.hpp

#pragma once
#ifdef _PROFILING

# include <stdlib.h>
# include <stdio.h>
# include <string.h>
# include <time.h>

#ifdef _MSC_VER
#  include <Windows.h>
#else
#  include <fcntl.h>
#  include <sys/stat.h>
#  include <sys/time.h>
#  include <sys/time.h>
#  include <stdint.h>
#  include <stdbool.h>
#  include <stddef.h>

/* Helpful conversion constants. */
static const unsigned usec_per_sec = 1000000;
static const unsigned usec_per_msec = 1000;

/* These functions are written to match the win32
   signatures and behavior as closely as possible.
*/
bool QueryPerformanceFrequency(int64_t *frequency)
{
    /* gettimeofday reports to microsecond accuracy. */
    *frequency = usec_per_sec;

    return true;
}

bool QueryPerformanceCounter(int64_t *performance_count)
{
    struct timeval time;

    /* Grab the current time. */
    gettimeofday(&time, NULL);
    *performance_count = time.tv_usec + /* Microseconds. */
        time.tv_sec * usec_per_sec; /* Seconds. */

    return true;
}
#endif


struct Profiler {
#ifdef _MSC_VER
    LARGE_INTEGER tmp_time;
#else
    int64_t tmp_time;
#endif

    // same for all threads
    inline static char file_template[40] = "\0";

    // thread local
    inline thread_local static bool   is_initialized = false;
    inline thread_local static size_t thread_id = -1;
    inline thread_local static int    call_depth = 0;
    inline thread_local static FILE*  out_file = nullptr;


    Profiler(const char* file, const char* name, const int line, const char* comment1, const char* comment2) {
        call_depth += 1;

        // if we never used this thread before
        if (!is_initialized) {
            thread_id = (size_t)&thread_id;
            time_t t = time(NULL);
            tm* tm_i = localtime(&t);

            // create folder
#ifdef _MSC_VER
            _mkdir("./profiler_reports/");
#else
            mkdir("./profiler_reports/", S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH);
#endif

            // if we never even created the shared file name template
            if (!file_template[0]) {
                // make sure folder exists
                sprintf(file_template, "./profiler_reports/%02d.%02d.%04d-%02d.%02d.%02d-%%zd-profiler.report",
                        tm_i->tm_mday, tm_i->tm_mon+1, tm_i->tm_year+1900,
                        tm_i->tm_hour, tm_i->tm_min, tm_i->tm_sec);
            }

            char file_name[100];
            sprintf(file_name, file_template, thread_id);
            // printf("Hello I am %zd\n", thread_id);
            out_file = fopen(file_name, "w");
            if (!out_file) {
                printf("could not open %s\n", file_name);
            }

            // initially write the performance frequency
            QueryPerformanceFrequency(&tmp_time);
#ifdef _MSC_VER
            fprintf(out_file, "%lld,,,,\n", tmp_time.QuadPart);
#else
            fprintf(out_file, "%ld,,,,\n", tmp_time);
#endif
            is_initialized = true;
        }

        QueryPerformanceCounter(&tmp_time);
#ifdef _MSC_VER
        fprintf(out_file, "->,%lld,%s,%s,%d,%s,%s\n",
                tmp_time.QuadPart, name, file,
                line, comment1, comment2);
#else
        fprintf(out_file, "->,%ld,%s,%s,%d,%s,%s\n",
                tmp_time, name, file,
                line, comment1, comment2);
#endif
    };

    ~Profiler() {
        call_depth -= 1;
        QueryPerformanceCounter(&tmp_time);
#ifdef _MSC_VER
        fprintf(out_file, "<-,%lld,,,,,\n", tmp_time.QuadPart);
#else
        fprintf(out_file, "<-,%ld,,,,,\n", tmp_time);
#endif
        if (call_depth == 0)
            fflush(out_file);
    };
};

# define profile_this()                             Profiler profiler(__FILE__, __FUNCTION__, __LINE__, "", "")
# define profile_with_name(name)                    Profiler profiler(__FILE__,         name, __LINE__, "", "")
# define profile_with_comment(c1)                   Profiler profiler(__FILE__, __FUNCTION__, __LINE__, c1, "")
# define profile_with_comments(c1,c2)               Profiler profiler(__FILE__, __FUNCTION__, __LINE__, c1, c2)
# define profile_with_name_and_comment(name,c1)     Profiler profiler(__FILE__,         name, __LINE__, c1, "")
# define profile_with_name_and_comments(name,c1,c2) Profiler profiler(__FILE__,         name, __LINE__, c1, c2)
#else
# define profile_this()                             do {} while(0)
# define profile_with_name(name)                    do {} while(0)
# define profile_with_comment(c1)                   do {} while(0)
# define profile_with_comments(c1,c2)               do {} while(0)
# define profile_with_name_and_comment(name,c1)     do {} while(0)
# define profile_with_name_and_comments(name,c1,c2) do {} while(0)
#endif