/*
*
* Template Numerical Toolkit (TNT): One-dimensional Fortran numerical array
*
* Mathematical and Computational Sciences Division
* National Institute of Technology,
* Gaithersburg, MD USA
*
*
* This software was developed at the National Institute of Standards and
* Technology (NIST) by employees of the Federal Government in the course
* of their official duties. Pursuant to title 17 Section 105 of the
* United States Code, this software is not subject to copyright protection
* and is in the public domain. NIST assumes no responsibility whatsoever for
* its use by other parties, and makes no guarantees, expressed or implied,
* about its quality, reliability, or any other characteristic.
*
*/



#ifndef TNT_FORTRAN_ARRAY1D_H
#define TNT_FORTRAN_ARRAY1D_H

#include <cstdlib>
#include <iostream>
#ifdef TNT_BOUNDS_CHECK
#include <assert.h>
#endif
#include "tnt_array1d.h"

namespace TNT
{

/**
    Templated one-dimensional, numerical array which
    looks like a conventional Fortran array.  This is
    useful when integrating with C/C++ codes translated
    from Fortran.  Indexing is via the A(i) notation and A(1)
    is the first element.

    <p>
    Array assignment is by reference (i.e. shallow assignment).
    That is, B=A implies that the A and B point to the
    same array, so modifications to the elements of A
    will be reflected in B. If an independent copy
    is required, then B = A.copy() can be used.  Note
    that this facilitates returning arrays from functions
    without relying on compiler optimizations to eliminate
    extensive data copying.

    <p>
    This class employs its own garbage collection via
    the use of reference counts.  That is, whenever
    an internal array storage no longer has any references
    to it, it is destoryed.
*/
template <class T>
class Fortran_Array1D
{


  private:
          Array1D<T> A_;

  public:

    typedef         T   value_type;

