diff options
Diffstat (limited to 'tracker-pt/module/module.cpp')
| -rw-r--r-- | tracker-pt/module/module.cpp | 17 |
1 files changed, 16 insertions, 1 deletions
diff --git a/tracker-pt/module/module.cpp b/tracker-pt/module/module.cpp index f665face..db3e8fac 100644 --- a/tracker-pt/module/module.cpp +++ b/tracker-pt/module/module.cpp @@ -2,6 +2,7 @@ #include "module.hpp" #include "camera.h" +#include "compat/camera-names.hpp" #include "frame.hpp" #include "point_extractor.h" #include "ftnoir_tracker_pt_dialog.h" @@ -18,6 +19,7 @@ static const QString module_name = "tracker-pt"; namespace pt_module { +// Traits for OpenCV VideoCapture camera struct pt_module_traits final : pt_runtime_traits { pointer<pt_camera> make_camera() const override @@ -48,9 +50,22 @@ struct pt_module_traits final : pt_runtime_traits struct tracker_pt : Tracker_PT { - tracker_pt() : Tracker_PT(pointer<pt_runtime_traits>(new pt_module_traits)) + tracker_pt() : Tracker_PT(module_name) { } + + pointer<pt_runtime_traits> create_traits() override + { + // Create different traits according to settings + if (s.camera_name().compare(KKinectIRSensor) == 0) + { + // Use Kinect IR trait + return pointer<pt_runtime_traits>(new pt_module_traits); + } + + return pointer<pt_runtime_traits>(new pt_module_traits); + } + }; struct dialog_pt : TrackerDialog_PT |