Changeset 447 in flair-src for trunk/tools

Timestamp:

Sep 16, 2021, 2:01:56 PM (3 years ago)

Author:

Sanahuja Guillaume

Message:

vrpnlite handle disconection

File:

• trunk/tools/VrpnLite/src/VrpnLite.cpp (modified) (2 diffs)

trunk/tools/VrpnLite/src/VrpnLite.cpp

@@ -23 +23 @@
 #include <Quaternion.h>
 #include <string.h>
+#include <bits/stdc++.h>
 
 using namespace std;
@@ -61 +62 @@
     ssize_t rcv=dataSocket->RecvMessage(msg,sizeof(msg),dataSocketTimeout,NULL,NULL,&src_id);
     if(rcv>0) {
-      string object=msg;
-      uint16_t id;
-      memcpy(&id,&msg[rcv-2],sizeof(id));
+      int16_t id;
+      string object=string(msg,rcv-sizeof(id));
+      memcpy(&id,&msg[rcv-sizeof(id)],sizeof(id));
       dataSocket->HostToNetwork((char*)&id,sizeof(id));
-      
-      if(src_id==vrpnobjects_list.size()) {
-        vector<VrpnObject*> vect;
-        vrpnobjects_list.push_back(vect);
-        //Printf("add vect %i\n",src_id);
-      }
-      //assume we receive it in the good order
-      if(id==vrpnobjects_list.at(src_id).size()) {
-        Printf("adding object %s with id %i from %i\n",object.c_str(),id,src_id);
-        VrpnObject* vrpnobject = new VrpnObject(object,vrpnclient->GetTabWidget());
-        vrpnobjects.push_back(vrpnobject);
-        vrpnobjects_list.at(src_id).push_back(vrpnobject);
-      }else {
-        Err("adding object %s failed, expected id %i, got %i\n",object.c_str(),vrpnobjects.size(),id);
+      Printf("src id %i\n",src_id);
+      if(id>0) { //add object
+        if(src_id==vrpnobjects_list.size()) {
+          vector<VrpnObject*> vect;
+          vrpnobjects_list.push_back(vect);
+          //Printf("add vect %i\n",src_id);
+        }
+        //assume we receive it in the good order
+        if(id==vrpnobjects_list.at(src_id).size()+1) {//id 0 is not used as we use positive and negative id for add/remove
+          Printf("adding object %s with id %i from client %i\n",object.c_str(),id,src_id);
+          VrpnObject* vrpnobject = new VrpnObject(object,vrpnclient->GetTabWidget());
+          vrpnobjects.push_back(vrpnobject);
+          vrpnobjects_list.at(src_id).push_back(vrpnobject);
+        }else {
+          Err("adding object %s failed, expected id %i, got %i\n",object.c_str(),vrpnobjects_list.at(src_id).size()+1,id);
+        }
+      } else { //remove object
+        for (vector<VrpnObject*>::iterator it = vrpnobjects.begin();it < vrpnobjects.end(); it++) {
+          if ((*it)->ObjectName() == object) {
+            Printf("removing object %s with id %i from client %i\n",object.c_str(),-id,src_id);
+            auto iteratorEnd=remove(vrpnobjects_list.at(src_id).begin(), vrpnobjects_list.at(src_id).end(),(*it));
+            vrpnobjects_list.at(src_id).erase(iteratorEnd, vrpnobjects_list.at(src_id).end());
+            if(vrpnobjects_list.at(src_id).size()==0) {
+              Printf("not removing empty vrpnobjects_list (need code rewrite\n");
+              //le src_id va continuer a augmenter cote udtsocket et la comparaison if(src_id==vrpnobjects_list.size()) {
+                //ne va pas marcher; il faut identifier autrement les src
+              //auto iteratorEnd=remove(vrpnobjects_list.begin(), vrpnobjects_list.end(),vrpnobjects_list.at(src_id));
+              //vrpnobjects_list.erase(iteratorEnd, vrpnobjects_list.end());
+            }
+            delete (*it);
+            vrpnobjects.erase(it);
+            break;
+          }
+        }
+        
       }
     }