darshan-core.h 2 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>

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
#define DARSHAN_CORE_MAX_RECORDS 4096
30

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

Shane Snyder's avatar
Shane Snyder committed
34 35 36
#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))
37

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

Shane Snyder's avatar
Shane Snyder committed
52 53 54 55 56 57
struct darshan_core_module
{
    darshan_module_id id;
    struct darshan_module_funcs mod_funcs;
};

58 59
struct darshan_core_record_ref
{
60
    struct darshan_record rec;
61 62
    uint64_t mod_flags;
    uint64_t global_mod_flags;
63
    UT_hash_handle hlink;
64
};
65

66 67 68
void darshan_core_initialize(int argc, char **argv);
void darshan_core_shutdown(void);

69 70 71
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);

72
#endif /* __DARSHAN_CORE_H */