@@ -7,11 +7,10 @@ from protoc_base_plugin.builder import build_tree, inject_docs
7
7
8
8
9
9
class UnityObjectParser (UnityParser ):
10
-
11
10
null_checks = {
12
11
# "StringVariable": "proto.{name} != default(string)",
13
12
"BoolVariable" : "true" ,
14
- "ByteArrayVariable" : "!proto.{name}.IsEmpty"
13
+ "ByteArrayVariable" : "!proto.{name}.IsEmpty" ,
15
14
# "UnsignedIntVariable": "proto.{name} != default(uint)",
16
15
# "UnsignedLongVariable": "proto.{name} != default(ulong)",
17
16
# "IntVariable": "proto.{name} != default(int)",
@@ -95,9 +94,7 @@ public {list_type} {name};
95
94
if field ["label" ] == "LABEL_REPEATED" :
96
95
if field ["type" ] == "ByteArrayVariable" :
97
96
value_setter = "value.ToByteArray()"
98
- value_getter = (
99
- "ByteString.CopyFrom(value.GetValue(), 0, value.GetValue().Length)"
100
- )
97
+ value_getter = "ByteString.CopyFrom(value.GetValue(), 0, value.GetValue().Length)"
101
98
else :
102
99
value_setter = "value"
103
100
value_getter = "value.GetValue()"
@@ -115,9 +112,7 @@ public {list_type} {name};
115
112
}}
116
113
{name}.Raise();
117
114
}}
118
- """ .format (
119
- name = field ["name" ], type = field ["type" ], value_setter = value_setter
120
- )
115
+ """ .format (name = field ["name" ], type = field ["type" ], value_setter = value_setter )
121
116
)
122
117
copy .append (
123
118
"""
@@ -130,27 +125,30 @@ public {list_type} {name};
130
125
{name}.Add(elem);
131
126
}}
132
127
{name}.Raise();
133
- }}""" .format (
134
- name = field ["name" ], type = field ["type" ]
135
- )
128
+ }}""" .format (name = field ["name" ], type = field ["type" ])
136
129
)
137
130
138
131
retrieval .append (
139
132
"""
140
133
foreach (var value in {name}.Items) {{ proto.{name}.Add({value_getter}); }}
141
134
""" .format (
142
- name = field ["name" ], type = field ["type" ], value_getter = value_getter
135
+ name = field ["name" ],
136
+ value_getter = value_getter ,
143
137
)
144
138
)
145
139
else :
146
140
if field ["type" ] == "ByteArrayVariable" :
147
- value_setter = "proto.{name}.ToByteArray()" .format (name = field ["name" ])
141
+ value_setter = "proto.{name}.ToByteArray()" .format (
142
+ name = field ["name" ]
143
+ )
148
144
value_getter = "({name} != null) ? ByteString.CopyFrom({name}.GetValue(), 0, {name}.GetValue().Length) : proto.{name}" .format (
149
145
name = field ["name" ]
150
146
)
151
147
else :
152
148
value_setter = "proto.{name}" .format (name = field ["name" ])
153
- value_getter = "{name}?.GetValue() ?? proto.{name}" .format (name = field ["name" ])
149
+ value_getter = "{name}?.GetValue() ?? proto.{name}" .format (
150
+ name = field ["name" ]
151
+ )
154
152
assignment .append (
155
153
"""
156
154
if ({name} == null) {{ {name} = ScriptableObject.CreateInstance<{type}>(); }}
@@ -161,7 +159,9 @@ public {list_type} {name};
161
159
name = field ["name" ],
162
160
type = field ["type" ],
163
161
value_setter = value_setter ,
164
- null_check = self .null_checks [field ["type" ]].format (name = field ["name" ])
162
+ null_check = self .null_checks [field ["type" ]].format (
163
+ name = field ["name" ]
164
+ )
165
165
if field ["type" ] in self .null_checks
166
166
else f"proto.{ field ['name' ]} != { field ['type' ]} .TypeDefault" ,
167
167
)
@@ -175,9 +175,7 @@ public {list_type} {name};
175
175
else if (other.{name} != null)
176
176
{{
177
177
{name}.SetValue(other.{name});
178
- }}""" .format (
179
- name = field ["name" ]
180
- )
178
+ }}""" .format (name = field ["name" ])
181
179
)
182
180
retrieval .append (
183
181
" proto.{name} = {value_getter};" .format (
@@ -208,7 +206,10 @@ public {list_type} {name};
208
206
def process (self , request ):
209
207
response = plugin .CodeGeneratorResponse ()
210
208
for proto_file in request .proto_file :
211
- if self .skip_dependencies and proto_file .name not in request .file_to_generate :
209
+ if (
210
+ self .skip_dependencies
211
+ and proto_file .name not in request .file_to_generate
212
+ ):
212
213
continue
213
214
result = build_tree (proto_file , self .with_options )
214
215
if self .with_comments :
0 commit comments