mpe_log_thread.c

fortran并行计算包

/*
     (C) 2007 by Argonne National Laboratory.
          See COPYRIGHT in top-level directory.
*/
#include "mpe_logging_conf.h"

#if defined( STDC_HEADERS ) || defined( HAVE_STDIO_H )
#include <stdio.h>
#endif
#if defined( STDC_HEADERS ) || defined( HAVE_STDLIB_H )
#include <stdlib.h>
#endif
#if defined( HAVE_PTHREAD_H )
#include <pthread.h>
#endif

#if defined( HAVE_LIBPTHREAD )

#include "mpe_callstack.h"

#define MPE_ThreadID_t      int
#define MPE_THREADID_NULL  -9999

#define MPE_LOG_THREAD_PRINTSTACK() \
        do { \
            MPE_CallStack_t  cstk; \
            MPE_CallStack_init( &cstk ); \
            MPE_CallStack_fancyprint( &cstk, 2, \
                                      "\t", 1, MPE_CALLSTACK_UNLIMITED ); \
        } while (0)

/* MPE coarse-grained lock support mechanism */
/*
    pthread_mutex_t  MPE_Thread_mutex = PTHREAD_MUTEX_INITIALIZER;

    Replaced PTHREAD_MUTEX_INITIALIZER by pthread_mutex_init() as
    Sun's cc for linux failed to compile PTHREAD_MUTEX_INITIALIZER because
    Sun's cc is a strict C compiler and PTHREAD_MUTEX_INITIALIZER
    implementation on 32bit linux uses C++ feature.
*/
pthread_mutex_t  MPE_Thread_mutex;
pthread_key_t    MPE_ThreadStm_key;
MPE_ThreadID_t   MPE_Thread_count = MPE_THREADID_NULL;

void MPE_ThreadStm_free( void *thdstm );
void MPE_ThreadStm_free( void *thdstm )
{
    if ( thdstm != NULL ) {
        free( thdstm );
        thdstm = NULL;
    }
}

void MPE_Log_thread_init( void );
void MPE_Log_thread_init( void )
{
    int   ierr;
    /*
         Check if MPE_Thread_count == MPE_THREADID_NULL to protect
         the initialization routines from invoking more than once.
    */
    if ( MPE_Thread_count == MPE_THREADID_NULL ) {
        MPE_Thread_count = 0;
        ierr = pthread_mutex_init( &MPE_Thread_mutex, NULL );
        if ( ierr != 0 ) {
            perror( "pthread_mutex_init() fails!" );
            MPE_LOG_THREAD_PRINTSTACK();
            pthread_exit( NULL );
        }
        ierr = pthread_key_create( &MPE_ThreadStm_key, MPE_ThreadStm_free );
        if ( ierr != 0 ) {
            perror( "pthread_key_create() fails!" );
            MPE_LOG_THREAD_PRINTSTACK();
            pthread_exit( NULL );
        }
    }
}

#else

void MPE_ThreadStm_free( void *thdstm );
void MPE_ThreadStm_free( void *thdstm )
{}

void MPE_Log_thread_init( void );
void MPE_Log_thread_init( void )
{}

#endif