#include "well.h"
#include "math.h"

double well::wav1d(int k, double x) {
	if (fabs(x)>width) return 0;
	k++;
	double pi=atan(1.0)*4.0;
	double wav=x*pi/width/2;
	if (k%2) wav=cos(k*wav);
	else wav=sin(k*wav);
	wav/=sqrt(width);
	return wav;
}
	

double well::secdiv(int k) {
	k++;
	double pi=atan(1.0)*4.0;
	double wav=-k*k*pi*pi/width/width/4.0;
	return wav;
}

double well::E(int k) {
  return -secdiv(k)/2/mass;
}