cooSparsityPattern.hpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2025 - 2026 NeoN authors
//
// SPDX-License-Identifier: MIT
 
#pragma once
 
#include "NeoN/core/copyTo.hpp"
#include "NeoN/core/vector/vector.hpp"
#include "NeoN/linearAlgebra/sparsityView.hpp"
#include "NeoN/mesh/unstructured/unstructuredMesh.hpp"
 
namespace NeoN::la
{
 
 
template<typename IndexType>
class CooSparsityPattern : public NeoN::SupportsCopyTo<CooSparsityPattern<IndexType>>
{
 
    void validate() const;
 
public:
 
    using SparsityIndexType = IndexType;
 
    CooSparsityPattern(const CooSparsityPattern& sp);
 
    CooSparsityPattern(Vector<IndexType>&& colIdx, Vector<IndexType>&& rowIdxs, Dimensions dim);
 
    [[nodiscard]] CooSparsityPattern copyToExecutor(Executor dstExec) const override
    {
        return CooSparsityPattern<IndexType>(
            colIdxs_.copyToExecutor(dstExec), rowIdxs_.copyToExecutor(dstExec), dimensions_
        );
    }
 
    ~CooSparsityPattern() = default;
 
    const Executor& exec() const { return rowOffs_.exec(); }
 
    [[nodiscard]] const Vector<IndexType>& colIdxs() const { return colIdxs_; };
 
    [[nodiscard]] Vector<IndexType>& colIdxs() { return colIdxs_; };
 
    [[nodiscard]] const Vector<IndexType>& rowIdxs() const { return rowIdxs_; };
 
    [[nodiscard]] Vector<IndexType>& rowIdxs() { return rowIdxs_; };
 
    [[nodiscard]] const Vector<IndexType>& rowOffs() const { return rowOffs_; };
 
    [[nodiscard]] Vector<IndexType>& rowOffs() { return rowOffs_; };
 
    [[nodiscard]] localIdx rows() const { return dimensions_.rows; };
 
    [[nodiscard]] localIdx nnz() const { return colIdxs_.size(); };
 
    [[nodiscard]] Dimensions dimension() const { return dimensions_; };
 
    [[nodiscard]] SparsityView<IndexType> view() const
    {
        return SparsityView<IndexType>(colIdxs_.view(), rowIdxs_.view());
    }
 
private:
 
    Dimensions dimensions_;
 
    Vector<IndexType> rowIdxs_; 
 
    Vector<IndexType> colIdxs_; 
 
    // NOTE for now the rowOffsets are kept since it is required for face based
    // matrix assembly.
    Vector<IndexType> rowOffs_; 
};
 
} // namespace NeoN::la