Generate_Potential.h

#ifndef QNANO_NEW_GENERATE_POTENTIAL_DEFINED_H
#define QNANO_NEW_GENERATE_POTENTIAL_DEFINED_H

#include "structure/Potential.h"
#include "structure/Atom_List.h"


class Generate_Potential: public Potential{
public:
  virtual void generate(const Atom_List &atlist) = 0;

  virtual void generate(const std::string &positionfile);

};



class Constant_Potential: public Generate_Potential{
public: 
  double strength;

  void generate(size_t size_);
  void generate(const Atom_List &atlist){ generate(atlist.size()); }
  Constant_Potential(double strength_=0, size_t size_=0) 
    : strength(strength_) {
    generate(size_);
  }
}; 


class Linear_Potential: public Generate_Potential{
public: 
  double strength;
  Vec3d direction;
  Vec3d center;

  void generate(const Atom_List &atlist);
  Linear_Potential(double strength_, const Vec3d &direction_, const Vec3d &center_=Vec3d())
    : strength(strength_),  direction(direction_), center(center_){
  }
}; 
#endif