Potential_from_GTE.h

#ifndef QNANO_NEW_POTENTIAL_FROM_GTE_DEFINED_H
#define QNANO_NEW_POTENTIAL_FROM_GTE_DEFINED_H

#include "tools/Tensor2.h"
#include "tools/List_Class.h"
#include "tools/Parameter_Map.h"
#include "structure/Generate_Potential.h"

class Single_GTE: public Tensor2d{
public:
  double Lx, Ly;
  double voltage, zcoord;

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


  Single_GTE &operator=(const Single_GTE &other){
    Tensor2d::operator=(other);
    Lx=other.Lx; Ly=other.Ly;
    voltage=other.voltage; zcoord=other.zcoord;
    return *this;
  }
  Single_GTE(): Tensor2d(), Lx(1.), Ly(1.), voltage(0.),zcoord(0.){ 
  }
  Single_GTE(const Single_GTE &other){
    operator=(other);
  }
  Single_GTE(const std::string &filename, double voltage_=0., double z_=0.)  
    : Lx(1.), Ly(1.), voltage(voltage_), zcoord(z_){ 
    read(filename);
  }
};

class Potential_from_GTE:  public Generate_Potential {
public:
  std::vector<Single_GTE> gates;
 
  void add_gate(const std::string &filename, double voltage, double z){
    gates.push_back(Single_GTE(filename, voltage, z));
  }
  void setup(Parameter_Map &param_map);

 
  virtual void generate(const Atom_List &atlist);

  Potential_from_GTE(){}
  Potential_from_GTE(Parameter_Map &param_map){
    setup(param_map);
  }

};



#endif