darshan-core.h 2.09 KB
Newer Older
1
/*
Shane Snyder's avatar
Shane Snyder committed
2 3 4
 * Copyright (C) 2015 University of Chicago.
 * See COPYRIGHT notice in top-level directory.
 *
5 6 7 8 9 10 11 12 13
 */

#ifndef __DARSHAN_CORE_H
#define __DARSHAN_CORE_H

#include <unistd.h>
#include <sys/types.h>
#include <stdint.h>

Shane Snyder's avatar
Shane Snyder committed
14 15
#include "uthash.h"
#include "darshan-log-format.h"
16

17 18
/* Environment variable to override __DARSHAN_JOBID */
#define DARSHAN_JOBID_OVERRIDE "DARSHAN_JOBID"
19

20 21
/* Environment variable to override __DARSHAN_LOG_PATH */
#define DARSHAN_LOG_PATH_OVERRIDE "DARSHAN_LOGPATH"
22

23 24
/* Environment variable to override __DARSHAN_LOG_HINTS */
#define DARSHAN_LOG_HINTS_OVERRIDE "DARSHAN_LOGHINTS"
25

26 27
/* Environment variable to override __DARSHAN_MEM_ALIGNMENT */
#define DARSHAN_MEM_ALIGNMENT_OVERRIDE "DARSHAN_MEMALIGN"
28

29 30 31
#ifdef __DARSHAN_MAX_RECORDS
#define DARSHAN_CORE_MAX_RECORDS __DARSHAN_MAX_RECORDS
#else
32
#define DARSHAN_CORE_MAX_RECORDS 2048
33
#endif
34

35
/* TODO: revisit this default size if we change memory per module */
Shane Snyder's avatar
Shane Snyder committed
36
#define DARSHAN_CORE_COMP_BUF_SIZE (2 * 1024 * 1024)
37

Shane Snyder's avatar
Shane Snyder committed
38 39 40
#define DARSHAN_CORE_MOD_SET(flags, id) (flags | (1 << id))
#define DARSHAN_CORE_MOD_UNSET(flags, id) (flags & ~(1 << id))
#define DARSHAN_CORE_MOD_ISSET(flags, id) (flags & (1 << id))
41

42
/* in memory structure to keep up with job level data */
43
struct darshan_core_runtime
44
{
45
    struct darshan_header log_header;
46
    struct darshan_job log_job;
47
    char exe[DARSHAN_EXE_LEN+1];
48
    struct darshan_core_record_ref *rec_hash;
Shane Snyder's avatar
Shane Snyder committed
49
    int rec_count;
50
    struct darshan_core_module* mod_array[DARSHAN_MAX_MODS];
Shane Snyder's avatar
Shane Snyder committed
51
    char comp_buf[DARSHAN_CORE_COMP_BUF_SIZE];
52 53
    double wtime_offset;
    char *trailing_data;
54 55
};

Shane Snyder's avatar
Shane Snyder committed
56 57 58 59 60 61
struct darshan_core_module
{
    darshan_module_id id;
    struct darshan_module_funcs mod_funcs;
};

62 63
struct darshan_core_record_ref
{
64
    struct darshan_record rec;
65 66
    uint64_t mod_flags;
    uint64_t global_mod_flags;
67
    UT_hash_handle hlink;
68
};
69

Shane Snyder's avatar
Shane Snyder committed
70 71 72
void darshan_core_initialize(int argc, char **argv);
void darshan_core_shutdown(void);

73 74 75
uint32_t darshan_hashlittle(const void *key, size_t length, uint32_t initval);
uint64_t darshan_hash(const register unsigned char *k, register uint64_t length, register uint64_t level);

76
#endif /* __DARSHAN_CORE_H */