           Fortran_Array1D();
           Fortran_Array1D(int n);
           Fortran_Array1D(int n,  T *a);
           Fortran_Array1D(int n, const T &a);
    inline Fortran_Array1D(const Fortran_Array1D &A);
    inline Fortran_Array1D & operator=(const T &a);
    inline Fortran_Array1D & operator=(const Fortran_Array1D &A);
    inline Fortran_Array1D & ref(const Fortran_Array1D &A);
           Fortran_Array1D copy();
           Fortran_Array1D & inject(const Fortran_Array1D & A);
    inline T& operator()(int i);
    inline const T& operator()(int i) const ;
    inline int dim() const;
    inline int dim1() const;
    inline int ref_count() const;
               ~Fortran_Array1D();


};

/**
    Create a null (0-length) array.
*/
template <class T>
Fortran_Array1D<T>::Fortran_Array1D() : A_() {}


/**
    Copy constructor. Array data is NOT copied, but shared.
    Thus, in Fortran_Array1D B(A), subsequent changes to A will
    be reflected in B.  For an indepent copy of A, use
    Fortran_Array1D B(A.copy()), or B = A.copy(), instead.
*/
template <class T>
Fortran_Array1D<T>::Fortran_Array1D(const Fortran_Array1D<T> &A) : A_(A.A_) {}



/**
    Create a new (n) array, WIHOUT initializing array elements.
    To create an initialized array of constants, see Fortran_Array1D(n, value).

    <p>
    This version avoids the O(n) initialization overhead and
    is used just before manual assignment.

    @param m the dimension of the new matrix.
*/
template <class T>
Fortran_Array1D<T>::Fortran_Array1D(int n) : A_(n) {}



/**
    Create a new n-length array,  initializing array elements to
    constant specified by argument.  Most often used to
    create an array of zeros, as in A(n, 0.0).

    @param m the dimension of the new matrix.
    @param val the constant value to set all elements of the new array to.
*/
template <class T>
Fortran_Array1D<T>::Fortran_Array1D(int n, const T &val) : A_(n, val) {}


/**
    Create a new  n-length array,  as a view of an existing one-dimensional
    C array.  (Note that the storage for this pre-existing array will
    never be garbage collected by the Fortran_Array1D class.)

    @param m the first dimension of the new matrix.
    @param n the second dimension of the new matrix.
    @param a the one dimensional C array to use as data storage for
        the array.
*/
template <class T>
Fortran_Array1D<T>::Fortran_Array1D(int n, T *a) : A_(n, a) {}




/**
    Elements are accessed via A(i) indexing.

    If TNT_BOUNDS_CHECK macro is defined, the indices
    are checked that they falls within the array bounds (via the
    assert() macro.)
*/
template <class T>
inline T& Fortran_Array1D<T>::operator()(int i)
{
#ifdef TNT_BOUNDS_CHECK
    assert(i >= 1);
    assert(i <= A_.dim1());
#endif

    return A_[i-1];

}

/**
    Read-only version of A(i,j) indexing.

    If TNT_BOUNDS_CHECK macro is defined, the indices
    are checked that they falls within the array bounds (via the
    assert() macro.)
*/
template <class T>
inline const T& Fortran_Array1D<T>::operator()(int i)  const
{
#ifdef TNT_BOUNDS_CHECK
    assert(i >= 1);
    assert(i <= A_.dim1());
#endif

    return A_[i-1];

}


/**
    Assign all elemnts of A to a constant scalar.
*/
template <class T>
Fortran_Array1D<T> & Fortran_Array1D<T>::operator=(const T &a)
{
    A_ = a;
    return *this;
}
/**
    Create a new of existing matrix.  Used in B = A.copy()
    or in the construction of B, e.g. Fortran_Array1D B(A.copy()),
    to create a new array that does not share data.

*/
template <class T>
Fortran_Array1D<T> Fortran_Array1D<T>::copy()
{

    Fortran_Array1D B;

    B.A_= A_.copy();
    return B;
}


/**
    Copy the elements to from one array to another, in place.
    That is B.inject(A), both A and B must conform (i.e. have
    identical dimensions).

    This differs from B = A.copy() in that references to B
    before this assignment are also affected.  That is, if
    we have
    <pre>
    Fortran_Array1D A(n);
    Fortran_Array1D C(n);
    Fortran_Array1D B(C);        // elements of B and C are shared.

</pre>
    then B.inject(A) affects both and C, while B=A.copy() creates
    a new array B which shares no data with C or A.

    @param A the array from elements will be copied
    @return an instance of the modifed array. That is, in B.inject(A),
    it returns B.  If A and B are not conformat, no modifications to
    B are made.

*/
template <class T>
Fortran_Array1D<T> & Fortran_Array1D<T>::inject(const Fortran_Array1D &A)
{
    A_.inject(A.A_);

    return *this;
}





/**
    Create a reference (shallow assignment) to another existing array.
    In B.ref(A), B and A shared the same data and subsequent changes
    to the array elements of one will be reflected in the other.
    <p>
    This is what operator= calls, and B=A and B.ref(A) are equivalent
    operations.

    @return The new referenced array: in B.ref(A), it returns B.
*/
template <class T>
Fortran_Array1D<T> & Fortran_Array1D<T>::ref(const Fortran_Array1D<T> &A)
{
    A_.ref(A.A_);
    return *this;
}

/**
    B = A is shorthand notation for B.ref(A).
*/
template <class T>
Fortran_Array1D<T> & Fortran_Array1D<T>::operator=(const Fortran_Array1D<T> &A)
{
    return ref(A);
}

/**
    @return the size (dimension of) the array.
*/
template <class T>
inline int Fortran_Array1D<T>::dim1() const { return A_.dim1(); }


/**
    @return the size (dimension of) the array.
*/
template <class T>
inline int Fortran_Array1D<T>::dim() const { return A_.dim(); }




/**
    @return the number of arrays that share the same storage area
    as this one.  (Must be at least one.)
*/
template <class T>
inline int Fortran_Array1D<T>::ref_count() const { return A_.ref_count(); }

template <class T>
Fortran_Array1D<T>::~Fortran_Array1D()
{
}




} /* namespace TNT */

#endif
/* TNT_FORTRAN_ARRAY1D_H */
