added addition to cvec3

David Mayerich
1 parent b28f312d
Showing 1 changed file with 13 additions and 0 deletions Show diff stats
stim/math/vec3.h
@@ -312,6 +312,19 @@ std::string str() const{
  
 			return result;
 		}
+
+		/// @param rhs is the right-hand-side operator for the addition
+		CUDA_CALLABLE cvec3<T> operator+(cvec3<T> rhs) const {
+			return vec3< std::complex<T> >::operator+(rhs);
+		}
+		/// @param rhs is the right-hand-side operator for adding a real vector to a complex vector
+		CUDA_CALLABLE cvec3<T> operator+(vec3<T> rhs) const {
+			cvec3<T> result;
+			result.ptr[0] = vec3< std::complex<T> >::ptr[0] + rhs[0];
+			result.ptr[1] = vec3< std::complex<T> >::ptr[1] + rhs[1];
+			result.ptr[2] = vec3< std::complex<T> >::ptr[2] + rhs[2];
+			return result;
+		}
 	};						//end class cvec3
 }							//end namespace tira