Remove debug printf

[hercules2020/kcf.git] / src / debug.h

diff --git a/src/debug.h b/src/debug.h
index 85a4890bbad684dd61e61b6a1b93ae05ea28e503..cce8327201b44732d131ff4f080a01bed7813180 100644 (file)
--- a/src/debug.h
+++ b/src/debug.h
@@ -7,10 +7,11 @@
 #include <stdio.h>
 #include <opencv2/opencv.hpp>
 #include "dynmem.hpp"
-#ifdef CUFFT
-#include "complexmat.cuh"
-#else
 #include "complexmat.hpp"
+
+#ifdef CUFFT
+#include <cufft.h>
+#include "nvToolsExt.h"
 #endif
 
 
@@ -41,6 +42,7 @@ class DbgTracer {
 
   public:
     bool debug = false;
+    static constexpr int precision = 2;
 
     std::string indent() { return std::string(indentLvl * 4, ' '); }
 
@@ -51,6 +53,9 @@ class DbgTracer {
       public:
         FTrace(DbgTracer &dt, const char *fn, const char *format, ...) : t(dt), funcName(fn)
         {
+#ifdef CUFFT
+            nvtxRangePushA(fn);
+#endif
             if (!t.debug) return;
             char *arg;
             va_list vl;
@@ -64,6 +69,9 @@ class DbgTracer {
         }
         ~FTrace()
         {
+#ifdef CUFFT
+            nvtxRangePop();
+#endif
             if (!t.debug) return;
             t.indentLvl--;
             std::cerr << t.indent() << "}" << std::endl;
@@ -74,8 +82,11 @@ class DbgTracer {
     void traceVal(const char *name, const T& obj, int line, bool always = false)
     {
         (void)line;
-        if (debug || always)
+        if (debug || always) {
+            IOSave s(std::cerr);
+            std::cerr << std::setprecision(precision);
             std::cerr << indent() << name /*<< " @" << line */ << " " << print(obj) << std::endl;
+        }
     }
 
     template <typename T> struct Printer {