adjoint.h

#ifndef __ADJOINT_H__
#define __ADJOINT_H__

/* includes */
#include <deal.II/distributed/solution_transfer.h>
#include <deal.II/fe/fe_q.h>
#include <deal.II/fe/fe_values.h>
#include <deal.II/lac/la_parallel_vector.h>

#include <iostream>
#include <vector>

#include "dg/dg_base.hpp"
#include "functional.h"
#include "parameters/all_parameters.h"
#include "physics/physics.h"

namespace PHiLiP {


 #if PHILIP_DIM==1 // dealii::parallel::distributed::Triangulation<dim> does not work for 1D
template <int dim, int nspecies, int nstate, typename real, typename MeshType = dealii::Triangulation<dim>>
#else
template <int dim, int nspecies, int nstate, typename real, typename MeshType = dealii::parallel::distributed::Triangulation<dim>>
#endif
class Adjoint
{
public:

    enum AdjointStateEnum{
        coarse, 
        fine,   
    };


    Adjoint(
        std::shared_ptr< DGBase<dim,nspecies,real,MeshType> > _dg,
        std::shared_ptr< Functional<dim, nspecies, nstate, real, MeshType> > _functional,
        std::shared_ptr< Physics::PhysicsBase<dim,nspecies,nstate,Sacado::Fad::DFad<real>> > _physics);


    void reinit();
    // to reinitialize with other pointers, just create a new class


    void convert_to_state(AdjointStateEnum state);


    void coarse_to_fine();


    void fine_to_coarse();


    dealii::LinearAlgebra::distributed::Vector<real> fine_grid_adjoint();


    dealii::LinearAlgebra::distributed::Vector<real> coarse_grid_adjoint();


    dealii::Vector<real> dual_weighted_residual();


    void output_results_vtk(const unsigned int cycle);

    std::shared_ptr< DGBase<dim,nspecies,real,MeshType> > dg;
    std::shared_ptr< Functional<dim, nspecies, nstate, real, MeshType> > functional;
    std::shared_ptr< Physics::PhysicsBase<dim,nspecies,nstate,Sacado::Fad::DFad<real>> > physics;
    
    std::shared_ptr<MeshType> triangulation;
    dealii::LinearAlgebra::distributed::Vector<real> solution_coarse;
    dealii::LinearAlgebra::distributed::Vector<real> dIdw_fine;
    dealii::LinearAlgebra::distributed::Vector<real> dIdw_coarse;
    dealii::LinearAlgebra::distributed::Vector<real> adjoint_fine;
    dealii::LinearAlgebra::distributed::Vector<real> adjoint_coarse;

    dealii::Vector<real> dual_weighted_residual_fine;
    
    dealii::Vector<real> coarse_fe_index;

    AdjointStateEnum adjoint_state;

protected:
    MPI_Comm mpi_communicator; 
    dealii::ConditionalOStream pcout; 

}; // Adjoint class


} // PHiLiP namespace

#endif // __ADJOINT_H__