Enable visualization by default for inference and replay

src/main.cpp

@@ -2,6 +2,7 @@
 #include "config.hpp"
 #include "stop_signals.hpp"
 #include "visualizer.hpp"
+#include <cstdlib>
 #include <functional>
 #include <iostream>
 #include <map>
@@ -10,7 +11,7 @@ int main(int argc,char** argv)try {
     auto root=std::filesystem::canonical("/proc/self/exe").parent_path().parent_path();
     auto configPath=root/"config/camera.json";
     cms::OutputOptions output;output.directory=root/"output";
-    bool visualize=false;
+    bool visualize=true;
     std::filesystem::path recordSession,replaySession;
     std::map<std::string,std::function<void(int&)>> arguments{
         {"--record-session",[&](int&i){if(++i>=argc)throw std::invalid_argument("--record-session needs a directory");recordSession=argv[i];}},
@@ -19,8 +20,9 @@ int main(int argc,char** argv)try {
         {"--save-annotated",[&](int&){output.annotated=true;}},
         {"--save-results",[&](int&){output.results=true;}},
         {"--visualize",[&](int&){visualize=true;}},
+        {"--no-visualize",[&](int&){visualize=false;}},
         {"--config",[&](int&i){if(++i>=argc)throw std::invalid_argument("--config needs a file");configPath=argv[i];}},
-        {"--help",[](int&){std::cout<<"camera_risk [--record-session NEW_DIR] [--replay-session DIR] [--save-original] [--save-annotated] [--save-results] [--config FILE] [--visualize]\nContinuous camera inference; replay runs to EOF without a camera.\nH.264 sessions require a new recording directory. Stop with Ctrl+C or SIGTERM to finalize video.\n";std::exit(0);}}
+        {"--help",[](int&){std::cout<<"camera_risk [--record-session NEW_DIR] [--replay-session DIR] [--save-original] [--save-annotated] [--save-results] [--config FILE] [--visualize|--no-visualize]\nVisualization is enabled by default for live inference, recording and replay.\nContinuous camera inference; replay runs to EOF without a camera.\nH.264 sessions require a new recording directory. Stop with Ctrl+C or SIGTERM to finalize video.\n";std::exit(0);}}
     };
     for(int i=1;i<argc;++i){auto handler=arguments.find(argv[i]);if(handler==arguments.end())throw std::invalid_argument("Unknown option: "+std::string(argv[i]));handler->second(i);}
 
@@ -43,7 +45,9 @@ int main(int argc,char** argv)try {
      */
     // Optional observer. Closing/failing the window never requests inference stop.
     if(visualize) {
-        try {cms::runLocalVisualization(pipeline);}
+        auto display=std::getenv("DISPLAY");auto wayland=std::getenv("WAYLAND_DISPLAY");
+        if((!display||!*display)&&(!wayland||!*wayland))std::cerr<<"No display session; inference continues without visualization.\n";
+        else try {cms::runLocalVisualization(pipeline);}
         catch(std::exception const& e){std::cerr<<"Visualization unavailable; inference continues: "<<e.what()<<std::endl;}
     }
     pipeline.join();