/*******************************************************************************
 * Copyright (C) 2017-2026 Theodore Chang
 *
 * This program 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.
 *
 * This program 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 this program.  If not, see <http://www.gnu.org/licenses/>.
 ******************************************************************************/

#include "ElementalLee.h"

ElementalLee::ElementalLee(const unsigned T, const double A, uvec&& ET)
    : ModifierDynamics(T, std::move(ET))
    , damping_ratio(std::abs(A)) {}

int ElementalLee::update_status() {
    suanpan::for_all(element_pool, [&](const std::weak_ptr<Element>& ele_ptr) {
        const auto t_ptr = ele_ptr.lock();

        if(nullptr == t_ptr || !t_ptr->if_update_viscous() || !t_ptr->allow_modify_viscous()) return;

        if(!t_ptr->get_current_mass().empty()) access::rw(t_ptr->get_mass_container()) = damping_ratio * t_ptr->get_current_mass();

        if(!t_ptr->get_current_stiffness().empty()) {
            mat t_stiffness(t_ptr->get_total_number(), t_ptr->get_total_number(), fill::zeros);
            t_stiffness += damping_ratio * t_ptr->get_current_stiffness();
            if(t_ptr->is_nlgeom() && !t_ptr->get_current_geometry().empty()) t_stiffness += damping_ratio * t_ptr->get_current_geometry();
            access::rw(t_ptr->get_stiffness_container()) = t_stiffness;
        }
    });

    return SUANPAN_SUCCESS;
}
