TimeWriter.cpp

Go to the documentation of this file.

/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
   This file is part of CEPS.
 
   CEPS is free software: you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation, either version 3 of the License, or
   (at your option) any later version.
 
   CEPS is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.
 
   You should have received a copy of the GNU General Public License
   along with CEPS (see file LICENSE at root of project).
   If not, see <https://www.gnu.org/licenses/>.
 
 
   Copyright 2019-2024 Inria, Universite de Bordeaux
 
   Authors, in alphabetical order:
 
     Pierre-Elliott BECUE, Florian CARO, Yves COUDIERE(*), Andjela DAVIDOVIC, 
     Charlie DOUANLA-LONTSI, Marc FUENTES, Mehdi JUHOOR, Michael LEGUEBE(*), 
     Pauline MIGERDITICHAN, Valentin PANNETIER(*), Nejib ZEMZEMI.
     * : currently active authors
 
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
#include "pde/common/writers/TimeWriter.hpp"
#include "pde/problem/AbstractPdeProblem.hpp"
#include "pde/common/AbstractDiscretization.hpp"
 
TimeWriter::TimeWriter(
  const CepsString&             fileName, 
  const CepsVector<CepsReal3D>& points,
  AbstractDiscretization*       discr
):
  FileWriter   (fileName),
  m_discr      (discr),
  m_points     (points),
  m_dofIndices ({}),
  m_customNames({}),
  m_customData ({}),
  m_canAddData (true)
{
  // Initialize 0D unknowns
  for (Unknown* u: discr->getProblem()->getZeroDUnknowns())
  {
    DegreeOfFreedom* dof = discr->getDegreesOfFreedomForUnknown(u)[0];
    m_dofIndices.push_back(ceps::getRank()==dof->getOwner() ? dof->getGlobalIndex() : -1);
  }
 
  // Initialize points
  for (CepsReal3D point: points)
    for (Unknown* u : discr->getProblem()->getSpatialUnknowns())
    {
      CepsDofGlobalIndex dofId;
      CepsProcId         owner;
      discr->findClosestDof(point,u,dofId,owner);
      m_dofIndices.push_back(owner == ceps::getRank()? dofId : -1);
    }
}
 
CepsBool
TimeWriter::hasSomethingToWrite() const
{
  return m_dofIndices.size() + m_customData.size() != 0;
}
 
void
TimeWriter::addCustomData(const CepsString& name, CepsReal* data)
{
  CEPS_ABORT_IF(not m_canAddData,
    "Cannot add custom data to TimeWriter after write() has been called"
  );
  m_customData .push_back(data);
  m_customNames.push_back(name);
}
 
void
TimeWriter::write(CepsReal t, DHVecPtr data)
{
 
  if (m_canAddData)
    writeHeader();
 
  // Extract data from vector
  data->getLocalData();
  CepsUInt n = m_dofIndices.size();
  CepsVector<CepsReal> myToWrite(n,0.),toWrite(n,0.);
  for (CepsUInt i=0; i<n; i++)
    if (m_dofIndices[i]!=-1)
    {
      myToWrite[i] = (*data)[m_dofIndices[i]];
    }
  data->releaseLocalData();
 
  MPI_Allreduce(myToWrite.data(),toWrite.data(),n,CEPS_MPI_REAL,MPI_SUM,ceps::getCommunicator()),
 
  m_file << std::setprecision(16);
  m_file << t;
  for (CepsReal rdata: toWrite)
    m_file << " " << rdata;
  for (CepsReal* data : m_customData)
    m_file << " " << *data;
  m_file << std::endl;
 
}
 
 
void
TimeWriter::writeHeader()
{
 
  open();
 
  // Write header
  m_file << "# Time dependant data generated by CEPS\n"
         << "# Columns are:\n"
         << "#   1. Time\n";
  CepsUInt i=1;
  for (Unknown* u: m_discr->getProblem()->getZeroDUnknowns())
  {
    i++;
    m_file << "#   " << i << ". " << u->getName() << "\n";
  }
  for (CepsReal3D point: m_points)
    for (Unknown* u : m_discr->getProblem()->getSpatialUnknowns())
    {
      i++;
      m_file << "#   " << i << ". " << u->getName() << " at closest point to (" << point[0]
             << "," << point[1] << "," << point[2] << ")" << "\n";
    }
  for (CepsString cn : m_customNames)
  {
    i++;
    m_file << "#   " << i << ". " << cn << "\n";
  }
  m_file << std::endl;
 
  m_canAddData = false;
}