Handle signals in sam tools #220

Author: mfe

applications/sam/src/sam/check/main.cpp

@@ -115,6 +115,8 @@ void checkSequence( Graph::Node& read, Graph::Node& stat, Graph& graph, const sp
 
 int main( int argc, char** argv )
 {
+	signal(SIGINT, signal_callback_handler);
+
 	using namespace tuttle::common;
 	using namespace sam;
 

applications/sam/src/sam/common/mvcp.hpp

@@ -121,6 +121,8 @@ void copy_sequence( const sp::Sequence& s, const bfs::path& d, const sp::Time of
 
 int sammvcp(int argc, char** argv)
 {
+	signal(SIGINT, signal_callback_handler);
+
 	using namespace tuttle::common;
 	using namespace sam;
 

applications/sam/src/sam/common/utility.hpp

@@ -4,10 +4,21 @@
 #include <boost/algorithm/string/predicate.hpp>
 
 #include <string>
+#include <iostream>
+#include <tuttle/common/utils/global.hpp>
+#include <csignal>
 
 bool string_to_boolean( const std::string& str )
 {
 	return ( str == "1" || boost::iequals(str, "y") || boost::iequals(str, "yes") || boost::iequals(str, "true") );
 }
 
+void signal_callback_handler( int signum )
+{
+	std::ostringstream os;
+	os << "Signal "<< strsignal(signum) << "(" << signum << ") was caught.\nTerminate sam.";
+	TUTTLE_LOG_WARNING(os.str());
+	exit(signum);
+}
+
 #endif

applications/sam/src/sam/diff/main.cpp

@@ -299,6 +299,8 @@ void displayHelp(bpo::options_description &desc)
 
 int main( int argc, char** argv )
 {
+	signal(SIGINT, signal_callback_handler);
+
 	using namespace tuttle::common;
 	using namespace sam;
 

applications/sam/src/sam/do/main.cpp

@@ -249,6 +249,8 @@ bool isContextSupported( const ttl::Graph::Node* node , const std::string& conte
 
 int main( int argc, char** argv )
 {
+	signal(SIGINT, signal_callback_handler);
+
 	using namespace tuttle::common;
 	using namespace sam;
 	using namespace sam::samdo;

applications/sam/src/sam/info/main.cpp

@@ -183,6 +183,8 @@ void dumpImageProperties( boost::ptr_vector<sp::FileObject>& listing )
 
 int main( int argc, char** argv )
 {
+	signal(SIGINT, signal_callback_handler);
+
 	using namespace tuttle::common;
 	using namespace sam;
 

applications/sam/src/sam/ls/main.cpp

@@ -49,6 +49,8 @@ void coutVec( const boost::ptr_vector<T>& v )
 
 int main( int argc, char** argv )
 {
+	signal(SIGINT, signal_callback_handler);
+
 	using namespace tuttle::common;
 	using namespace sam;
 

applications/sam/src/sam/plugins/main.cpp

@@ -45,6 +45,8 @@ std::string getDefaultValues(const tth::ofx::property::OfxhProperty& prop)
 
 void printProperties( const tth::ofx::property::OfxhSet properties, std::string context="" )
 {
+	signal(SIGINT, signal_callback_handler);
+
 	using namespace tuttle::common;
 	boost::shared_ptr<Color> color( Color::get() );
 	if( context.size() == 0 )

applications/sam/src/sam/rm/main.cpp

@@ -136,6 +136,8 @@ void removeFiles( std::vector<boost::filesystem::path> &listing )
 
 int main( int argc, char** argv )
 {
+	signal(SIGINT, signal_callback_handler);
+
 	using namespace tuttle::common;
 	using namespace sam;
 

applications/sam/src/sam/sam/main.cpp

@@ -99,6 +99,8 @@ std::vector<bfs::path> retrieveAllSamCommands(const std::vector<bfs::path>& dirs
 
 int main( int argc, char** argv )
 {
+	signal(SIGINT, signal_callback_handler);
+
 	using namespace tuttle::common;
 	using namespace sam;