NewtonRaphson.C

Go to the documentation of this file.

/*
 * Copyright (c) 2024 Saud Zahir
 *
 * This file is part of deltaFlow.
 *
 * deltaFlow is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public
 * License as published by the Free Software Foundation; either
 * version 3 of the License, or (at your option) any later version.
 *
 * deltaFlow is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public
 * License along with deltaFlow.  If not, see
 * <https://www.gnu.org/licenses/>.
 */
 
#include <cmath>
#include <limits>
#include <iostream>
#include <vector>
 
#include "NewtonRaphson.H"
#include "Jacobian.H"
#include "PowerMismatch.H"
#include "Data.H"
#include "Utils.H"
 
bool NewtonRaphson::solve(SolverContext& ctx) {
    return NewtonRaphson::run(
        *ctx.G,
        *ctx.B,
        *ctx.P,
        *ctx.Q,
        *ctx.V,
        *ctx.delta,
        ctx.N,
        ctx.n_pq,
        *ctx.pq_bus_id,
        ctx.maxIter,
        ctx.tolerance,
        ctx.iterHistory,
        ctx.observer
    );
}
 
bool NewtonRaphson::run(
    const Eigen::MatrixXd& G,
    const Eigen::MatrixXd& B,
    const Eigen::VectorXd& Ps,
    const Eigen::VectorXd& Qs,
    Eigen::VectorXd& V,
    Eigen::VectorXd& delta,
    int n_bus,
    int n_pq,
    const std::vector<int>& pq_bus_id,
    int maxIter,
    double tolerance,
    std::vector<std::pair<int, double>>* iterHistory,
    ISolverObserver* observer
) {
    // Compute initial mismatch
    Eigen::VectorXd P(n_bus), Q(n_bus);
    Eigen::VectorXd mismatch = PowerMismatch::compute(Ps, Qs, G, B, V, delta, n_bus, pq_bus_id, P, Q);
 
    double error = mismatch.cwiseAbs().maxCoeff();
    int iter = 0;
 
    if (iterHistory) {
        iterHistory->clear();
        iterHistory->emplace_back(0, error);
    }
 
    while (error >= tolerance) {
        if (iter >= maxIter) {
            if (observer) observer->onFinished("Newton-Raphson", false, iter, maxIter, error, tolerance);
            return false;
        }
        iter++;
 
        // Build Jacobian
        Eigen::MatrixXd J = Jacobian::compute(V, delta, n_bus, n_pq, pq_bus_id, G, B, P, Q);
 
        // Solve J * correction = mismatch
        Eigen::VectorXd correction = J.colPivHouseholderQr().solve(mismatch);
 
        // Update delta for non-slack buses (indices 1..N-1)
        for (int i = 1; i < n_bus; ++i) {
            delta(i) += correction(i - 1);
        }
 
        // Update V for PQ buses only
        for (int k = 0; k < n_pq; ++k) {
            V(pq_bus_id[k]) += correction(n_bus - 1 + k);
        }
 
        // Recompute mismatch
        mismatch = PowerMismatch::compute(Ps, Qs, G, B, V, delta, n_bus, pq_bus_id, P, Q);
 
        error = mismatch.cwiseAbs().maxCoeff();
        if (iterHistory) iterHistory->emplace_back(iter, error);
        if (observer) observer->onIteration("Newton-Raphson", iter, maxIter, error, tolerance);
    }
 
    if (observer) observer->onFinished("Newton-Raphson", true, iter, maxIter, error, tolerance);
    return true;
}