Cosmetics.
[selector.git] / selector.c
index effd014..afd5ce7 100644 (file)
@@ -3,7 +3,7 @@
  *  selector is a simple command line utility for selection of strings
  *  with a dynamic pattern-matching.
  *
- *  Copyright (c) 2009, 2010, 2011 Francois Fleuret
+ *  Copyright (c) 2009-2013 Francois Fleuret
  *  Written by Francois Fleuret <francois@fleuret.org>
  *
  *  This file is part of selector.
@@ -45,9 +45,9 @@
 #include <getopt.h>
 #include <limits.h>
 
-#define VERSION "1.1.5"
+#define VERSION "1.1.7"
 
-#define BUFFER_SIZE 4096
+#define BUFFER_SIZE 16384
 
 /* Yeah, global variables! */
 
@@ -79,7 +79,7 @@ void *safe_malloc(size_t n) {
   void *p = malloc(n);
   if(!p && n != 0) {
     fprintf(stderr,
-            "selector: can not allocate memory: %s\n", strerror(errno));
+            "selector: cannot allocate memory: %s\n", strerror(errno));
     exit(EXIT_FAILURE);
   }
   return p;
@@ -191,7 +191,7 @@ void usage(FILE *out) {
   fprintf(out, " -a, --case-sensitive\n");
   fprintf(out, "         start in case sensitive mode\n");
   fprintf(out, " -j, --show-long-lines\n");
-  fprintf(out, "         print three dots at the end of truncated lines\n");
+  fprintf(out, "         print a long-line indicator at the end of truncated lines\n");
   fprintf(out, " -y, --show-hits\n");
   fprintf(out, "         highlight the matching substrings\n");
   fprintf(out, " -u, --upper-case-makes-case-sensitive\n");
@@ -203,9 +203,12 @@ void usage(FILE *out) {
   fprintf(out, "         make a flash instead of a beep on an edition error\n");
   fprintf(out, " --bash\n");
   fprintf(out, "         setting for bash history search, same as -b -i -d -v -w -l ${HISTSIZE}\n");
-  fprintf(out, " --      all following arguments are filenames\n");
+  fprintf(out, " --\n");
+  fprintf(out, "         all following arguments are filenames\n");
   fprintf(out, " -t <title>, --title <title>\n");
   fprintf(out, "         add a title in the modeline\n");
+  fprintf(out, " -r <pattern>, --pattern <pattern>\n");
+  fprintf(out, "         set an initial pattern\n");
   fprintf(out, " -c <colors>, --colors <colors>\n");
   fprintf(out, "         set the display colors with an argument of the form\n");
   fprintf(out, "         <fg_modeline>,<bg_modeline>,<fg_highlight>,<bg_highlight>\n");
@@ -214,7 +217,8 @@ void usage(FILE *out) {
   fprintf(out, " -s <pattern separator>, --pattern-separator <pattern separator>\n");
   fprintf(out, "         set the symbol to separate substrings in the pattern\n");
   fprintf(out, " -x <label separator>, --label-separator <label separator>\n");
-  fprintf(out, "         set the symbol to terminate the label\n");
+  fprintf(out, "         set the character to separate the label to show from the\n");
+  fprintf(out, "         string to return\n");
   fprintf(out, " -l <max number of lines>, --number-of-lines <max number of lines>\n");
   fprintf(out, "         set the maximum number of lines to take into account\n");
   fprintf(out, "\n");
@@ -320,8 +324,9 @@ struct matcher {
 };
 
 /* Routine to add an interval to a sorted list of intervals
-   extermities. Returns the number of extremities. This is an effing
-   nightmare */
+   extremities. Returns the resulting number of extremities.
+
+   This routine is an effing nightmare */
 
 int add_interval(int n, int *switches, int start, int end) {
   int f, g, k;
@@ -334,8 +339,8 @@ int add_interval(int n, int *switches, int start, int end) {
   while(g < n && switches[g] <= end) { g++; }
 
   if(f == n) {
-    /* switches[n]   start  end  */
-    /* XXXXXXXXXX|               */
+    /* switches[n-1]   start  end  */
+    /* XXXXXXXXXXXX|               */
     switches[f] = start;
     switches[f+1] = end;
     return n + 2;
@@ -570,7 +575,6 @@ int next_visible(int current_line, int nb_lines, char **lines,
 /*********************************************************************/
 
 void print_string_with_switches(char *buffer, int line_width,
-                                int console_width,
                                 int nb_patterns, int *switches) {
   int w, current = 0, next;
   if(switches) {
@@ -589,10 +593,10 @@ void print_string_with_switches(char *buffer, int line_width,
       }
     }
     if(current < line_width) {
-      addnstr(buffer + current, console_width - current);
+      addnstr(buffer + current, line_width - current);
     }
   } else {
-    addnstr(buffer, console_width);
+    addnstr(buffer, line_width);
   }
 }
 
@@ -740,36 +744,37 @@ void update_screen(int *current_focus_line, int *displayed_focus_line,
 
           if(l == new_focus_line) {
             if(show_long_lines && k >= console_width) {
-              if(console_width >= 4) {
-                buffer[console_width - 4] = ' ';
-                buffer[console_width - 3] = '.';
-                buffer[console_width - 2] = '.';
-                buffer[console_width - 1] = '.';
-              }
+              attron(attr_focus_line);
+              print_string_with_switches(buffer, console_width-1,
+                                         nb_switches / 2, switches);
+              /* attron(attr_error); */
+              addnstr("\\", 1);
+              /* attroff(attr_error); */
+              attroff(attr_focus_line);
             } else {
               while(k < console_width) {
                 buffer[k++] = ' ';
               }
+              attron(attr_focus_line);
+              print_string_with_switches(buffer, k,
+                                         nb_switches / 2, switches);
+              attroff(attr_focus_line);
             }
-            attron(attr_focus_line);
-            print_string_with_switches(buffer, k, console_width,
-                                       nb_switches / 2, switches);
-            attroff(attr_focus_line);
           } else {
             if(show_long_lines && k >= console_width) {
-              if(console_width >= 4) {
-                buffer[console_width - 4] = ' ';
-                buffer[console_width - 3] = '.';
-                buffer[console_width - 2] = '.';
-                buffer[console_width - 1] = '.';
-              }
+              print_string_with_switches(buffer, console_width-1,
+                                         nb_switches / 2, switches);
+              attron(attr_focus_line);
+              addnstr("\\", 1);
+              attroff(attr_focus_line);
             } else {
-              buffer[k++] = '\n';
-              buffer[k++] = '\0';
+              if(k < console_width) {
+                buffer[k++] = '\n';
+                buffer[k++] = '\0';
+              }
+              print_string_with_switches(buffer, k,
+                                         nb_switches / 2, switches);
             }
-
-            print_string_with_switches(buffer, k, console_width,
-                                       nb_switches / 2, switches);
           }
 
           nb_printed_lines++;
@@ -920,19 +925,31 @@ void read_file(struct hash_table_t *hash_table,
   char raw_line[BUFFER_SIZE];
   char *s;
   FILE *file;
+  int l;
 
   file = fopen(input_filename, "r");
 
   if(!file) {
-    fprintf(stderr, "selector: Can not open `%s'.\n", input_filename);
+    fprintf(stderr, "selector: Cannot open `%s'.\n", input_filename);
     exit(EXIT_FAILURE);
   }
 
-  while(*nb_lines < nb_lines_max && fgets(raw_line, BUFFER_SIZE, file)) {
-    for(s = raw_line + strlen(raw_line) - 1; s > raw_line && *s == '\n'; s--) {
-      *s = '\0';
+  if(label_separator == '\n') {
+    while(*nb_lines < nb_lines_max && fgets(raw_line, BUFFER_SIZE, file)) {
+      l = strlen(raw_line);
+      fgets(raw_line + l, BUFFER_SIZE - l, file);
+      for(s = raw_line + strlen(raw_line) - 1; s > raw_line && *s == '\n'; s--) {
+        *s = '\0';
+      }
+      store_line(hash_table, raw_line, nb_lines, lines);
+    }
+  } else {
+    while(*nb_lines < nb_lines_max && fgets(raw_line, BUFFER_SIZE, file)) {
+      for(s = raw_line + strlen(raw_line) - 1; s > raw_line && *s == '\n'; s--) {
+        *s = '\0';
+      }
+      store_line(hash_table, raw_line, nb_lines, lines);
     }
-    store_line(hash_table, raw_line, nb_lines, lines);
   }
 
   fclose(file);
@@ -965,6 +982,7 @@ static struct option long_options[] = {
   { "show-hits", no_argument, 0, 'j'},
   { "upper-case-makes-case-sensitive", no_argument, 0, 'u' },
   { "title", 1, 0, 't' },
+  { "pattern", 1, 0, 'r' },
   { "number-of-lines", 1, 0, 'l' },
   { "colors", 1, 0, 'c' },
   { "bash", no_argument, 0, OPT_BASH_MODE },
@@ -996,6 +1014,8 @@ int main(int argc, char **argv) {
     exit(EXIT_FAILURE);
   }
 
+  pattern[0] = '\0';
+
   color_fg_modeline  = COLOR_WHITE;
   color_bg_modeline  = COLOR_BLACK;
   color_fg_highlight = COLOR_BLACK;
@@ -1005,7 +1025,7 @@ int main(int argc, char **argv) {
 
   strcpy(output_filename, "");
 
-  while ((c = getopt_long(argc, argv, "o:s:x:vwmqf:ibzdeajyunt:l:c:-h",
+  while ((c = getopt_long(argc, argv, "o:s:x:vwmqf:ibzdeajyunt:r:l:c:-h",
                           long_options, NULL)) != -1) {
 
     switch(c) {
@@ -1019,7 +1039,11 @@ int main(int argc, char **argv) {
       break;
 
     case 'x':
-      label_separator = optarg[0];
+      if(strcmp(optarg, "\\n") == 0) {
+        label_separator = '\n';
+      } else {
+        label_separator = optarg[0];
+      }
       break;
 
     case 'v':
@@ -1080,6 +1104,10 @@ int main(int argc, char **argv) {
       strcpy(title, optarg);
       break;
 
+    case 'r':
+      strcpy(pattern, optarg);
+      break;
+
     case 'l':
       str_to_positive_integers(optarg, &nb_lines_max, 1);
       break;
@@ -1198,8 +1226,6 @@ int main(int argc, char **argv) {
     *s = '\0';
   }
 
-  pattern[0] = '\0';
-
   cursor_position = 0;
 
   /* Here we start to display with curse */
@@ -1245,6 +1271,7 @@ int main(int argc, char **argv) {
 
   current_focus_line = 0;
   displayed_focus_line = 0;
+  cursor_position = strlen(pattern);
 
   update_screen(&current_focus_line, &displayed_focus_line,
                 0,
@@ -1392,7 +1419,7 @@ int main(int argc, char **argv) {
         fprintf(out, "\n");
       } else {
         fprintf(stderr,
-                "selector: Can not open %s for writing.\n",
+                "selector: Cannot open %s for writing.\n",
                 output_filename);
         exit(EXIT_FAILURE);
       }