darshan-core.h 2.68 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
 */

#ifndef __DARSHAN_CORE_H
#define __DARSHAN_CORE_H

#include <unistd.h>
#include <sys/types.h>
#include <stdint.h>
13
#include <limits.h>
14

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

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

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

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

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

30 31
/* Environment variable to override memory per module */
#define DARSHAN_MOD_MEM_OVERRIDE "DARSHAN_MODMEM"
32

33 34 35
/* Environment variable to override default mmap log path */
#define DARSHAN_MMAP_LOG_PATH_OVERRIDE "DARSHAN_MMAP_LOGPATH"

36 37 38
/* Maximum amount of memory per instrumentation module in MiB */
#ifdef __DARSHAN_MOD_MEM_MAX
#define DARSHAN_MOD_MEM_MAX (__DARSHAN_MOD_MEM_MAX * 1024 * 1024)
39
#else
40
#define DARSHAN_MOD_MEM_MAX (2 * 1024 * 1024) /* 2 MiB default */
41
#endif
42

43 44 45
/* default path for storing mmap log files is '/tmp' */
#define DARSHAN_DEF_MMAP_LOG_PATH "/tmp"

46 47
/* default name record buf can store 2048 records of size 100 bytes */
#define DARSHAN_NAME_RECORD_BUF_SIZE (2048 * 100)
48

49 50
/* Default runtime compression buffer size */
#define DARSHAN_COMP_BUF_SIZE DARSHAN_MOD_MEM_MAX
51

52 53 54 55 56 57 58 59 60 61 62 63 64 65 66
/* structure to track registered modules */
struct darshan_core_module
{
    struct darshan_module_funcs mod_funcs;
    int mem_avail;
};

struct darshan_core_name_record_ref
{
    struct darshan_name_record *name_record;
    uint64_t mod_flags;
    uint64_t global_mod_flags;
    UT_hash_handle hlink;
};

67
/* in memory structure to keep up with job level data */
68
struct darshan_core_runtime
69
{
70
    /* pointers to each log file component */
71 72 73
    struct darshan_header *log_hdr_p;
    struct darshan_job *log_job_p;
    char *log_exemnt_p;
74
    void *log_name_p;
75
    void *log_mod_p;
76

77
    /* darshan-core internal data structures */
78
    struct darshan_core_module* mod_array[DARSHAN_MAX_MODS];
79
    int mod_mem_used;
80 81
    struct darshan_core_name_record_ref *name_hash;
    int name_hash_cnt;
82
    double wtime_offset;
83
    char *comp_buf;
84 85 86
#ifdef __DARSHAN_ENABLE_MMAP_LOGS
    char mmap_log_name[PATH_MAX];
#endif
87 88
};

Shane Snyder's avatar
Shane Snyder committed
89 90 91
void darshan_core_initialize(int argc, char **argv);
void darshan_core_shutdown(void);

92 93 94
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);

95
#endif /* __DARSHAN_CORE_H */