Planet
navi homePPSaboutscreenshotsdownloaddevelopmentforum

source: code/branches/network4/src/libraries/network/packet/Gamestate.cc @ 7497

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

merging network3 into network4 branch

  • 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
[7497]45#define PACKET_FLAG_GAMESTATE  0 //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
[7497]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
[7163]472Gamestate* Gamestate::diffVariables(Gamestate *base)
[1701]473{
[7163]474  assert(this && base); assert(data_ && base->data_);
475  assert(!header_->isCompressed() && !base->header_->isCompressed());
[2662]476  assert(!header_->isDiffed());
[7163]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_);
[7497]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();
[7163]487
[7497]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);
[7163]493
[7497]494//   uint32_t baseOffset = 0; //offset in the diffed stream
495//   uint32_t origOffset = 0; //offset in the new stream with removed 0's
[7163]496  std::vector<uint32_t>::iterator sizes = this->sizes_.begin();
497
[7497]498  while( origData < origDataEnd )
[7163]499  {
500    //iterate through all objects
501
[7497]502    SynchronisableHeader h(origData);
[7163]503
504    // Find (if possible) the current object in the datastream of the old gamestate
505    // Start at the current offset position
[7497]506    if(baseData >= baseDataEnd)
507      baseData = GAMESTATE_START(base->data_);
508//     uint8_t* temp = baseData + baseOffset;
[7163]509    uint32_t objectID = h.getObjectID();
[7497]510    assert(baseData < baseDataEnd);
511    assert(dest < newData + newDataSize);
[7163]512    assert(sizes != this->sizes_.end());
[7497]513    bool diffedObject = false;
514    while ( baseData < baseDataEnd )
[7163]515    {
[7497]516      SynchronisableHeader htemp(baseData);
[7163]517      assert( htemp.getDataSize()!=0 );
518      if ( htemp.getObjectID() == objectID )
519      {
520        assert( h.getClassID() == htemp.getClassID() );
[7497]521//         goto DODIFF;
522        diffObject(dest, origData, baseData, h, sizes);
523        diffedObject = true;
524        break;
[1701]525      }
[7497]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     
[1701]539    }
[7163]540    // If not found start looking at the beginning
[7497]541    assert( diffedObject || baseData == baseDataEnd );
542    baseData = GAMESTATE_START(base->data_);
[7163]543    {
[7497]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);
[7163]555      if ( htemp.getObjectID() == objectID )
556      {
557        assert( h.getClassID() == htemp.getClassID() );
[7497]558//         goto DODIFF;
559        diffObject(dest, origData, baseData, h, sizes);
560        break;
[7163]561      }
562//       {
[7497]563//         SynchronisableHeader htemp2(baseData+htemp.getDataSize()+SynchronisableHeader::getSize());
564//         if( baseData+htemp.getDataSize()+SynchronisableHeader::getSize() < baseDataEnd )
[7163]565//         {
566//           assert(htemp2.getClassID()<500);
567//           assert(htemp2.getDataSize()!=0 && htemp2.getDataSize()<1000);
568//           assert(htemp2.isDiffed()==false);
569//         }
570//       }
[7497]571      baseData += htemp.getDataSize()+SynchronisableHeader::getSize();
[7163]572    }
573    // Object is new, thus never transmitted -> just copy over
[7497]574//     goto DOCOPY;
575    if ( baseData >= baseDataEnd )
[7163]576    {
[7497]577      assert(baseData == baseDataEnd);
578      copyObject(dest, origData, baseData, h, sizes);
579      assert(sizes != this->sizes_.end() || origData>=origDataEnd);
[7163]580    }
[1701]581  }
[7497]582  assert(sizes==this->sizes_.end());
[1701]583
[7163]584
[7497]585  Gamestate *g = new Gamestate(newData, getClientID());
[7163]586  assert(g->header_);
[2662]587  *(g->header_) = *header_;
588  g->header_->setBaseID( base->getID() );
[7497]589  g->header_->setDataSize(dest - newData - GamestateHeader::getSize());
[1751]590  g->flags_=flags_;
591  g->packetDirection_ = packetDirection_;
[7163]592  assert(!g->isCompressed());
[1701]593  return g;
[7163]594}
[3198]595
[7163]596
597Gamestate* Gamestate::diffData(Gamestate *base)
[3198]598{
599  assert(this && base); assert(data_ && base->data_);
600  assert(!header_->isCompressed() && !base->header_->isCompressed());
601  assert(!header_->isDiffed());
[6417]602
[3198]603  uint8_t *basep = GAMESTATE_START(base->data_);
604  uint8_t *gs = GAMESTATE_START(this->data_);
605  uint32_t dest_length = header_->getDataSize();
[6417]606
[3198]607  if(dest_length==0)
608    return NULL;
[6417]609
[3198]610  uint8_t *ndata = new uint8_t[dest_length*sizeof(uint8_t)+GamestateHeader::getSize()];
611  uint8_t *dest = GAMESTATE_START(ndata);
[6417]612
[3198]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 );
[5929]618  delete dest2;
[3198]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;
[1701]631}
632
[7163]633
634Gamestate* Gamestate::undiff(Gamestate *base)
[3198]635{
636  assert(this && base); assert(data_ && base->data_);
637  assert(!header_->isCompressed() && !base->header_->isCompressed());
638  assert(header_->isDiffed());
[6417]639
[3198]640  uint8_t *basep = GAMESTATE_START(base->data_);
641  uint8_t *gs = GAMESTATE_START(this->data_);
642  uint32_t dest_length = header_->getDataSize();
[6417]643
[3198]644  if(dest_length==0)
645    return NULL;
[6417]646
[3198]647  uint8_t *ndata = new uint8_t[dest_length*sizeof(uint8_t)+GamestateHeader::getSize()];
648  uint8_t *dest = ndata + GamestateHeader::getSize();
[6417]649
[3198]650  rawDiff( dest, gs, basep, header_->getDataSize(), base->header_->getDataSize() );
[6417]651
[3198]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;
[6417]669
[3198]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
[7163]690
[2662]691Gamestate* Gamestate::doSelection(unsigned int clientID, unsigned int targetSize){
[1907]692  assert(data_);
[2662]693  std::list<obj>::iterator it;
[2087]694
[1907]695  // allocate memory for new data
[2662]696  uint8_t *gdata = new uint8_t[header_->getDataSize()+GamestateHeader::getSize()];
[1907]697  // create a gamestate out of it
698  Gamestate *gs = new Gamestate(gdata);
[2662]699  uint8_t *newdata = gdata + GamestateHeader::getSize();
[1907]700  uint8_t *origdata = GAMESTATE_START(data_);
[2087]701
[1907]702  //copy the GamestateHeader
[2662]703  assert(gs->header_);
704  *(gs->header_) = *header_;
[2087]705
[2662]706  uint32_t objectOffset;
707  unsigned int objectsize, destsize=0;
708  // TODO: Why is this variable not used?
709  //Synchronisable *object;
[2087]710
[2662]711  //call TrafficControl
[3084]712  TrafficControl::getInstance()->processObjectList( clientID, header_->getID(), dataVector_ );
[2662]713
[1907]714  //copy in the zeros
[3084]715//   std::list<obj>::iterator itt;
716//   COUT(0) << "myvector contains:";
717//   for ( itt=dataVector_.begin() ; itt!=dataVector_.end(); itt++ )
[7163]718//     COUT(0) << " " << (*itt).objID;
[3084]719//   COUT(0) << endl;
720  for(it=dataVector_.begin(); it!=dataVector_.end();){
[2662]721    SynchronisableHeader oldobjectheader(origdata);
722    SynchronisableHeader newobjectheader(newdata);
[7163]723    if ( (*it).objSize == 0 )
[2662]724    {
725      ++it;
726      continue;
727    }
[7163]728    objectsize = oldobjectheader.getDataSize()+SynchronisableHeader::getSize();
[2662]729    objectOffset=SynchronisableHeader::getSize(); //skip the size and the availableData variables in the objectheader
[7163]730    if ( (*it).objID == oldobjectheader.getObjectID() ){
[2662]731      memcpy(newdata, origdata, objectsize);
732      ++it;
[1907]733    }else{
[2662]734      newobjectheader = oldobjectheader;
[2171]735      memset(newdata+objectOffset, 0, objectsize-objectOffset);
[1907]736    }
737    newdata += objectsize;
738    origdata += objectsize;
[2662]739    destsize += objectsize;
[1907]740  }
[2662]741#ifndef NDEBUG
742  uint32_t origsize = destsize;
743  while ( origsize < header_->getDataSize() )
744  {
745    SynchronisableHeader oldobjectheader(origdata);
[7163]746    objectsize = oldobjectheader.getDataSize()+SynchronisableHeader::getSize();
[2662]747    origdata += objectsize;
748    origsize += objectsize;
[1907]749  }
[2662]750  assert(origsize==header_->getDataSize());
751  assert(destsize!=0);
752#endif
753  gs->header_->setDataSize( destsize );
[1907]754  return gs;
755}
756
[2087]757
[2662]758uint32_t Gamestate::calcGamestateSize(int32_t id, uint8_t mode)
[1701]759{
[7163]760  uint32_t size = 0;
761  uint32_t nrOfVariables = 0;
[1701]762    // get the start of the Synchronisable list
[2171]763  ObjectList<Synchronisable>::iterator it;
[1701]764    // get total size of gamestate
[7163]765  for(it = ObjectList<Synchronisable>::begin(); it; ++it){
[1907]766    size+=it->getSize(id, mode); // size of the actual data of the synchronisable
[7163]767    nrOfVariables += it->getNrOfVariables();
768  }
769//   COUT(0) << "allocating " << nrOfVariables << " ints" << endl;
770  this->sizes_.reserve(nrOfVariables);
[1701]771  return size;
772}
773
[7163]774
[2662]775} //namespace packet
776} //namespace orxonox
Note: See TracBrowser for help on using the repository browser.