ODEintSolver.cpp

/* Copyright 2017-2021 Institute for Automation of Complex Power Systems,
 *                     EONERC, RWTH Aachen University
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/.
 *********************************************************************************/
 
#include <dpsim/ODEintSolver.h>
 
using namespace DPsim;
 
ODEintSolver::ODEintSolver(String name, CPS::ODEintInterface::Ptr comp, Real dt,
                           Real t0)
    : mComponent(comp), mTimestep(dt) {
  times.push_back(t0);
  self = this; // sets static pointer to current object
  ProbDim = comp->num_states();
 
  curSolution.resize(ProbDim);
 
  //register all system functions
  system.push_back([comp](const double y[], double ydot[], const double t) {
    comp->odeint(y, ydot, t);
  });
}
 
Real ODEintSolver::step(Real time) {
 
  mComponent->pre_step(); 
  curSolution.assign(mComponent->state_vector(),
                     mComponent->state_vector() + ProbDim);
  Real NextTime = time + mTimestep;
 
  std::cout << "Current Time " << NextTime << std::endl;
 
  stepper.do_step(
      &ODEintSolver::StateSpaceWrapper, curSolution, time,
      mTimestep); 
  mComponent->set_state_vector(
      curSolution); 
  solution.push_back(curSolution); 
  times.push_back(time);           
 
  // TODO: add proper logging
  mComponent->post_step();
  return NextTime;
}
 
void ODEintSolver::StateSpaceWrapper(const std::vector<double> &y,
                                     std::vector<double> ydot, double t) {
 
  for (
      auto comp :
      self->system) { // call system functions of the components with the state vector
    comp(y.data(), ydot.data(), t);
  }
}
 
ODEintSolver::~ODEintSolver() {}