ung2r.hpp

Go to the documentation of this file.

 
//
// Copyright (c) 2025, University of Colorado Denver. All rights reserved.
//
// This file is part of <T>LAPACK.
// <T>LAPACK is free software: you can redistribute it and/or modify it under
// the terms of the BSD 3-Clause license. See the accompanying LICENSE file.
 
#ifndef TLAPACK_UNG2R_HH
#define TLAPACK_UNG2R_HH
 
#include "tlapack/base/utils.hpp"
#include "tlapack/blas/scal.hpp"
#include "tlapack/lapack/ungq_level2.hpp"
 
namespace tlapack {
 
template <class T, TLAPACK_SMATRIX matrix_t, TLAPACK_VECTOR vector_t>
constexpr WorkInfo ung2r_worksize(const matrix_t& A, const vector_t& tau)
{
    using idx_t = size_type<matrix_t>;
    using range = pair<idx_t, idx_t>;
 
    // constants
    const idx_t n = ncols(A);
 
    if (n > 1) {
        auto&& C = cols(A, range{1, n});
        return larf_worksize<T>(LEFT_SIDE, FORWARD, COLUMNWISE_STORAGE,
                                col(A, 0), tau[0], C);
    }
    return WorkInfo(0);
}
 
template <TLAPACK_SMATRIX matrix_t,
          TLAPACK_VECTOR vector_t,
          TLAPACK_WORKSPACE work_t>
int ung2r_work(matrix_t& A, const vector_t& tau, work_t& work)
{
    return ungq_level2_work(FORWARD, COLUMNWISE_STORAGE, A, tau, work);
}
 
template <TLAPACK_SMATRIX matrix_t, TLAPACK_VECTOR vector_t>
int ung2r(matrix_t& A, const vector_t& tau)
{
    using T = type_t<matrix_t>;
    using idx_t = size_type<matrix_t>;
 
    // Functor
    Create<matrix_t> new_matrix;
 
    // constants
    const idx_t n = ncols(A);
 
    // quick return
    if (n <= 0) return 0;
 
    // Allocates workspace
    WorkInfo workinfo = ung2r_worksize<T>(A, tau);
    std::vector<T> work_;
    auto work = new_matrix(work_, workinfo.m, workinfo.n);
 
    return ung2r_work(A, tau, work);
}
 
}  // namespace tlapack
 
#endif  // TLAPACK_UNG2R_HH