https://github.com/akkartik/mu/blob/master/029tools.cc
  1 //: Allow Mu programs to log facts just like we've been doing in C++ so far.
  2 
  3 :(scenario trace)
  4 def main [
  5   trace 1, [foo], [this is a trace in Mu]
  6 ]
  7 +foo: this is a trace in Mu
  8 
  9 :(before "End Primitive Recipe Declarations")
 10 TRACE,
 11 :(before "End Primitive Recipe Numbers")
 12 put(Recipe_ordinal, "trace", TRACE);
 13 :(before "End Primitive Recipe Checks")
 14 case TRACE: {
 15   if (SIZE(inst.ingredients) < 3) {
 16     raise << maybe(get(Recipe, r).name) << "'trace' takes three or more ingredients rather than '" << to_original_string(inst) << "'\n" << end();
 17     break;
 18   }
 19   if (!is_mu_number(inst.ingredients.at(0))) {
 20     raise << maybe(get(Recipe, r).name) << "first ingredient of 'trace' should be a number (depth), but got '" << inst.ingredients.at(0).original_string << "'\n" << end();
 21     break;
 22   }
 23   if (!is_literal_text(inst.ingredients.at(1))) {
 24     raise << maybe(get(Recipe, r).name) << "second ingredient of 'trace' should be a literal string (label), but got '" << inst.ingredients.at(1).original_string << "'\n" << end();
 25     break;
 26   }
 27   break;
 28 }
 29 :(before "End Primitive Recipe Implementations")
 30 case TRACE: {
 31   int depth = ingredients.at(0).at(0);
 32   string label = current_instruction().ingredients.at(1).name;
 33   ostringstream out;
 34   for (int i = 2;  i < SIZE(current_instruction().ingredients);  ++i) {
 35     if (i > 2) out << ' ';
 36     out << inspect(current_instruction().ingredients.at(i), ingredients.at(i));
 37   }
 38   trace(depth, label) << out.str() << end();
 39   break;
 40 }
 41 
 42 //: simpler limited version of 'trace'
 43 
 44 :(before "End Primitive Recipe Declarations")
 45 STASH,
 46 :(before "End Primitive Recipe Numbers")
 47 put(Recipe_ordinal, "stash", STASH);
 48 :(before "End Primitive Recipe Checks")
 49 case STASH: {
 50   break;
 51 }
 52 :(before "End Primitive Recipe Implementations")
 53 case STASH: {
 54   ostringstream out;
 55   for (int i = 0;  i < SIZE(current_instruction().ingredients);  ++i) {
 56     if (i) out << ' ';
 57     out << inspect(current_instruction().ingredients.at(i), ingredients.at(i));
 58   }
 59   trace(2, "app") << out.str() << end();
 60   break;
 61 }
 62 
 63 :(scenario stash_literal_string)
 64 def main [
 65   stash [foo]
 66 ]
 67 +app: foo
 68 
 69 :(scenario stash_literal_number)
 70 def main [
 71   stash [foo:], 4
 72 ]
 73 +app: foo: 4
 74 
 75 :(scenario stash_number)
 76 def main [
 77   1:num <- copy 34
 78   stash [foo:], 1:num
 79 ]
 80 +app: foo: 34
 81 
 82 :(code)
 83 string inspect(const reagent& r, const vector<double>& data) {
 84   if (is_literal(r))
 85     return r.name;
 86   // End inspect Special-cases(r, data)
 87   ostringstream out;
 88   for (long long i = 0;  i < SIZE(data);  ++i) {
 89     if (i) out << ' ';
 90     out << no_scientific(data.at(i));
 91   }
 92   return out.str();
 93 }
 94 
 95 :(before "End Primitive Recipe Declarations")
 96 HIDE_ERRORS,
 97 :(before "End Primitive Recipe Numbers")
 98 put(Recipe_ordinal, "hide-errors", HIDE_ERRORS);
 99 :(before "End Primitive Recipe Checks")
