vbvector.hh

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright © DUNE Project contributors, see file LICENSE.md in module root
// SPDX-License-Identifier: LicenseRef-GPL-2.0-only-with-DUNE-exception
// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:
#ifndef DUNE_ISTL_VBVECTOR_HH
#define DUNE_ISTL_VBVECTOR_HH
 
#include <cmath>
#include <complex>
#include <iostream>
#include <iterator>
#include <memory>
 
#include <dune/common/ftraits.hh>
#include <dune/common/iteratorfacades.hh>
#include "istlexception.hh"
#include "bvector.hh"
 
#include <dune/istl/blocklevel.hh>
 
namespace Dune {
 
  template<class B, class A=std::allocator<B> >
  class VariableBlockVector : public Imp::block_vector_unmanaged<B,typename A::size_type>
                              // this derivation gives us all the blas level 1 and norms
                              // on the large array. However, access operators have to be
                              // overwritten.
  {
    // just a shorthand
    typedef Imp::BlockVectorWindow<B,A> window_type;
 
  public:
 
    //===== type definitions and constants
 
    using field_type = typename Imp::BlockTraits<B>::field_type;
 
    typedef A allocator_type;
 
    typedef window_type& reference;
 
    typedef const window_type& const_reference;
 
    typedef typename A::size_type size_type;
 
    typedef BlockVector<B,A> value_type;
 
    typedef BlockVector<B,A> block_type;
 
    //===== constructors and such
 
    VariableBlockVector () : Imp::block_vector_unmanaged<B,size_type>()
    {
      // nothing is known ...
      nblocks = 0;
      block = nullptr;
      initialized = false;
    }
 
    explicit VariableBlockVector (size_type _nblocks) : Imp::block_vector_unmanaged<B,size_type>()
    {
      // we can allocate the windows now
      nblocks = _nblocks;
      if (nblocks>0)
      {
        block = windowAllocator_.allocate(nblocks);
        new (block) window_type[nblocks];
      }
      else
      {
        nblocks = 0;
        block = nullptr;
      }
 
      // Note: memory in base class still not allocated
      // the vector not usable
      initialized = false;
    }
 
    VariableBlockVector (size_type _nblocks, size_type m) : Imp::block_vector_unmanaged<B,size_type>()
    {
      // and we can allocate the big array in the base class
      this->n = _nblocks*m;
      if (this->n>0)
      {
        this->p = allocator_.allocate(this->n);
        new (this->p)B[this->n];
      }
      else
      {
        this->n = 0;
        this->p = nullptr;
      }
 
      // we can allocate the windows now
      nblocks = _nblocks;
      if (nblocks>0)
      {
        // allocate and construct the windows
        block = windowAllocator_.allocate(nblocks);
        new (block) window_type[nblocks];
 
        // set the windows into the big array
        for (size_type i=0; i<nblocks; ++i)
          block[i].set(m,this->p+(i*m));
      }
      else
      {
        nblocks = 0;
        block = nullptr;
      }
 
      // and the vector is usable
      initialized = true;
    }
 
    VariableBlockVector (const VariableBlockVector& a)
    {
      // allocate the big array in the base class
      this->n = a.n;
      if (this->n>0)
      {
        // allocate and construct objects
        this->p = allocator_.allocate(this->n);
        new (this->p)B[this->n];
 
        // copy data
        for (size_type i=0; i<this->n; i++) this->p[i]=a.p[i];
      }
      else
      {
        this->n = 0;
        this->p = nullptr;
      }
 
      // we can allocate the windows now
      nblocks = a.nblocks;
      if (nblocks>0)
      {
        // alloc
        block = windowAllocator_.allocate(nblocks);
        new (block) window_type[nblocks];
 
        // and we must set the windows
        block[0].set(a.block[0].getsize(),this->p);           // first block
        for (size_type i=1; i<nblocks; ++i)                         // and the rest
          block[i].set(a.block[i].getsize(),block[i-1].getptr()+block[i-1].getsize());
      }
      else
      {
        nblocks = 0;
        block = nullptr;
      }
 
      // and we have a usable vector
      initialized = true;
    }
 
    ~VariableBlockVector ()
    {
      if (this->n>0) {
        size_type i=this->n;
        while (i)
          this->p[--i].~B();
        allocator_.deallocate(this->p,this->n);
      }
      if (nblocks>0) {
        size_type i=nblocks;
        while (i)
          block[--i].~window_type();
        windowAllocator_.deallocate(block,nblocks);
      }
 
    }
 
 
    void resize (size_type _nblocks)
    {
      // deconstruct objects and deallocate memory if necessary
      if (this->n>0) {
        size_type i=this->n;
        while (i)
          this->p[--i].~B();
        allocator_.deallocate(this->p,this->n);
      }
      if (nblocks>0) {
        size_type i=nblocks;
        while (i)
          block[--i].~window_type();
        windowAllocator_.deallocate(block,nblocks);
      }
      this->n = 0;
      this->p = nullptr;
 
      // we can allocate the windows now
      nblocks = _nblocks;
      if (nblocks>0)
      {
        block = windowAllocator_.allocate(nblocks);
        new (block) window_type[nblocks];
      }
      else
      {
        nblocks = 0;
        block = nullptr;
      }
 
      // and the vector not fully usable
      initialized = false;
    }
 
    void resize (size_type _nblocks, size_type m)
    {
      // deconstruct objects and deallocate memory if necessary
      if (this->n>0) {
        size_type i=this->n;
        while (i)
          this->p[--i].~B();
        allocator_.deallocate(this->p,this->n);
      }
      if (nblocks>0) {
        size_type i=nblocks;
        while (i)
          block[--i].~window_type();
        windowAllocator_.deallocate(block,nblocks);
      }
 
      // and we can allocate the big array in the base class
      this->n = _nblocks*m;
      if (this->n>0)
      {
        this->p = allocator_.allocate(this->n);
        new (this->p)B[this->n];
      }
      else
      {
        this->n = 0;
        this->p = nullptr;
      }
 
      // we can allocate the windows now
      nblocks = _nblocks;
      if (nblocks>0)
      {
        // allocate and construct objects
        block = windowAllocator_.allocate(nblocks);
        new (block) window_type[nblocks];
 
        // set the windows into the big array
        for (size_type i=0; i<nblocks; ++i)
          block[i].set(m,this->p+(i*m));
      }
      else
      {
        nblocks = 0;
        block = nullptr;
      }
 
      // and the vector is usable
      initialized = true;
    }
 
    VariableBlockVector& operator= (const VariableBlockVector& a)
    {
      if (&a!=this)     // check if this and a are different objects
      {
        // reallocate arrays if necessary
        // Note: still the block sizes may vary !
        if (this->n!=a.n || nblocks!=a.nblocks)
        {
          // deconstruct objects and deallocate memory if necessary
          if (this->n>0) {
            size_type i=this->n;
            while (i)
              this->p[--i].~B();
            allocator_.deallocate(this->p,this->n);
          }
          if (nblocks>0) {
            size_type i=nblocks;
            while (i)
              block[--i].~window_type();
            windowAllocator_.deallocate(block,nblocks);
          }
 
          // allocate the big array in the base class
          this->n = a.n;
          if (this->n>0)
          {
            // allocate and construct objects
            this->p = allocator_.allocate(this->n);
            new (this->p)B[this->n];
          }
          else
          {
            this->n = 0;
            this->p = nullptr;
          }
 
          // we can allocate the windows now
          nblocks = a.nblocks;
          if (nblocks>0)
          {
            // alloc
            block = windowAllocator_.allocate(nblocks);
            new (block) window_type[nblocks];
          }
          else
          {
            nblocks = 0;
            block = nullptr;
          }
        }
 
        // copy block structure, might be different although
        // sizes are the same !
        if (nblocks>0)
        {
          block[0].set(a.block[0].getsize(),this->p);                 // first block
          for (size_type i=1; i<nblocks; ++i)                               // and the rest
            block[i].set(a.block[i].getsize(),block[i-1].getptr()+block[i-1].getsize());
        }
 
        // and copy the data
        for (size_type i=0; i<this->n; i++) this->p[i]=a.p[i];
      }
 
      // and we have a usable vector
      initialized = true;
 
      return *this;     // Return reference to make constructions like a=b=c; work.
    }
 
 
    //===== assignment from scalar
 
    VariableBlockVector& operator= (const field_type& k)
    {
      (static_cast<Imp::block_vector_unmanaged<B,size_type>&>(*this)) = k;
      return *this;
    }
 
 
    //===== the creation interface
 
    class CreateIterator;
 
#ifndef DOXYGEN
 
    // The window_type does not hand out a reference to its size,
    // so in order to provide a valid iterator, we need a workaround
    // to make assignment possible. This proxy enables just that by
    // implicitly converting to the stored size for read access and
    // tunneling assignment to the accessor method of the window.
    struct SizeProxy
    {
 
      operator size_type() const
      {
        return target->getsize();
      }
 
      SizeProxy& operator=(size_type size)
      {
        target->setsize(size);
        return *this;
      }
 
    private:
 
      friend class CreateIterator;
 
      SizeProxy(window_type& t)
        : target(&t)
      {}
 
      window_type* target;
    };
 
#endif // DOXYGEN
 
    class CreateIterator
    {
    public:
      using iterator_category = std::output_iterator_tag;
 
      using value_type = size_type;
 
      using difference_type = void;
 
      using pointer = size_type*;
 
      using reference = SizeProxy;
 
      CreateIterator (VariableBlockVector& _v, int _i, bool _isEnd) :
        v(_v),
        i(_i),
        isEnd(_isEnd) {}
 
      ~CreateIterator() {
        // When the iterator gets destructed, we allocate the memory
        // for the VariableBlockVector if
        // 1. the current iterator was not created as enditerator
        // 2. we're at the last block
        // 3. the vector hasn't been initialized earlier
        if (not isEnd && i==v.nblocks && not v.initialized)
          v.allocate();
      }
 
      CreateIterator& operator++()
      {
        // go to next block
        ++i;
 
        return *this;
      }
 
      CreateIterator operator++ (int)
      {
        CreateIterator tmp(*this);
        this->operator++();
        return tmp;
      }
 
      bool operator!= (const CreateIterator& it) const
      {
        return (i!=it.i) || (&v!=&it.v);
      }
 
      bool operator== (const CreateIterator& it) const
      {
        return (i==it.i) && (&v==&it.v);
      }
 
      size_type index () const
      {
        return i;
      }
 
      void setblocksize (size_type _k)
      {
        v.block[i].setsize(_k);
      }
 
#ifdef DOXYGEN
      size_type&
#else
      SizeProxy
#endif
      operator*()
      {
        return {v.block[i]};
      }
 
    private:
      VariableBlockVector& v;     // my vector
      size_type i;                      // current block to be defined
      const bool isEnd; // flag if this object was created as the end iterator.
    };
 
    // CreateIterator wants to set all the arrays ...
    friend class CreateIterator;
 
    CreateIterator createbegin ()
    {
#ifdef DUNE_ISTL_WITH_CHECKING
      if (initialized) DUNE_THROW(ISTLError,"no CreateIterator in initialized state");
#endif
      return CreateIterator(*this,0, false);
    }
 
    CreateIterator createend ()
    {
      return CreateIterator(*this,nblocks, true);
    }
 
 
    //===== access to components
    // has to be overwritten from base class because it must
    // return access to the windows
 
    window_type& operator[] (size_type i)
    {
#ifdef DUNE_ISTL_WITH_CHECKING
      if (i>=nblocks) DUNE_THROW(ISTLError,"index out of range");
#endif
      return block[i];
    }
 
    const window_type& operator[] (size_type i) const
    {
#ifdef DUNE_ISTL_WITH_CHECKING
      if (i<0 || i>=nblocks) DUNE_THROW(ISTLError,"index out of range");
#endif
      return block[i];
    }
 
    template <class T, class R>
    class RealIterator
    : public RandomAccessIteratorFacade<RealIterator<T,R>, T, R>
    {
    public:
      RealIterator ()
      {
        p = nullptr;
        i = 0;
      }
 
      RealIterator (window_type* _p, size_type _i)
      : p(_p), i(_i)
      {}
 
      void increment()
      {
        ++i;
      }
 
      void decrement()
      {
        --i;
      }
 
      bool equals (const RealIterator& it) const
      {
        return (p+i)==(it.p+it.i);
      }
 
      window_type& dereference () const
      {
        return p[i];
      }
 
      void advance(std::ptrdiff_t d)
      {
        i+=d;
      }
 
      std::ptrdiff_t distanceTo(const RealIterator& o) const
      {
        return o.i-i;
      }
 
      // Needed for operator[] of the iterator
      window_type& elementAt (std::ptrdiff_t offset) const
      {
        return p[i+offset];
      }
 
      size_type index() const
      {
        return i;
      }
 
    private:
      window_type* p;
      size_type i;
    };
 
    using Iterator = RealIterator<value_type,window_type&>;
 
    Iterator begin ()
    {
      return Iterator(block,0);
    }
 
    Iterator end ()
    {
      return Iterator(block,nblocks);
    }
 
    Iterator beforeEnd ()
    {
      return Iterator(block,nblocks-1);
    }
 
    Iterator beforeBegin () const
    {
      return Iterator(block,-1);
    }
 
    using iterator = Iterator;
 
    using ConstIterator = RealIterator<const value_type, const window_type&>;
 
    using const_iterator = ConstIterator;
 
    ConstIterator begin () const
    {
      return ConstIterator(block,0);
    }
 
    ConstIterator end () const
    {
      return ConstIterator(block,nblocks);
    }
 
    ConstIterator beforeEnd() const
    {
      return ConstIterator(block,nblocks-1);
    }
 
    ConstIterator rend () const
    {
      return ConstIterator(block,-1);
    }
 
    Iterator find (size_type i)
    {
      return Iterator(block,std::min(i,nblocks));
    }
 
    ConstIterator find (size_type i) const
    {
      return ConstIterator(block,std::min(i,nblocks));
    }
 
    //===== sizes
 
    size_type N () const
    {
      return nblocks;
    }
 
    size_type size () const
    {
      return nblocks;
    }
 
 
  private:
 
    void allocate() {
      if (this->initialized)
        DUNE_THROW(ISTLError, "Attempt to re-allocate already initialized VariableBlockVector");
 
      // calculate space needed:
      this->n=0;
      for(size_type i = 0; i < nblocks; i++) {
        this->n += block[i].size();
      }
 
      // now we can allocate the big array in the base class of v
      if (this->n>0)
      {
        // allocate and construct objects
        this->p = allocator_.allocate(this->n);
        new (this->p)B[this->n];
      }
      else
      {
        this->p = nullptr;
      }
 
      // and we set the window pointers
      this->block[0].setptr(this->p); // pointer to first block
      for (size_type j=1; j<nblocks; ++j) // and the rest
        block[j].setptr(block[j-1].getptr()+block[j-1].getsize());
 
      // and the vector is ready
      this->initialized = true;
    }
 
    size_type nblocks;            // number of blocks in vector
    window_type* block;     // array of blocks pointing to the array in the base class
    bool initialized;       // true if vector has been initialized
 
    A allocator_;
 
    typename std::allocator_traits<A>::template rebind_alloc<window_type> windowAllocator_;
  };
 
  template<class B, class A>
  struct FieldTraits< VariableBlockVector<B, A> >
  {
    typedef typename FieldTraits<B>::field_type field_type;
    typedef typename FieldTraits<B>::real_type real_type;
  };
} // end namespace
 
#endif