optimproblem.hpp

Go to the documentation of this file.

#include "math.h"
#include <assert.h>
#include <petsctao.h>
#include "defs.hpp"
#include "timestepper.hpp"
#include <iostream>
#include <algorithm>
#include "optimtarget.hpp"
#pragma once
 
/* if Petsc version < 3.17: Change interface for Tao Optimizer */
#if PETSC_VERSION_MAJOR<4 && PETSC_VERSION_MINOR<17
#define TaoSetObjective TaoSetObjectiveRoutine
#define TaoSetGradient(tao, NULL, TaoEvalGradient,this)  TaoSetGradientRoutine(tao, TaoEvalGradient,this) 
#define TaoSetObjectiveAndGradient(tao, NULL, TaoEvalObjectiveAndGradient,this) TaoSetObjectiveAndGradientRoutine(tao, TaoEvalObjectiveAndGradient,this)
#define TaoSetSolution(tao, xinit) TaoSetInitialVector(tao, xinit)
#define TaoGetSolution(tao, params) TaoGetSolutionVector(tao, params) 
#endif
/* if Petsc version < 3.21: Change interface for Tao Monitor */
#if PETSC_VERSION_MAJOR<4 && PETSC_VERSION_MINOR<21
#define TaoMonitorSet TaoSetMonitor 
#endif
 
class OptimProblem {
  protected:
 
  size_t ninit; 
  int ninit_local; 
  Vec rho_t0; 
  Vec rho_t0_bar; 
  std::vector<Vec> store_finalstates; 
 
  OptimTarget* optim_target; 
 
  MPI_Comm comm_init; 
  MPI_Comm comm_optim; 
  int mpirank_optim, mpisize_optim; 
  int mpirank_petsc, mpisize_petsc; 
  int mpirank_world, mpisize_world; 
  int mpirank_init, mpisize_init; 
 
  bool quietmode; 
 
  std::vector<double> obj_weights; 
  int ndesign; 
  double objective; 
  double obj_cost; 
  double obj_regul; 
  double obj_penal_leakage; 
  double obj_penal_weightedcost; 
  double obj_penal_dpdm; 
  double obj_penal_variation; 
  double obj_penal_energy; 
  double fidelity; 
  double gnorm; 
  double gamma_tikhonov; 
  bool tikhonov_use_x0; 
  double gamma_penalty_leakage; 
  double gamma_penalty_weightedcost; 
  double gamma_penalty_dpdm; 
  double gamma_penalty_energy; 
  double gamma_penalty_variation; 
  double tol_grad_abs; 
  double tol_grad_rel; 
  double tol_final_cost; 
  double tol_infidelity; 
  int maxiter; 
  Tao tao; 
  double* mygrad; 
    
  Vec xtmp; 
  
  public: 
    Output* output; 
    TimeStepper* timestepper; 
    Vec xlower, xupper; 
    Vec xprev; 
    Vec xinit; 
 
  OptimProblem(const Config& config, TimeStepper* timestepper_, MPI_Comm comm_init_, MPI_Comm comm_optim, Output* output_, bool quietmode=false);
 
  ~OptimProblem();
 
  int getNdesign(){ return ndesign; };
 
  double getObjective(){ return objective; };
 
  double getCostT()    { return obj_cost; };
 
  double getRegul()    { return obj_regul; };
 
  double getPenaltyLeakage()  { return obj_penal_leakage; };
 
  double getPenaltyWeightedCost()  { return obj_penal_weightedcost; };
 
 
 
  double getPenaltyDpDm()  { return obj_penal_dpdm; };
 
  double getPenaltyVariation()  { return obj_penal_variation; };
 
  double getPenaltyEnergy()  { return obj_penal_energy; };
 
  double getFidelity() { return fidelity; };
 
  double getTolFinalCost()    { return tol_final_cost; };
 
  double getTolGradAbs()    { return tol_grad_abs; };
 
  double getTolInfidelity()   { return tol_infidelity; };
 
  int getMPIrank_world() { return mpirank_world;};
 
  int getMaxIter()     { return maxiter; };
 
  double evalF(const Vec x);
 
  void evalGradF(const Vec x, Vec G);
 
  void solve(Vec xinit);
 
  void getStartingPoint(Vec x);
 
  void getSolution(Vec* opt);
};
 
PetscErrorCode TaoMonitor(Tao tao,void*ptr);
 
PetscErrorCode TaoEvalObjective(Tao tao, Vec x, PetscReal *f, void*ptr);
 
PetscErrorCode TaoEvalGradient(Tao tao, Vec x, Vec G, void*ptr);
 
PetscErrorCode TaoEvalObjectiveAndGradient(Tao tao, Vec x, PetscReal *f, Vec G, void*ptr);