Parabolic_Potential.h

#ifndef QNANO_NEW_PARABOLIC_POTENTIAL_DEFINED_H
#define QNANO_NEW_PARABOLIC_POTENTIAL_DEFINED_H

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

class Parabolic_Potential: public Generate_Potential{
public:
  double strength;
  Vec3d direction;
  Vec3d center;
  double truncate_radius;

  
  virtual void generate(const Atom_List &atlist);
  
  void setup(Parameter_Map& param_map);


  Parabolic_Potential(double strength_, const Vec3d &direction_, const Vec3d &center_=Vec3d(), double truncate_radius_=-1.) 
    :  strength(strength_), direction(direction_), center(center_), truncate_radius(truncate_radius_) {

  }

};


class Parabolic_Potential_2d: public Generate_Potential{
public:
  double strength;
  Vec3d direction;
  Vec3d center;
  double truncate_radius;
  bool set_truncate_offset_zero;

  virtual void setup(Parameter_Map &param_map);
  virtual void generate(const Atom_List &atlist);

  Parabolic_Potential_2d(double strength_, const Vec3d &direction_, \
         const Vec3d &center_=Vec3d(), double truncate_radius_=-1., \
         bool set_truncate_offset_zero_=false) 
         :  strength(strength_), direction(direction_), \
            center(center_), truncate_radius(truncate_radius_), \
            set_truncate_offset_zero(set_truncate_offset_zero_) {
     
  }
  Parabolic_Potential_2d(Parameter_Map &param_map){
    setup(param_map);
  }

};


#endif