Single_Particle_Spins.h

#ifndef QNANO_NEW_SINGLE_PARTICLE_SPINS_DEFINED_H
#define QNANO_NEW_SINGLE_PARTICLE_SPINS_DEFINED_H

#include "tools/Reader.h"

class Single_Particle_Spins{
public:
  int nr;
 
  std::complex<double> *s;

  void allocate(){ 
    s=new std::complex<double>[3*nr*nr];
  }
  void deallocate(){ 
    delete [] s;
  }
  void resize(int nr_){
    nr=nr_;
    deallocate();
    allocate();
  }

  std::complex<double> operator()(int i, int j, int k)const{
    return s[(k*nr+i)*nr+j];
  }
  std::complex<double> & operator()(int i, int j, int k){
    return s[(k*nr+i)*nr+j];
  }
  std::complex<double> get_sx(int i,int j)const{ return s[i*nr+j]; }
  std::complex<double> get_sy(int i,int j)const{ return s[(nr+i)*nr+j]; }
  std::complex<double> get_sz(int i,int j)const{ return s[(2*nr+i)*nr+j]; }

  void read(const std::string &filename);


  Single_Particle_Spins(){
    nr=0; 
    allocate();
  }
  Single_Particle_Spins(const std::string &filename){
    nr=0;
    allocate();
    read(filename);
  }
};

#endif