Stream_MGR.h

Go to the documentation of this file.

/* ############################ MALP License ############################## */
/* # Fri Jan 18 14:00:00 CET 2013                                         # */
/* # Copyright or (C) or Copr. Commissariat a l'Energie Atomique          # */
/* #                                                                      # */
/* # This software is governed by the CeCILL-C license under French law   # */
/* # and abiding by the rules of distribution of free software.  You can  # */
/* # use, modify and/ or redistribute the software under the terms of     # */
/* # the CeCILL-C license as circulated by CEA, CNRS and INRIA at the     # */
/* # following URL http://www.cecill.info.                                # */
/* #                                                                      # */
/* # The fact that you are presently reading this means that you have     # */
/* # had knowledge of the CeCILL-C license and that you accept its        # */
/* # terms.                                                               # */
/* #                                                                      # */
/* # Authors:                                                             # */
/* #   - BESNARD Jean-Baptiste jean-baptiste.besnard@cea.fr               # */
/* #                                                                      # */
/* ######################################################################## */

#ifndef STREAM_MGT_H
#define STREAM_MGT_H

#include <stdint.h>

#include "VMPI.h"
#include "Posix_Wrapp.h"

#include "Event_Desc.h"
#include "Spinlock.h"
#include "Chained_List.h"
#include "Buffered_FIFO.h"

struct Stream_Manager
{
        struct Chained_List streams;    
        uint64_t current_id;            
        uint64_t partition_id;          
        int rank;                       
        MALP_Spinlock lock;             
};

extern struct Stream_Manager __stream_Stream_Manager;

static inline struct Stream_Manager * Stream_Manager()
{
        return &__stream_Stream_Manager;
}

void Stream_Manager_init();

void Stream_Manager_release();

struct Stream_Infos * Stream_MGR_new_stream( struct Stream_Descriptor *parent, Stream_type type );

void Stream_MGR_end_stream();

#define MAX_PENDING_EVENT 4096

struct Stream_Infos
{
        struct Stream_Descriptor infos;                 
        volatile uint8_t is_in_mpi_call;                
        struct Buffered_FIFO mpi_wrapped_event_buffer;  
};

void Stream_Infos_init( struct Stream_Infos *info );

void Stream_Infos_release( struct Stream_Infos *info );

extern __thread struct Stream_Infos * __trace_module;

static inline struct Stream_Infos * Stream_infos_get()
{
        return __trace_module;
}

static inline void Stream_infos_set(struct Stream_Infos *sd)
{
        __trace_module = sd;
}

static inline struct Stream_Descriptor * Stream_desc_get()
{
        return &__trace_module->infos;
}

static inline void Stream_Infos_enter_mpi()
{
        if( !__trace_module )
                return;

        __trace_module->is_in_mpi_call = 1;
}

static inline void Stream_Infos_leave_mpi()
{
        if( !__trace_module)
                return;

        __trace_module->is_in_mpi_call = 0;
}

static inline int Stream_Infos_is_in_mpi()
{
        if( !__trace_module )
                return 0;

        return __trace_module->is_in_mpi_call;
}

static inline void MALP_Trace_Event_fill( struct MALP_Trace_Event *event,
                                                                                        MALP_Trace_event_type type,
                                                                                        uint64_t timestamp,
                                                                                        uint64_t duration )
{
        struct Stream_Descriptor *stream_infos = Stream_desc_get();

        event->type = type;
        event->timestamp = timestamp;
        event->duration = duration;

        event->partition_id = VMPI_Get_partition_id();
        
        if( !stream_infos )
        {
                event->tid = 0; /* Undefined IDS are set to 0 */
                return;
        }

        event->tid = stream_infos->tid;
        
        event->rank = stream_infos->mpi_rank;
}


#endif /* STREAM_MGT_H */