MemoryDataReaderWriter.cc

// -*- tab-width: 2; indent-tabs-mode: nil; coding: utf-8-with-signature -*-
//-----------------------------------------------------------------------------
// Copyright 2000-2026 CEA (www.cea.fr) IFPEN (www.ifpenergiesnouvelles.com)
// See the top-level COPYRIGHT file for details.
// SPDX-License-Identifier: Apache-2.0
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* MemoryDataReaderWriter.cc                                   (C) 2000-2009 */
/*                                                                           */
/* Reading/writing data in memory.                                           */
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include "arcane/utils/Collection.h"
#include "arcane/utils/Enumerator.h"
#include "arcane/utils/NotImplementedException.h"
#include "arcane/utils/FatalErrorException.h"
#include "arcane/utils/TraceInfo.h"
#include "arcane/utils/Ref.h"
 
#include "arcane/core/IData.h"
#include "arcane/core/IVariable.h"
#include "arcane/core/VariableCollection.h"
 
#include "arcane/impl/MemoryDataReaderWriter.h"
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Arcane
{
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
MemoryDataReaderWriter::
~MemoryDataReaderWriter()
{
  free();
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
void MemoryDataReaderWriter::
free()
{
  m_vars_to_data.clear();
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
void MemoryDataReaderWriter::
beginWrite(const VariableCollection& vars)
{
  // Copies the current table and removes all references to the variables
  // that will be saved. Once this is done, only references to
  // more frequently used variables remain. Therefore,
  // the corresponding IData is released.
  VarToDataMap vars_to_data = m_vars_to_data;
  for (VariableCollection::Enumerator ivar(vars); ++ivar;) {
    IVariable* var = *ivar;
    VarToDataMap::iterator i = vars_to_data.find(var->fullName());
    if (i != vars_to_data.end())
      vars_to_data.erase(i);
  }
 
  for (VarToDataMap::iterator i = vars_to_data.begin(); i != vars_to_data.end(); ++i) {
    // Removes the reference to the current table and destroys the data
    m_vars_to_data.erase(i->first);
  }
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
void MemoryDataReaderWriter::
write(IVariable* var, IData* data)
{
  Ref<IData> cdata = _findData(var);
  if (!cdata.get()) {
    cdata = data->cloneRef();
    m_vars_to_data.insert(std::make_pair(var->fullName(), cdata));
  }
  else {
    cdata->copy(data);
  }
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
void MemoryDataReaderWriter::
read(IVariable* var, IData* data)
{
  Ref<IData> cdata = _findData(var);
  if (!cdata.get()) {
    warning() << A_FUNCNAME << ": "
              << String::format("can not find data for variable '{0}': variable will not be restored",
                                var->fullName());
    return;
  }
  data->copy(cdata.get());
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
Ref<IData> MemoryDataReaderWriter::
_findData(IVariable* var)
{
  auto i = m_vars_to_data.find(var->fullName());
  if (i == m_vars_to_data.end())
    return {};
  return i->second;
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // namespace Arcane
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/