BasicDistributionLibrary.cc

Go to the documentation of this file.

#include "BasicDistributionLibrary.h"

#include "Schedule.h"

BasicDistributionLibrary::BasicDistributionLibrary(Fabrique* f) { 
  setFabric(f);
  _darray = f->creer();
  _vdarray = f->vcreer(); 
  _vdarray->clear(); 
  
  _clttopo.total = 123; // debug
  _srvtopo.total = 123; // debug
}

BasicDistributionLibrary::BasicDistributionLibrary(Fabrique* f, void* comm ) : _comm(comm) { 
  setFabric(f);
  _darray = f->creer();
  _vdarray = f->vcreer(); 
  _vdarray->clear(); 
  
  _clttopo.total = 123; // debug
  _srvtopo.total = 123; // debug
}

BasicDistributionLibrary::~BasicDistributionLibrary() { 
  delete _fab; delete _darray; delete _vdarray;
} 

void BasicDistributionLibrary::setCommunicator(void* comm) { _comm = comm; }

void BasicDistributionLibrary::setFabric(Fabrique* f) { _fab = f; }

void BasicDistributionLibrary::reset() { _vdarray->clear(); }

const PaCO::distLoc_t& BasicDistributionLibrary::getMode() const { return _mode; }

bool BasicDistributionLibrary::setMode(const PaCO::distLoc_t mode) { _mode = mode; return true; } // true if ok

bool BasicDistributionLibrary::setClientConfiguration(const PaCO::PacoTopology_t & ctopo) { _clttopo = ctopo; return true; }
bool BasicDistributionLibrary::setServerConfiguration(const PaCO::PacoTopology_t & stopo) { _srvtopo = stopo; return true; }

bool BasicDistributionLibrary::setGlobalDataConfiguration(const PaCO::PacoGlobalData_t& gd) { _gd = gd; return true; }
bool BasicDistributionLibrary::setLocalDataConfiguration (const PaCO::PacoLocalData_t&  ld) {
  _ld.rank  = ld.rank;
  _ld.start = ld.start;
  _ld.len  = ld.len;
  return true;
}

void BasicDistributionLibrary::computeSend(const void* data, vAbstrait& vdarray, vector<unsigned>& destid) {
  _ld.base = (char*) data;

  computeSendDataBlock1D(_gd, _ld, _clttopo, _srvtopo, vdarray, destid, _mode, _comm);
}

bool BasicDistributionLibrary::computeReceive(Abstrait* darray) {

  this->setClientConfiguration(darray->topo());  

  // Get the mode of the client
  this->setMode(darray->mode());

  unsigned nb  = nbofPart(_mode, _clttopo.total, _srvtopo.total, _ld.rank);

  unsigned pos;

  // In server side redistribution, empty dist for server nodes whose id > max client node
  if (darray->dist().length() != 0 ) {
    pos = posofPart(_mode, _srvtopo.total, darray->dist()[0].rank);
  } else {
    pos = 0;
  }

  // Initialize _vdarray & remaining (assumed clean)
  if (_vdarray->size() == 0) {
    // _vdarray is empty: so let's define its size
    _vdarray->size(nb);
    _remaining = nb;
  }


  fprintf(stderr, "computeReceive: Node %d : set entry %d/%d to %p\n", _ld.rank, pos, nb, darray);
  _vdarray->setAbstrait(pos, darray);
  _remaining--;

  fprintf(stderr, "computeReceive: Node %d : got %d of %d data\n", _ld.rank, _vdarray->size() - _remaining, _vdarray->size());

  if (_remaining == 0) {
    computeReceiveDataBlock1D(_vdarray, _mode, _srvtopo.total, _ld.rank, _darray, _comm);
    return true;
  } else {
    return false;
  }
}

Abstrait* BasicDistributionLibrary::getResult() const { return _darray; }