pintos-os.org Git - pspp/blob - src/language/lexer/macro.h

   1 /* PSPP - a program for statistical analysis.
   2    Copyright (C) 2021 Free Software Foundation, Inc.
   3
   4    This program is free software: you can redistribute it and/or modify
   5    it under the terms of the GNU General Public License as published by
   6    the Free Software Foundation, either version 3 of the License, or
   7    (at your option) any later version.
   8
   9    This program is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12    GNU General Public License for more details.
  13
  14    You should have received a copy of the GNU General Public License
  15    along with this program.  If not, see <http://www.gnu.org/licenses/>. */
  16
  17 #ifndef MACRO_H
  18 #define MACRO_H 1
  19
  20 #include <stdbool.h>
  21 #include <stddef.h>
  22
  23 #include "libpspp/hmap.h"
  24 #include "libpspp/str.h"
  25 #include "language/lexer/segment.h"
  26 #include "language/lexer/token.h"
  27
  28 struct msg_location;
  29
  30 /* A token along with the syntax that was tokenized to produce it.  The syntax
  31    allows the token to be turned back into syntax accurately. */
  32 struct macro_token
  33   {
  34     struct token token;
  35     struct substring syntax;
  36   };
  37
  38 void macro_token_copy (struct macro_token *, const struct macro_token *);
  39 void macro_token_uninit (struct macro_token *);
  40
  41 void macro_token_to_syntax (struct macro_token *, struct string *);
  42
  43 bool is_macro_keyword (struct substring);
  44 \f
  45 /* A dynamic array of macro tokens.
  46
  47    The syntax for the tokens doesn't include white space, etc. between them. */
  48 struct macro_tokens
  49   {
  50     struct macro_token *mts;
  51     size_t n;
  52     size_t allocated;
  53   };
  54
  55 void macro_tokens_copy (struct macro_tokens *, const struct macro_tokens *);
  56 void macro_tokens_uninit (struct macro_tokens *);
  57 struct macro_token *macro_tokens_add_uninit (struct macro_tokens *);
  58 void macro_tokens_add (struct macro_tokens *, const struct macro_token *);
  59
  60 void macro_tokens_from_string (struct macro_tokens *, const struct substring,
  61                                enum segmenter_mode);
  62
  63 void macro_tokens_to_syntax (struct macro_tokens *, struct string *,
  64                              size_t *ofs, size_t *len);
  65
  66 void macro_tokens_print (const struct macro_tokens *, FILE *);
  67 \f
  68 /* A parameter to a macro. */
  69 struct macro_param
  70   {
  71     bool positional;            /* Is this a positional parameter? */
  72     char *name;                 /* "!1" or "!name". */
  73     struct macro_tokens def;    /* Default expansion. */
  74     bool expand_arg;            /* Macro-expand the argument? */
  75
  76     enum
  77       {
  78         ARG_N_TOKENS,
  79         ARG_CHAREND,
  80         ARG_ENCLOSE,
  81         ARG_CMDEND
  82       }
  83     arg_type;
  84     union
  85       {
  86         int n_tokens;            /* ARG_N_TOKENS. */
  87         struct token charend;    /* ARG_CHAREND. */
  88         struct token enclose[2]; /* ARG_ENCLOSE. */
  89       };
  90   };
  91
  92 /* A macro. */
  93 struct macro
  94   {
  95     struct hmap_node hmap_node; /* Indexed by 'name'. */
  96     char *name;
  97
  98     /* Source code location of macro definition, for error reporting. */
  99     struct msg_location *location;
 100
 101     /* Parameters. */
 102     struct macro_param *params;
 103     size_t n_params;
 104
 105     /* Body. */
 106     struct macro_tokens body;
 107   };
 108
 109 void macro_destroy (struct macro *);
 110 \f
 111 /* A collection of macros. */
 112 struct macro_set
 113   {
 114     struct hmap macros;
 115   };
 116
 117 struct macro_set *macro_set_create (void);
 118 void macro_set_destroy (struct macro_set *);
 119 const struct macro *macro_set_find (const struct macro_set *,
 120                                     const char *);
 121 void macro_set_add (struct macro_set *, struct macro *);
 122
 123 static inline bool
 124 macro_set_is_empty (const struct macro_set *set)
 125 {
 126   return hmap_is_empty (&set->macros);
 127 }
 128 \f
 129 /* Parsing and expanding macro calls. */
 130
 131 struct macro_call;
 132
 133 int macro_call_create (const struct macro_set *, const struct token *,
 134                        struct macro_call **);
 135 int macro_call_add (struct macro_call *, const struct macro_token *,
 136                     const struct msg_location *);
 137
 138 void macro_call_expand (struct macro_call *, enum segmenter_mode segmenter_mode,
 139                         const struct msg_location *call_loc, struct macro_tokens *);
 140
 141 void macro_call_destroy (struct macro_call *);
 142
 143 #endif /* macro.h */