TransferFromPETSc.cpp

Go to the documentation of this file.

#ifdef with_petsc

#include <stdlib.h>
#include <vector>
#include <basic.h>
#include <arrayfunctions.h>
#include <simulation_object.h>
#include <petscinterface_struct.h>

#undef __FUNCT__
#define __FUNCT__ "TransferVecFromPETSc"

int TransferVecFromPETSc( double* const u, 
                          const Vec Y, 
                          void* ctxt, 
                          const int sim_idx,
                          const int offset   )
{
  PETScContext* context = (PETScContext*) ctxt;
  SimulationObject* sim = (SimulationObject*) context->simobj;
  const double* Yarr;

  PetscFunctionBegin;
  VecGetArrayRead(Y,&Yarr);
  std::vector<int> index(sim[sim_idx].solver.ndims,0);
  ArrayCopynD(  sim[sim_idx].solver.ndims,
                (Yarr+offset),
                u,
                sim[sim_idx].solver.dim_local,
                0,
                sim[sim_idx].solver.ghosts,
                index.data(),
                sim[sim_idx].solver.nvars );
  VecRestoreArrayRead(Y,&Yarr);

  PetscFunctionReturn(0);
}

#endif