simulation/cubesat_ui_panel.py at main · Bronco-Space/simulation · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
import bpy
import os
import sys
import numpy
from bpy.props import (StringProperty,
                       BoolProperty,
                       IntProperty,
                       FloatProperty,
                       FloatVectorProperty,
                       EnumProperty,
                       PointerProperty,
                       )
from bpy.types import (Panel,
                       Operator,
                       AddonPreferences,
                       PropertyGroup,
                       )

#needed to locate file containing functions
dir = os.path.dirname(bpy.data.filepath)
if not dir in sys.path:
    sys.path.append(dir)

import gravity_calc as grav
import mag_calc as mag
import magnetorquer

# Custom Properties for cubesat
# Accessible from bpy.context.scene['cubesat_props']
# Reference: https://blender.stackexchange.com/questions/35007/how-can-i-add-a-checkbox-in-the-tools-ui
class cubesat_property_group(PropertyGroup):
    # Accessible from bpy.data.scenes['Scene'].cubesat_props.logging
    logging : BoolProperty(
        name="Enable or Disable",
        description = "Enable logging",
        default = False
    )

class VIEW3D_PT_cubesat(bpy.types.Panel):
    """Creates a Panel in the scene context of the properties editor"""
    bl_label = "CubeSat Sim Controls" #Visual full name of menu
    bl_idname = "SCENE_PT_layout" #Name of menu on the backend
    bl_space_type = "VIEW_3D"   #Viewable ONLY in the 3d view port
    bl_region_type = "UI" #Specifies that it appears in the tools bar
    bl_category = "CubeSat" #Name on the side bar of the tools bar
    bl_context = "objectmode" #Viewable ONLY in object mode

    def draw(self, context):
        layout = self.layout
        scene = context.scene
        cubesat_props = scene.cubesat_props

        # Logging checkbox
        row = layout.row()
        row.prop(cubesat_props, "logging", text="Enable Logging")

	    #Reload scripts
        row = layout.row()
        row.operator("myops.reload_scripts")

        # Cubesat readout. Note: assumes cubesats name is 'cubesat'
        col = layout.column(align=True)
        col.label(text="Cubesat")
        col.prop(bpy.data.objects['cubesat'], "location")

        # Altitude
        col = layout.column(align=True)
        diffVector = bpy.data.objects['Earth'].matrix_world.translation - bpy.data.objects['cubesat'].matrix_world.translation
        col.label(text='Altitude: {:>10.3f}'.format(numpy.linalg.norm(diffVector)))

	    # Center of mass readout. Note: assumes center of mass's name is 'centerofmass'
        col = layout.column(align=True)
        col.label(text="Center of Mass")
        col.prop(bpy.data.objects['centerofmass'], "location")

        # Readout of gravitational properties
        force = grav.gravForce()
        acc = grav.gravAccel()
        col = layout.column(align=True)
        col.label(text= "Grav Force Experienced (N):")
        col.label(text=f'x:{round(force.x, 3)}')
        col.label(text=f'y:{round(force.y, 3)}')
        col.label(text=f'z:{round(force.z, 3)}')
        col.label(text= "Grav Acc Experienced (m/sec^2):")
        col.label(text=f'x:{round(acc.x, 3)}')
        col.label(text=f'y:{round(acc.y, 3)}')
        col.label(text=f'z:{round(acc.z, 3)}')

        # Readout of magnetic properties
        magf = mag.get_magnetic_force(bpy.data.objects['cubesat'].location)
        col = layout.column(align=True)
        col.label(text='North component: {:>+10.3f}'.format(magf[0]))
        col.label(text='East component: {:>+10.3f}'.format(magf[1]))
        col.label(text='Vertical component (+ve down): {:>+10.3f}'.format(magf[2]))
        col.label(text='Total intensity: {:>+10.3f} nT'.format(magf[3]))
        stuff = magnetorquer.calcTorque(magf)
        col.label(text=f'stuff: ({stuff[0]:>+.0f},{stuff[1]:>+.0f},{stuff[2]:>+.0f})')

        # Time step
        col = layout.column(align=True)
        col.label(text="Time Step:")
        col.prop(bpy.data.scenes["Scene"], "frame_current", text="Timestep (s/frame)")


#Stuff that is required for the script to work
def register():
    bpy.utils.register_class(cubesat_property_group)
    bpy.utils.register_class(VIEW3D_PT_cubesat)

    bpy.types.Scene.cubesat_props = PointerProperty(type=cubesat_property_group)


def unregister():
    bpy.utils.unregister_class(VIEW3D_PT_cubesat)
    bpy.utils.unregister_class(cubesat_property_group)

    del bpy.types.Scene.cubesat_props


if __name__ == "__main__":
    register()