array.hpp

/*
 * array.hpp
 *
 * Created on: 23 Jan 2007
 * Author: Holger Schmitz
 * Email: holger@notjustphysics.com
 *
 * Copyright 2012 Holger Schmitz
 *
 * This file is part of Schnek.
 *
 * Schnek is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Schnek 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Schnek.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

#ifndef SCHNEK_ARRAY_HPP_
#define SCHNEK_ARRAY_HPP_

#include "arraycheck.hpp"

#include "../config.hpp"
#include "../macros.hpp"

#include <initializer_list>

namespace schnek {

template<class Operator, size_t Length>
class ArrayExpression;


template<
  class T, 
  size_t Length, 
  template<size_t> class CheckingPolicy = ArrayNoArgCheck
>
class Array :
    public CheckingPolicy<Length>
{
  private:
    T data[Length];

    typedef Array<T,Length,CheckingPolicy> ThisType;
  public:
    typedef T value_type;
    static constexpr size_t length = Length;
    
    SCHNEK_INLINE Array();
    template<template<size_t> class CheckingPolicy2>
    SCHNEK_INLINE Array(const Array<T, Length, CheckingPolicy2> &);

    template<class Operator>
    SCHNEK_INLINE Array(const ArrayExpression<Operator, Length> &);

    // Array(std::initializer_list<T> l);

    SCHNEK_INLINE Array(const T&);
    SCHNEK_INLINE Array(const T&, const T&);
    SCHNEK_INLINE Array(const T&, const T&, const T&);
    SCHNEK_INLINE Array(const T&, const T&, const T&, const T&);
    SCHNEK_INLINE Array(const T&, const T&, const T&, const T&, const T&);
    SCHNEK_INLINE Array(const T&, const T&, const T&, const T&, const T&, 
               const T&);
    SCHNEK_INLINE Array(const T&, const T&, const T&, const T&, const T&, 
               const T&, const T&);
    SCHNEK_INLINE Array(const T&, const T&, const T&, const T&, const T&, 
               const T&, const T&, const T&);
    SCHNEK_INLINE Array(const T&, const T&, const T&, const T&, const T&, 
               const T&, const T&, const T&, const T&);
    SCHNEK_INLINE Array(const T&, const T&, const T&, const T&, const T&, 
               const T&, const T&, const T&, const T&, const T&);

    SCHNEK_INLINE ~Array() {}

    SCHNEK_INLINE T& operator[](size_t);
    SCHNEK_INLINE T operator[](size_t) const;

    SCHNEK_INLINE T& at(size_t);
    SCHNEK_INLINE T at(size_t) const;

  public:
    template<class T2, template <size_t> class CheckingPolicy2>
    SCHNEK_INLINE Array<T,Length,CheckingPolicy> &operator=(const Array<T2,Length,CheckingPolicy2>&);

    template<class Operator>
    SCHNEK_INLINE Array<T,Length,CheckingPolicy> &operator=(const ArrayExpression<Operator, Length> &);

    template<class T2, template <size_t> class CheckingPolicy2>
    SCHNEK_INLINE Array<T,Length,CheckingPolicy> &operator+=(const Array<T2,Length,CheckingPolicy2>&);

    template<class T2, template <size_t> class CheckingPolicy2>
    SCHNEK_INLINE Array<T,Length,CheckingPolicy> &operator*=(const Array<T2,Length,CheckingPolicy2>&);

    template<class T2, template <size_t> class CheckingPolicy2>
    SCHNEK_INLINE Array<T,Length,CheckingPolicy> &operator/=(const Array<T2,Length,CheckingPolicy2>&);

    template<class T2, template <size_t> class CheckingPolicy2>
    SCHNEK_INLINE Array<T,Length,CheckingPolicy> &operator-=(const Array<T2,Length,CheckingPolicy2>&);

    template<typename T2>
    SCHNEK_INLINE Array<T,Length,CheckingPolicy> &operator+=(const T2);

    template<typename T2>
    SCHNEK_INLINE Array<T,Length,CheckingPolicy> &operator-=(const T2);

    template<typename T2>
    SCHNEK_INLINE Array<T,Length,CheckingPolicy> &operator*=(const T2);

    template<typename T2>
    SCHNEK_INLINE Array<T,Length,CheckingPolicy> &operator/=(const T2);


    template<class Operator>
    SCHNEK_INLINE Array<T,Length,CheckingPolicy> &operator+=(const ArrayExpression<Operator, Length> &);

    template<class Operator>
    SCHNEK_INLINE Array<T,Length,CheckingPolicy> &operator-=(const ArrayExpression<Operator, Length> &);

    template<class Operator>
    SCHNEK_INLINE Array<T,Length,CheckingPolicy> &operator*=(const ArrayExpression<Operator, Length> &);

    template<class Operator>
    SCHNEK_INLINE Array<T,Length,CheckingPolicy> &operator/=(const ArrayExpression<Operator, Length> &);

  public:
    Array<T,Length,CheckingPolicy>& clear();
    Array<T,Length,CheckingPolicy>& fill(const T&);
    
    template<size_t destLength>
    Array<T,destLength,CheckingPolicy> project() const;

    Array<T,Length-1,CheckingPolicy> projectDim(size_t dim) const;


    static Array<T,Length,CheckingPolicy> Zero();
    
    static Array<T,Length,CheckingPolicy> Ones();

    T product() const;

    T sum() const;

    T sqr() const;
};



} // namespace

template<
  class T1, class T2,
  size_t Length, 
  template<size_t> class CheckingPolicy1, template<size_t> class CheckingPolicy2
>
SCHNEK_INLINE bool operator==(
  const schnek::Array<T1,Length,CheckingPolicy1>&,
  const schnek::Array<T2,Length,CheckingPolicy2>&
);

template<
  class T1, class T2,
  size_t Length, 
  template<size_t> class CheckingPolicy1, template<size_t> class CheckingPolicy2
>
SCHNEK_INLINE bool operator!=(
  const schnek::Array<T1,Length,CheckingPolicy1>&,
  const schnek::Array<T2,Length,CheckingPolicy2>&
);

template<
  class T1, class T2,
  size_t Length, 
  template<size_t> class CheckingPolicy1, template<size_t> class CheckingPolicy2
>
SCHNEK_INLINE bool operator<(
  const schnek::Array<T1,Length,CheckingPolicy1>&,
  const schnek::Array<T2,Length,CheckingPolicy2>&
);

template<
  class T1, class T2,
  size_t Length, 
  template<size_t> class CheckingPolicy1, template<size_t> class CheckingPolicy2
>
SCHNEK_INLINE bool operator<=(
  const schnek::Array<T1,Length,CheckingPolicy1>&,
  const schnek::Array<T2,Length,CheckingPolicy2>&
);


#include "array.t"

#endif // SCHNEK_ARRAY_HPP_