source: flair-src/trunk/lib/FlairCore/src/IODevice_impl.cpp@ 16

// %flair:license{
// This file is part of the Flair framework distributed under the
// CECILL-C License, Version 1.0.
// %flair:license}
//  created:    2011/05/01
//  filename:   IODevice.cpp
//
//  author:     Guillaume Sanahuja
//              Copyright Heudiasyc UMR UTC/CNRS 7253
//
//  version:    $Id: $
//
//  purpose:    classe de base avec entrees sorties
//
//
/*********************************************************************/

#include "IODevice_impl.h"
#include "IODevice.h"
#include "io_data.h"
#include "io_data_impl.h"
#include "Thread.h"
#include "Mutex.h"
#include "FrameworkManager.h"
#include "FrameworkManager_impl.h"
#include "SharedMem.h"
#include <string.h>

using std::string;
using std::fstream;
using namespace flair::core;

IODevice_impl::IODevice_impl(const IODevice *self) {
  this->self = self;
  thread_to_wake = NULL;
  wake_mutex = new Mutex(self);
  framework_impl = ::getFrameworkManagerImpl();
  framework = getFrameworkManager();
  tobelogged = false;
  outputtoshm = false;
}

IODevice_impl::~IODevice_impl() {}

void IODevice_impl::OutputToShMem(bool enabled) {
  if (framework->IsLogging()) {
    self->Err("impossible while logging\n");
  } else {
    if (LogSize() == 0) {
      self->Warn("log size is null, not enabling output to shared memory.");
      return;
    }

    if (enabled) {
      string dev_name =
          getFrameworkManager()->ObjectName() + "-" + self->ObjectName();
      shmem = new SharedMem(self, dev_name.c_str(), LogSize());
      outputtoshm = true;

      Printf("Created %s shared memory for object %s; size: %i\n",
             dev_name.c_str(), self->ObjectName().c_str(), LogSize());
      PrintLogsDescriptors();
    } else {
      outputtoshm = false;
      delete shmem;
    }
  }
}

void IODevice_impl::PrintLogsDescriptors(void) {
  // own logs
  for (size_t i = 0; i < datasToLog.size(); i++)
    datasToLog.at(i)->pimpl_->PrintLogDescriptor();
  // childs logs
  for (size_t i = 0; i < self->TypeChilds()->size(); i++)
    ((IODevice *)self->TypeChilds()->at(i))->pimpl_->PrintLogsDescriptors();
  // manually added logs
  for (size_t i = 0; i < devicesToLog.size(); i++)
    devicesToLog.at(i)->pimpl_->PrintLogsDescriptors();
}

void IODevice_impl::WriteToShMem(void) {
  if (outputtoshm) {
    size_t size = LogSize();

    char *buf = framework_impl->GetBuffer(size);
    char *buf_orig = buf;
    if (buf == NULL) {
      self->Err("err GetBuffer\n");
      return;
    }

    AppendLog(&buf);

    shmem->Write(buf_orig, size);
    framework_impl->ReleaseBuffer(buf_orig);
  }
}

void IODevice_impl::AddDeviceToLog(const IODevice *device) {
  if (framework->IsLogging()) {
    self->Err("impossible while logging\n");
  } else {
    devicesToLog.push_back(device);
  }
}

bool IODevice_impl::SetToBeLogged(void) {
  if (!tobelogged) {
    tobelogged = true;
    return true;
  } else {
    self->Warn("already added to log\n");
    return false;
  }
}

void IODevice_impl::WriteLogsDescriptors(fstream &desc_file, int *index) {
  // own descriptor
  for (size_t i = 0; i < datasToLog.size(); i++)
    datasToLog.at(i)->pimpl_->WriteLogDescriptor(desc_file, index);
  // childs descriptors
  for (size_t i = 0; i < self->TypeChilds()->size(); i++)
    ((IODevice *)self->TypeChilds()->at(i))
        ->pimpl_->WriteLogsDescriptors(desc_file, index);
  // manually added logs
  for (size_t i = 0; i < devicesToLog.size(); i++)
    devicesToLog.at(i)->pimpl_->WriteLogsDescriptors(desc_file, index);
}

void IODevice_impl::ResumeThread(void) {
  wake_mutex->GetMutex();
  if (thread_to_wake != NULL) {
    thread_to_wake->Resume();
    thread_to_wake = NULL;
  }
  wake_mutex->ReleaseMutex();
}

void IODevice_impl::AddDataToLog(const io_data *data) {
  if (framework->IsLogging()) {
    self->Err("impossible while logging\n");
  } else {
    datasToLog.push_back(data);
  }
}

size_t IODevice_impl::LogSize() const {
  size_t value = 0;

  for (size_t i = 0; i < datasToLog.size(); i++) {
    value += datasToLog.at(i)->GetDataType().GetSize();
  }

  // childs logs
  for (size_t i = 0; i < self->TypeChilds()->size(); i++) {
    value += ((IODevice *)self->TypeChilds()->at(i))->pimpl_->LogSize();
  }
  // manually added logs
  for (size_t i = 0; i < devicesToLog.size(); i++) {
    value += devicesToLog.at(i)->pimpl_->LogSize();
  }

  return value;
}

void IODevice_impl::WriteLog(Time time) {
  if (tobelogged == false)
    return;

  size_t size = LogSize();

  char *buf = framework_impl->GetBuffer(
      sizeof(FrameworkManager_impl::log_header_t) + size);
  char *buf_orig = buf;
  if (buf == NULL) {
    self->Err("err GetBuffer\n");
    return;
  }

  FrameworkManager_impl::log_header_t header;
  header.device = self;
  header.size = size;
  header.time = time;

  memcpy(buf, &header, sizeof(FrameworkManager_impl::log_header_t));
  buf += sizeof(FrameworkManager_impl::log_header_t);
  AppendLog(&buf);

  framework_impl->WriteLog(buf_orig,
                           sizeof(FrameworkManager_impl::log_header_t) + size);
  framework_impl->ReleaseBuffer(buf_orig);
}

void IODevice_impl::AppendLog(char **ptr) {
  // Printf("AppendLog %s %x\n",self->ObjectName().c_str(),*ptr);

  // copy state to buf
  for (size_t i = 0; i < datasToLog.size(); i++) {
    // printf("copy\n");
    datasToLog.at(i)->CopyDatas(*ptr);
    (*ptr) += datasToLog.at(i)->GetDataType().GetSize();
  }

  // copy linked states to buf
  for (size_t i = 0; i < self->TypeChilds()->size(); i++) {
    ((IODevice *)self->TypeChilds()->at(i))->pimpl_->AppendLog(ptr);
  }
  // copy manually added logs to buf
  for (size_t i = 0; i < devicesToLog.size(); i++) {
    devicesToLog.at(i)->pimpl_->AppendLog(ptr);
    // devices.at(i)->DataToLog()->CopyDatas(*ptr);
    //(*ptr)+=devices.at(i)->DataToLog()->Size();
  }
  // Printf("AppendLog %s ok\n",self->ObjectName().c_str());
}

int IODevice_impl::SetToWake(const Thread *thread) {
  int status = 0;

  wake_mutex->GetMutex();
  if (thread_to_wake == NULL) {
    thread_to_wake = (Thread *)thread;
  } else {
    status = -1;
  }
  wake_mutex->ReleaseMutex();

  return status;
}