introduced BR_DISABLE_ALGORITHMS to ship a limited version of br command line

Josh Klontz
1 parent 3fdc911d
Showing 3 changed files with 18 additions and 4 deletions
app/CMakeLists.txt
app/br/CMakeLists.txt
app/br/br.cpp
 # Build the command line interface
+add_subdirectory(br)
+
 if(NOT BR_PACKAGE_THIRDPARTY)
   if(NOT BR_EMBEDDED)
-    add_subdirectory(br)
-
     # Build examples/tests
     add_subdirectory(examples)
@@ -2,6 +2,10 @@ if(UNIX)
   find_package(Threads REQUIRED)
 endif()
+if(BR_DISABLE_ALGORITHMS)
+  add_definitions(-DBR_DISABLE_ALGORITHMS)
+endif()
+
 add_executable(br br.cpp ${BR_RESOURCES})
 target_link_libraries(br openbr ${CMAKE_THREAD_LIBS_INIT})
 qt5_use_modules(br ${QT_DEPENDENCIES})
@@ -90,6 +90,7 @@ public:
             argv = &argv[parc+1];
             // Core Tasks
+#ifndef BR_DISABLE_ALGORITHMS
             if (!strcmp(fun, "train")) {
                 check(parc >= 1, "Insufficient parameter count for 'train'.");
                 br_train_n(parc == 1 ? 1 : parc-1, parv, parc == 1 ? "" : parv[parc-1]);
@@ -100,7 +101,9 @@ public:
             } else if (!strcmp(fun, "compare")) {
                 check((parc >= 2) && (parc <= 3), "Incorrect parameter count for 'compare'.");
                 br_compare(parv[0], parv[1], parc == 3 ? parv[2] : "");
-            } else if (!strcmp(fun, "eval")) {
+            } else
+#endif // !BR_DISABLE_ALGORITHMS
+            if (!strcmp(fun, "eval")) {
                 check((parc >= 1) && (parc <= 4), "Incorrect parameter count for 'eval'.");
                 if (parc == 1) {
                     br_eval(parv[0], "", "", 0);
@@ -192,7 +195,9 @@ public:
             } else if (!strcmp(fun, "plotKNN")) {
                 check(parc >=2, "Incorrect parameter count for 'plotKNN'.");
                 br_plot_knn(parc-1, parv, parv[parc-1], true);
-            } else if (!strcmp(fun, "project")) {
+            }
+#ifndef BR_DISABLE_ALGORITHMS
+            else if (!strcmp(fun, "project")) {
                 check(parc == 2, "Insufficient parameter count for 'project'.");
                 br_project(parv[0], parv[1]);
             } else if (!strcmp(fun, "deduplicate")) {
@@ -202,6 +207,7 @@ public:
                 check(parc == 3, "Incorrect parameter count for 'likely'.");
                 br_likely(parv[0], parv[1], parv[2]);
             }
+#endif // !BR_DISABLE_ALGORITHMS
             // Miscellaneous
             else if (!strcmp(fun, "help")) {
@@ -271,9 +277,11 @@ private:
         printf("<arg> = Input; {arg} = Output; [arg] = Optional; (arg0|...|argN) = Choice\n"
                "\n"
                "==== Core Commands ====\n"
+#ifndef BR_DISABLE_ALGORITHMS
                "-train <gallery> ... <gallery> [{model}]\n"
                "-enroll <input_gallery> ... <input_gallery> {output_gallery}\n"
                "-compare <target_gallery> <query_gallery> [{output}]\n"
+#endif // !BR_DISABLE_ALGORITHMS
                "-eval <simmat> [<mask>] [{csv}] [{matches}]\n"
                "-plot <csv> ... <csv> {destination}\n"
                "\n"
@@ -298,9 +306,11 @@ private:
                "-plotLandmarking <file> ... <file> {destination}\n"
                "-plotMetadata <file> ... <file> <columns>\n"
                "-plotKNN <file> ... <file> {destination}\n"
+#ifndef BR_DISABLE_ALGORITHMS
                "-project <input_gallery> {output_gallery}\n"
                "-deduplicate <input_gallery> <output_gallery> <threshold>\n"
                "-likely <input_type> <output_type> <output_likely_source>\n"
+#endif // !BR_DISABLE_ALGORITHMS
                "-getHeader <matrix>\n"
                "-setHeader {<matrix>} <target_gallery> <query_gallery>\n"
                "-<key> <value>\n"