forked from usnistgov/trec_eval
-
Notifications
You must be signed in to change notification settings - Fork 0
/
functions.h
103 lines (92 loc) · 4.48 KB
/
functions.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
/*
Copyright (c) 2008 - Chris Buckley.
Permission is granted for use and modification of this file for
research, non-commercial purposes.
*/
#ifndef FUNCTIONSH
#define FUNCTIONSH
#include "trec_eval.h"
/* Utility functions */
void * te_chk_and_malloc (void *ptr, long *current_bound,
const long needed, const size_t size);
void * te_chk_and_realloc (void *ptr, long *current_bound,
const long needed, const int size);
/* Functions for dealing with zscores */
int te_get_zscores (const EPI *epi, const char *zscores_file,
ALL_ZSCORES *zscores);
int te_get_zscores_cleanup ();
int te_convert_to_zscore (const ALL_ZSCORES *all_zscores, TREC_EVAL *q_eval);
/* ------------------- Generic Routines for Measures ------------------------ */
/* -------- Initialize measure -------- */
/* Code is in meas_init.c */
/* Measure does not require initialization or storage */
int te_init_meas_empty (EPI *epi, TREC_MEAS *tm, TREC_EVAL *eval);
/* Measure is a single float measure, no parameters */
int te_init_meas_s_float (EPI *epi, TREC_MEAS *tm, TREC_EVAL *eval);
/* Measure is a single long measure, no parameters. */
int te_init_meas_s_long (EPI *epi, TREC_MEAS *tm, TREC_EVAL *eval);
/* Measure is a float array with long cutoffs */
int te_init_meas_a_float_cut_long (EPI *epi, TREC_MEAS *tm, TREC_EVAL *eval);
/* Measure is a float array with float cutoffs */
int te_init_meas_a_float_cut_float (EPI *epi, TREC_MEAS *tm, TREC_EVAL *eval);
/* Measure is a single float with float params */
int te_init_meas_s_float_p_float (EPI *epi, TREC_MEAS *tm, TREC_EVAL *eval);
/* Measure is a single float with paired name=float params */
int te_init_meas_s_float_p_pair (EPI *epi, TREC_MEAS *tm, TREC_EVAL *eval);
/* -------- Accumulate Measure -------- */
/* Code is in meas_acc.c */
/* Measure does not require accumulation */
int te_acc_meas_empty (const EPI *epi, const TREC_MEAS *tm,
const TREC_EVAL *q_eval, TREC_EVAL *accum_eval);
/* Accumulate a single value */
int te_acc_meas_s (const EPI *epi, const TREC_MEAS *tm,
const TREC_EVAL *q_eval, TREC_EVAL *accum_eval);
/* Accumulate an array of values (one per cutoff) */
int te_acc_meas_a_cut (const EPI *epi, const TREC_MEAS *tm,
const TREC_EVAL *q_eval, TREC_EVAL *accum_eval);
/* ----- Calculate Average measure ---- */
/* Code is in meas_calc_avg.c */
/* Measure does not require final averaging */
int te_calc_avg_meas_empty (const EPI *epi, const TREC_MEAS *tm,
const ALL_REL_INFO *all_rel_info, TREC_EVAL *eval);
/* Measure is a single value with averaging */
int te_calc_avg_meas_s (const EPI *epi, const TREC_MEAS *tm,
const ALL_REL_INFO *all_rel_info, TREC_EVAL *accum_eval);
/* Measure is an array of values (one per cutoff) with averaging */
int te_calc_avg_meas_a_cut (const EPI *epi, const TREC_MEAS *tm,
const ALL_REL_INFO *all_rel_info, TREC_EVAL *accum_eval);
/* Measure is a single value using geometric mean */
int te_calc_avg_meas_s_gm (const EPI *epi, const TREC_MEAS *tm,
const ALL_REL_INFO *all_rel_info, TREC_EVAL *accum_eval);
/* ----- Print single query for measure ---- */
/* Code is in meas_print_single.c */
/* Measure does not require printing */
int te_print_single_meas_empty (const EPI *epi, const TREC_MEAS *tm,
const TREC_EVAL *eval);
/* Measure is a single float measure with no parameters, */
int te_print_single_meas_s_float (const EPI *epi, const TREC_MEAS *tm,
const TREC_EVAL *eval);
/* Measure is a single long measure with no parameters. */
int te_print_single_meas_s_long (const EPI *epi, const TREC_MEAS *tm,
const TREC_EVAL *eval);
/* Measure is a float array with cutoffs */
int te_print_single_meas_a_cut (const EPI *epi, const TREC_MEAS *tm,
const TREC_EVAL *eval);
/* ----- Print final query for measure ---- */
/* Code is in meas_print_final.c */
/* Measure does not require printing or storage */
int te_print_final_meas_empty (const EPI *epi, TREC_MEAS *tm,
TREC_EVAL *eval);
/* Measure is a single float measure with no parameters */
int te_print_final_meas_s_float (const EPI *epi, TREC_MEAS *tm,
TREC_EVAL *eval);
/* Measure is a single long measure with no parameters. */
int te_print_final_meas_s_long (const EPI *epi, TREC_MEAS *tm,
TREC_EVAL *eval);
/* Measure is a float array with cutoffs */
int te_print_final_meas_a_cut (const EPI *epi, TREC_MEAS *tm,
TREC_EVAL *eval);
/* Measure is a single float with float params */
int te_print_final_meas_s_float_p (const EPI *epi, TREC_MEAS *tm,
TREC_EVAL *eval);
#endif /* FUNCTIONSH */