ftb/profiler.hpp

#pragma once
#ifdef _PROFILING
# include <stdlib.h>
# include <stdio.h>
# include <string.h>
# include <time.h>
// # include <bits/stdc++.h>
// # include <iostream>
// # include <sys/stat.h>
// # include <sys/types.h>
#ifdef _MSC_VER
// if windows
# include <Windows.h>
#else
# include <sys/time.h>
#endif


struct Profiler {
    LARGE_INTEGER tmp_time;

    // 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* func, const int line, char* custom_name) {
        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);
            }

            is_initialized = true;
        }
        QueryPerformanceCounter(&tmp_time);
        fprintf(out_file, "->,%lld,%s,%s,%d\n",
                tmp_time.QuadPart,
                (custom_name ?
                 custom_name :
                 func), file, line);
    };
    ~Profiler() {
        call_depth -= 1;
        QueryPerformanceCounter(&tmp_time);
        fprintf(out_file, "<-,%lld,,,\n", tmp_time.QuadPart);
        if (call_depth == 0)
            fflush(out_file);
    };
};

# define profile_this()          Profiler profiler(__FILE__, __FUNCTION__, __LINE__,    0)
# define profile_with_name(name) Profiler profiler(__FILE__, __FUNCTION__, __LINE__, name)
#else
# define profile_this()      do {} while(0)
# define profile_with_name() do {} while(0)
#endif