dlib C++ Library - stack_trace.cpp

    // Copyright (C) 2008  Davis E. King (davisking@users.sourceforge.net)
    // License: Boost Software License   See LICENSE.txt for the full license.
    #ifndef DLIB_STACK_TRACE_CPp_
    #define DLIB_STACK_TRACE_CPp_

    #if defined(DLIB_ENABLE_STACK_TRACE) && !defined(NO_MAKEFILE)

    #include <sstream>
    #include <cstring>
    #include "stack_trace.h"
    #include "threads.h"
    #include "stack.h"
    #include "memory_manager.h"

    namespace dlib
    {

    // ----------------------------------------------------------------------------------------

        namespace
        {
            struct stack_tracer_data
            {
                stack_tracer_data(
                ) :  funct_name(0),
                     file_name(0),
                     line_number(0)
                {}
                const char* funct_name;
                const char* file_name;
                int line_number;
            };

            thread_specific_data<stack<stack_tracer_data,memory_manager<char>::kernel_2a>::kernel_1a>& get_dlib_stack_trace_stack()
            {
                static thread_specific_data<stack<stack_tracer_data,memory_manager<char>::kernel_2a>::kernel_1a> a;
                return a;
            }
        }

    // ----------------------------------------------------------------------------------------

        stack_tracer::
        stack_tracer (
            const char* funct_name,
            const char* file_name,
            const int line_number
        )
        {
            stack_tracer_data data;
            data.funct_name = funct_name;
            data.file_name = file_name;
            data.line_number = line_number;

            // pop the info onto the function stack trace
            get_dlib_stack_trace_stack().data().push(data);
        }

    // ----------------------------------------------------------------------------------------

        stack_tracer::
        ~stack_tracer()
        {
            stack_tracer_data temp;
            get_dlib_stack_trace_stack().data().pop(temp);
        }

    // ----------------------------------------------------------------------------------------

        const std::string get_stack_trace()
        {
            std::ostringstream sout;
            get_dlib_stack_trace_stack().data().reset();
            while (get_dlib_stack_trace_stack().data().move_next())
            {
                stack_tracer_data data = get_dlib_stack_trace_stack().data().element();
                sout << data.file_name << ":" << data.line_number << "\n    " << data.funct_name << "\n";
            }
            return sout.str();
        }

    // ----------------------------------------------------------------------------------------

    }
    #endif

    #endif // DLIB_STACK_TRACE_CPp_