DP_SSNComp.h

// SPDX-FileCopyrightText: 2026 Institute for Automation of Complex Power Systems, EONERC, RWTH Aachen University
// SPDX-License-Identifier: MPL-2.0
 
#pragma once
 
#include <dpsim-models/MNASimPowerComp.h>
#include <dpsim-models/Solver/MNAInterface.h>
 
namespace CPS {
namespace DP {

class SSNComp : public MNASimPowerComp<Complex> {
private:
  const Int mInputSize;
  const Int mOutputSize;
 
protected:
  Real mTimeStep;

  Matrix mA;
  Matrix mB;
  Matrix mC;
  Matrix mD;

  MatrixComp mDiscreteA;
  MatrixComp mDiscreteB;

  MatrixComp mW;
  MatrixComp mYHist;

  const Attribute<MatrixComp>::Ptr mX;
 
  SSNComp(String uid, String name, Int inputSize, Int outputSize,
          Logger::Level logLevel = Logger::Level::off);

  Matrix buildAugmentedA(Real omega) const;
  Matrix buildAugmentedB() const;

  virtual void recomputeDiscreteModel(Real omega);
 
  virtual MatrixComp calculateHistoryVector() const;

  virtual MatrixComp calculateSteadyStateStateFromInput(const MatrixComp &u,
                                                        Real omega) const;
  virtual MatrixComp
  calculateSteadyStateOutputFromInput(const MatrixComp &x,
                                      const MatrixComp &u) const;
 
  virtual void updateState(const MatrixComp &uOld, const MatrixComp &uNew);

  virtual void updateStateSpaceModel();
 
  virtual Attribute<MatrixComp>::Ptr inputAttribute() const = 0;
  virtual Attribute<MatrixComp>::Ptr outputAttribute() const = 0;
 
public:
  UInt getStateCount() const;

  const MatrixComp &getDiscreteA() const;
  const MatrixComp &getDiscreteB() const;
  const Matrix &getC() const;
 
  void setParameters(const Matrix &A, const Matrix &B, const Matrix &C,
                     const Matrix &D);
 
  void mnaCompInitialize(Real omega, Real timeStep,
                         Attribute<Matrix>::Ptr leftVector) override final;
 
  void mnaCompAddPreStepDependencies(
      AttributeBase::List &prevStepDependencies,
      AttributeBase::List &attributeDependencies,
      AttributeBase::List &modifiedAttributes) override final;
 
  void mnaCompPreStep(Real time, Int timeStepCount) override final;
 
  void mnaCompAddPostStepDependencies(
      AttributeBase::List &prevStepDependencies,
      AttributeBase::List &attributeDependencies,
      AttributeBase::List &modifiedAttributes,
      Attribute<Matrix>::Ptr &leftVector) override final;
};
 
} // namespace DP
} // namespace CPS