LegacyMatrix.hpp

Go to the documentation of this file.

 
//
// Copyright (c) 2021-2023, 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_LEGACY_MATRIX_HH
#define TLAPACK_LEGACY_MATRIX_HH
 
#include <cassert>
 
#include "tlapack/base/exceptionHandling.hpp"
#include "tlapack/base/types.hpp"
 
namespace tlapack {
 
template <class T,
          class idx_t = std::size_t,
          Layout L = Layout::ColMajor,
          std::enable_if_t<(L == Layout::RowMajor) || (L == Layout::ColMajor),
                           int> = 0>
struct LegacyMatrix {
    idx_t m, n;  
    T* ptr;      
    idx_t ldim;  
 
    static constexpr Layout layout = L;
 
    constexpr const T& operator()(idx_t i, idx_t j) const noexcept
    {
        assert(i >= 0);
        assert(i < m);
        assert(j >= 0);
        assert(j < n);
        return (layout == Layout::ColMajor) ? ptr[i + j * ldim]
                                            : ptr[i * ldim + j];
    }
 
    constexpr T& operator()(idx_t i, idx_t j) noexcept
    {
        assert(i >= 0);
        assert(i < m);
        assert(j >= 0);
        assert(j < n);
        return (layout == Layout::ColMajor) ? ptr[i + j * ldim]
                                            : ptr[i * ldim + j];
    }
 
    constexpr LegacyMatrix(idx_t m, idx_t n, T* ptr, idx_t ldim)
        : m(m), n(n), ptr(ptr), ldim(ldim)
    {
        tlapack_check(m >= 0);
        tlapack_check(n >= 0);
        tlapack_check(ldim >= ((layout == Layout::ColMajor) ? m : n));
    }
 
    constexpr LegacyMatrix(idx_t m, idx_t n, T* ptr)
        : m(m), n(n), ptr(ptr), ldim((layout == Layout::ColMajor) ? m : n)
    {
        tlapack_check(m >= 0);
        tlapack_check(n >= 0);
    }
};
 
}  // namespace tlapack
 
#endif  // TLAPACK_LEGACY_MATRIX_HH