104 lines
2.6 KiB
C++
104 lines
2.6 KiB
C++
|
// Codetastic 2024
|
||
|
|
// ChatGPT was used to create this.
|
||
|
|
|
||
|
|
#ifndef PatternSphereSpin_H
|
||
|
|
#define PatternSphereSpin_H
|
||
|
|
|
||
|
|
#include <vector>
|
||
|
|
#include <cmath>
|
||
|
|
|
||
|
|
struct Point3D {
|
||
|
|
float x, y, z;
|
||
|
|
};
|
||
|
|
|
||
|
|
struct Point2D {
|
||
|
|
int x, y;
|
||
|
|
};
|
||
|
|
|
||
|
|
class PatternSphereSpin : public Drawable {
|
||
|
|
|
||
|
|
private:
|
||
|
|
|
||
|
|
std::vector<Point3D> points;
|
||
|
|
|
||
|
|
float angleX = 0.0f;
|
||
|
|
float angleY = 0.0f;
|
||
|
|
float distance = 3.0f;
|
||
|
|
|
||
|
|
|
||
|
|
Point2D project(Point3D point, float distance) {
|
||
|
|
Point2D projected;
|
||
|
|
projected.x = static_cast<int>((point.x / (distance - point.z)) * VPANEL_W / 2 + VPANEL_W / 2);
|
||
|
|
projected.y = static_cast<int>((point.y / (distance - point.z)) * VPANEL_H / 2 + VPANEL_H / 2);
|
||
|
|
return projected;
|
||
|
|
}
|
||
|
|
|
||
|
|
std::vector<Point3D> generateSpherePoints(int numPoints) {
|
||
|
|
std::vector<Point3D> points;
|
||
|
|
for (int i = 0; i < numPoints; ++i) {
|
||
|
|
float theta = 2 * PI * (i / static_cast<float>(numPoints));
|
||
|
|
for (int j = 0; j < numPoints / 2; ++j) {
|
||
|
|
float phi = PI * (j / static_cast<float>(numPoints / 2));
|
||
|
|
points.push_back({ cos(theta) * sin(phi), sin(theta) * sin(phi), cos(phi) });
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return points;
|
||
|
|
}
|
||
|
|
|
||
|
|
void rotatePoints(std::vector<Point3D>& points, float angleX, float angleY) {
|
||
|
|
for (auto& point : points) {
|
||
|
|
// Rotate around the X axis
|
||
|
|
float y = point.y * cos(angleX) - point.z * sin(angleX);
|
||
|
|
float z = point.y * sin(angleX) + point.z * cos(angleX);
|
||
|
|
point.y = y;
|
||
|
|
point.z = z;
|
||
|
|
|
||
|
|
// Rotate around the Y axis
|
||
|
|
float x = point.x * cos(angleY) + point.z * sin(angleY);
|
||
|
|
z = -point.x * sin(angleY) + point.z * cos(angleY);
|
||
|
|
point.x = x;
|
||
|
|
point.z = z;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
public:
|
||
|
|
PatternSphereSpin() {
|
||
|
|
name = (char *)"Sphere Spin";
|
||
|
|
}
|
||
|
|
|
||
|
|
void start()
|
||
|
|
{
|
||
|
|
points = generateSpherePoints(30);
|
||
|
|
angleX = 0.0f;
|
||
|
|
angleY = 0.0f;
|
||
|
|
distance = 3.0f;
|
||
|
|
};
|
||
|
|
|
||
|
|
unsigned int drawFrame() {
|
||
|
|
|
||
|
|
effects.ClearFrame();
|
||
|
|
|
||
|
|
// Rotate points
|
||
|
|
rotatePoints(points, angleX, angleY);
|
||
|
|
|
||
|
|
// Project and draw lines between points
|
||
|
|
for (size_t i = 0; i < points.size(); ++i) {
|
||
|
|
Point2D p1 = project(points[i], distance);
|
||
|
|
for (size_t j = i + 1; j < points.size(); ++j) {
|
||
|
|
Point2D p2 = project(points[j], distance);
|
||
|
|
effects.drawLine(p1.x, p1.y, p2.x, p2.y, CRGB(255,255,255));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Update angles for rotation
|
||
|
|
angleX += 0.05f;
|
||
|
|
angleY += 0.05f;
|
||
|
|
|
||
|
|
effects.ShowFrame();
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif
|