Array.h

// Copyright (C) 2009-2012 Garth N. Wells
//
// This file is part of DOLFIN.
//
// DOLFIN is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// DOLFIN is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with DOLFIN. If not, see <http://www.gnu.org/licenses/>.
//
// Modified by Anders Logg 2010-2011
// Modified by Joachim B Haga 2012
//
// First added:  2009-12-06
// Last changed: 2012-03-12

#ifndef __DOLFIN_ARRAY_H
#define __DOLFIN_ARRAY_H

#include <cstddef>
#include <sstream>
#include <string>

#include <dolfin/common/constants.h>
#include <dolfin/log/log.h>

namespace dolfin
{


  template <typename T> class Array
  {

  public:

    explicit Array(std::size_t N) : _size(N), _x(new T[N]), _owner(true) {}

    Array(std::size_t N, T* x) : _size(N), _x(x), _owner(false) {}

    ~Array()
    {
      if (_owner)
        delete [] _x;
    }

    std::string str(bool verbose) const
    {
      std::stringstream s;

      if (verbose)
      {
        s << str(false) << std::endl << std::endl;

        for (std::size_t i = 0; i < size(); i++)
          s << i << ": " << (*this)[i] << std::endl;
      }
      else
        s << "<Array<T> of size " << size() << ">";

      return s.str();
    }

    std::size_t size() const
    { return _size; }

    const T& operator[] (std::size_t i) const
    { dolfin_assert(i < _size); return _x[i]; }

    T& operator[] (std::size_t i)
    { dolfin_assert(i < _size); return _x[i]; }

    const T* data() const
    { return _x; }

    T* data()
    { return _x; }

   private:

    Array(const Array& other) = delete;

  private:

    const std::size_t _size;

    T* _x;

    bool _owner;

  };

}

#endif