Planet
navi homePPSaboutscreenshotsdownloaddevelopmentforum

source: code/branches/network3/src/libraries/network/packet/Gamestate.cc @ 7344

Last change on this file since 7344 was 7344, checked in by scheusso, 14 years ago

fixed a bug in gamestate diff which wasted bandwidth

  • Property svn:eol-style set to native
File size: 25.5 KB
Line 
1/*
2 *   ORXONOX - the hottest 3D action shooter ever to exist
3 *                    > www.orxonox.net <
4 *
5 *
6 *   License notice:
7 *
8 *   This program is free software; you can redistribute it and/or
9 *   modify it under the terms of the GNU General Public License
10 *   as published by the Free Software Foundation; either version 2
11 *   of the License, or (at your option) any later version.
12 *
13 *   This program is distributed in the hope that it will be useful,
14 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
15 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16 *   GNU General Public License for more details.
17 *
18 *   You should have received a copy of the GNU General Public License
19 *   along with this program; if not, write to the Free Software
20 *   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
21 *
22 *   Author:
23 *      Oliver Scheuss
24 *   Co-authors:
25 *      ...
26 *
27 */
28
29#include "Gamestate.h"
30
31#include <zlib.h>
32
33#include "util/Debug.h"
34#include "core/GameMode.h"
35#include "core/ObjectList.h"
36#include "network/synchronisable/Synchronisable.h"
37#include "network/GamestateHandler.h"
38
39namespace orxonox {
40
41namespace packet {
42
43#define GAMESTATE_START(data) (data + GamestateHeader::getSize())
44
45#define PACKET_FLAG_GAMESTATE  PacketFlag::Reliable
46
47inline bool memzero( uint8_t* data, uint32_t datalength)
48{
49  uint64_t* d = (uint64_t*)data;
50
51  for( unsigned int i=0; i<datalength/8; i++ )
52  {
53    if( *(d+i) != 0 )
54      return false;
55  }
56  // now process the rest (when datalength isn't a multiple of 4)
57  for( unsigned int j = 8*(datalength/8); j<datalength; j++ )
58  {
59    if( *(data+j) != 0 )
60      return false;
61  }
62  return true;
63}
64
65
66Gamestate::Gamestate():
67  header_(0)
68{
69  flags_ = flags_ | PACKET_FLAG_GAMESTATE;
70}
71
72
73Gamestate::Gamestate(uint8_t *data, unsigned int clientID):
74  Packet(data, clientID)
75{
76  flags_ = flags_ | PACKET_FLAG_GAMESTATE;
77  header_ = new GamestateHeader(data_);
78}
79
80
81Gamestate::Gamestate(uint8_t *data)
82{
83  flags_ = flags_ | PACKET_FLAG_GAMESTATE;
84  data_ = data;
85  header_ = new GamestateHeader(data_);
86}
87
88
89Gamestate::Gamestate(const Gamestate& g) :
90    Packet( *(Packet*)&g ), nrOfVariables_(0)
91{
92  flags_ = flags_ | PACKET_FLAG_GAMESTATE;
93  header_ = new GamestateHeader(data_);
94  sizes_ = g.sizes_;
95}
96
97
98Gamestate::~Gamestate()
99{
100  if( header_ )
101    delete header_;
102}
103
104
105bool Gamestate::collectData(int id, uint8_t mode)
106{
107  assert(this->header_==0); // make sure the header didn't exist before
108  uint32_t tempsize=0, currentsize=0;
109  assert(data_==0);
110  uint32_t size = calcGamestateSize(id, mode);
111
112  COUT(4) << "G.ST.Man: producing gamestate with id: " << id << std::endl;
113  if(size==0)
114    return false;
115  data_ = new uint8_t[size + GamestateHeader::getSize()];
116  if(!data_)
117  {
118    COUT(2) << "GameStateManager: could not allocate memory" << std::endl;
119    return false;
120  }
121
122  // create the header object
123  assert( header_ == 0 );
124  header_ = new GamestateHeader(data_);
125
126  //start collect data synchronisable by synchronisable
127  uint8_t *mem = data_; // in this stream store all data of the variables and the headers of the synchronisable
128  mem += GamestateHeader::getSize();
129  ObjectList<Synchronisable>::iterator it;
130  for(it = ObjectList<Synchronisable>::begin(); it; ++it)
131  {
132
133//     tempsize=it->getSize(id, mode);
134
135    tempsize = it->getData(mem, this->sizes_, id, mode);
136    if ( tempsize != 0 )
137      dataVector_.push_back( obj(it->getObjectID(), it->getCreatorID(), tempsize, mem-data_) );
138
139#ifndef NDEBUG
140    if(currentsize+tempsize > size)
141    {
142      assert(0); // if we don't use multithreading this part shouldn't be neccessary
143      // start allocate additional memory
144      COUT(3) << "G.St.Man: need additional memory" << std::endl;
145      ObjectList<Synchronisable>::iterator temp = it;
146      uint32_t addsize=tempsize;
147      while(++temp)
148        addsize+=temp->getSize(id, mode);
149      data_ = (uint8_t *)realloc(data_, GamestateHeader::getSize() + currentsize + addsize);
150      if(!data_)
151        return false;
152      size = currentsize+addsize;
153    }// stop allocate additional memory
154#endif
155//     if(!it->getData(mem, id, mode))
156//       return false; // mem pointer gets automatically increased because of call by reference
157    // increase size counter by size of current synchronisable
158    currentsize+=tempsize;
159  }
160
161
162  //start write gamestate header
163  header_->setDataSize( currentsize );
164  header_->setID( id );
165  header_->setBaseID( GAMESTATEID_INITIAL );
166  header_->setDiffed( false );
167  header_->setComplete( true );
168  header_->setCompressed( false );
169  //stop write gamestate header
170
171  COUT(5) << "G.ST.Man: Gamestate size: " << currentsize << std::endl;
172  COUT(5) << "G.ST.Man: 'estimated' (and corrected) Gamestate size: " << size << std::endl;
173  return true;
174}
175
176
177bool Gamestate::spreadData(uint8_t mode)
178{
179  COUT(4) << "processing gamestate with id " << header_->getID() << endl;
180  assert(data_);
181  assert(!header_->isCompressed());
182  uint8_t *mem=data_+GamestateHeader::getSize();
183  Synchronisable *s;
184
185  // update the data of the objects we received
186  while(mem < data_+GamestateHeader::getSize()+header_->getDataSize())
187  {
188    SynchronisableHeader objectheader(mem);
189
190    s = Synchronisable::getSynchronisable( objectheader.getObjectID() );
191    if(!s)
192    {
193      if (!GameMode::isMaster())
194      {
195        Synchronisable::fabricate(mem, mode);
196      }
197      else
198      {
199        mem += objectheader.getDataSize() + ( objectheader.isDiffed() ? SynchronisableHeaderLight::getSize() : SynchronisableHeader::getSize() );
200      }
201    }
202    else
203    {
204      bool b = s->updateData(mem, mode);
205      assert(b);
206    }
207  }
208   // In debug mode, check first, whether there are no duplicate objectIDs
209#ifndef NDEBUG
210  if(this->getID()%1000==1)
211  {
212    std::list<uint32_t> v1;
213    ObjectList<Synchronisable>::iterator it;
214    for (it = ObjectList<Synchronisable>::begin(); it != ObjectList<Synchronisable>::end(); ++it)
215    {
216      if (it->getObjectID() == OBJECTID_UNKNOWN)
217      {
218        if (it->objectMode_ != 0x0)
219        {
220          COUT(0) << "Found object with OBJECTID_UNKNOWN on the client with objectMode != 0x0!" << std::endl;
221          COUT(0) << "Possible reason for this error: Client created a synchronized object without the Server's approval." << std::endl;
222          COUT(0) << "Objects class: " << it->getIdentifier()->getName() << std::endl;
223          assert(false);
224        }
225      }
226      else
227      {
228        std::list<uint32_t>::iterator it2;
229        for (it2 = v1.begin(); it2 != v1.end(); ++it2)
230        {
231          if (it->getObjectID() == *it2)
232          {
233            COUT(0) << "Found duplicate objectIDs on the client!" << std::endl
234                    << "Are you sure you don't create a Sychnronisable objcect with 'new' \
235                        that doesn't have objectMode = 0x0?" << std::endl;
236            assert(false);
237          }
238        }
239        v1.push_back(it->getObjectID());
240      }
241    }
242  }
243#endif
244  return true;
245}
246
247
248uint32_t Gamestate::getSize() const
249{
250  assert(data_);
251  if(header_->isCompressed())
252    return header_->getCompSize()+GamestateHeader::getSize();
253  else
254  {
255    return header_->getDataSize()+GamestateHeader::getSize();
256  }
257}
258
259
260bool Gamestate::operator==(packet::Gamestate gs)
261{
262  uint8_t *d1 = data_+GamestateHeader::getSize();
263  uint8_t *d2 = gs.data_+GamestateHeader::getSize();
264  GamestateHeader* h1 = new GamestateHeader(data_);
265  GamestateHeader* h2 = new GamestateHeader(gs.data_);
266  assert(h1->getDataSize() == h2->getDataSize());
267  assert(!isCompressed());
268  assert(!gs.isCompressed());
269  return memcmp(d1, d2, h1->getDataSize())==0;
270}
271
272
273bool Gamestate::process()
274{
275  return GamestateHandler::addGamestate(this, getClientID());
276}
277
278
279bool Gamestate::compressData()
280{
281  assert(data_);
282  assert(!header_->isCompressed());
283  uLongf buffer = (uLongf)(((header_->getDataSize() + 12)*1.01)+1);
284  if(buffer==0)
285    return false;
286
287  uint8_t *ndata = new uint8_t[buffer+GamestateHeader::getSize()];
288  uint8_t *dest = ndata + GamestateHeader::getSize();
289  uint8_t *source = data_ + GamestateHeader::getSize();
290  int retval;
291  retval = compress( dest, &buffer, source, (uLong)(header_->getDataSize()) );
292  switch ( retval )
293  {
294    case Z_OK: COUT(5) << "G.St.Man: compress: successfully compressed" << std::endl; break;
295    case Z_MEM_ERROR: COUT(1) << "G.St.Man: compress: not enough memory available in gamestate.compress" << std::endl; return false;
296    case Z_BUF_ERROR: COUT(2) << "G.St.Man: compress: not enough memory available in the buffer in gamestate.compress" << std::endl; return false;
297    case Z_DATA_ERROR: COUT(2) << "G.St.Man: compress: data corrupted in gamestate.compress" << std::endl; return false;
298  }
299
300  //copy and modify header
301  GamestateHeader *temp = header_;
302  header_ = new GamestateHeader(ndata, temp);
303  delete temp;
304  //delete old data
305  delete[] data_;
306  //save new data
307  data_ = ndata;
308  header_->setCompSize( buffer );
309  header_->setCompressed( true );
310  COUT(0) << "gamestate compress datasize: " << header_->getDataSize() << " compsize: " << header_->getCompSize() << std::endl;
311  return true;
312}
313
314
315bool Gamestate::decompressData()
316{
317  assert(data_);
318  assert(header_->isCompressed());
319  COUT(4) << "GameStateClient: uncompressing gamestate. id: " << header_->getID() << ", baseid: " << header_->getBaseID() << ", datasize: " << header_->getDataSize() << ", compsize: " << header_->getCompSize() << std::endl;
320  uint32_t datasize = header_->getDataSize();
321  uint32_t compsize = header_->getCompSize();
322  uint32_t bufsize;
323  bufsize = datasize;
324  assert(bufsize!=0);
325  uint8_t *ndata = new uint8_t[bufsize + GamestateHeader::getSize()];
326  uint8_t *dest = ndata + GamestateHeader::getSize();
327  uint8_t *source = data_ + GamestateHeader::getSize();
328  int retval;
329  uLongf length=bufsize;
330  retval = uncompress( dest, &length, source, (uLong)compsize );
331  switch ( retval )
332  {
333    case Z_OK: COUT(5) << "successfully decompressed" << std::endl; break;
334    case Z_MEM_ERROR: COUT(1) << "not enough memory available" << std::endl; return false;
335    case Z_BUF_ERROR: COUT(2) << "not enough memory available in the buffer" << std::endl; return false;
336    case Z_DATA_ERROR: COUT(2) << "data corrupted (zlib)" << std::endl; return false;
337  }
338
339  //copy over the header
340  GamestateHeader *temp = header_;
341  header_ = new GamestateHeader( data_, header_ );
342  delete temp;
343
344  if (this->bDataENetAllocated_)
345  {
346    // Memory was allocated by ENet. --> We let it be since enet_packet_destroy will
347    // deallocated it anyway. So data and packet stay together.
348    this->bDataENetAllocated_ = false;
349  }
350  else
351  {
352    // We allocated the memory in the first place (unlikely). So we destroy the old data
353    // and overwrite it with the new decompressed data.
354    delete[] this->data_;
355  }
356
357  //set new pointers
358  data_ = ndata;
359  header_->setCompressed( false );
360  assert(header_->getDataSize()==datasize);
361  assert(header_->getCompSize()==compsize);
362  return true;
363}
364
365
366inline void /*Gamestate::*/diffObject( uint8_t*& newData, uint8_t*& origData, uint8_t*& baseData, SynchronisableHeader& objectHeader, std::vector<uint32_t>::iterator& sizes )
367{
368  //       COUT(4) << "dodiff" << endl;
369  //       if(baseOffset==0)
370  //       {
371  //         assert(origOffset==0);
372  //       }
373  uint32_t objectOffset = SynchronisableHeader::getSize(); // offset inside the object in the origData and baseData
374  // Check whether the whole object stayed the same
375  if( memcmp( origData+objectOffset, baseData+objectOffset, objectHeader.getDataSize()) == 0 )
376  {
377    //         COUT(4) << "skip object" << Synchronisable::getSynchronisable(h.getObjectID())->getIdentifier()->getName() << endl;
378    origData += objectOffset + objectHeader.getDataSize(); // skip the whole object
379    baseData += objectOffset + objectHeader.getDataSize();
380    sizes += Synchronisable::getSynchronisable(objectHeader.getObjectID())->getNrOfVariables();
381  }
382  else
383  {
384    //         if( Synchronisable::getSynchronisable(h.getObjectID())->getIdentifier()->getName() == "Bot" )
385    //           COUT(0) << "blub" << endl;
386    //         COUT(4) << "object diff: " << Synchronisable::getSynchronisable(h.getObjectID())->getIdentifier()->getName() << endl;
387    //         COUT(4) << "diff " << h.getObjectID() << ":";
388    // Now start to diff the Object
389    SynchronisableHeaderLight h2(newData);
390    h2 = objectHeader; // copy over the objectheader
391    VariableID variableID = 0;
392    uint32_t newObjectOffset = SynchronisableHeaderLight::getSize();
393    // iterate through all variables
394    while( objectOffset < objectHeader.getDataSize()+SynchronisableHeader::getSize() )
395    {
396      // check whether variable changed and write id and copy over variable to the new stream
397      // otherwise skip variable
398//       assert(sizes != this->sizes_.end());
399      uint32_t varSize = *sizes;
400      assert( varSize == Synchronisable::getSynchronisable(objectHeader.getObjectID())->getVarSize(variableID) );
401      if ( varSize != 0 )
402      {
403        if ( memcmp(origData+objectOffset, baseData+objectOffset, varSize) != 0 )
404        {
405          //               COUT(4) << "copy variable" << endl;
406          *(VariableID*)(newData+newObjectOffset) = variableID; // copy over the variableID
407          newObjectOffset += sizeof(VariableID);
408          memcpy( newData+newObjectOffset, origData+objectOffset, varSize );
409          newObjectOffset += varSize;
410          objectOffset += varSize;
411        }
412        else
413        {
414          //               COUT(4) << "skip variable" << endl;
415          objectOffset += varSize;
416        }
417      }
418//           else
419//             COUT(4) << "varsize 0" << endl;
420
421      ++variableID;
422      ++sizes;
423    }
424           
425    if( Synchronisable::getSynchronisable(objectHeader.getObjectID())->getNrOfVariables() != variableID )
426      sizes += Synchronisable::getSynchronisable(objectHeader.getObjectID())->getNrOfVariables() - variableID;
427    //         COUT(4) << endl;
428   
429    h2.setDiffed(true);
430    h2.setDataSize(newObjectOffset-SynchronisableHeaderLight::getSize());
431    assert(objectOffset == objectHeader.getDataSize()+SynchronisableHeader::getSize());
432    assert(h2.getDataSize()>0);
433    origData += objectOffset;
434    //         baseOffset += temp + h.getDataSize()+SynchronisableHeader::getSize() - baseData;
435    //baseOffset += objectOffset;
436    //         SynchronisableHeader htemp(temp);
437    //         baseOffset += SynchronisableHeader::getSize() + htemp.getDataSize();
438    //         {
439      //           SynchronisableHeader htemp2( baseData+(temp-baseData+objectOffset) );
440    //           if( baseData+(temp-baseData+objectOffset) < baseData+baseLength )
441    //           {
442      //             assert(htemp2.getClassID()<500);
443    //             assert(htemp2.getDataSize()!=0 && htemp2.getDataSize()<1000);
444    //             assert(htemp2.isDiffed()==false);
445    //           }
446    //         }
447    baseData += objectOffset;
448    newData += newObjectOffset;
449  }
450}
451
452inline void /*Gamestate::*/copyObject( uint8_t*& newData, uint8_t*& origData, uint8_t*& baseData, SynchronisableHeader& objectHeader, std::vector<uint32_t>::iterator& sizes )
453{
454  //       COUT(4) << "docopy" << endl;
455  // Just copy over the whole Object
456  memcpy( newData, origData, objectHeader.getDataSize()+SynchronisableHeader::getSize() );
457  newData += objectHeader.getDataSize()+SynchronisableHeader::getSize();
458  origData += objectHeader.getDataSize()+SynchronisableHeader::getSize();
459  SynchronisableHeader baseHeader( baseData );
460  baseData += baseHeader.getDataSize()+SynchronisableHeader::getSize();
461  //       COUT(4) << "copy " << h.getObjectID() << endl;
462  //       COUT(4) << "copy " << h.getObjectID() << ":";
463  sizes += Synchronisable::getSynchronisable(objectHeader.getObjectID())->getNrOfVariables();
464//   for( unsigned int i = 0; i < Synchronisable::getSynchronisable(objectHeader.getObjectID())->getNrOfVariables(); ++i )
465//   {
466//     //         COUT(4) << " " << *sizes;
467//     ++sizes;
468//   }
469    //       COUT(4) << endl;
470}
471
472Gamestate* Gamestate::diffVariables(Gamestate *base)
473{
474  assert(this && base); assert(data_ && base->data_);
475  assert(!header_->isCompressed() && !base->header_->isCompressed());
476  assert(!header_->isDiffed());
477
478
479  // *** first do a raw diff of the two gamestates
480
481  uint8_t *baseData = GAMESTATE_START(base->data_);
482  uint8_t *origData = GAMESTATE_START(this->data_);
483  uint8_t *origDataEnd = origData + header_->getDataSize();
484  uint8_t *baseDataEnd = baseData + base->header_->getDataSize();
485//   uint32_t origLength = header_->getDataSize();
486//   uint32_t baseLength = base->header_->getDataSize();
487
488  assert( header_->getDataSize() && base->header_->getDataSize() );
489 
490  uint32_t newDataSize = header_->getDataSize() + GamestateHeader::getSize() + sizeof(uint32_t)*this->nrOfVariables_;
491  uint8_t *newData = new uint8_t[newDataSize]; // this is the maximum size needed in the worst case
492  uint8_t *dest = GAMESTATE_START(newData);
493
494//   uint32_t baseOffset = 0; //offset in the diffed stream
495//   uint32_t origOffset = 0; //offset in the new stream with removed 0's
496  std::vector<uint32_t>::iterator sizes = this->sizes_.begin();
497
498  while( origData < origDataEnd )
499  {
500    //iterate through all objects
501
502    SynchronisableHeader h(origData);
503
504    // Find (if possible) the current object in the datastream of the old gamestate
505    // Start at the current offset position
506    if(baseData >= baseDataEnd)
507      baseData = GAMESTATE_START(base->data_);
508//     uint8_t* temp = baseData + baseOffset;
509    uint32_t objectID = h.getObjectID();
510    assert(baseData < baseDataEnd);
511    assert(dest < newData + newDataSize);
512    assert(sizes != this->sizes_.end());
513    bool diffedObject = false;
514    while ( baseData < baseDataEnd )
515    {
516      SynchronisableHeader htemp(baseData);
517      assert( htemp.getDataSize()!=0 );
518      if ( htemp.getObjectID() == objectID )
519      {
520        assert( h.getClassID() == htemp.getClassID() );
521//         goto DODIFF;
522        diffObject(dest, origData, baseData, h, sizes);
523        diffedObject = true;
524        break;
525      }
526      {
527        SynchronisableHeader htemp2(baseData+htemp.getDataSize()+SynchronisableHeader::getSize());
528        if( baseData+htemp.getDataSize()+SynchronisableHeader::getSize() < baseDataEnd )
529        {
530          assert(htemp2.getClassID()<500);
531          assert(htemp2.getDataSize()!=0 && htemp2.getDataSize()<1000);
532          assert(htemp2.isDiffed()==false);
533        }
534      }
535      baseData += htemp.getDataSize()+SynchronisableHeader::getSize();
536//       assert( baseData+htemp.getDataSize()+SynchronisableHeader::getSize() <=baseData+baseLength );
537//       temp += htemp.getDataSize()+SynchronisableHeader::getSize();
538     
539    }
540    // If not found start looking at the beginning
541    assert( diffedObject || baseData == baseDataEnd );
542    baseData = GAMESTATE_START(base->data_);
543    {
544      SynchronisableHeader htemp2(baseData);
545      if( baseData < baseDataEnd )
546      {
547        assert(htemp2.getClassID()<500);
548        assert(htemp2.getDataSize()!=0 && htemp2.getDataSize()<1000);
549        assert(htemp2.isDiffed()==false);
550      }
551    }
552    while ( !diffedObject && baseData < baseDataEnd )
553    {
554      SynchronisableHeader htemp(baseData);
555      if ( htemp.getObjectID() == objectID )
556      {
557        assert( h.getClassID() == htemp.getClassID() );
558//         goto DODIFF;
559        diffObject(dest, origData, baseData, h, sizes);
560        break;
561      }
562//       {
563//         SynchronisableHeader htemp2(baseData+htemp.getDataSize()+SynchronisableHeader::getSize());
564//         if( baseData+htemp.getDataSize()+SynchronisableHeader::getSize() < baseDataEnd )
565//         {
566//           assert(htemp2.getClassID()<500);
567//           assert(htemp2.getDataSize()!=0 && htemp2.getDataSize()<1000);
568//           assert(htemp2.isDiffed()==false);
569//         }
570//       }
571      baseData += htemp.getDataSize()+SynchronisableHeader::getSize();
572    }
573    // Object is new, thus never transmitted -> just copy over
574//     goto DOCOPY;
575    if ( baseData >= baseDataEnd )
576    {
577      assert(baseData == baseDataEnd);
578      copyObject(dest, origData, baseData, h, sizes);
579      assert(sizes != this->sizes_.end() || origData>=origDataEnd);
580    }
581  }
582  assert(sizes==this->sizes_.end());
583
584
585  Gamestate *g = new Gamestate(newData, getClientID());
586  assert(g->header_);
587  *(g->header_) = *header_;
588  g->header_->setBaseID( base->getID() );
589  g->header_->setDataSize(dest - newData - GamestateHeader::getSize());
590  g->flags_=flags_;
591  g->packetDirection_ = packetDirection_;
592  assert(!g->isCompressed());
593  return g;
594}
595
596
597Gamestate* Gamestate::diffData(Gamestate *base)
598{
599  assert(this && base); assert(data_ && base->data_);
600  assert(!header_->isCompressed() && !base->header_->isCompressed());
601  assert(!header_->isDiffed());
602
603  uint8_t *basep = GAMESTATE_START(base->data_);
604  uint8_t *gs = GAMESTATE_START(this->data_);
605  uint32_t dest_length = header_->getDataSize();
606
607  if(dest_length==0)
608    return NULL;
609
610  uint8_t *ndata = new uint8_t[dest_length*sizeof(uint8_t)+GamestateHeader::getSize()];
611  uint8_t *dest = GAMESTATE_START(ndata);
612
613  rawDiff( dest, gs, basep, header_->getDataSize(), base->header_->getDataSize() );
614#ifndef NDEBUG
615  uint8_t *dest2 = new uint8_t[dest_length];
616  rawDiff( dest2, dest, basep, header_->getDataSize(), base->header_->getDataSize() );
617  assert( memcmp( dest2, gs, dest_length) == 0 );
618  delete dest2;
619#endif
620
621  Gamestate *g = new Gamestate(ndata, getClientID());
622  assert(g->header_);
623  *(g->header_) = *header_;
624  g->header_->setDiffed( true );
625  g->header_->setBaseID( base->getID() );
626  g->flags_=flags_;
627  g->packetDirection_ = packetDirection_;
628  assert(g->isDiffed());
629  assert(!g->isCompressed());
630  return g;
631}
632
633
634Gamestate* Gamestate::undiff(Gamestate *base)
635{
636  assert(this && base); assert(data_ && base->data_);
637  assert(!header_->isCompressed() && !base->header_->isCompressed());
638  assert(header_->isDiffed());
639
640  uint8_t *basep = GAMESTATE_START(base->data_);
641  uint8_t *gs = GAMESTATE_START(this->data_);
642  uint32_t dest_length = header_->getDataSize();
643
644  if(dest_length==0)
645    return NULL;
646
647  uint8_t *ndata = new uint8_t[dest_length*sizeof(uint8_t)+GamestateHeader::getSize()];
648  uint8_t *dest = ndata + GamestateHeader::getSize();
649
650  rawDiff( dest, gs, basep, header_->getDataSize(), base->header_->getDataSize() );
651
652  Gamestate *g = new Gamestate(ndata, getClientID());
653  assert(g->header_);
654  *(g->header_) = *header_;
655  g->header_->setDiffed( false );
656  g->flags_=flags_;
657  g->packetDirection_ = packetDirection_;
658  assert(!g->isDiffed());
659  assert(!g->isCompressed());
660  return g;
661}
662
663
664void Gamestate::rawDiff( uint8_t* newdata, uint8_t* data, uint8_t* basedata, uint32_t datalength, uint32_t baselength)
665{
666  uint64_t* gd = (uint64_t*)data;
667  uint64_t* bd = (uint64_t*)basedata;
668  uint64_t* nd = (uint64_t*)newdata;
669
670  unsigned int i;
671  for( i=0; i<datalength/8; i++ )
672  {
673    if( i<baselength/8 )
674      *(nd+i) = *(gd+i) ^ *(bd+i);  // xor the data
675    else
676      *(nd+i) = *(gd+i); // just copy over the data
677  }
678  unsigned int j;
679  // now process the rest (when datalength isn't a multiple of 4)
680  for( j = 8*(datalength/8); j<datalength; j++ )
681  {
682    if( j<baselength )
683      *(newdata+j) = *(data+j) ^ *(basedata+j); // xor
684    else
685      *(newdata+j) = *(data+j); // just copy
686  }
687  assert(j==datalength);
688}
689
690
691Gamestate* Gamestate::doSelection(unsigned int clientID, unsigned int targetSize){
692  assert(data_);
693  std::list<obj>::iterator it;
694
695  // allocate memory for new data
696  uint8_t *gdata = new uint8_t[header_->getDataSize()+GamestateHeader::getSize()];
697  // create a gamestate out of it
698  Gamestate *gs = new Gamestate(gdata);
699  uint8_t *newdata = gdata + GamestateHeader::getSize();
700  uint8_t *origdata = GAMESTATE_START(data_);
701
702  //copy the GamestateHeader
703  assert(gs->header_);
704  *(gs->header_) = *header_;
705
706  uint32_t objectOffset;
707  unsigned int objectsize, destsize=0;
708  // TODO: Why is this variable not used?
709  //Synchronisable *object;
710
711  //call TrafficControl
712  TrafficControl::getInstance()->processObjectList( clientID, header_->getID(), dataVector_ );
713
714  //copy in the zeros
715//   std::list<obj>::iterator itt;
716//   COUT(0) << "myvector contains:";
717//   for ( itt=dataVector_.begin() ; itt!=dataVector_.end(); itt++ )
718//     COUT(0) << " " << (*itt).objID;
719//   COUT(0) << endl;
720  for(it=dataVector_.begin(); it!=dataVector_.end();){
721    SynchronisableHeader oldobjectheader(origdata);
722    SynchronisableHeader newobjectheader(newdata);
723    if ( (*it).objSize == 0 )
724    {
725      ++it;
726      continue;
727    }
728    objectsize = oldobjectheader.getDataSize()+SynchronisableHeader::getSize();
729    objectOffset=SynchronisableHeader::getSize(); //skip the size and the availableData variables in the objectheader
730    if ( (*it).objID == oldobjectheader.getObjectID() ){
731      memcpy(newdata, origdata, objectsize);
732      ++it;
733    }else{
734      newobjectheader = oldobjectheader;
735      memset(newdata+objectOffset, 0, objectsize-objectOffset);
736    }
737    newdata += objectsize;
738    origdata += objectsize;
739    destsize += objectsize;
740  }
741#ifndef NDEBUG
742  uint32_t origsize = destsize;
743  while ( origsize < header_->getDataSize() )
744  {
745    SynchronisableHeader oldobjectheader(origdata);
746    objectsize = oldobjectheader.getDataSize()+SynchronisableHeader::getSize();
747    origdata += objectsize;
748    origsize += objectsize;
749  }
750  assert(origsize==header_->getDataSize());
751  assert(destsize!=0);
752#endif
753  gs->header_->setDataSize( destsize );
754  return gs;
755}
756
757
758uint32_t Gamestate::calcGamestateSize(int32_t id, uint8_t mode)
759{
760  uint32_t size = 0;
761  uint32_t nrOfVariables = 0;
762    // get the start of the Synchronisable list
763  ObjectList<Synchronisable>::iterator it;
764    // get total size of gamestate
765  for(it = ObjectList<Synchronisable>::begin(); it; ++it){
766    size+=it->getSize(id, mode); // size of the actual data of the synchronisable
767    nrOfVariables += it->getNrOfVariables();
768  }
769//   COUT(0) << "allocating " << nrOfVariables << " ints" << endl;
770  this->sizes_.reserve(nrOfVariables);
771  return size;
772}
773
774
775} //namespace packet
776} //namespace orxonox
Note: See TracBrowser for help on using the repository browser.