X-Git-Url: http://git.tdb.fi/?p=libs%2Fvr.git;a=blobdiff_plain;f=source%2Fopenvr%2Fopenvrsystem.cpp;h=d631f48f8dece03c7a27c074924eafc0ded64b86;hp=50d2ecfec54b05380c41bcfa24084aca5e14738a;hb=2acdfcc95c317f30e725e17f0d372fce050830ef;hpb=7ad8d41683649816d3d162f614dc8bb585053311

diff --git a/source/openvr/openvrsystem.cpp b/source/openvr/openvrsystem.cpp
index 50d2ecf..d631f48 100644
--- a/source/openvr/openvrsystem.cpp
+++ b/source/openvr/openvrsystem.cpp
@@ -23,7 +23,8 @@ namespace VR {
 
 unsigned OpenVRSystem::n_instances = 0;
 
-OpenVRSystem::OpenVRSystem()
+OpenVRSystem::OpenVRSystem():
+	n_tracked_devices(0)
 {
 	if(!n_instances)
 	{
@@ -39,6 +40,12 @@ OpenVRSystem::OpenVRSystem()
 		throw runtime_error("OpenVR compositor initialization failed");
 
 	vr::VRCompositor()->SetTrackingSpace(vr::TrackingUniverseSeated);
+
+	for(unsigned i=0; i<vr::k_unMaxTrackedDeviceCount; ++i)
+		if(vr_sys->IsTrackedDeviceConnected(i))
+			n_tracked_devices = i+1;
+
+	tracking_matrices.resize(n_tracked_devices);
 }
 
 OpenVRSystem::~OpenVRSystem()
@@ -94,15 +101,44 @@ OpenVRCombiner *OpenVRSystem::create_combiner(GL::View &v)
 	return new OpenVRCombiner(*this, v);
 }
 
+void OpenVRSystem::tick()
+{
+	vr::IVRSystem *vr_sys = vr::VRSystem();
+
+	vr::VREvent_t event;
+	while(vr_sys->PollNextEvent(&event, sizeof(event)))
+	{
+		if(event.eventType==vr::VREvent_TrackedDeviceActivated)
+			if(event.trackedDeviceIndex>=n_tracked_devices)
+			{
+				n_tracked_devices = event.trackedDeviceIndex+1;
+				tracking_matrices.resize(n_tracked_devices);
+			}
+	}
+}
+
 void OpenVRSystem::update_pose_matrices()
 {
 	vector<vr::TrackedDevicePose_t> poses;
-	poses.resize(vr::k_unTrackedDeviceIndex_Hmd+1);
+	poses.resize(n_tracked_devices);
 	vr::VRCompositor()->WaitGetPoses(&poses[0], poses.size(), 0, 0);
 
-	vr::TrackedDevicePose_t &hmd_pose = poses[vr::k_unTrackedDeviceIndex_Hmd];
-	if(hmd_pose.bPoseIsValid)
-		hmd_matrix = convert_matrix(hmd_pose.mDeviceToAbsoluteTracking);
+	for(unsigned i=0; i<n_tracked_devices; ++i)
+		if(poses[i].bPoseIsValid)
+			tracking_matrices[i] = convert_matrix(poses[i].mDeviceToAbsoluteTracking);
+}
+
+const GL::Matrix &OpenVRSystem::get_tracking_matrix(unsigned index) const
+{
+	if(index>=tracking_matrices.size())
+		throw out_of_range("OpenVRSystem::get_tracking_matrix");
+
+	return tracking_matrices[index];
+}
+
+const GL::Matrix &OpenVRSystem::get_hmd_matrix() const
+{
+	return get_tracking_matrix(vr::k_unTrackedDeviceIndex_Hmd);
 }
 
 } // namespace VR