#include "driving.h"
#include <math.h>

_Complex double g_test(
  int kx,
  int ky
){
  //return sqrt(kx*kx*ky*ky)*exp(-(kx*kx+ky*ky));
  if(kx==2 && ky==-1){
    return 0.5+sqrt(3)/2*I;
  }
  else if(kx==-2 && ky==1){
     return 0.5-sqrt(3)/2*I;
  }
  return 0.;
}

#define UNUSED(x) (void)(x)
_Complex double g_zero(
  int kx,
  int ky
){
  // avoid unused variable warnings
  UNUSED(kx);
  UNUSED(ky);
  return 0.;
}