cool_lines/scene_data.py at main · BramVR/cool_lines · 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
124
125
126
127
128
129
130
131
132
133
134
135
136
import json
from maya import cmds
import pprint

from PySide2.QtWidgets import *
from PySide2.QtGui import *
from PySide2.QtCore import *


class GlobalSceneData(QObject):
    request_ui_sync = Signal()
    request_ui_clear = Signal()

    def __init__(self, *args, **kwargs):
        super(GlobalSceneData, self).__init__()

        self.global_lines_data= {}
        self.scene_data_node= None
        self.scene_data_node_default_name= 'CoolLines_SceneData'

        self.checkSceneDataNode()


    def checkSceneDataNode(self):

        if cmds.objExists(self.scene_data_node_default_name):
            print('SceneDataNode found!')
            self.scene_data_node= self.scene_data_node_default_name
            self.loadSceneData()
            return True

        else:
            print('Cant find data! Creating a new node so you can draw... :)')
            self.createSceneDataNode()
            return False


    def createSceneDataNode(self):
        # Create Node to store data. Data will be empty at first.
        self.scene_data_node= cmds.scriptNode( st=0, bs='', n=self.scene_data_node_default_name, stp='python')


    def addLineData(self, line_data):
        self.global_lines_data[line_data["name"]]= line_data
        self.request_ui_sync.emit()
        self.saveSceneData()


    def getGlobalLinesData(self):
        return self.global_lines_data


    def removeLineData(self, line_key):
        self.global_lines_data.pop(line_key)
        self.saveSceneData()
        self.request_ui_sync.emit()


    def clearGlobalLinesData(self):
        self.global_lines_data.clear()
        self.saveSceneData()
        self.request_ui_clear.emit()


    def updateLineData(self, line_key, new_line_data, update_name= False):

        print('*********')
        print('Updating Line Data:')

        if update_name:
            self.global_lines_data.pop(line_key)
            self.global_lines_data[new_line_data["name"]]= new_line_data
            pprint.pprint(self.global_lines_data[new_line_data["name"]])

        else:
            self.global_lines_data[line_key]= new_line_data
            pprint.pprint(self.global_lines_data[line_key])

        self.saveSceneData()
        self.request_ui_sync.emit()


    def saveSceneData(self):
        #Dump data to node:
        if self.scene_data_node:
            cmds.scriptNode(self.scene_data_node, e=True, bs=json.dumps(self.global_lines_data))
        else:
            cmds.error("Cannot save scene data because there is no scene data node..")


    def loadSceneData(self):
        #Attempt Loading data from node:
        if self.scene_data_node:
            scene_data_str= cmds.scriptNode(self.scene_data_node, q=True, bs=True)
        else:
            cmds.error("Cannot load scene data because there is no scene data node..")
            return


        if scene_data_str:
            scene_data= json.loads(scene_data_str)

            print('******************')
            print('Loaded Data:')
            print(scene_data)

            self.rebuildData(scene_data)
            self.request_ui_sync.emit()
        else:
            print("Skipping loading, data empty !")
            self.global_lines_data.clear()
            self.request_ui_clear.emit()


    def rebuildData(self, scene_data):

        for current_line_name in list(scene_data.keys()):
            # Safe data rebuilding:
            if cmds.objExists(scene_data[current_line_name]["group"]):
                self.global_lines_data[current_line_name]= dict(scene_data[current_line_name])
            else:
                print(f"Cannot find {current_line_name}'s Data ! Removing...")


        print('Data rebuilt successfully')
        self.saveSceneData()


    def detectSceneChange(self):

        # A new scene has been opened..
        if self.checkSceneDataNode():
            pass
        else:
            self.global_lines_data.clear() # if create new scene while script still opened
            self.request_ui_clear.emit()