Planet
navi homePPSaboutscreenshotsdownloaddevelopmentforum

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

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

fix in gamestate diff

  • Property svn:eol-style set to native
File size: 25.5 KB
RevLine 
[1711]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:
[3084]23 *      Oliver Scheuss
[1711]24 *   Co-authors:
25 *      ...
26 *
27 */
28
[1701]29#include "Gamestate.h"
[3214]30
[2773]31#include <zlib.h>
[3214]32
33#include "util/Debug.h"
[2896]34#include "core/GameMode.h"
[3214]35#include "core/ObjectList.h"
36#include "network/synchronisable/Synchronisable.h"
37#include "network/GamestateHandler.h"
[1701]38
[2171]39namespace orxonox {
[1701]40
41namespace packet {
42
[2662]43#define GAMESTATE_START(data) (data + GamestateHeader::getSize())
[1740]44
[3214]45#define PACKET_FLAG_GAMESTATE  PacketFlag::Reliable
[2087]46
[7163]47inline bool memzero( uint8_t* data, uint32_t datalength)
48{
49  uint64_t* d = (uint64_t*)data;
[2662]50
[7163]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)
[1701]68{
[1907]69  flags_ = flags_ | PACKET_FLAG_GAMESTATE;
[1701]70}
71
[7163]72
[1907]73Gamestate::Gamestate(uint8_t *data, unsigned int clientID):
[7163]74  Packet(data, clientID)
[1701]75{
[1907]76  flags_ = flags_ | PACKET_FLAG_GAMESTATE;
[2662]77  header_ = new GamestateHeader(data_);
[1701]78}
79
[7163]80
[1907]81Gamestate::Gamestate(uint8_t *data)
82{
83  flags_ = flags_ | PACKET_FLAG_GAMESTATE;
[7163]84  data_ = data;
[2662]85  header_ = new GamestateHeader(data_);
[1907]86}
[1701]87
[7163]88
[2662]89Gamestate::Gamestate(const Gamestate& g) :
[7163]90    Packet( *(Packet*)&g ), nrOfVariables_(0)
[2662]91{
92  flags_ = flags_ | PACKET_FLAG_GAMESTATE;
93  header_ = new GamestateHeader(data_);
[7163]94  sizes_ = g.sizes_;
[2662]95}
[1907]96
[2662]97
[1701]98Gamestate::~Gamestate()
99{
[3198]100  if( header_ )
101    delete header_;
[1701]102}
103
[7163]104
[2171]105bool Gamestate::collectData(int id, uint8_t mode)
[1701]106{
[2662]107  assert(this->header_==0); // make sure the header didn't exist before
108  uint32_t tempsize=0, currentsize=0;
[1751]109  assert(data_==0);
[2662]110  uint32_t size = calcGamestateSize(id, mode);
[1740]111
[1701]112  COUT(4) << "G.ST.Man: producing gamestate with id: " << id << std::endl;
113  if(size==0)
114    return false;
[2662]115  data_ = new uint8_t[size + GamestateHeader::getSize()];
[7163]116  if(!data_)
117  {
[1701]118    COUT(2) << "GameStateManager: could not allocate memory" << std::endl;
119    return false;
120  }
[6417]121
[2662]122  // create the header object
[3198]123  assert( header_ == 0 );
[2662]124  header_ = new GamestateHeader(data_);
[2087]125
[1701]126  //start collect data synchronisable by synchronisable
[7163]127  uint8_t *mem = data_; // in this stream store all data of the variables and the headers of the synchronisable
[2662]128  mem += GamestateHeader::getSize();
[2171]129  ObjectList<Synchronisable>::iterator it;
[7163]130  for(it = ObjectList<Synchronisable>::begin(); it; ++it)
131  {
[6417]132
[3084]133//     tempsize=it->getSize(id, mode);
134
[7163]135    tempsize = it->getData(mem, this->sizes_, id, mode);
[3084]136    if ( tempsize != 0 )
137      dataVector_.push_back( obj(it->getObjectID(), it->getCreatorID(), tempsize, mem-data_) );
[6417]138
[2662]139#ifndef NDEBUG
[7163]140    if(currentsize+tempsize > size)
141    {
[2171]142      assert(0); // if we don't use multithreading this part shouldn't be neccessary
[1701]143      // start allocate additional memory
144      COUT(3) << "G.St.Man: need additional memory" << std::endl;
[2171]145      ObjectList<Synchronisable>::iterator temp = it;
[2662]146      uint32_t addsize=tempsize;
[1701]147      while(++temp)
[1907]148        addsize+=temp->getSize(id, mode);
[2662]149      data_ = (uint8_t *)realloc(data_, GamestateHeader::getSize() + currentsize + addsize);
[1701]150      if(!data_)
151        return false;
152      size = currentsize+addsize;
153    }// stop allocate additional memory
[2662]154#endif
[3084]155//     if(!it->getData(mem, id, mode))
156//       return false; // mem pointer gets automatically increased because of call by reference
[1701]157    // increase size counter by size of current synchronisable
158    currentsize+=tempsize;
159  }
[1740]160
161
[1701]162  //start write gamestate header
[2662]163  header_->setDataSize( currentsize );
164  header_->setID( id );
[3198]165  header_->setBaseID( GAMESTATEID_INITIAL );
[2662]166  header_->setDiffed( false );
167  header_->setComplete( true );
168  header_->setCompressed( false );
[1701]169  //stop write gamestate header
[1740]170
[1701]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
[7163]176
[2171]177bool Gamestate::spreadData(uint8_t mode)
[1701]178{
[2662]179  COUT(4) << "processing gamestate with id " << header_->getID() << endl;
[1751]180  assert(data_);
[2662]181  assert(!header_->isCompressed());
182  uint8_t *mem=data_+GamestateHeader::getSize();
[1907]183  Synchronisable *s;
[1740]184
[1907]185  // update the data of the objects we received
[7163]186  while(mem < data_+GamestateHeader::getSize()+header_->getDataSize())
187  {
[2662]188    SynchronisableHeader objectheader(mem);
[1701]189
[2662]190    s = Synchronisable::getSynchronisable( objectheader.getObjectID() );
[1907]191    if(!s)
[1701]192    {
[2896]193      if (!GameMode::isMaster())
[2662]194      {
195        Synchronisable::fabricate(mem, mode);
196      }
197      else
198      {
[7163]199        mem += objectheader.getDataSize() + ( objectheader.isDiffed() ? SynchronisableHeaderLight::getSize() : SynchronisableHeader::getSize() );
[2662]200      }
[1701]201    }
[1907]202    else
203    {
204      bool b = s->updateData(mem, mode);
205      assert(b);
206    }
[1701]207  }
[2662]208   // In debug mode, check first, whether there are no duplicate objectIDs
209#ifndef NDEBUG
[7163]210  if(this->getID()%1000==1)
211  {
[3084]212    std::list<uint32_t> v1;
213    ObjectList<Synchronisable>::iterator it;
[7163]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        {
[3084]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        }
[2662]225      }
[7163]226      else
227      {
[3084]228        std::list<uint32_t>::iterator it2;
[7163]229        for (it2 = v1.begin(); it2 != v1.end(); ++it2)
230        {
231          if (it->getObjectID() == *it2)
232          {
[3084]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          }
[2662]238        }
[3084]239        v1.push_back(it->getObjectID());
[2662]240      }
241    }
242  }
243#endif
[1701]244  return true;
245}
246
[7163]247
[2662]248uint32_t Gamestate::getSize() const
[1701]249{
[1711]250  assert(data_);
[2662]251  if(header_->isCompressed())
252    return header_->getCompSize()+GamestateHeader::getSize();
[1701]253  else
254  {
[2662]255    return header_->getDataSize()+GamestateHeader::getSize();
[1701]256  }
257}
258
[7163]259
260bool Gamestate::operator==(packet::Gamestate gs)
261{
[2662]262  uint8_t *d1 = data_+GamestateHeader::getSize();
263  uint8_t *d2 = gs.data_+GamestateHeader::getSize();
[3084]264  GamestateHeader* h1 = new GamestateHeader(data_);
265  GamestateHeader* h2 = new GamestateHeader(gs.data_);
266  assert(h1->getDataSize() == h2->getDataSize());
[1751]267  assert(!isCompressed());
268  assert(!gs.isCompressed());
[3084]269  return memcmp(d1, d2, h1->getDataSize())==0;
[1751]270}
271
[7163]272
[1701]273bool Gamestate::process()
274{
[1705]275  return GamestateHandler::addGamestate(this, getClientID());
[1701]276}
277
[1907]278
[1701]279bool Gamestate::compressData()
280{
[2662]281  assert(data_);
282  assert(!header_->isCompressed());
283  uLongf buffer = (uLongf)(((header_->getDataSize() + 12)*1.01)+1);
[1701]284  if(buffer==0)
285    return false;
[1740]286
[2662]287  uint8_t *ndata = new uint8_t[buffer+GamestateHeader::getSize()];
288  uint8_t *dest = ndata + GamestateHeader::getSize();
289  uint8_t *source = data_ + GamestateHeader::getSize();
[1701]290  int retval;
[2662]291  retval = compress( dest, &buffer, source, (uLong)(header_->getDataSize()) );
[7163]292  switch ( retval )
293  {
[1701]294    case Z_OK: COUT(5) << "G.St.Man: compress: successfully compressed" << std::endl; break;
[1751]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;
[1701]298  }
299
300  //copy and modify header
[2662]301  GamestateHeader *temp = header_;
302  header_ = new GamestateHeader(ndata, temp);
303  delete temp;
[1701]304  //delete old data
305  delete[] data_;
306  //save new data
307  data_ = ndata;
[2662]308  header_->setCompSize( buffer );
309  header_->setCompressed( true );
[7163]310  COUT(0) << "gamestate compress datasize: " << header_->getDataSize() << " compsize: " << header_->getCompSize() << std::endl;
[1701]311  return true;
312}
[7163]313
314
[1701]315bool Gamestate::decompressData()
316{
[2662]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;
[1907]323  bufsize = datasize;
[1751]324  assert(bufsize!=0);
[2662]325  uint8_t *ndata = new uint8_t[bufsize + GamestateHeader::getSize()];
326  uint8_t *dest = ndata + GamestateHeader::getSize();
327  uint8_t *source = data_ + GamestateHeader::getSize();
[1701]328  int retval;
[1751]329  uLongf length=bufsize;
330  retval = uncompress( dest, &length, source, (uLong)compsize );
[7163]331  switch ( retval )
332  {
[1701]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  }
[1752]338
[1701]339  //copy over the header
[2662]340  GamestateHeader *temp = header_;
341  header_ = new GamestateHeader( data_, header_ );
342  delete temp;
[2087]343
[7163]344  if (this->bDataENetAllocated_)
345  {
[2087]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  }
[7163]350  else
351  {
[2087]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
[1751]357  //set new pointers
[1701]358  data_ = ndata;
[2662]359  header_->setCompressed( false );
360  assert(header_->getDataSize()==datasize);
361  assert(header_->getCompSize()==compsize);
[1701]362  return true;
363}
364
[7163]365
[7333]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, newData+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    origData += objectOffset;
433    //         baseOffset += temp + h.getDataSize()+SynchronisableHeader::getSize() - baseData;
434    //baseOffset += objectOffset;
435    //         SynchronisableHeader htemp(temp);
436    //         baseOffset += SynchronisableHeader::getSize() + htemp.getDataSize();
437    //         {
438      //           SynchronisableHeader htemp2( baseData+(temp-baseData+objectOffset) );
439    //           if( baseData+(temp-baseData+objectOffset) < baseData+baseLength )
440    //           {
441      //             assert(htemp2.getClassID()<500);
442    //             assert(htemp2.getDataSize()!=0 && htemp2.getDataSize()<1000);
443    //             assert(htemp2.isDiffed()==false);
444    //           }
445    //         }
446    baseData += objectOffset;
447    newData += newObjectOffset;
448  }
449}
450
451inline void /*Gamestate::*/copyObject( uint8_t*& newData, uint8_t*& origData, uint8_t*& baseData, SynchronisableHeader& objectHeader, std::vector<uint32_t>::iterator& sizes )
452{
453  //       COUT(4) << "docopy" << endl;
454  // Just copy over the whole Object
455  memcpy( newData, origData, objectHeader.getDataSize()+SynchronisableHeader::getSize() );
456  newData += objectHeader.getDataSize()+SynchronisableHeader::getSize();
457  origData += objectHeader.getDataSize()+SynchronisableHeader::getSize();
458  SynchronisableHeader baseHeader( baseData );
459  baseData += baseHeader.getDataSize()+SynchronisableHeader::getSize();
460  //       COUT(4) << "copy " << h.getObjectID() << endl;
461  //       COUT(4) << "copy " << h.getObjectID() << ":";
462  sizes += Synchronisable::getSynchronisable(objectHeader.getObjectID())->getNrOfVariables();
463//   for( unsigned int i = 0; i < Synchronisable::getSynchronisable(objectHeader.getObjectID())->getNrOfVariables(); ++i )
464//   {
465//     //         COUT(4) << " " << *sizes;
466//     ++sizes;
467//   }
468    //       COUT(4) << endl;
469}
470
[7163]471Gamestate* Gamestate::diffVariables(Gamestate *base)
[1701]472{
[7163]473  assert(this && base); assert(data_ && base->data_);
474  assert(!header_->isCompressed() && !base->header_->isCompressed());
[2662]475  assert(!header_->isDiffed());
[7163]476
477
478  // *** first do a raw diff of the two gamestates
479
480  uint8_t *baseData = GAMESTATE_START(base->data_);
481  uint8_t *origData = GAMESTATE_START(this->data_);
[7333]482  uint8_t *origDataEnd = origData + header_->getDataSize();
483  uint8_t *baseDataEnd = baseData + base->header_->getDataSize();
484//   uint32_t origLength = header_->getDataSize();
485//   uint32_t baseLength = base->header_->getDataSize();
[7163]486
[7333]487  assert( header_->getDataSize() && base->header_->getDataSize() );
488 
489  uint32_t newDataSize = header_->getDataSize() + GamestateHeader::getSize() + sizeof(uint32_t)*this->nrOfVariables_;
490  uint8_t *newData = new uint8_t[newDataSize]; // this is the maximum size needed in the worst case
491  uint8_t *dest = GAMESTATE_START(newData);
[7163]492
[7333]493//   uint32_t baseOffset = 0; //offset in the diffed stream
494//   uint32_t origOffset = 0; //offset in the new stream with removed 0's
[7163]495  std::vector<uint32_t>::iterator sizes = this->sizes_.begin();
496
[7333]497  while( origData < origDataEnd )
[7163]498  {
499    //iterate through all objects
500
[7333]501    SynchronisableHeader h(origData);
[7163]502
503    // Find (if possible) the current object in the datastream of the old gamestate
504    // Start at the current offset position
[7333]505    if(baseData >= baseDataEnd)
506      baseData = GAMESTATE_START(base->data_);
507//     uint8_t* temp = baseData + baseOffset;
[7163]508    uint32_t objectID = h.getObjectID();
[7333]509    assert(baseData < baseDataEnd);
510    assert(dest < newData + newDataSize);
[7163]511    assert(sizes != this->sizes_.end());
[7337]512    bool diffedObject = false;
[7333]513    while ( baseData < baseDataEnd )
[7163]514    {
[7333]515      SynchronisableHeader htemp(baseData);
[7163]516      assert( htemp.getDataSize()!=0 );
517      if ( htemp.getObjectID() == objectID )
518      {
519        assert( h.getClassID() == htemp.getClassID() );
[7333]520//         goto DODIFF;
521        diffObject(dest, origData, baseData, h, sizes);
[7337]522        diffedObject = true;
523        break;
[1701]524      }
[7333]525      {
526        SynchronisableHeader htemp2(baseData+htemp.getDataSize()+SynchronisableHeader::getSize());
527        if( baseData+htemp.getDataSize()+SynchronisableHeader::getSize() < baseDataEnd )
528        {
529          assert(htemp2.getClassID()<500);
530          assert(htemp2.getDataSize()!=0 && htemp2.getDataSize()<1000);
531          assert(htemp2.isDiffed()==false);
532        }
533      }
[7337]534      baseData += htemp.getDataSize()+SynchronisableHeader::getSize();
[7333]535//       assert( baseData+htemp.getDataSize()+SynchronisableHeader::getSize() <=baseData+baseLength );
536//       temp += htemp.getDataSize()+SynchronisableHeader::getSize();
[7337]537     
[1701]538    }
[7163]539    // If not found start looking at the beginning
[7337]540    assert( diffedObject || baseData == baseDataEnd );
[7333]541    baseData = GAMESTATE_START(base->data_);
[7163]542    {
[7333]543      SynchronisableHeader htemp2(baseData);
544      if( baseData < baseDataEnd )
545      {
546        assert(htemp2.getClassID()<500);
547        assert(htemp2.getDataSize()!=0 && htemp2.getDataSize()<1000);
548        assert(htemp2.isDiffed()==false);
549      }
550    }
[7337]551    while ( !diffedObject && baseData < baseDataEnd )
[7333]552    {
553      SynchronisableHeader htemp(baseData);
[7163]554      if ( htemp.getObjectID() == objectID )
555      {
556        assert( h.getClassID() == htemp.getClassID() );
[7333]557//         goto DODIFF;
558        diffObject(dest, origData, baseData, h, sizes);
559        break;
[7163]560      }
561//       {
[7333]562//         SynchronisableHeader htemp2(baseData+htemp.getDataSize()+SynchronisableHeader::getSize());
563//         if( baseData+htemp.getDataSize()+SynchronisableHeader::getSize() < baseDataEnd )
[7163]564//         {
565//           assert(htemp2.getClassID()<500);
566//           assert(htemp2.getDataSize()!=0 && htemp2.getDataSize()<1000);
567//           assert(htemp2.isDiffed()==false);
568//         }
569//       }
[7333]570      baseData += htemp.getDataSize()+SynchronisableHeader::getSize();
[7163]571    }
572    // Object is new, thus never transmitted -> just copy over
[7333]573//     goto DOCOPY;
574    if ( baseData >= baseDataEnd )
[7163]575    {
[7333]576      assert(baseData == baseDataEnd);
577      copyObject(dest, origData, baseData, h, sizes);
578      assert(sizes != this->sizes_.end() || origData>=origDataEnd);
[7163]579    }
[1701]580  }
[7337]581  assert(sizes==this->sizes_.end());
[1701]582
[7163]583
[7333]584  Gamestate *g = new Gamestate(newData, getClientID());
[7163]585  assert(g->header_);
[2662]586  *(g->header_) = *header_;
587  g->header_->setBaseID( base->getID() );
[7333]588  g->header_->setDataSize(dest - newData - GamestateHeader::getSize());
[1751]589  g->flags_=flags_;
590  g->packetDirection_ = packetDirection_;
[7163]591  assert(!g->isCompressed());
[1701]592  return g;
[7163]593}
[3198]594
[7163]595
596Gamestate* Gamestate::diffData(Gamestate *base)
[3198]597{
598  assert(this && base); assert(data_ && base->data_);
599  assert(!header_->isCompressed() && !base->header_->isCompressed());
600  assert(!header_->isDiffed());
[6417]601
[3198]602  uint8_t *basep = GAMESTATE_START(base->data_);
603  uint8_t *gs = GAMESTATE_START(this->data_);
604  uint32_t dest_length = header_->getDataSize();
[6417]605
[3198]606  if(dest_length==0)
607    return NULL;
[6417]608
[3198]609  uint8_t *ndata = new uint8_t[dest_length*sizeof(uint8_t)+GamestateHeader::getSize()];
610  uint8_t *dest = GAMESTATE_START(ndata);
[6417]611
[3198]612  rawDiff( dest, gs, basep, header_->getDataSize(), base->header_->getDataSize() );
613#ifndef NDEBUG
614  uint8_t *dest2 = new uint8_t[dest_length];
615  rawDiff( dest2, dest, basep, header_->getDataSize(), base->header_->getDataSize() );
616  assert( memcmp( dest2, gs, dest_length) == 0 );
[5929]617  delete dest2;
[3198]618#endif
619
620  Gamestate *g = new Gamestate(ndata, getClientID());
621  assert(g->header_);
622  *(g->header_) = *header_;
623  g->header_->setDiffed( true );
624  g->header_->setBaseID( base->getID() );
625  g->flags_=flags_;
626  g->packetDirection_ = packetDirection_;
627  assert(g->isDiffed());
628  assert(!g->isCompressed());
629  return g;
[1701]630}
631
[7163]632
633Gamestate* Gamestate::undiff(Gamestate *base)
[3198]634{
635  assert(this && base); assert(data_ && base->data_);
636  assert(!header_->isCompressed() && !base->header_->isCompressed());
637  assert(header_->isDiffed());
[6417]638
[3198]639  uint8_t *basep = GAMESTATE_START(base->data_);
640  uint8_t *gs = GAMESTATE_START(this->data_);
641  uint32_t dest_length = header_->getDataSize();
[6417]642
[3198]643  if(dest_length==0)
644    return NULL;
[6417]645
[3198]646  uint8_t *ndata = new uint8_t[dest_length*sizeof(uint8_t)+GamestateHeader::getSize()];
647  uint8_t *dest = ndata + GamestateHeader::getSize();
[6417]648
[3198]649  rawDiff( dest, gs, basep, header_->getDataSize(), base->header_->getDataSize() );
[6417]650
[3198]651  Gamestate *g = new Gamestate(ndata, getClientID());
652  assert(g->header_);
653  *(g->header_) = *header_;
654  g->header_->setDiffed( false );
655  g->flags_=flags_;
656  g->packetDirection_ = packetDirection_;
657  assert(!g->isDiffed());
658  assert(!g->isCompressed());
659  return g;
660}
661
662
663void Gamestate::rawDiff( uint8_t* newdata, uint8_t* data, uint8_t* basedata, uint32_t datalength, uint32_t baselength)
664{
665  uint64_t* gd = (uint64_t*)data;
666  uint64_t* bd = (uint64_t*)basedata;
667  uint64_t* nd = (uint64_t*)newdata;
[6417]668
[3198]669  unsigned int i;
670  for( i=0; i<datalength/8; i++ )
671  {
672    if( i<baselength/8 )
673      *(nd+i) = *(gd+i) ^ *(bd+i);  // xor the data
674    else
675      *(nd+i) = *(gd+i); // just copy over the data
676  }
677  unsigned int j;
678  // now process the rest (when datalength isn't a multiple of 4)
679  for( j = 8*(datalength/8); j<datalength; j++ )
680  {
681    if( j<baselength )
682      *(newdata+j) = *(data+j) ^ *(basedata+j); // xor
683    else
684      *(newdata+j) = *(data+j); // just copy
685  }
686  assert(j==datalength);
687}
688
[7163]689
[2662]690Gamestate* Gamestate::doSelection(unsigned int clientID, unsigned int targetSize){
[1907]691  assert(data_);
[2662]692  std::list<obj>::iterator it;
[2087]693
[1907]694  // allocate memory for new data
[2662]695  uint8_t *gdata = new uint8_t[header_->getDataSize()+GamestateHeader::getSize()];
[1907]696  // create a gamestate out of it
697  Gamestate *gs = new Gamestate(gdata);
[2662]698  uint8_t *newdata = gdata + GamestateHeader::getSize();
[1907]699  uint8_t *origdata = GAMESTATE_START(data_);
[2087]700
[1907]701  //copy the GamestateHeader
[2662]702  assert(gs->header_);
703  *(gs->header_) = *header_;
[2087]704
[2662]705  uint32_t objectOffset;
706  unsigned int objectsize, destsize=0;
707  // TODO: Why is this variable not used?
708  //Synchronisable *object;
[2087]709
[2662]710  //call TrafficControl
[3084]711  TrafficControl::getInstance()->processObjectList( clientID, header_->getID(), dataVector_ );
[2662]712
[1907]713  //copy in the zeros
[3084]714//   std::list<obj>::iterator itt;
715//   COUT(0) << "myvector contains:";
716//   for ( itt=dataVector_.begin() ; itt!=dataVector_.end(); itt++ )
[7163]717//     COUT(0) << " " << (*itt).objID;
[3084]718//   COUT(0) << endl;
719  for(it=dataVector_.begin(); it!=dataVector_.end();){
[2662]720    SynchronisableHeader oldobjectheader(origdata);
721    SynchronisableHeader newobjectheader(newdata);
[7163]722    if ( (*it).objSize == 0 )
[2662]723    {
724      ++it;
725      continue;
726    }
[7163]727    objectsize = oldobjectheader.getDataSize()+SynchronisableHeader::getSize();
[2662]728    objectOffset=SynchronisableHeader::getSize(); //skip the size and the availableData variables in the objectheader
[7163]729    if ( (*it).objID == oldobjectheader.getObjectID() ){
[2662]730      memcpy(newdata, origdata, objectsize);
731      ++it;
[1907]732    }else{
[2662]733      newobjectheader = oldobjectheader;
[2171]734      memset(newdata+objectOffset, 0, objectsize-objectOffset);
[1907]735    }
736    newdata += objectsize;
737    origdata += objectsize;
[2662]738    destsize += objectsize;
[1907]739  }
[2662]740#ifndef NDEBUG
741  uint32_t origsize = destsize;
742  while ( origsize < header_->getDataSize() )
743  {
744    SynchronisableHeader oldobjectheader(origdata);
[7163]745    objectsize = oldobjectheader.getDataSize()+SynchronisableHeader::getSize();
[2662]746    origdata += objectsize;
747    origsize += objectsize;
[1907]748  }
[2662]749  assert(origsize==header_->getDataSize());
750  assert(destsize!=0);
751#endif
752  gs->header_->setDataSize( destsize );
[1907]753  return gs;
754}
755
[2087]756
[2662]757uint32_t Gamestate::calcGamestateSize(int32_t id, uint8_t mode)
[1701]758{
[7163]759  uint32_t size = 0;
760  uint32_t nrOfVariables = 0;
[1701]761    // get the start of the Synchronisable list
[2171]762  ObjectList<Synchronisable>::iterator it;
[1701]763    // get total size of gamestate
[7163]764  for(it = ObjectList<Synchronisable>::begin(); it; ++it){
[1907]765    size+=it->getSize(id, mode); // size of the actual data of the synchronisable
[7163]766    nrOfVariables += it->getNrOfVariables();
767  }
768//   COUT(0) << "allocating " << nrOfVariables << " ints" << endl;
769  this->sizes_.reserve(nrOfVariables);
[1701]770  return size;
771}
772
[7163]773
[2662]774} //namespace packet
775} //namespace orxonox
Note: See TracBrowser for help on using the repository browser.