1 //: A debugging helper that lets you zoom in/out on a trace.
  2 //: Warning: this tool has zero automated tests.
  3 //:
  4 //: To try it out, first create an example trace:
  5 //:   mu --trace nqueens.mu
  6 //: Then to browse the trace, which was stored in a file called 'last_run':
  7 //:   mu browse-trace last_run
  8 //:
  9 //: You should now find yourself in a UI showing a subsequence of lines from
 10 //: the trace, each line starting with a numeric depth, and ending with a
 11 //: parenthetical count of trace lines hidden after it with greater depths.
 12 //:
 13 //: For example, this line:
 14 //:   2 app: line1 (30)
 15 //: indicates that it was logged with depth 2, and that 30 following lines
 16 //: have been hidden at a depth greater than 2.
 17 //:
 18 //: As an experiment, hidden counts of 1000 or more are in red to highlight
 19 //: where you might be particularly interested in expanding.
 20 //:
 21 //: The UI provides the following hotkeys:
 22 //:
 23 //:   `q` or `ctrl-c`: Quit.
 24 //:
 25 //:   `Enter`: 'Zoom into' this line. Expand some or all of the hidden lines
 26 //:   at the next higher level, updating parenthetical counts of hidden lines.
 27 //:
 28 //:   `Backspace`: 'Zoom out' on a line after zooming in, collapsing expanded
 29 //:   lines below by some series of <Enter> commands.
 30 //:
 31 //:   `j` or `down-arrow`: Move/scroll cursor down one line.
 32 //:   `k` or `up-arrow`: Move/scroll cursor up one line.
 33 //:   `J` or `ctrl-f` or `page-down`: Scroll cursor down one page.
 34 //:   `K` or `ctrl-b` or `page-up`: Scroll cursor up one page.
 35 //:   `h` or `left-arrow`: Scroll cursor left one character.
 36 //:   `l` or `right-arrow`: Scroll cursor right one character.
 37 //:   `H`: Scroll cursor left one screen-width.
 38 //:   `L`: Scroll cursor right one screen-width.
 39 //:
 40 //:   `g` or `home`: Move cursor to start of trace.
 41 //:   `G` or `end`: Move cursor to end of trace.
 42 //:
 43 //:   `t`: Move cursor to top line on screen.
 44 //:   `c`: Move cursor to center line on screen.
 45 //:   `b`: Move cursor to bottom line on screen.
 46 //:   `T`: Scroll line at cursor to top of screen.
 47 //:
 48 //:   `/`: Search forward for a pattern.
 49 //:   '?': Search backward for a pattern.
 50 //:   `n`: Repeat the previous '/' or '?'.
 51 //:   `N`: Repeat the previous '/' or '?' in the opposite direction.
 52 //:
 53 //:   After hitting `/`, a small editor on the bottom-most line supports the
 54 //:   following hotkeys:
 55 //:     ascii characters: add the key to the pattern.
 56 //:     `Enter`: search for the pattern.
 57 //:     `Esc` or `ctrl-c`: cancel the current search, setting the screen back
 58 //:       to its state before the search.
 59 //:     `left-arrow`: move cursor left.
 60 //:     `right-arrow`: move cursor right.
 61 //:     `ctrl-a` or `home`: move cursor to start of search pattern.
 62 //:     `ctrl-e` or `end`: move cursor to end of search pattern.
 63 //:     `ctrl-u`: clear search pattern before cursor
 64 //:     `ctrl-k`: clear search pattern at and after cursor
 65 
 66 :(before "End Primitive Recipe Declarations")
 67 _BROWSE_TRACE,
 68 :(before "End Primitive Recipe Numbers")
 69 put(Recipe_ordinal, "$browse-trace", _BROWSE_TRACE);
 70 :(before "End Primitive Recipe Checks")
 71 case _BROWSE_TRACE: {
 72   break;
 73 }
 74 :(before "End Primitive Recipe Implementations")
 75 case _BROWSE_TRACE: {
 76   start_trace_browser();
 77   break;
 78 }
 79 
 80 //: browse a trace loaded from a file
 81 :(after "Commandline Parsing")
 82 if (argc == 3 && is_equal(argv[1], "browse-trace")) {
 83   load_trace(argv[2]);
 84   start_trace_browser();
 85   return 0;
 86 }
 87 
 88 :(before "End Globals")
 89 set<int> Visible;
 90 int Top_of_screen = 0;
 91 int Left_of_screen = 0;
 92 int Last_printed_row = 0;
 93 map<int, int> Trace_index;  // screen row -> trace index
 94 string Current_search_pattern = "";
 95 :(before "End Types")
 96 enum search_direction { FORWARD, BACKWARD };
 97 :(before "End Globals")
 98 search_direction Current_search_direction = FORWARD;
 99 
