ParisBasicExample.cc

Go to the documentation of this file.

#include "ParisBasicExample.h"
#include <string.h>
#include <stdlib.h>
#include <iostream>

ParisBasicExample::ParisBasicExample() : _data_size(0), _data_elt_nb(0),
                _my_rank(-1), _server_nb_nodes(-1)
{
  _user_data  = 0;
  temp_buffer = 0;
  _length  = 0;
}

ParisBasicExample::~ParisBasicExample()
{
  delete position;
  delete _data_rst;
}

void
ParisBasicExample::setEltSize(long size) 
{
  _data_size = size;
  if (_data_elt_nb && _server_nb_nodes)
    {
      for (int i = 0; i < _server_nb_nodes; i++)
   {
     _data_rst[i] = _data_size * _data_elt_nb;
   }
    }
}

void
ParisBasicExample::setNbElt(long elt_nb) 
{
  _data_elt_nb = elt_nb;
  if (_data_size && _server_nb_nodes)
    {
      for (int i = 0; i < _server_nb_nodes; i++)
   {
     _data_rst[i] = _data_size * _data_elt_nb;
   }
    }
}

void 
ParisBasicExample::setSourceTopology(PaCO::PacoTopology_t topo)
{
  _srcTopo = topo;
}

PaCO::PacoTopology_t 
ParisBasicExample::getSourceTopology()
{
  return _srcTopo;
}

void 
ParisBasicExample::setDestTopology(PaCO::PacoTopology_t topo)
{
  _destTopo = topo;
  _server_nb_nodes = _destTopo.total;
  position  = new long[_server_nb_nodes];
  _data_rst = new long[_server_nb_nodes];
  for (int i = 0; i < _server_nb_nodes; i++)
    {
      position[i]  = 0;
      _data_rst[i] = 0;
    }
  if (_data_size && _data_elt_nb)
    {
      for (int i = 0; i < _server_nb_nodes; i++)
   {
     _data_rst[i] = _data_size * _data_elt_nb;
   }
    }
}

PaCO::PacoTopology_t 
ParisBasicExample::getDestTopology()
{
  return _destTopo;
}

void
ParisBasicExample::setNodeRank(long Rank)
{
  _my_rank = Rank;
}

long 
ParisBasicExample::getNodeRank()
{
  return _my_rank;
}

PieceToSend *
ParisBasicExample::computePiecesToSend()
{
  // Compute total data length
  long total_length = _data_size * _data_elt_nb;
  // We send at each server node the 
  // same data
  PieceToSend * comm_vector = new PieceToSend[_server_nb_nodes];
  for (int i = 0; i < _server_nb_nodes; i++)
    {
      comm_vector[i].sourceNode = _my_rank;
      comm_vector[i].destNode = i;
      if (i == _my_rank)
   {
     comm_vector[i].size = total_length;
   }
      else
   {
     comm_vector[i].size = 0;
   }
    }
  return comm_vector;
}

void
ParisBasicExample::setDataPtr(void* dataPtr)
{
  _user_data = (char*) dataPtr;
  cerr << "_data_size " <<_data_size << endl;
  cerr << "_data_elt_nb " <<_data_elt_nb << endl;
  cerr << "_my_rank " <<_my_rank << endl;
  cerr << "_server_nb_nodes " <<_server_nb_nodes << endl;
  cerr << "_data_rst " <<_data_rst << endl;
  cerr << " position" <<  position << endl;
}

void *
ParisBasicExample::getClientData(int server_node, long & size, long & length, bool & end)
{
  if (size > 0)
    {
      if (temp_buffer) clientFree(server_node);
      // First get position
      long actual_pos = position[server_node];
      // Test if size > _data_rst
      long real_size = 0;
      if (size > _data_rst[server_node])
   {
     real_size = _data_rst[server_node];
     size = size - _data_rst[server_node];
     _data_rst[server_node] = 0;
   }
      else
   {
     real_size = size;
     _data_rst[server_node] = _data_rst[server_node] - size;
     size = 0;
   }
      length = real_size / _data_size;
      _length  = length;
      end = true;
      position[server_node] = actual_pos + real_size;
      if (_data_rst[server_node] == 0) 
   {
     _data_rst[server_node] = _data_size * _data_elt_nb;
     position[server_node] = 0;
   }
      return &_user_data[actual_pos];
    }
  else
    {
      length = 0;
      _length  = length;
      end = true;
      return NULL;
    }
  return NULL;
}

void
ParisBasicExample::clientFree()
{
  free(temp_buffer);
}

void
ParisBasicExample::clientFree(int server_node)
{
  free(temp_buffer);
}

bool 
ParisBasicExample::insertData(void* rcptBuf)
{
  _user_data = (char *) rcptBuf;
  return true;
}

void *
ParisBasicExample::getServerData(long & length)
{
  length = _length;
  return _user_data;
}

void 
ParisBasicExample::setDescr(void * descr)
{
  _length = *((CORBA::Short*) descr);
  cerr << " Set Descr  : "<< _length << endl;
}

void *
ParisBasicExample::getDescr()
{
  return &_length;
}