Bound unique sort case

.github/workflows/tests.yml

@@ -10,7 +10,7 @@ jobs:
       - name: Install dependencies
         run: |
           sudo apt-get update
-          sudo apt-get install -y build-essential libboost-test-dev cmake pkg-config libpcre2-dev libncursesw5-dev
+          sudo apt-get install -y build-essential libboost-test-dev cmake pkg-config libpcre2-dev libncursesw5-dev libtbb-dev
       - name: Build project
         run: |
           cd build

CMakeLists.txt

@@ -14,7 +14,7 @@ project(choose)
 add_executable(choose src/main.cpp)
 target_include_directories(choose PRIVATE src)
 
-target_compile_options(choose PRIVATE -Wall -Wextra -O3 -fno-rtti)
+target_compile_options(choose PRIVATE -Wall -Wextra -O3)
 
 set(CURSES_NEED_NCURSES TRUE)
 set(CURSES_NEED_WIDE TRUE)
@@ -29,6 +29,13 @@ pkg_check_modules(PCRE REQUIRED libpcre2-8)
 target_include_directories(choose PRIVATE ${PCRE_INCLUDEDIR})
 target_link_libraries(choose PRIVATE ${PCRE_LIBRARIES})
 
+# https://stackoverflow.com/a/74755391/15534181
+# optional link here since sometimes it's ok not to have. this is the least invasive
+find_package(TBB QUIET)
+if(TBB_FOUND)
+  target_link_libraries(choose PRIVATE TBB::tbb)
+endif()
+
 if (NO_SCROLL_BORDER)
   target_compile_definitions(choose PRIVATE
     CHOOSE_NO_SCROLL_BORDER
@@ -52,7 +59,7 @@ if(BUILD_TESTING)
   add_executable(unit_tests src/test.cpp)
   target_include_directories(unit_tests PRIVATE src)
   # -O0 gives better coverage info than -Og
-  target_compile_options(unit_tests PRIVATE -Wall -Wextra -O0 -g -fno-rtti)
+  target_compile_options(unit_tests PRIVATE -Wall -Wextra -O0 -g)
 
   target_include_directories(unit_tests PRIVATE ${Boost_UNIT_TEST_FRAMEWORK_HEADER_NAME})
   target_link_libraries(unit_tests PRIVATE ${Boost_UNIT_TEST_FRAMEWORK_LIBRARY})
@@ -64,6 +71,10 @@ if(BUILD_TESTING)
   target_include_directories(unit_tests PRIVATE ${PCRE_INCLUDEDIR})
   target_link_libraries(unit_tests PRIVATE ${PCRE_LIBRARIES})
 
+  if(TBB_FOUND)
+    target_link_libraries(unit_tests PRIVATE TBB::tbb)
+  endif()
+
   if (DISABLE_FIELD)
     target_compile_definitions(unit_tests PRIVATE
       CHOOSE_DISABLE_FIELD

perf/gen_perf_stats.bash

@@ -5,6 +5,9 @@
 set -e
 sudo echo -n '' # do nothing. perf requires sudo. doing the prompt at the beginning
 
+# fairer performance for sort
+export LC_ALL=C
+
 # e.g. -n makes the benchmarks apply sorting and uniqueness numerically (but not for just uniqueness since awk doesn't support this)
 COMP_FLAGS=
 

perf/perf.md

@@ -20,7 +20,7 @@ Also note the compile time option called `DISABLE_FIELD`. It disables the `--fie
 
 ### Sorting, and Sorting + Uniqueness
 
-In most cases, `choose` is faster than `sort` and `sort -u` at sorting and sorting + uniqueness, respectively. Note however that the c locale isn't used for comparisons, and again it is the task clock that is reported.
+`sort` is using naive byte order (via `LC_ALL=C`), as this is the fairest. `sort` is faster than `choose` at sorting. If truncation is leveraged, or if there are many duplicates (when applying uniqueness as well), then `choose` is faster than `sort`.
 
 ## Input Data
 
@@ -82,80 +82,80 @@ garbage,5,garbage
 
 ### Versions
 ```txt
-choose 0.3.0, ncurses 6.1.20180127, pcre2 10.42
-pcre2grep version 10.42 2022-12-11
-sed (GNU sed) 4.4
-GNU Awk 4.1.4, API: 1.1 (GNU MPFR 4.0.1, GNU MP 6.1.2)
-sort (GNU coreutils) 8.28
+choose 0.3.0, ncurses 6.2.20200212, pcre2 10.43
+pcre2grep version 10.43-DEV 2023-04-14
+sed (GNU sed) 4.7
+GNU Awk 5.0.1, API: 2.0 (GNU MPFR 4.0.2, GNU MP 6.2.0)
+sort (GNU coreutils) 8.30
 ```
 ### Specs
 ```txt
 5.15.90.1-microsoft-standard-WSL2
-Intel(R) Core(TM) i5-8600K CPU @ 3.60GHz
-ram: 8116584 kB
+AMD Ryzen 7 3800X 8-Core Processor
+ram: 16331032 kB
 ```
 
 ### Grepping
 
 | (ms)             | choose | pcre2grep  |
 |------------------|--------|------------|
-| plain_text       | 238.334100 | 246.104400 | 
-| test_repeated    | 1536.390100 | 1446.540000 | 
-| no_duplicates    | 321.083200 | 313.054700 | 
+| plain_text       | 247.17 | 269.69 | 
+| test_repeated    | 1620.34 | 1583.77 | 
+| no_duplicates    | 323.59 | 370.16 | 
 
 ### Stream Editing
 
 | (ms)             | choose | sed  |
 |------------------|--------|------|
-| plain_text       | 173.019600 | 156.455300 | 
-| test_repeated    | 2563.258500 | 1024.358400 | 
-| no_duplicates    | 8.424300 | 46.834200 | 
+| plain_text       | 179.57 | 135.57 | 
+| test_repeated    | 2725.50 | 1157.39 | 
+| no_duplicates    | 5.10 | 44.00 | 
 
 (here is a cherry picked great case for choose compared to sed)
 
 | (ms)             | choose | sed (with newline delimiter) |
 |------------------|--------|------|
-| no_duplicates    | 8.245600 | 437.878300 | 
+| no_duplicates    | 5.13 | 543.93 | 
 
 (a special case, where choose cheats by using a literal replacement string)
 
 | (ms)             | choose (delimiter sub) | sed |
 |------------------|------------------------|-----|
-| test_repeated    | 1457.271000 | 1010.783600 | 
+| test_repeated    | 1521.93 | 1156.64 | 
 
 ### Sorting 
 
 | (ms)             | choose | sort |
 |------------------|--------|------|
-| plain_text       | 694.556000 | 1905.257700 | 
-| test_repeated    | 2226.087400 | 1987.113500 | 
-| no_duplicates    | 2120.992700 | 5092.179100 | 
+| plain_text       | 1628.88 | 448.06 | 
+| test_repeated    | 1850.89 | 1616.13 | 
+| no_duplicates    | 3714.94 | 1036.93 | 
 
 (a special case that leverages truncation)
 
 | (ms)             | choose -s --out 5 | sort \| head -n 5 |
 |------------------|--------|------|
-| no_duplicates    | 251.069600 | 5063.083100 | 
+| no_duplicates    | 354.20 | 1059.81 | 
 
 ### Uniqueness 
 
 | (ms)             | choose | awk |
 |------------------|--------|-----|
-| plain_text       | 114.649800 | 208.971700 | 
-| test_repeated    | 578.412600 | 972.325200 | 
-| no_duplicates    | 2480.435700 | 1477.912300 | 
+| plain_text       | 111.95 | 214.41 | 
+| test_repeated    | 565.31 | 1147.75 | 
+| no_duplicates    | 2340.37 | 1496.42 | 
 
 ### Sorting and Uniqueness   -u
 
 | (ms)             | choose | sort |
 |------------------|--------|------|
-| plain_text       | 106.970100 | 1906.801600 | 
-| test_repeated    | 574.516000 | 1961.279100 | 
-| no_duplicates    | 4165.485200 | 5670.807600 | 
+| plain_text       | 122.80 | 440.57 | 
+| test_repeated    | 558.86 | 1640.79 | 
+| no_duplicates    | 5742.11 | 1168.84 | 
 
 
 ### Sorting and Uniqueness based on field   -u
 
 | (ms)             | choose | sort |
 |------------------|--------|------|
-| csv_field        | 1779.289000 | 1987.503500 | 
+| csv_field        | 2770.27 | 474.02 | 

readme.md

@@ -18,7 +18,7 @@ Also it's fast. See benchmarks [here](./perf/perf.md) comparing choose to other
 
 ## Install
 ```bash
-sudo apt-get install cmake pkg-config libpcre2-dev libncursesw5-dev
+sudo apt-get install cmake pkg-config libpcre2-dev libncursesw5-dev libtbb-dev
 git clone https://github.com/jagprog5/choose.git && cd choose
 make install
 source ~/.bashrc

src/algo_utils.hpp

@@ -46,6 +46,7 @@ void stable_partial_sort(ExecutionPolicy&& policy, it begin, it middle, it end,
 bool general_numeric_compare(const char* lhs_begin, const char* lhs_end, const char* rhs_begin, const char* rhs_end) { //
   float lhs, rhs;
   // if from_chars isn't found, get a newer compiler. e.g.
+  //    add-apt-repository -y ppa:ubuntu-toolchain-r/test
   //    apt-get install g++-11
   //    cd build && cmake .. -DCMAKE_C_COMPILER=gcc-11 -DCMAKE_CXX_COMPILER=g++-11
   std::from_chars_result lhs_ret = std::from_chars(lhs_begin, lhs_end, lhs, std::chars_format::general);

src/args.hpp

@@ -117,6 +117,13 @@ struct Arguments {
     return is_direct_output() && !unique;
   }
 
+  // the elements in output vector are being inserted with any excess being discarded
+  bool mem_is_bounded() const {
+    return out_end.has_value()   //
+           && !truncate_no_bound //
+           && (unique ? sort && unique_type == sort_type : true);
+  }
+
   void drop_warning() {
     if (this->can_drop_warn) {
       this->can_drop_warn = false;
@@ -134,8 +141,9 @@ struct Arguments {
 namespace {
 
 struct UncompiledCodes {
-  // all args must be parsed before the args are compiled
-  // the uncompiled args are stored here before transfer to the Arguments output.
+  // all args must be parsed before the args are compiled. the uncompiled args
+  // are stored here before transfer to the Arguments output. this also contains
+  // fields that aren't needed in the rest of the program, past the arg parsing
   uint32_t re_options = PCRE2_LITERAL;
   std::vector<uncompiled::UncompiledOrderedOp> ordered_ops;
 
@@ -152,6 +160,8 @@ struct UncompiledCodes {
   bool bout_delimiter_set = false;
   bool primary_set = false;
 
+  bool is_bounded_query = false;
+
   void compile(Arguments& output) const {
     for (const uncompiled::UncompiledOrderedOp& op : ordered_ops) {
       OrderedOp oo = uncompiled::compile(op, re_options);
@@ -315,6 +325,9 @@ void print_help_message() {
       "                must have at least one digit. parse failures are smallest\n"
       "        -i, --ignore-case\n"
       "                make the positional argument case-insensitive\n"
+      "        --is-bounded\n"
+      "                prints a line indicating if the memory usage is bounded due to\n"
+      "                truncation, then exits. not related to --unique-limit\n"
       "        --load-factor <positive float, default: " choose_xstr(UNIQUE_LOAD_FACTOR_DEFAULT) ">\n"
       "                if a hash table is used for uniqueness, set the max load factor\n"
       "        --locale <locale>\n"
@@ -375,12 +388,11 @@ void print_help_message() {
       "                on tui confirmed selection, do not exit; but still flush the\n"
       "                current selection to the output as a batch\n"
       "        --truncate-no-bound\n"
-      "                if truncation is specified (--out/--tail) and uniqueness is not\n"
-      "                specified, then choose only retains the relevant n values in\n"
-      "                memory. This is only faster for small values of n, as elements\n"
-      "                are shifted within this storage space. If n is large, this\n"
-      "                option should be used to disable this optimization, leading to\n"
-      "                faster speed but more space used\n"
+      "                if truncation is specified (--out/--tail) then choose may retain\n"
+      "                only the relevant n values in memory. see --is-bounded. this is\n"
+      "                faster for small values of n, as elements are shifted within\n"
+      "                this storage space. If n is large, this option should be used to\n"
+      "                disable this optimization\n"
       "        -u, --unique\n"
       "                remove duplicate input tokens. leaves first occurrences. applied\n"
       "                before sorting\n"
@@ -396,7 +408,8 @@ void print_help_message() {
       "        --unique-general-numeric\n"
       "                apply uniqueness general numerically. implies -u\n"
       "        --unique-limit <#tokens>\n"
-      "                implies -u. forget least recently used tokens\n"
+      "                implies -u. forget least recently used tokens. ignored if memory\n"
+      "                is bounded due to truncation (see --is-bounded)\n"
       "        --unique-use-set\n"
       "                implies -u. apply uniqueness with a tree instead of a hash table\n"
       "        --use-delimiter\n"
@@ -527,6 +540,7 @@ Arguments handle_args(int argc, char* const* argv, FILE* input = NULL, FILE* out
         {"flip", no_argument, NULL, 0},
         {"flush", no_argument, NULL, 0},
         {"ignore-case", no_argument, NULL, 'i'},
+        {"is-bounded", no_argument, NULL, 0},
         {"multi", no_argument, NULL, 'm'},
         {"multiline", no_argument, NULL, 0},
         {"match", no_argument, NULL, 0},
@@ -776,6 +790,8 @@ Arguments handle_args(int argc, char* const* argv, FILE* input = NULL, FILE* out
           } else if (strcmp("unique-general-numeric", name) == 0) {
             ret.unique = true;
             ret.unique_type = general_numeric;
+          } else if (strcmp("is-bounded", name) == 0) {
+            uncompiled_output.is_bounded_query = true;
           } else if (strcmp("multiline", name) == 0) {
             uncompiled_output.re_options &= ~PCRE2_LITERAL;
             uncompiled_output.re_options |= PCRE2_MULTILINE;
@@ -998,6 +1014,11 @@ Arguments handle_args(int argc, char* const* argv, FILE* input = NULL, FILE* out
     }
   }
 
+  if (uncompiled_output.is_bounded_query) {
+    int exit_code = puts(ret.mem_is_bounded() ? "yes" : "no") < 0 ? EXIT_FAILURE : EXIT_SUCCESS;
+    exit(exit_code);
+  }
+
   if (isatty(fileno(ret.input))) {
     int exit_code = puts("Try 'choose --help' for more information.") < 0 ? EXIT_FAILURE : EXIT_SUCCESS;
     exit(exit_code);