Vector.h

// Copyright (C) 2020 Garth N. Wells
//
// This file is part of DOLFINx (https://www.fenicsproject.org)
//
// SPDX-License-Identifier:    LGPL-3.0-or-later
 
#pragma once
 
#include <dolfinx/common/IndexMap.h>
#include <memory>
 
namespace dolfinx::la
{
 
 
template <typename T>
class Vector
{
public:
  Vector(const std::shared_ptr<const common::IndexMap>& map, int bs)
      : _map(map), _bs(bs)
  {
    assert(map);
    const std::int32_t local_size
        = bs * (map->size_local() + map->num_ghosts());
    _x.resize(local_size);
  }
 
  Vector(const Vector& x) = default;
 
  Vector(Vector&& x) noexcept = default;
 
  ~Vector() = default;
 
  // Assignment operator (disabled)
  Vector& operator=(const Vector& x) = delete;
 
  Vector& operator=(Vector&& x) = default;
 
  std::shared_ptr<const common::IndexMap> map() const { return _map; }
 
  constexpr int bs() const { return _bs; }
 
  const std::vector<T>& array() const { return _x; }
 
  std::vector<T>& mutable_array() { return _x; }
 
private:
  // Map describing the data layout
  std::shared_ptr<const common::IndexMap> _map;
 
  // Block size
  int _bs;
 
  // Data
  std::vector<T> _x;
};
} // namespace dolfinx::la