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