100 :(code)
101 void start_trace_browser() {
102   if (!Trace_stream) return;
103   cerr << "computing min depth to display\n";
104   int min_depth = 9999;
105   for (int i = 0;  i < SIZE(Trace_stream->past_lines);  ++i) {
106   ¦ trace_line& curr_line = Trace_stream->past_lines.at(i);
107   ¦ if (curr_line.depth < min_depth) min_depth = curr_line.depth;
108   }
109   cerr << "min depth is " << min_depth << '\n';
110   cerr << "computing lines to display\n";
111   for (int i = 0;  i < SIZE(Trace_stream->past_lines);  ++i) {
112   ¦ if (Trace_stream->past_lines.at(i).depth == min_depth)
113   ¦ ¦ Visible.insert(i);
114   }
115   tb_init();
116   Display_row = Display_column = 0;
117   Top_of_screen = 0;
118   refresh_screen_rows();
119   while (true) {
120   ¦ render();
121   ¦ int key = read_key();
122   ¦ if (key == 'q' || key == 'Q' || key == TB_KEY_CTRL_C) break;
123   ¦ if (key == 'j' || key == TB_KEY_ARROW_DOWN) {
124   ¦ ¦ // move cursor one line down
125   ¦ ¦ if (Display_row < Last_printed_row) ++Display_row;
126   ¦ }
127   ¦ else if (key == 'k' || key == TB_KEY_ARROW_UP) {
128   ¦ ¦ // move cursor one line up
129   ¦ ¦ if (Display_row > 0) --Display_row;
130   ¦ }
131   ¦ else if (key == 't') {
132   ¦ ¦ // move cursor to top of screen
133   ¦ ¦ Display_row = 0;
134   ¦ }
135   ¦ else if (key == 'c') {
136   ¦ ¦ // move cursor to center of screen
137   ¦ ¦ Display_row = tb_height()/2;
138   ¦ ¦ while (!contains_key(Trace_index, Display_row))
139   ¦ ¦ ¦ --Display_row;
140   ¦ }
141   ¦ else if (key == 'b') {
142   ¦ ¦ // move cursor to bottom of screen
143   ¦ ¦ Display_row = tb_height()-1;
144   ¦ ¦ while (!contains_key(Trace_index, Display_row))
145   ¦ ¦ ¦ --Display_row;
146   ¦ }
147   ¦ else if (key == 'T') {
148   ¦ ¦ // scroll line at cursor to top of screen
149   ¦ ¦ Top_of_screen = get(Trace_index, Display_row);
150   ¦ ¦ Display_row = 0;
151   ¦ ¦ refresh_screen_rows();
152   ¦ }
153   ¦ else if (key == 'h' || key == TB_KEY_ARROW_LEFT) {
154   ¦ ¦ // pan screen one character left
155   ¦ ¦ if (Left_of_screen > 0) --Left_of_screen;
156   ¦ }
157   ¦ else if (key == 'l' || key == TB_KEY_ARROW_RIGHT) {
158   ¦ ¦ // pan screen one character right
159   ¦ ¦ ++Left_of_screen;
160   ¦ }
161   ¦ else if (key == 'H') {
162   ¦ ¦ // pan screen one screen-width left
163   ¦ ¦ Left_of_screen -= (tb_width() - 5);
164   ¦ ¦ if (Left_of_screen < 0) Left_of_screen = 0;
165   ¦ }
166   ¦ else if (key == 'L') {
167   ¦ ¦ // pan screen one screen-width right
168   ¦ ¦ Left_of_screen += (tb_width() - 5);
169   ¦ }
170   ¦ else if (key == 'J' || key == TB_KEY_PGDN || key == TB_KEY_CTRL_F) {
171   ¦ ¦ // page-down
172   ¦ ¦ if (Trace_index.find(tb_height()-1) != Trace_index.end()) {
173   ¦ ¦ ¦ Top_of_screen = get(Trace_index, tb_height()-1) + 1;
174   ¦ ¦ ¦ refresh_screen_rows();
175   ¦ ¦ }
176   ¦ }
177   ¦ else if (key == 'K' || key == TB_KEY_PGUP || key == TB_KEY_CTRL_B) {
178   ¦ ¦ // page-up is more convoluted
179   ¦ ¦ for (int screen_row = tb_height();  screen_row > 0 && Top_of_screen > 0;  --screen_row) {
180   ¦ ¦ ¦ --Top_of_screen;
181   ¦ ¦ ¦ if (Top_of_screen <= 0) break;
182   ¦ ¦ ¦ while (Top_of_screen > 0 && !contains_key(Visible, Top_of_screen))
183   ¦ ¦ ¦ ¦ --Top_of_screen;
184   ¦ ¦ }
185   ¦ ¦ if (Top_of_screen >= 0)
186   ¦ ¦ ¦ refresh_screen_rows();
187   ¦ }
188   ¦ else if (key == 'g' || key == TB_KEY_HOME) {
189   ¦ ¦ ¦ Top_of_screen = 0;
190   ¦ ¦ ¦ Last_printed_row = 0;
191   ¦ ¦ ¦ Display_row = 0;
192   ¦ ¦ ¦ refresh_screen_rows();
193   ¦ }
194   ¦ else if (key == 'G' || key == TB_KEY_END) {
195   ¦ ¦ // go to bottom of screen; largely like page-up, interestingly
196   ¦ ¦ Top_of_screen = SIZE(Trace_stream->past_lines)-1;
197   ¦ ¦ for (int screen_row = tb_height();  screen_row > 0 && Top_of_screen > 0;  --screen_row) {
198   ¦ ¦ ¦ --Top_of_screen;
199   ¦ ¦ ¦ if (Top_of_screen <= 0) break;
200   ¦ ¦ ¦ while (Top_of_screen > 0 && !contains_key(Visible, Top_of_screen))
201   ¦ ¦ ¦ ¦ --Top_of_screen;
202   ¦ ¦ }
203   ¦ ¦ refresh_screen_rows();
204   ¦ ¦ // move cursor to bottom
205   ¦ ¦ Display_row = Last_printed_row;
206   ¦ ¦ refresh_screen_rows();
207   ¦ }
208   ¦ else if (key == TB_KEY_CARRIAGE_RETURN) {
209   ¦ ¦ // expand lines under current by one level
210   ¦ ¦ assert(contains_key(Trace_index, Display_row));
211   ¦ ¦ int start_index = get(Trace_index, Display_row);
212   ¦ ¦ int index = 0;
213   ¦ ¦ // simultaneously compute end_index and min_depth
214   ¦ ¦ int min_depth = 9999;
215   ¦ ¦ for (index = start_index+1;  index < SIZE(Trace_stream->past_lines);  ++index) {
216   ¦ ¦ ¦ if (contains_key(Visible, index)) break;
217   ¦ ¦ ¦ trace_line& curr_line = Trace_stream->past_lines.at(index);
218   ¦ ¦ ¦ assert(curr_line.depth > Trace_stream->past_lines.at(start_index).depth);
219   ¦ ¦ ¦ if (curr_line.depth < min_depth) min_depth = curr_line.depth;
220   ¦ ¦ }
221   ¦ ¦ int end_index = index;
222   ¦ ¦ // mark as visible all intervening indices at min_depth
223   ¦ ¦ for (index = start_index;  index < end_index;  ++index) {
224   ¦ ¦ ¦ trace_line& curr_line = Trace_stream->past_lines.at(index);
225   ¦ ¦ ¦ if (curr_line.depth == min_depth) {
226   ¦ ¦ ¦ ¦ Visible.insert(index);
227   ¦ ¦ ¦ }
228   ¦ ¦ }
229   ¦ ¦ refresh_screen_rows();
230   ¦ }
231   ¦ else if (key == TB_KEY_BACKSPACE || key == TB_KEY_BACKSPACE2) {
232   ¦ ¦ // collapse all lines under current
233   ¦ ¦ assert(contains_key(Trace_index, Display_row));
234   ¦ ¦ int start_index = get(Trace_index, Display_row);
235   ¦ ¦ int index = 0;
236   ¦ ¦ // end_index is the next line at a depth same as or lower than start_index
237   ¦ ¦ int initial_depth = Trace_stream->past_lines.at(start_index).depth;
238   ¦ ¦ for (index = start_index+1;  index < SIZE(Trace_stream->past_lines);  ++index) {
239   ¦ ¦ ¦ if (!contains_key(Visible, index)) continue;
240   ¦ ¦ ¦ trace_line& curr_line = Trace_stream->past_lines.at(index);
241   ¦ ¦ ¦ if (curr_line.depth <= initial_depth) break;
242   ¦ ¦ }
243   ¦ ¦ int end_index = index;
244   ¦ ¦ // mark as visible all intervening indices at min_depth
245   ¦ ¦ for (index = start_index+1;  index < end_index;  ++index) {
246   ¦ ¦ ¦ Visible.erase(index);
247   ¦ ¦ }
248   ¦ ¦ refresh_screen_rows();
249   ¦ }
250   ¦ else if (key == '/') {
251   ¦ ¦ if (start_search_editor(FORWARD))
252   ¦ ¦ ¦ search(Current_search_pattern, Current_search_direction);
253   ¦ }
254   ¦ else if (key == '?') {
255   ¦ ¦ if (start_search_editor(BACKWARD))
256   ¦ ¦ ¦ search(Current_search_pattern, Current_search_direction);
257   ¦ }
258   ¦ else if (key == 'n') {
259   ¦ ¦ if (!Current_search_pattern.empty())
260   ¦ ¦ ¦ search(Current_search_pattern, Current_search_direction);
261   ¦ }
262   ¦ else if (key == 'N') {
263   ¦ ¦ if (!Current_search_pattern.empty())
264   ¦ ¦ ¦ search(Current_search_pattern, opposite(Current_search_direction));
265   ¦ }
266   }
267   tb_shutdown();
268 }
269 
270 bool start_search_editor(search_direction dir) {
271   const int bottom_screen_line = tb_height()-1;
272   // run a little editor just in the last line of the screen
273   clear_line(bottom_screen_line);
274   tb_set_cursor(0, bottom_screen_line);
275   int col = 0;  // screen column of cursor on bottom line. also used to update pattern.
276   tb_change_cell(col, bottom_screen_line, '/', TB_WHITE, TB_BLACK);
277   ++col;
278   tb_set_cursor(col, bottom_screen_line);
279   tb_present();
280   string pattern;
281   while (true) {
282   ¦ int key = read_key();
283   ¦ if (key == TB_KEY_ENTER) {
284   ¦ ¦ if (!pattern.empty()) {
285   ¦ ¦ ¦ Current_search_pattern = pattern;
286   ¦ ¦ ¦ Current_search_direction = dir;
287   ¦ ¦ }
288   ¦ ¦ return true;
289   ¦ }
290   ¦ else if (key == TB_KEY_ESC || key == TB_KEY_CTRL_C) {
291   ¦ ¦ return false;
292   ¦ }
293   ¦ else if (key == TB_KEY_ARROW_LEFT) {
294   ¦ ¦ if (col > /*slash*/1) {
295   ¦ ¦ ¦ --col;
296   ¦ ¦ ¦ tb_set_cursor(col, bottom_screen_line);
297   ¦ ¦ ¦ tb_present();
298   ¦ ¦ }
299   ¦ }
300   ¦ else if (key == TB_KEY_ARROW_RIGHT) {
301   ¦ ¦ if (col-/*slash*/1 < SIZE(pattern)) {
302   ¦ ¦ ¦ ++col;
303   ¦ ¦ ¦ tb_set_cursor(col, bottom_screen_line);
304   ¦ ¦ ¦ tb_present();
305   ¦ ¦ }
306   ¦ }
307   ¦ else if (key == TB_KEY_HOME || key == TB_KEY_CTRL_A) {
308   ¦ ¦ col = /*skip slash*/1;
309   ¦ ¦ tb_set_cursor(col, bottom_screen_line);
310   ¦ ¦ tb_present();
311   ¦ }
312   ¦ else if (key == TB_KEY_END || key == TB_KEY_CTRL_E) {
313   ¦ ¦ col = SIZE(pattern)+/*skip slash*/1;
314   ¦ ¦ tb_set_cursor(col, bottom_screen_line);
315   ¦ ¦ tb_present();
316   ¦ }
317   ¦ else if (key == TB_KEY_BACKSPACE || key == TB_KEY_BACKSPACE2) {
318   ¦ ¦ if (col > /*slash*/1) {
319   ¦ ¦ ¦ --col;
320   ¦ ¦ ¦ // update pattern
321   ¦ ¦ ¦ pattern.erase(col-/*slash*/1, /*len*/1);
322   ¦ ¦ ¦ // update screen
323   ¦ ¦ ¦ if (col > SIZE(pattern)) {
324   ¦ ¦ ¦ ¦ tb_change_cell(col, bottom_screen_line, ' ', TB_WHITE, TB_BLACK);
325   ¦ ¦ ¦ }
326   ¦ ¦ ¦ else {
327   ¦ ¦ ¦ ¦ assert(col <= SIZE(pattern));
328   ¦ ¦ ¦ ¦ for (int x = col;  x < SIZE(pattern)+/*skip slash*/1;  ++x)
329   ¦ ¦ ¦ ¦ ¦ tb_change_cell(x, bottom_screen_line, pattern.at(x-/*slash*/1), TB_WHITE, TB_BLACK);
330   ¦ ¦ ¦ ¦ tb_change_cell(SIZE(pattern)+/*skip slash*/1, bottom_screen_line, ' ', TB_WHITE, TB_BLACK);
331   ¦ ¦ ¦ }
332   ¦ ¦ ¦ tb_set_cursor(col, bottom_screen_line);
333   ¦ ¦ ¦ tb_present();
334   ¦ ¦ }
335   ¦ }
336   ¦ else if (key == TB_KEY_CTRL_K) {
337   ¦ ¦ int old_pattern_size = SIZE(pattern);
338   ¦ ¦ pattern.erase(col-/*slash*/1, SIZE(pattern) - (col-/*slash*/1));
339   ¦ ¦ for (int x = col;  x < old_pattern_size+/*slash*/1;  ++x)
340   ¦ ¦ ¦ tb_change_cell(x, bottom_screen_line, ' ', TB_WHITE, TB_BLACK);
341   ¦ ¦ tb_set_cursor(col, bottom_screen_line);
342   ¦ ¦ tb_present();
343   ¦ }
344   ¦ else if (key == TB_KEY_CTRL_U) {
345   ¦ ¦ int old_pattern_size = SIZE(pattern);
346   ¦ ¦ pattern.erase(0, col-/*slash*/1);
347   ¦ ¦ for (int x = /*slash*/1;  x < SIZE(pattern)+/*skip slash*/1;  ++x)
348   ¦ ¦ ¦ tb_change_cell(x, bottom_screen_line, pattern.at(x-/*slash*/1), TB_WHITE, TB_BLACK);
349   ¦ ¦ for (int x = SIZE(pattern)+/*slash*/1;  x < old_pattern_size+/*skip slash*/1;  ++x)
350   ¦ ¦ ¦ tb_change_cell(x, bottom_screen_line, ' ', TB_WHITE, TB_BLACK);
351   ¦ ¦ tb_set_cursor(/*start of pattern skipping slash*/1, bottom_screen_line);
352   ¦ ¦ tb_present();
353   ¦ }
354   ¦ else if (key < 128) {  // ascii only
355   ¦ ¦ // update pattern
356   ¦ ¦ char c = static_cast<char>(key);
357   ¦ ¦ pattern.insert(col-/*slash*/1, /*num*/1, c);
358   ¦ ¦ // update screen
359   ¦ ¦ for (int x = col;  x < SIZE(pattern)+/*skip slash*/1;  ++x)
360   ¦ ¦ ¦ tb_change_cell(x, bottom_screen_line, pattern.at(x-/*slash*/1), TB_WHITE, TB_BLACK);
361   ¦ ¦ ++col;
362   ¦ ¦ tb_set_cursor(col, bottom_screen_line);
363   ¦ ¦ tb_present();
364   ¦ }
365   }
366 }
367 
368 void search(const string& pat, search_direction dir) {
369   if (dir == FORWARD) search_next(pat);
370   else search_previous(pat);
371 }
372 
373 search_direction opposite(search_direction dir) {
374   if (dir == FORWARD) return BACKWARD;
375   else return FORWARD;
376 }
377 
378 void search_next(const string& pat) {
379   for (int trace_index = get(Trace_index, Display_row)+1;  trace_index < SIZE(Trace_stream->past_lines);  ++trace_index) {
380   ¦ if (!contains_key(Visible, trace_index)) continue;
381   ¦ const trace_line& line = Trace_stream->past_lines.at(trace_index);
382   ¦ if (line.label.find(pat) == string::npos && line.contents.find(pat) == string::npos) continue;
383   ¦ Top_of_screen = trace_index;
384   ¦ Display_row = 0;
385   ¦ refresh_screen_rows();
386   ¦ return;
387   }
388 }
389 
390 void search_previous(const string& pat) {
391   for (int trace_index = get(Trace_index, Display_row)-1;  trace_index >= 0;  --trace_index) {
392   ¦ if (!contains_key(Visible, trace_index)) continue;
393   ¦ const trace_line& line = Trace_stream->past_lines.at(trace_index);
394   ¦ if (line.label.find(pat) == string::npos && line.contents.find(pat) == string::npos) continue;
395   ¦ Top_of_screen = trace_index;
396   ¦ Display_row = 0;
397   ¦ refresh_screen_rows();
398   ¦ return;
399   }
400 }
401 
402 void clear_line(int screen_row) {
403   for (int col = 0;  col < tb_width();  ++col)
404   ¦ tb_change_cell(col, screen_row, ' ', TB_WHITE, TB_BLACK);
405 }
406 
407 // update Trace_indices for each screen_row on the basis of Top_of_screen and Visible
408 void refresh_screen_rows() {
409   int screen_row = 0, index = 0;
410   Trace_index.clear();
411   for (screen_row = 0, index = Top_of_screen;  screen_row < tb_height() && index < SIZE(Trace_stream->past_lines);  ++screen_row, ++index) {
412   ¦ // skip lines without depth for now
413   ¦ while (!contains_key(Visible, index)) {
414   ¦ ¦ ++index;
415   ¦ ¦ if (index >= SIZE(Trace_stream->past_lines)) goto done;
416   ¦ }
417   ¦ assert(index < SIZE(Trace_stream->past_lines));
418   ¦ put(Trace_index, screen_row, index);
419   }
420 done:;
421 }
422 
423 void render() {
424   int screen_row = 0;
425   for (screen_row = 0;  screen_row < tb_height();  ++screen_row) {
426   ¦ if (!contains_key(Trace_index, screen_row)) break;
427   ¦ trace_line& curr_line = Trace_stream->past_lines.at(get(Trace_index, screen_row));
428   ¦ ostringstream out;
429   ¦ out << std::setw(4) << curr_line.depth << ' ' << curr_line.label << ": " << curr_line.contents;
430   ¦ if (screen_row < tb_height()-1) {
431   ¦ ¦ int delta = lines_hidden(screen_row);
432   ¦ ¦ // home-brew escape sequence for red
433   ¦ ¦ if (delta > 1) {
434   ¦ ¦ ¦ if (delta > 999) out << static_cast<char>(1);
435   ¦ ¦ ¦ out << " (" << delta << ")";
436   ¦ ¦ ¦ if (delta > 999) out << static_cast<char>(2);
437   ¦ ¦ }
438   ¦ }
439   ¦ render_line(screen_row, out.str(), screen_row == Display_row);
440   }
441   // clear rest of screen
442   Last_printed_row = screen_row-1;
443   for (;  screen_row < tb_height();  ++screen_row)
444   ¦ render_line(screen_row, "~", /*cursor_line?*/false);
445   // move cursor back to display row at the end
446   tb_set_cursor(0, Display_row);
447   tb_present();
448 }
449 
450 int lines_hidden(int screen_row) {
451   assert(contains_key(Trace_index, screen_row));
452   if (!contains_key(Trace_index, screen_row+1))
453   ¦ return SIZE(Trace_stream->past_lines) - get(Trace_index, screen_row);
454   else
455   ¦ return get(Trace_index, screen_row+1) - get(Trace_index, screen_row);
456 }
457 
458 void render_line(int screen_row, const string& s, bool cursor_line) {
459   int col = 0;
460   int color = TB_WHITE;
461   int background_color = cursor_line ? /*subtle grey*/240 : TB_BLACK;
462   vector<pair<size_t, size_t> > highlight_ranges = find_all_occurrences(s, Current_search_pattern);
463   for (col = 0;  col < tb_width() && col+Left_of_screen < SIZE(s);  ++col) {
464   ¦ char c = s.at(col+Left_of_screen);  // todo: unicode
465   ¦ if (c == '\n') c = ';';  // replace newlines with semi-colons
466   ¦ // escapes. hack: can't start a line with them.
467   ¦ if (c == '\1') { color = /*red*/1;  c = ' '; }
468   ¦ if (c == '\2') { color = TB_WHITE;  c = ' '; }
469   ¦ if (in_range(highlight_ranges, col+Left_of_screen))
470   ¦ ¦ tb_change_cell(col, screen_row, c, TB_BLACK, /*yellow*/11);
471   ¦ else
472   ¦ ¦ tb_change_cell(col, screen_row, c, color, background_color);
473   }
474   for (;  col < tb_width();  ++col)
475   ¦ tb_change_cell(col, screen_row, ' ', TB_WHITE, background_color);
476 }
477 
478 vector<pair<size_t, size_t> > find_all_occurrences(const string& s, const string& pat) {
479   vector<pair<size_t, size_t> > result;
480   if (pat.empty()) return result;
481   size_t idx = 0;
482   while (true) {
483   ¦ size_t next_idx = s.find(pat, idx);
484   ¦ if (next_idx == string::npos) break;
485   ¦ result.push_back(pair<size_t, size_t>(next_idx, next_idx+SIZE(pat)));
486   ¦ idx = next_idx+SIZE(pat);
487   }
488   return result;
489 }
490 
491 bool in_range(const vector<pair<size_t, size_t> >& highlight_ranges, size_t idx) {
492   for (int i = 0;  i < SIZE(highlight_ranges);  ++i) {
493   ¦ if (idx >= highlight_ranges.at(i).first && idx < highlight_ranges.at(i).second)
494   ¦ ¦ return true;
495   ¦ if (idx < highlight_ranges.at(i).second) break;
496   }
497   return false;
498 }
499 
500 void load_trace(const char* filename) {
501   ifstream tin(filename);
502   if (!tin) {
503   ¦ cerr << "no such file: " << filename << '\n';
504   ¦ exit(1);
505   }
506   Trace_stream = new trace_stream;
507   while (has_data(tin)) {
508   ¦ tin >> std::noskipws;
509   ¦ ¦ skip_whitespace_but_not_newline(tin);
510   ¦ ¦ if (!isdigit(tin.peek())) {
511   ¦ ¦ ¦ string dummy;
512   ¦ ¦ ¦ getline(tin, dummy);
513   ¦ ¦ ¦ continue;
514   ¦ ¦ }
515   ¦ tin >> std::skipws;
516   ¦ int depth;
517   ¦ tin >> depth;
518   ¦ string label;
519   ¦ tin >> label;
520   ¦ if (*--label.end() == ':') label.erase(--label.end());
521   ¦ string line;
522   ¦ getline(tin, line);
523   ¦ Trace_stream->past_lines.push_back(trace_line(depth, label, line));
524   }
525   cerr << "lines read: " << Trace_stream->past_lines.size() << '\n';
526 }
527 
528 int read_key() {
529   tb_event event;
530   do {
531   ¦ tb_poll_event(&event);
532   } while (event.type != TB_EVENT_KEY);
533   return event.key ? event.key : event.ch;
534 }