mirror of
https://github.com/johnkerl/miller.git
synced 2026-07-30 03:01:39 +00:00
189 lines
10 KiB
C
189 lines
10 KiB
C
#ifndef RVAL_EVALUATORS_H
|
|
#define RVAL_EVALUATORS_H
|
|
#include <stdio.h>
|
|
#include "containers/mlr_dsl_ast.h"
|
|
#include "mapping/rval_evaluator.h"
|
|
|
|
// ================================================================
|
|
// NOTES:
|
|
//
|
|
// * Code here evaluates right-hand-side values (rvals) and return mlrvals (mv_t).
|
|
//
|
|
// * This is used by mlr filter and mlr put.
|
|
//
|
|
// * Unlike most files in Miller which are read top-down (with sufficient
|
|
// static prototypes at the top of the file to keep the compiler happy),
|
|
// please read this one from the bottom up.
|
|
//
|
|
// * Comparison to mlrval.c: the latter is functions from mlrval(s) to
|
|
// mlrval; in this file we have the higher-level notion of evaluating lrec
|
|
// objects, using mlrval.c to do so.
|
|
//
|
|
// * There are two kinds of lrec-evaluators here: those with _x_ in their names
|
|
// which accept various types of mlrval, with disposition-matrices in
|
|
// mlrval.c functions, and those with _i_/_f_/_b_/_s_ (int, float, boolean,
|
|
// string) which either type-check or type-coerce their arguments, invoking
|
|
// type-specific functions in mlrval.c. Those with _n_ take int or float
|
|
// and also use disposition matrices. In all cases it's the job of
|
|
// rval_evaluators.c to invoke functions here with mlrvals of the correct
|
|
// type(s). See also comments in containers/mlrval.h.
|
|
// ================================================================
|
|
|
|
#define TYPE_INFER_STRING_FLOAT_INT 0xce08
|
|
#define TYPE_INFER_STRING_FLOAT 0xce09
|
|
#define TYPE_INFER_STRING_ONLY 0xce0a
|
|
|
|
typedef enum _func_class_t {
|
|
FUNC_CLASS_ARITHMETIC,
|
|
FUNC_CLASS_MATH,
|
|
FUNC_CLASS_BOOLEAN,
|
|
FUNC_CLASS_STRING,
|
|
FUNC_CLASS_CONVERSION,
|
|
FUNC_CLASS_TIME
|
|
} func_class_t;
|
|
|
|
typedef enum _arity_check_t {
|
|
ARITY_CHECK_PASS,
|
|
ARITY_CHECK_FAIL,
|
|
ARITY_CHECK_NO_SUCH
|
|
} arity_check_t;
|
|
|
|
typedef struct _function_lookup_t {
|
|
func_class_t function_class;
|
|
char* function_name;
|
|
int arity;
|
|
char* usage_string;
|
|
} function_lookup_t;
|
|
|
|
extern function_lookup_t FUNCTION_LOOKUP_TABLE[];
|
|
|
|
// ----------------------------------------------------------------
|
|
// rval_expr_evaluators.c
|
|
|
|
// Topmost function:
|
|
rval_evaluator_t* rval_evaluator_alloc_from_ast(mlr_dsl_ast_node_t* past, int type_inferencing, int context_flags);
|
|
|
|
// Next level:
|
|
rval_evaluator_t* rval_evaluator_alloc_from_field_name(char* field_name, int type_inferencing);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_indirect_field_name(mlr_dsl_ast_node_t* pnode,
|
|
int type_inferencing, int context_flags);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_oosvar_keylist(mlr_dsl_ast_node_t* pnode,
|
|
int type_inferencing, int context_flags);
|
|
|
|
// This is used for evaluating strings and numbers in literal expressions, e.g. '$x = "abc"'
|
|
// or '$x = "left_\1". The values are subject to replacement with regex captures. See comments
|
|
// in mapper_put for more information.
|
|
//
|
|
// Compare rval_evaluator_alloc_from_string which doesn't do regex replacement: it is intended for
|
|
// oosvar names on expression left-hand sides (outside of this file).
|
|
rval_evaluator_t* rval_evaluator_alloc_from_strnum_literal(char* string, int type_inferencing);
|
|
|
|
// This is intended only for oosvar names on expression left-hand sides.
|
|
// Compare rval_evaluator_alloc_from_strnum_literal.
|
|
rval_evaluator_t* rval_evaluator_alloc_from_string(char* string);
|
|
|
|
rval_evaluator_t* rval_evaluator_alloc_from_boolean_literal(char* string);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_boolean(int boolval);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_environment(mlr_dsl_ast_node_t* pnode,
|
|
int type_inferencing, int context_flags);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_NF();
|
|
rval_evaluator_t* rval_evaluator_alloc_from_NR();
|
|
rval_evaluator_t* rval_evaluator_alloc_from_FNR();
|
|
rval_evaluator_t* rval_evaluator_alloc_from_FILENAME();
|
|
rval_evaluator_t* rval_evaluator_alloc_from_FILENUM();
|
|
rval_evaluator_t* rval_evaluator_alloc_from_PI();
|
|
rval_evaluator_t* rval_evaluator_alloc_from_E();
|
|
rval_evaluator_t* rval_evaluator_alloc_from_context_variable(char* variable_name);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_bound_variable(char* variable_name);
|
|
|
|
// For unit test:
|
|
rval_evaluator_t* rval_evaluator_alloc_from_mlrval(mv_t* pval);
|
|
|
|
// ----------------------------------------------------------------
|
|
// rval_func_evaluators.c
|
|
|
|
void rval_evaluator_list_functions(FILE* output_stream, char* leader);
|
|
// Pass function_name == NULL to get usage for all functions:
|
|
void rval_evaluator_function_usage(FILE* output_stream, char* function_name);
|
|
void rval_evaluator_list_all_functions_raw(FILE* output_stream);
|
|
void check_arity_with_report(function_lookup_t fcn_lookup_table[], char* function_name,
|
|
int user_provided_arity);
|
|
|
|
rval_evaluator_t* rval_evaluator_alloc_from_zary_func_name(char* function_name);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_unary_func_name(char* fnnm, rval_evaluator_t* parg1);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_binary_func_name(char* fnnm,
|
|
rval_evaluator_t* parg1, rval_evaluator_t* parg2);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_binary_regex_arg2_func_name(char* fnnm,
|
|
rval_evaluator_t* parg1, char* regex_string, int ignore_case);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_ternary_func_name(char* fnnm,
|
|
rval_evaluator_t* parg1, rval_evaluator_t* parg2, rval_evaluator_t* parg3);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_ternary_regex_arg2_func_name(char* fnnm,
|
|
rval_evaluator_t* parg1, char* regex_string, int ignore_case, rval_evaluator_t* parg3);
|
|
|
|
// These have some shared code that would otherwise be duplicated per-function in containers/mlrval.c.
|
|
|
|
rval_evaluator_t* rval_evaluator_alloc_from_b_b_func(mv_unary_func_t* pfunc, rval_evaluator_t* parg1);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_b_bb_and_func(rval_evaluator_t* parg1, rval_evaluator_t* parg2);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_b_bb_or_func(rval_evaluator_t* parg1, rval_evaluator_t* parg2);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_b_bb_xor_func(rval_evaluator_t* parg1, rval_evaluator_t* parg2);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_x_z_func(mv_zary_func_t* pfunc);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_f_f_func(mv_unary_func_t* pfunc, rval_evaluator_t* parg1);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_x_n_func(mv_unary_func_t* pfunc, rval_evaluator_t* parg1);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_i_i_func(mv_unary_func_t* pfunc, rval_evaluator_t* parg1);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_f_ff_func(mv_binary_func_t* pfunc,
|
|
rval_evaluator_t* parg1, rval_evaluator_t* parg2);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_x_xx_func(mv_binary_func_t* pfunc,
|
|
rval_evaluator_t* parg1, rval_evaluator_t* parg2);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_x_xx_nullable_func(mv_binary_func_t* pfunc,
|
|
rval_evaluator_t* parg1, rval_evaluator_t* parg2);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_f_fff_func(mv_ternary_func_t* pfunc,
|
|
rval_evaluator_t* parg1, rval_evaluator_t* parg2, rval_evaluator_t* parg3);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_i_ii_func(mv_binary_func_t* pfunc,
|
|
rval_evaluator_t* parg1, rval_evaluator_t* parg2);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_i_iii_func(mv_ternary_func_t* pfunc,
|
|
rval_evaluator_t* parg1, rval_evaluator_t* parg2, rval_evaluator_t* parg3);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_ternop(rval_evaluator_t* parg1, rval_evaluator_t* parg2,
|
|
rval_evaluator_t* parg3);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_s_s_func(mv_unary_func_t* pfunc, rval_evaluator_t* parg1);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_s_f_func(mv_unary_func_t* pfunc, rval_evaluator_t* parg1);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_s_i_func(mv_unary_func_t* pfunc, rval_evaluator_t* parg1);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_f_s_func(mv_unary_func_t* pfunc, rval_evaluator_t* parg1);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_i_s_func(mv_unary_func_t* pfunc, rval_evaluator_t* parg1);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_x_x_func(mv_unary_func_t* pfunc, rval_evaluator_t* parg1);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_x_ns_func(mv_binary_func_t* pfunc,
|
|
rval_evaluator_t* parg1, rval_evaluator_t* parg2);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_x_ss_func(mv_binary_func_t* pfunc,
|
|
rval_evaluator_t* parg1, rval_evaluator_t* parg2);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_x_ssc_func(mv_binary_arg3_capture_func_t* pfunc,
|
|
rval_evaluator_t* parg1, rval_evaluator_t* parg2);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_x_sr_func(mv_binary_arg2_regex_func_t* pfunc,
|
|
rval_evaluator_t* parg1, char* regex_string, int ignore_case);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_s_xs_func(mv_binary_func_t* pfunc,
|
|
rval_evaluator_t* parg1, rval_evaluator_t* parg2);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_s_sss_func(mv_ternary_func_t* pfunc,
|
|
rval_evaluator_t* parg1, rval_evaluator_t* parg2, rval_evaluator_t* parg3);
|
|
rval_evaluator_t* rval_evaluator_alloc_from_x_srs_func(mv_ternary_arg2_regex_func_t* pfunc,
|
|
rval_evaluator_t* parg1, char* regex_string, int ignore_case, rval_evaluator_t* parg3);
|
|
|
|
// ----------------------------------------------------------------
|
|
// rval_list_evaluators.c
|
|
|
|
// Nominally for oosvar multikeys.
|
|
sllmv_t* evaluate_list(sllv_t* pevaluators, variables_t* pvars, int* pall_non_null_or_error);
|
|
sllmv_t** evaluate_lists(sllv_t** ppevaluators, int num_evaluators, variables_t* pvars, int* pall_non_null_or_error);
|
|
|
|
// ----------------------------------------------------------------
|
|
// Type-inferenced srec-field getters for the expression-evaluators, as well as for boundvars in srec for-loops.
|
|
|
|
// For RHS evaluation:
|
|
mv_t get_srec_value_string_only(char* field_name, lrec_t* pinrec, lhmsmv_t* ptyped_overlay);
|
|
mv_t get_srec_value_string_float(char* field_name, lrec_t* pinrec, lhmsmv_t* ptyped_overlay);
|
|
mv_t get_srec_value_string_float_int(char* field_name, lrec_t* pinrec, lhmsmv_t* ptyped_overlay);
|
|
|
|
// For boundvars in for-srec:
|
|
typedef mv_t type_infererenced_srec_field_getter_t(lrece_t* pentry, lhmsmv_t* ptyped_overlay);
|
|
mv_t get_srec_value_string_only_aux(lrece_t* pentry, lhmsmv_t* ptyped_overlay);
|
|
mv_t get_srec_value_string_float_aux(lrece_t* pentry, lhmsmv_t* ptyped_overlay);
|
|
mv_t get_srec_value_string_float_int_aux(lrece_t* pentry, lhmsmv_t* ptyped_overlay);
|
|
|
|
#endif // LREC_FEVALUATORS_H
|