basetensor.h

Go to the documentation of this file.

/*
  This file is part of MADNESS.

  Copyright (C) 2007,2010 Oak Ridge National Laboratory

  This program 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 2 of the License, or
  (at your option) any later version.

  This program 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 this program; if not, write to the Free Software
  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA

  For more information please contact:

  Robert J. Harrison
  Oak Ridge National Laboratory
  One Bethel Valley Road
  P.O. Box 2008, MS-6367

  email: harrisonrj@ornl.gov
  tel:   865-241-3937
  fax:   865-572-0680


  $Id$
*/


#ifndef MADNESS_TENSOR_BASETENSOR_H__INCLUDED
#define MADNESS_TENSOR_BASETENSOR_H__INCLUDED


#include <complex>

typedef std::complex<float> float_complex;
typedef std::complex<double> double_complex;

// These probably have to be included in this order
#include <madness/tensor/tensor_macros.h>
#include <madness/tensor/type_data.h>
#include <madness/tensor/slice.h>
#include <madness/tensor/vector_factory.h>

#include <madness/madness_config.h>
#include <madness/tensor/slice.h>
#include <madness/tensor/tensor_macros.h>

#ifdef TENSOR_INSTANCE_COUNT
#include <madness/world/atomicint.h>
#endif

namespace madness {
    class BaseTensor {
    private:
#ifdef TENSOR_INSTANCE_COUNT
        static madness::AtomicInt instance_count; 
#endif

    protected:

        long _size;                     
        long _ndim;                     
        long _id;                       
        long _dim[TENSOR_MAXDIM];       
        long _stride[TENSOR_MAXDIM];     

        void set_dims_and_size(long nd, const long d[]) {
            _ndim = nd;
            _size = 1;
            if (_ndim < 0) _size=0;
            for (long i=_ndim-1; i>=0; --i) {
                _dim[i] = d[i];
                _stride[i] = _size;
                _size *= d[i];
            }
            for (long i=std::max(_ndim,0L); i<TENSOR_MAXDIM; ++i) { // So can iterate over missing dimensions
                _dim[i] = 1;
                _stride[i] = 0;
            }
        }

    public:

        BaseTensor() : _size(0), _ndim(-1) {
#ifdef TENSOR_INSTANCE_COUNT
            instance_count++;
#endif
        }

        virtual ~BaseTensor() {
#ifdef TENSOR_INSTANCE_COUNT
            instance_count--;
#endif
        }

        static inline int get_instance_count() {
#ifdef TENSOR_INSTANCE_COUNT
            return instance_count;
#else
            return 0;
#endif
        }

        long size() const {return _size;}

        long id() const {return _id;}

        long ndim() const {return _ndim;}

        long dim(int i) const {return _dim[i];}

        long stride(int i) const {return _stride[i];}

        const long* dims() const {return _dim;}

        const long* strides() const {return _stride;}

        bool conforms(const BaseTensor *t) const {
            if (_ndim != t->_ndim) return false;
            for (long i=0; i<_ndim; ++i) {
                if (_dim[i] != t->_dim[i]) return false;
            }
            return true;
        }

        bool iscontiguous() const {
            if (_size <= 0) return true;
            long sz = 1;
            for (long i=_ndim-1; i>=0; --i) {
                if (_stride[i] != sz) return false;
                sz *= _dim[i];
            }
            return true;
        }

    protected:

        void reshape_inplace(const std::vector<long>& d);

        void reshape_inplace(int ndimnew, const long* d);

        void flat_inplace();

        void splitdim_inplace(long i, long dimi0, long dimi1);

        void fusedim_inplace(long i);

        void swapdim_inplace(long i, long j);

        void cycledim_inplace(long shift, long start, long end);

        void mapdim_inplace(const std::vector<long>& map);
    };

}

#endif // MADNESS_TENSOR_BASETENSOR_H__INCLUDED