added a grid sampling function to spharmonics

David Mayerich
1 parent 00a7e339
Showing 1 changed file with 11 additions and 0 deletions Show diff stats
stim/math/spharmonics.h
@@ -231,6 +231,17 @@ public:
  
 		return fx;
 	}
+
+	/// Fill an NxN grid with the spherical function for theta = [0 2pi] and phi = [0 pi]
+	void grid(T* data, size_t N){
+		double dt = stim::TAU / (double)N;			//calculate the step size in each direction
+		double dp = stim::PI / (double)(N - 1);
+		for(size_t ti = 0; ti < N; ti++){
+			for(size_t pi = 0; pi < N){
+				data[pi * N + ti] = (*this)((double)ti * dt, (double)pi * dp);
+			}
+		}
+	}
 /*
 	//overload arithmetic operations
 	spharmonics<T> operator*(T rhs) const {