100 case HIDE_ERRORS: {
101   break;
102 }
103 :(before "End Primitive Recipe Implementations")
104 case HIDE_ERRORS: {
105   Hide_errors = true;
106   break;
107 }
108 
109 :(before "End Primitive Recipe Declarations")
110 SHOW_ERRORS,
111 :(before "End Primitive Recipe Numbers")
112 put(Recipe_ordinal, "show-errors", SHOW_ERRORS);
113 :(before "End Primitive Recipe Checks")
114 case SHOW_ERRORS: {
115   break;
116 }
117 :(before "End Primitive Recipe Implementations")
118 case SHOW_ERRORS: {
119   Hide_errors = false;
120   break;
121 }
122 
123 :(before "End Primitive Recipe Declarations")
124 TRACE_UNTIL,
125 :(before "End Primitive Recipe Numbers")
126 put(Recipe_ordinal, "trace-until", TRACE_UNTIL);
127 :(before "End Primitive Recipe Checks")
128 case TRACE_UNTIL: {
129   break;
130 }
131 :(before "End Primitive Recipe Implementations")
132 case TRACE_UNTIL: {
133   if (Trace_stream) {
134     Trace_stream->collect_depth = ingredients.at(0).at(0);
135   }
136   break;
137 }
138 
139 :(before "End Primitive Recipe Declarations")
140 _DUMP_TRACE,
141 :(before "End Primitive Recipe Numbers")
142 put(Recipe_ordinal, "$dump-trace", _DUMP_TRACE);
143 :(before "End Primitive Recipe Checks")
144 case _DUMP_TRACE: {
145   break;
146 }
147 :(before "End Primitive Recipe Implementations")
148 case _DUMP_TRACE: {
149   if (ingredients.empty()) {
150     DUMP("");
151   }
152   else {
153     DUMP(current_instruction().ingredients.at(0).name);
154   }
155   break;
156 }
157 
158 :(before "End Primitive Recipe Declarations")
159 _CLEAR_TRACE,
160 :(before "End Primitive Recipe Numbers")
161 put(Recipe_ordinal, "$clear-trace", _CLEAR_TRACE);
162 :(before "End Primitive Recipe Checks")
163 case _CLEAR_TRACE: {
164   break;
165 }
166 :(before "End Primitive Recipe Implementations")
167 case _CLEAR_TRACE: {
168   if (Trace_stream) Trace_stream->past_lines.clear();
169   break;
170 }
171 
172 :(before "End Primitive Recipe Declarations")
173 _SAVE_TRACE,
174 :(before "End Primitive Recipe Numbers")
175 put(Recipe_ordinal, "$save-trace", _SAVE_TRACE);
176 :(before "End Primitive Recipe Checks")
177 case _SAVE_TRACE: {
178   break;
179 }
180 :(before "End Primitive Recipe Implementations")
181 case _SAVE_TRACE: {
182   if (Save_trace) Trace_stream->save();
183   break;
184 }
185 
186 //:: 'cheating' by using the host system
187 
188 :(before "End Primitive Recipe Declarations")
189 _PRINT,
190 :(before "End Primitive Recipe Numbers")
191 put(Recipe_ordinal, "$print", _PRINT);
192 :(before "End Primitive Recipe Checks")
193 case _PRINT: {
194   break;
195 }
196 :(before "End Primitive Recipe Implementations")
197 case _PRINT: {
198   for (int i = 0;  i < SIZE(ingredients);  ++i) {
199     if (is_literal(current_instruction().ingredients.at(i))) {
200       trace(9998, "run") << "$print: " << current_instruction().ingredients.at(i).name << end();
201       if (!has_property(current_instruction().ingredients.at(i), "newline")) {
202         cout << current_instruction().ingredients.at(i).name;
203       }
204       // hack: '$print 10' prints '10', but '$print 10/newline' prints '\n'
205       // End $print 10/newline Special-cases
206       else {
207         cout << '\n';
208       }
209     }
210     // End $print Special-cases
211     else {
212       for (int j = 0;  j < SIZE(ingredients.at(i));  ++j) {
213         trace(9998, "run") << "$print: " << ingredients.at(i).at(j) << end();
214         if (j > 0) cout << " ";
215         cout << no_scientific(ingredients.at(i).at(j));
216       }
217     }
218   }
219   cout.flush();
220   break;
221 }
222 
223 :(before "End Primitive Recipe Declarations")
224 _EXIT,
225 :(before "End Primitive Recipe Numbers")
226 put(Recipe_ordinal, "$exit", _EXIT);
227 :(before "End Primitive Recipe Checks")
228 case _EXIT: {
229   break;
230 }
231 :(before "End Primitive Recipe Implementations")
232 case _EXIT: {
233   exit(0);
234   break;
235 }
236 
237 :(before "End Primitive Recipe Declarations")
238 _SYSTEM,
239 :(before "End Primitive Recipe Numbers")
240 put(Recipe_ordinal, "$system", _SYSTEM);
241 :(before "End Primitive Recipe Checks")
242 case _SYSTEM: {
243   if (SIZE(inst.ingredients) != 1) {
244     raise << maybe(get(Recipe, r).name) << "'$system' requires exactly one ingredient, but got '" << to_string(inst) << "'\n" << end();
245     break;
246   }
247   if (!is_literal_text(inst.ingredients.at(0))) {
248     raise << maybe(get(Recipe, r).name) << "ingredient to '$system' must be a literal text, but got '" << to_string(inst) << "'\n" << end();
249   }
250   break;
251 }
252 :(before "End Primitive Recipe Implementations")
253 case _SYSTEM: {
254   int status = system(current_instruction().ingredients.at(0).name.c_str());
255   products.resize(1);
256   products.at(0).push_back(status);
257   break;
258 }
259 
260 :(before "End Primitive Recipe Declarations")
261 _DUMP_MEMORY,
262 :(before "End Primitive Recipe Numbers")
263 put(Recipe_ordinal, "$dump-memory", _DUMP_MEMORY);
264 :(before "End Primitive Recipe Checks")
265 case _DUMP_MEMORY: {
266   break;
267 }
268 :(before "End Primitive Recipe Implementations")
269 case _DUMP_MEMORY: {
270   dump_memory();
271   break;
272 }
273 
274 //: In times of real extremis we need to create a whole new modality for debug
275 //: logs, independent of other changes to the screen or Trace_stream.
276 
277 :(before "End Globals")
278 ofstream LOG;
279 :(before "End One-time Setup")
280 //? LOG.open("log");
281 
282 :(before "End Primitive Recipe Declarations")
283 _LOG,
284 :(before "End Primitive Recipe Numbers")
285 put(Recipe_ordinal, "$log", _LOG);
286 :(before "End Primitive Recipe Checks")
287 case _LOG: {
288   break;
289 }
290 :(before "End Primitive Recipe Implementations")
291 case _LOG: {
292   ostringstream out;
293   for (int i = 0;  i < SIZE(current_instruction().ingredients);  ++i) {
294     out << inspect(current_instruction().ingredients.at(i), ingredients.at(i));
295   }
296   LOG << out.str() << '\n';
297   break;
298 }
299 
300 //: set a variable from within Mu code
301 //: useful for selectively tracing or printing after some point
302 :(before "End Globals")
303 bool Foo = false;
304 :(before "End Primitive Recipe Declarations")
305 _FOO,
306 :(before "End Primitive Recipe Numbers")
307 put(Recipe_ordinal, "$foo", _FOO);
308 :(before "End Primitive Recipe Checks")
309 case _FOO: {
310   break;
311 }
312 :(before "End Primitive Recipe Implementations")
313 case _FOO: {
314   Foo = true;
315   break;
316 }