LisVector.h

Go to the documentation of this file.

 
#pragma once
 
#include <cassert>
#include <span>
#include <string>
#include <vector>
 
#include "BaseLib/Error.h"
#include "LisWrapper.h"
 
namespace MathLib
{
class LisVector
{
public:
    using IndexType = LIS_INT;
 
public:
    explicit LisVector(std::size_t length);
 
    LisVector(std::size_t length, double* data);
 
    LisVector(LisVector const& src);
 
    virtual ~LisVector();
 
    std::size_t size() const;
 
    std::size_t getRangeBegin() const { return 0; }
    std::size_t getRangeEnd() const { return this->size(); }
 
    // TODO preliminary
    void setZero() { lis_vector_set_all(0.0, vec_); }
 
    double operator[](IndexType rowId) const { return get(rowId); }
    double get(IndexType rowId) const
    {
        double v = .0;
        lis_vector_get_value(vec_, rowId, &v);
        return v;
    }
 
    void set(IndexType rowId, double v)
    {
        lis_vector_set_value(LIS_INS_VALUE, rowId, v, vec_);
    }
 
    void add(IndexType rowId, double v)
    {
        lis_vector_set_value(LIS_ADD_VALUE, rowId, v, vec_);
    }
 
    void write(const std::string& filename) const;
 
    LIS_VECTOR& getRawVector() { return vec_; }
 
    template <class T_SUBVEC>
    void set(const std::vector<IndexType>& pos, const T_SUBVEC& sub_vec)
    {
        for (std::size_t i = 0; i < pos.size(); ++i)
        {
            set(pos[i], sub_vec[i]);
        }
    }
 
    template <class T_SUBVEC>
    void add(const std::vector<IndexType>& pos, const T_SUBVEC& sub_vec)
    {
        for (std::size_t i = 0; i < pos.size(); ++i)
        {
            add(pos[i], sub_vec[i]);
        }
    }
 
    void copyValues(std::vector<double>& u) const
    {
        u.resize(size());
        lis_vector_get_values(vec_, 0, size(), u.data());
    }
 
    void copyValues(std::span<double> u) const
    {
        if (u.size() != size())
        {
            OGS_FATAL(
                "LisVector::copyValues() size mismatch. Trying to copy a "
                "vector "
                "of size {:d} to a span of size {:d}.",
                size(), u.size());
        }
        lis_vector_get_values(vec_, 0, size(), u.data());
    }
 
private:
    LIS_VECTOR vec_;
};
 
}  // namespace MathLib