ScatterMessage.cc

Go to the documentation of this file.

  /*
   *
   * Copyright 2006,2007 Savarese Software Research Corporation
   *
   * Licensed under the Apache License, Version 2.0 (the "License");
   * you may not use this file except in compliance with the License.
   * You may obtain a copy of the License at
   *
   * http://www.savarese.com/software/ApacheLicense-2.0
   *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  
  #include <ssrc/spread/ScatterMessage.h>
  
  __BEGIN_NS_SSRC_SPREAD
  
  void ScatterMessage::init_pre_receive() {
   int size = _messages.size();
  
   for(int i = 0; i < size; ++i) {
   value_type & value = _messages[i];
   Message *m = value.first;
  
   if(value.second >= _scatter.num_elements)
   break;
  
   _size-=m->size();
   m->resize(m->capacity());
   _size+=m->size();
   _scatter.elements[value.second].len = m->size();
   }
  }
  
  void ScatterMessage::init_post_receive(int bytes_received) {
   int index = 0, i, len;
   int size = _messages.size();
   Message *m;
  
   if(bytes_received > 0)
   _size = bytes_received;
   else
   _size = 0;
  
   for(i = 0; i < size && bytes_received > 0; ++i) {
   value_type & value = _messages[i];
   int mindex = value.second;
   if(mindex >= _scatter.num_elements)
   break;
  
   while(index < mindex)
   bytes_received-=_scatter.elements[index++].len;
  
   if(bytes_received <= 0)
   break;
  
   len = _scatter.elements[mindex].len;
   m = value.first;
  
   m->set_type(type());
   m->set_service(service());
   m->set_endian_mismatch(endian_mismatch());
   m->set_sender(_sender);
  
   if(len <= bytes_received)
   m->resize(len);
   else
   m->resize(bytes_received);
  
   bytes_received-=len;
   }
  
   // Anything left over gets a size of zero.
   for(; i < size; ++i) {
   m = _messages[i].first;
   m->resize(0);
   m->set_type(type());
   m->set_service(service());
   m->set_endian_mismatch(endian_mismatch());
   m->set_sender(_sender);
   }
  }
  
  bool ScatterMessage::add(const void *data, const unsigned int size) {
   int i = _scatter.num_elements;
  
   if(i >= MaxScatterElements)
   return false;
  
   ++_scatter.num_elements;
   _scatter.elements[i].buf =
   const_cast<char *>(static_cast<const char *>(data));
   _scatter.elements[i].len = size;
   _size+=size;
  
   return true;
  }
  
  __END_NS_SSRC_SPREAD