Added some operators to the vector classes

This commit is contained in:
trotFunky 2019-10-14 18:46:32 +02:00
parent 38cc784ec8
commit bb674cdbb6

View file

@ -15,6 +15,48 @@ struct CoordinatesVector
{
T coordinates[N];
// Use SFINAE to declare the function if and only if the type is arithmetic
template <typename Scalar>
typename std::enable_if<std::is_arithmetic<Scalar>::value,CoordinatesVector<T,N>>::type
operator*(const Scalar& scalar)
{
CoordinatesVector result;
for (unsigned int i = 0;i<N;i++)
{
result.coordinates[i] = coordinates[i] * scalar;
}
}
CoordinatesVector operator+(const CoordinatesVector<T,N>& op)
{
CoordinatesVector result;
for (unsigned int i = 0;i<N;i++)
{
result.coordinates[i] = coordinates[i] + op.coordinates[i];
}
return result;
}
CoordinatesVector operator-(const CoordinatesVector<T,N>& op)
{
CoordinatesVector result;
for (unsigned int i = 0;i<N;i++)
{
result.coordinates[i] = coordinates[i] - op.coordinates[i];
}
return result;
}
CoordinatesVector& operator=(const CoordinatesVector<T,N>& original)
{
if (this != &original)
{
std::copy(std::begin(original.coordinates), std::end(original.coordinates), std::begin(coordinates));
}
return *this;
}
friend std::fstream& operator>>(std::fstream& stream, CoordinatesVector<T,N>& vector)
{
for (unsigned int i = 0;i<N;i++)
@ -33,6 +75,45 @@ struct CoordinatesVector<T,2>
T& x = coordinates[0];
T& y = coordinates[1];
// Use SFINAE to declare the function if and only if the type is a scalar
template <typename Scalar>
typename std::enable_if<std::is_arithmetic<Scalar>::value,CoordinatesVector<T,2>>::type
operator*(const Scalar& scalar)
{
CoordinatesVector result;
for (unsigned int i = 0;i<2;i++)
{
result.coordinates[i] = coordinates[i] * scalar;
}
}
CoordinatesVector operator+(const CoordinatesVector<T,2>& op)
{
CoordinatesVector result;
result.x = x + op.x;
result.y = y + op.y;
return result;
}
CoordinatesVector operator-(const CoordinatesVector<T,2>& op)
{
CoordinatesVector<T,2> result;
result.x = x - op.x;
result.y = y - op.y;
return result;
}
CoordinatesVector& operator=(const CoordinatesVector<T,2>& original)
{
if (this != &original)
{
std::copy(std::begin(original.coordinates), std::end(original.coordinates), std::begin(coordinates));
}
return *this;
}
friend std::fstream& operator>>(std::fstream& stream, CoordinatesVector<T,2>& vec2)
{
stream >> vec2.x >> vec2.y;
@ -49,6 +130,47 @@ struct CoordinatesVector<T,3>
T& y = coordinates[1];
T& z = coordinates[2];
// Use SFINAE to declare the function if and only if the type is a scalar
template <typename Scalar>
typename std::enable_if<std::is_arithmetic<Scalar>::value,CoordinatesVector<T,3>>::type
operator*(const Scalar& scalar)
{
CoordinatesVector result;
for (unsigned int i = 0;i<3;i++)
{
result.coordinates[i] = coordinates[i] * scalar;
}
}
CoordinatesVector operator+(const CoordinatesVector<T,3>& op)
{
CoordinatesVector result;
result.x = x + op.x;
result.y = y + op.y;
result.z = z + op.z;
return result;
}
CoordinatesVector operator-(const CoordinatesVector<T,3>& op)
{
CoordinatesVector<T,3> result;
result.x = x - op.x;
result.y = y - op.y;
result.z = z - op.z;
return result;
}
CoordinatesVector& operator=(const CoordinatesVector<T,3>& original)
{
if (this != &original)
{
std::copy(std::begin(original.coordinates), std::end(original.coordinates), std::begin(coordinates));
}
return *this;
}
friend std::fstream& operator>>(std::fstream& stream, CoordinatesVector<T,3>& vec3)
{
stream >> vec3.x >> vec3.y >> vec3.z;