threading: Set thread names for perf

Author: xlz

include/internal/libfreenect2/async_packet_processor.h

@@ -125,6 +125,7 @@ class AsyncPacketProcessor : public PacketProcessor<PacketT>
   /** Asynchronously process a provided packet. */
   void execute()
   {
+    this_thread::set_name(processor_->name());
     libfreenect2::unique_lock l(packet_mutex_);
 
     while(!shutdown_)

include/internal/libfreenect2/depth_packet_processor.h

@@ -130,6 +130,7 @@ class OpenGLDepthPacketProcessor : public DepthPacketProcessor
   virtual void loadXZTables(const float *xtable, const float *ztable);
   virtual void loadLookupTable(const short *lut);
 
+  virtual const char *name() { return "OpenGL"; }
   virtual void process(const DepthPacket &packet);
 private:
   OpenGLDepthPacketProcessorImpl *impl_;
@@ -152,6 +153,7 @@ class CpuDepthPacketProcessor : public DepthPacketProcessor
   virtual void loadXZTables(const float *xtable, const float *ztable);
   virtual void loadLookupTable(const short *lut);
 
+  virtual const char *name() { return "CPU"; }
   virtual void process(const DepthPacket &packet);
 private:
   CpuDepthPacketProcessorImpl *impl_;
@@ -173,6 +175,7 @@ class OpenCLDepthPacketProcessor : public DepthPacketProcessor
   virtual void loadXZTables(const float *xtable, const float *ztable);
   virtual void loadLookupTable(const short *lut);
 
+  virtual const char *name() { return "OpenCL"; }
   virtual void process(const DepthPacket &packet);
 private:
   OpenCLDepthPacketProcessorImpl *impl_;
@@ -195,6 +198,7 @@ class CudaDepthPacketProcessor : public DepthPacketProcessor
   virtual void loadLookupTable(const short *lut);
 
   virtual bool good();
+  virtual const char *name() { return "CUDA"; }
 
   virtual void process(const DepthPacket &packet);
 protected:

include/internal/libfreenect2/packet_processor.h

@@ -52,6 +52,8 @@ class PacketProcessor
 
   virtual bool good() { return true; }
 
+  virtual const char *name() { return "a packet processor"; }
+
   /**
    * A new packet has arrived, process it.
    * @param packet Packet to process.

include/internal/libfreenect2/rgb_packet_processor.h

@@ -87,6 +87,7 @@ class TurboJpegRgbPacketProcessor : public RgbPacketProcessor
   TurboJpegRgbPacketProcessor();
   virtual ~TurboJpegRgbPacketProcessor();
   virtual void process(const libfreenect2::RgbPacket &packet);
+  virtual const char *name() { return "TurboJPEG"; }
 private:
   TurboJpegRgbPacketProcessorImpl *impl_; ///< Decoder implementation.
 };
@@ -101,6 +102,7 @@ class VTRgbPacketProcessor : public RgbPacketProcessor
   VTRgbPacketProcessor();
   virtual ~VTRgbPacketProcessor();
   virtual void process(const libfreenect2::RgbPacket &packet);
+  virtual const char *name() { return "VideoToolbox"; }
 private:
   VTRgbPacketProcessorImpl *impl_;
 };
@@ -115,6 +117,7 @@ class VaapiRgbPacketProcessor : public RgbPacketProcessor
   VaapiRgbPacketProcessor();
   virtual ~VaapiRgbPacketProcessor();
   virtual bool good();
+  virtual const char *name() { return "VAAPI"; }
   virtual void process(const libfreenect2::RgbPacket &packet);
 protected:
   virtual Allocator *getAllocator();
@@ -132,6 +135,7 @@ class TegraJpegRgbPacketProcessor : public RgbPacketProcessor
   TegraJpegRgbPacketProcessor();
   virtual ~TegraJpegRgbPacketProcessor();
   virtual bool good();
+  virtual const char *name() { return "TegraJPEG"; }
   virtual void process(const libfreenect2::RgbPacket &packet);
 private:
   TegraJpegRgbPacketProcessorImpl *impl_;

include/internal/libfreenect2/threading.h

@@ -94,5 +94,25 @@ using namespace tthread::this_thread;
 
 #endif
 
+#if defined(__linux__)
+#include <sys/prctl.h>
+#elif defined(__APPLE__)
+#include <pthread.h>
+#endif
+
+namespace libfreenect2
+{
+namespace this_thread
+{
+  static inline void set_name(const char *name)
+  {
+#if defined(__linux__)
+    prctl(PR_SET_NAME, name);
+#elif defined(__APPLE__)
+    pthread_setname_np(name);
+#endif
+  }
+}
+}
 
 #endif /* THREADING_H_ */

src/event_loop.cpp

@@ -91,6 +91,7 @@ void EventLoop::stop()
 /** Execute the job, until shut down. */
 void EventLoop::execute()
 {
+  this_thread::set_name("USB");
   timeval t;
   t.tv_sec = 0;
   t.tv_usec = 100000;