Parametrization.cpp

Go to the documentation of this file.

#include "Parametrization.hpp"
#include <polyfem/utils/Logger.hpp>
 
namespace polyfem::solver
{
    int CompositeParametrization::size(const int x_size) const
    {
        int cur_size = x_size;
        for (const auto &p : parametrizations_)
            cur_size = p->size(cur_size);
 
        return cur_size;
    }
 
    int CompositeParametrization::inverse_size(int y_size) const
    {
        if (parametrizations_.empty())
        {
            return y_size;
        };
 
        int result = y_size;
        for (int i = parametrizations_.size() - 1; i >= 0; i--)
        {
            result = parametrizations_[i]->inverse_size(result);
        }
        return result;
    }
 
    Eigen::VectorXd CompositeParametrization::inverse_eval(const Eigen::VectorXd &y) const
    {
        if (parametrizations_.empty())
        {
            return y;
        }
 
        Eigen::VectorXd x = y;
        for (int i = parametrizations_.size() - 1; i >= 0; i--)
        {
            x = parametrizations_[i]->inverse_eval(x);
        }
 
        return x;
    }
 
    Eigen::VectorXd CompositeParametrization::eval(const Eigen::VectorXd &x) const
    {
        if (parametrizations_.empty())
            return x;
 
        Eigen::VectorXd y = x;
        for (const auto &p : parametrizations_)
        {
            y = p->eval(y);
        }
 
        return y;
    }
    Eigen::VectorXd CompositeParametrization::apply_jacobian(const Eigen::VectorXd &grad_full, const Eigen::VectorXd &x) const
    {
        Eigen::VectorXd gradv = grad_full;
 
        if (parametrizations_.empty())
            return gradv;
 
        std::vector<Eigen::VectorXd> ys;
        auto y = x;
        for (const auto &p : parametrizations_)
        {
            ys.emplace_back(y);
            y = p->eval(y);
        }
 
        for (int i = parametrizations_.size() - 1; i >= 0; --i)
            gradv = parametrizations_[i]->apply_jacobian(gradv, ys[i]);
 
        return gradv;
    }
} // namespace polyfem::solver