import pyvista
from pyvista import examples
import numpy as np
from pyvistaqt import BackgroundPlotter
import os
import datetime
import math

class RenderRPO:

    def __init__(self):
        self.timer_count = 0

        self.chaser_stl_path, self.target_stl_path = 'models/DART_STL.stl', 'models/DART_STL.stl'
        self.pl, self.earth, self.chaser_mesh, self.target_mesh = self._BuildScene()
        self.light = self._set_light_position_by_date(datetime.datetime.now())
        self.pl.add_light(self.light)

        self.earthaxis = pyvista.AxesActor()
        self.earthaxis.origin = self.earth.center
        scale = 1000
        self.earthaxis.SetTotalLength(6371+scale, 6371+scale, 6371+scale)
        self.earthaxis.GetXAxisCaptionActor2D().GetCaptionTextProperty().SetColor((1, 0, 0))
        self.earthaxis.GetYAxisCaptionActor2D().GetCaptionTextProperty().SetColor((0, 1, 0))
        self.earthaxis.GetZAxisCaptionActor2D().GetCaptionTextProperty().SetColor((0, 0, 1))
        
        self.pl.add_axes()

        print('Building scene...')
        print('The units of the scene are in km.')
        print('Scene built.')
        print('Rendering scene...')

    def _BuildScene(self):
        earth, earth_texture = self._BuildEci()

        print('Building satellites...')
        print(f'Chaser: {self.chaser_stl_path}')
        print(f'Target: {self.target_stl_path}')
        chaser_mesh, target_mesh = self._BuildSatellites(self.chaser_stl_path, self.target_stl_path)

        pl = BackgroundPlotter(lighting="none")
        cubemap = examples.download_cubemap_space_16k()
        _ = pl.add_actor(cubemap.to_skybox())
        pl.set_environment_texture(cubemap, True)

        pl.add_mesh(earth, texture=earth_texture, smooth_shading=True)
        pl.add_mesh(chaser_mesh, color='red')
        pl.add_mesh(target_mesh, color='blue')

        print('Returning plotter and objects.')
        return [pl, earth, chaser_mesh, target_mesh]

    # ... (Other methods continue in a similar fashion)