Tensor2.h

#ifndef QNANO_NEW_TENSOR2_DEFINED_H
#define QNANO_NEW_TENSOR2_DEFINED_H

#include <complex>
#include <iostream>
#include <cstdlib>
#include "tools/Reader.h"

template <typename T> class template_Tensor2{
public:
  //sizes: 
  size_t size[2];
  //data block:
  T *data;
  
  inline size_t get_index(size_t i, size_t j)const{
    return i*size[1]+j;
  }

public: 
  inline size_t get_totalsize()const{ return size[0]*size[1];  }

protected:

  void allocate(){  data=new T[get_totalsize()];  }
  void deallocate(){ delete[] data; }
public:
  inline size_t get_size(size_t i)const{return size[i];}
  void resize(size_t i, size_t j){
    size[0]=i; size[1]=j; 
    deallocate();
    allocate();
  }
  void set_zero(){ 
    size_t ts=get_totalsize();
    for(size_t i=0; i<ts; i++)data[i]=0.;
  }
  
  const T &operator()(size_t i, size_t j)const{
    return data[get_index(i,j)];
  }
  T &operator()(size_t i, size_t j){
    return data[get_index(i,j)];
  }
  const T &get_check(int i, int j){
    if(i<0 || i>=(int)size[0] || j<0 || j>=(int)size[1] ){
      std::cerr<<"Tensor2: out of bounds!"<<std::endl;
      exit(1);
    }
    return data[get_index(i,j)];
  }
  
  void swap_lines(size_t l1, size_t l2){
    for(size_t k=0; k<size[1]; k++){
      T tmp=data[get_index(l1,k)];
      data[get_index(l1,k)]=data[get_index(l2,k)];
      data[get_index(l2,k)]=tmp;
    }
  }
 

  //Constructors and copy operators
  void copy(const template_Tensor2 &other){
    resize(other.get_size(0),other.get_size(1));
    for(size_t i=0; i<get_totalsize(); i++){
      data[i]=other.data[i];
    }
  } 
  template_Tensor2 &operator=(const template_Tensor2 &other){
    copy(other);
    return *this;
  }
  template_Tensor2(const template_Tensor2 &other){
    copy(other);
  }
  template_Tensor2(){
    size[0]=size[1]=0;
    allocate();
  } 
  template_Tensor2(size_t i, size_t j){
    size[0]=i; size[1]=j;
    allocate();
    set_zero();
  }
  ~template_Tensor2(){
    deallocate();
  }
  

};


typedef template_Tensor2<std::complex<double> >Tensor2;
typedef template_Tensor2<double> Tensor2d;

#endif