Procedural-Village-Generation/GameObject.py at main · pramesh025/Procedural-Village-Generation · 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

import numpy as np
import pygame as pg
import pyrr


class GameObject:
    def __init__(self, name = "DefaultObject", position = [0,0,0], eulers = [0,0,0], eulerVelocity = [0,0,0], mesh = None, material = None, parent = None):
        self.name = name
        self.position = np.array(position, dtype=np.float16)
        self.eulers = np.array(eulers, dtype=np.float16)
        self.eulerVelocity = np.array(eulerVelocity, dtype=np.float16)
        self.setMesh(mesh)
        self.setMaterial(material)
        self.setParent(parent)
        self.children = []

        self.eulermat = pyrr.matrix44.create_from_eulers(eulers = np.radians(self.eulers), dtype=np.float16)
        self.posmat = pyrr.matrix44.create_from_translation(vec=np.array(self.position),dtype=np.float16)

    def setPosition(self, iposition):
        self.position = np.array(iposition, dtype=np.float16)
        for child in self.children:
            child.setPosition([iposition[i] + child.position[i] for i in range(3)])
        self.posmat = pyrr.matrix44.create_from_translation(vec=np.array(self.position),dtype=np.float16)


    def setEulers(self, ieulers):
        self.eulers = np.array(ieulers, dtype=np.float16)
        for child in self.children:
            child.setEulers([ieulers[i] + child.eulers[i] for i in range(3)])
        self.eulermat = pyrr.matrix44.create_from_eulers(eulers = np.radians(self.eulers), dtype=np.float16)

    def setOrientation(self, orientation):
        self.eulers[1] = 90*orientation
        for child in self.children:
            child.setChildOrientation(orientation, self.eulers[1] + child.eulers[1])
        self.eulermat = pyrr.matrix44.create_from_eulers(eulers = np.radians(self.eulers), dtype=np.float16)

    def setChildOrientation(self, orientation, euler1):
        self.eulers[1] = euler1
        if orientation == 1:
            t = self.position[0]
            self.position[0] = self.position[1]
            self.position[1] = -t
        elif orientation == 2:
            self.position[0] = - self.position[0]
            self.position[1] = - self.position[1]
        elif orientation == 3:
            t = self.position[0]
            self.position[0] = -self.position[1]
            self.position[1] = t
        self.eulermat = pyrr.matrix44.create_from_eulers(eulers = np.radians(self.eulers), dtype=np.float16)


    def setMesh(self, mesh):
        self.mesh = mesh


    def setMaterial(self, material):
        self.material = material

    def setParent(self, parentGameObject):
        self.parent = parentGameObject

    def setChild(self, childGameObject):
        self.children.append(childGameObject)
        childGameObject.parent = self


    def placeGameObject(self):
        if self.mesh != None:
            self.mesh.assignGameObject(self)

        self.mesh.transformsCreated = False
        for child in self.children:
            child.placeGameObject()
        pass

    def unplaceGameObject(self):
        for child in self.children:
            child.unplaceGameObject()
        self.mesh.transformscreated = False
        if self.mesh != None:
            self.mesh.unassignGameObject(self)
        del self
        pass