PetscPostStage.cpp File Reference

Post-time-integration-stage function. More...

#include <stdio.h>
#include <basic.h>
#include <simulation_object.h>
#include <petscinterface.h>

Go to the source code of this file.

Macros
#define	__FUNCT__   "PetscPostStage"

Functions
PetscErrorCode	PetscPostStage (TS ts, PetscReal stagetime, PetscInt stageindex, Vec *Y)

Detailed Description

Post-time-integration-stage function.

Author

Debojyoti Ghosh

Definition in file PetscPostStage.cpp.

Macro Definition Documentation

__FUNCT__

#define __FUNCT__   "PetscPostStage"

Definition at line 13 of file PetscPostStage.cpp.

Function Documentation

PetscPostStage()

PetscErrorCode PetscPostStage	(	TS	ts,
		PetscReal	stagetime,
		PetscInt	stageindex,
		Vec *	Y
	)

Function called after every stage in a multi-stage time-integration method

Parameters

ts	Time integrator of PETSc type TS
stagetime	Current stage time
stageindex	Stage
Y	Stage solutions (all stages) - be careful what you access

Definition at line 16 of file PetscPostStage.cpp.

{
  PETScContext* context(nullptr);

  PetscFunctionBegin;

  TSGetApplicationContext(ts,&context);
  if (!context) {
    fprintf(stderr,"Error in PetscPreTimeStep: Null context!\n");
    return(1);
  }
  SimulationObject* sim = (SimulationObject*) context->simobj;
  int nsims = context->nsims;

  TSType time_scheme;
  TSGetType(ts,&time_scheme);

  TSGetTimeStep(ts,&(context->dt));
  context->stage_index = stageindex;
  if (context->stage_times.size() == stageindex) {
    context->stage_times.push_back(stagetime/context->dt);
  }

  for (int ns = 0; ns < nsims; ns++) {

    HyPar* solver = &(sim[ns].solver);
    MPIVariables* mpi = &(sim[ns].mpi);
  
    /* get solution */
    TransferVecFromPETSc(solver->u,Y[stageindex],context,ns,context->offsets[ns]);
  
    /* apply immersed boundaries */
    solver->ApplyIBConditions(solver,mpi,solver->u,stagetime);
  
    /* If using a non-linear scheme with ARKIMEX methods, 
       compute the non-linear finite-difference operator */
    if (!strcmp(time_scheme,TSARKIMEX)) {
      solver->NonlinearInterp(  solver->u,
                                solver,
                                mpi,
                                (double)stagetime,
                                solver->FFunction );
    }
  
    /* Call any physics-specific post-stage function, if available */
    if (solver->PostStage) {
      solver->PostStage(solver->u,solver,mpi,stagetime);
    }
  
    TransferVecToPETSc(solver->u,Y[stageindex],context,ns,context->offsets[ns]);

  }

  PetscFunctionReturn(0);
}