Addedd solar module and update to full degrees
This commit is contained in:
1
.gitignore
vendored
Normal file
1
.gitignore
vendored
Normal file
@@ -0,0 +1 @@
|
|||||||
|
__pycache__/
|
||||||
27
motor.py
27
motor.py
@@ -3,6 +3,9 @@ import RPi.GPIO as GPIO
|
|||||||
import time
|
import time
|
||||||
import os
|
import os
|
||||||
|
|
||||||
|
# Solar module for simulation of world
|
||||||
|
import solar
|
||||||
|
|
||||||
# Constants
|
# Constants
|
||||||
SERVO1_PIN = 18
|
SERVO1_PIN = 18
|
||||||
SERVO2_PIN = 19
|
SERVO2_PIN = 19
|
||||||
@@ -13,9 +16,9 @@ BUTTON2_FWD = 17
|
|||||||
BUTTON2_BWD = 27
|
BUTTON2_BWD = 27
|
||||||
SHUTDOWN_BTN = 26
|
SHUTDOWN_BTN = 26
|
||||||
|
|
||||||
MIN_PULSE = 1000 # In ms
|
MIN_PULSE = 500 # In ms
|
||||||
MAX_PULSE = 2000
|
MAX_PULSE = 2500
|
||||||
INIT_PULSE = 1500
|
INIT_PULSE = 1000
|
||||||
STEP = 10
|
STEP = 10
|
||||||
LOOP_DELAY = 0.01 # In seconds
|
LOOP_DELAY = 0.01 # In seconds
|
||||||
|
|
||||||
@@ -42,6 +45,24 @@ def move_servo(current, target, step=STEP):
|
|||||||
current = max(current - step, target)
|
current = max(current - step, target)
|
||||||
return current
|
return current
|
||||||
|
|
||||||
|
# Testing embedding the mirrors in the world
|
||||||
|
world = solar.World(tilt_deg=15) # The world is tilted 15 degrees around y-axis
|
||||||
|
|
||||||
|
source = solar.Source(world, pos=(100, 100, 100))
|
||||||
|
target = solar.Target(world, pos=(50, 50, 0))
|
||||||
|
|
||||||
|
# Create mirrors in a 9x9 grid
|
||||||
|
for x in range(3):
|
||||||
|
for y in range(3):
|
||||||
|
mirror = solar.Mirror(world, cluster_x=x, cluster_y=y)
|
||||||
|
world.add_mirror(mirror)
|
||||||
|
|
||||||
|
world.update_mirrors_from_source_target(source, target)
|
||||||
|
|
||||||
|
for i, mirror in enumerate(world.mirrors):
|
||||||
|
pitch, yaw = mirror.get_angles()
|
||||||
|
print(f"Mirror {i} ({mirror.cluster_x}, {mirror.cluster_y}) angles -> pitch: {pitch:.2f}°, yaw: {yaw:.2f}°")
|
||||||
|
|
||||||
# Main
|
# Main
|
||||||
try:
|
try:
|
||||||
while True:
|
while True:
|
||||||
|
|||||||
105
solar.py
Normal file
105
solar.py
Normal file
@@ -0,0 +1,105 @@
|
|||||||
|
"""Alle gemessenen Koordinaten der Quelle und der Sonne haben den Ursprung in der linken unteren Ecke des Clusters in einem rechtshaendigen flachen System.
|
||||||
|
"""
|
||||||
|
|
||||||
|
import math
|
||||||
|
|
||||||
|
class MovingEntity:
|
||||||
|
"""Embedded entity in the world with a position."""
|
||||||
|
|
||||||
|
def __init__(self, world):
|
||||||
|
self.world = world
|
||||||
|
self.pos = (0.0, 0.0, 0.0) # (x, y, z) in local untilted coordinates
|
||||||
|
|
||||||
|
def get_pos_rotated(self):
|
||||||
|
"""Return position rotated by world's tilt around y-axis."""
|
||||||
|
return self.world.rotate_point_y(self.pos)
|
||||||
|
|
||||||
|
class Target(MovingEntity):
|
||||||
|
def __init__(self, world, pos=(0.0, 0.0, 0.0)):
|
||||||
|
super().__init__(world)
|
||||||
|
self.pos = pos
|
||||||
|
|
||||||
|
class Source(MovingEntity):
|
||||||
|
def __init__(self, world, pos=(10.0, 10.0, 10.0)):
|
||||||
|
super().__init__(world)
|
||||||
|
self.pos = pos
|
||||||
|
|
||||||
|
class Mirror:
|
||||||
|
def __init__(self, world, cluster_x=0, cluster_y=0):
|
||||||
|
self.world = world
|
||||||
|
self.cluster_x = cluster_x
|
||||||
|
self.cluster_y = cluster_y
|
||||||
|
self.angle_x = 0.0
|
||||||
|
self.angle_y = 0.0
|
||||||
|
|
||||||
|
# Position in un-tilted coordinate system
|
||||||
|
self.pos = (cluster_x * self.world.grid_size,
|
||||||
|
cluster_y * self.world.grid_size,
|
||||||
|
0.0)
|
||||||
|
|
||||||
|
def get_pos_rotated(self):
|
||||||
|
return self.world.rotate_point_y(self.pos)
|
||||||
|
|
||||||
|
def set_angle_from_source_target(self, source: Source, target: Target):
|
||||||
|
# Get rotated positions
|
||||||
|
pos_mirror = self.get_pos_rotated()
|
||||||
|
pos_source = source.get_pos_rotated()
|
||||||
|
pos_target = target.get_pos_rotated()
|
||||||
|
|
||||||
|
v_source = (
|
||||||
|
pos_source[0] - pos_mirror[0],
|
||||||
|
pos_source[1] - pos_mirror[1],
|
||||||
|
pos_source[2] - pos_mirror[2],
|
||||||
|
)
|
||||||
|
v_target = (
|
||||||
|
pos_target[0] - pos_mirror[0],
|
||||||
|
pos_target[1] - pos_mirror[1],
|
||||||
|
pos_target[2] - pos_mirror[2],
|
||||||
|
)
|
||||||
|
|
||||||
|
def normalize(v):
|
||||||
|
length = math.sqrt(v[0] ** 2 + v[1] ** 2 + v[2] ** 2)
|
||||||
|
if length == 0:
|
||||||
|
return (0, 0, 0)
|
||||||
|
return (v[0] / length, v[1] / length, v[2] / length)
|
||||||
|
|
||||||
|
v_source_n = normalize(v_source)
|
||||||
|
v_target_n = normalize(v_target)
|
||||||
|
|
||||||
|
mirror_normal = (
|
||||||
|
v_source_n[0] + v_target_n[0],
|
||||||
|
v_source_n[1] + v_target_n[1],
|
||||||
|
v_source_n[2] + v_target_n[2],
|
||||||
|
)
|
||||||
|
mirror_normal = normalize(mirror_normal)
|
||||||
|
|
||||||
|
# Update the angles based on the normals in rotated positions
|
||||||
|
self.angle_y = math.degrees(math.atan2(mirror_normal[0], mirror_normal[2]))
|
||||||
|
self.angle_x = math.degrees(math.atan2(mirror_normal[1], mirror_normal[2]))
|
||||||
|
|
||||||
|
def get_angles(self):
|
||||||
|
return self.angle_x, self.angle_y
|
||||||
|
|
||||||
|
class World:
|
||||||
|
def __init__(self, tilt_deg=0.0):
|
||||||
|
self.grid_size = 10 # In cm
|
||||||
|
self.tilt_deg = tilt_deg # Tilt of the grid system around y-axis
|
||||||
|
self.mirrors = []
|
||||||
|
|
||||||
|
def add_mirror(self, mirror):
|
||||||
|
self.mirrors.append(mirror)
|
||||||
|
|
||||||
|
def update_mirrors_from_source_target(self, source: Source, target: Target):
|
||||||
|
for mirror in self.mirrors:
|
||||||
|
mirror.set_angle_from_source_target(source, target)
|
||||||
|
|
||||||
|
def rotate_point_y(self, point):
|
||||||
|
"""Rotate a point around the y-axis by the world's tilt angle."""
|
||||||
|
x, y, z = point
|
||||||
|
theta = math.radians(self.tilt_deg)
|
||||||
|
cos_t = math.cos(theta)
|
||||||
|
sin_t = math.sin(theta)
|
||||||
|
x_rot = x * cos_t + z * sin_t
|
||||||
|
y_rot = y
|
||||||
|
z_rot = -x * sin_t + z * cos_t
|
||||||
|
return (x_rot, y_rot, z_rot)
|
||||||
Reference in New Issue
Block a user