backend_open62541_nodes.py 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524
  1. #!/usr/bin/env/python
  2. # -*- coding: utf-8 -*-
  3. ###
  4. ### Author: Chris Iatrou (ichrispa@core-vector.net)
  5. ### Version: rev 13
  6. ###
  7. ### This program was created for educational purposes and has been
  8. ### contributed to the open62541 project by the author. All licensing
  9. ### terms for this source is inherited by the terms and conditions
  10. ### specified for by the open62541 project (see the projects readme
  11. ### file for more information on the LGPL terms and restrictions).
  12. ###
  13. ### This program is not meant to be used in a production environment. The
  14. ### author is not liable for any complications arising due to the use of
  15. ### this program.
  16. ###
  17. from nodes import *
  18. from backend_open62541_datatypes import *
  19. import re
  20. import logging
  21. logger = logging.getLogger(__name__)
  22. #################
  23. # Generate Code #
  24. #################
  25. def generateNodeIdPrintable(node):
  26. if isinstance(node.id, NodeId):
  27. CodePrintable = node.__class__.__name__ + "_" + str(node.id)
  28. else:
  29. CodePrintable = node.__class__.__name__ + "_unknown_nid"
  30. return re.sub('[^0-9a-z_]+', '_', CodePrintable.lower())
  31. def generateNodeValueInstanceName(node, parent, recursionDepth, arrayIndex):
  32. return generateNodeIdPrintable(parent) + "_" + str(node.alias) + "_" + str(arrayIndex) + "_" + str(recursionDepth)
  33. def generateReferenceCode(reference):
  34. if reference.isForward:
  35. return "retVal |= UA_Server_addReference(server, %s, %s, %s, true);" % \
  36. (generateNodeIdCode(reference.source),
  37. generateNodeIdCode(reference.referenceType),
  38. generateExpandedNodeIdCode(reference.target))
  39. else:
  40. return "retVal |= UA_Server_addReference(server, %s, %s, %s, false);" % \
  41. (generateNodeIdCode(reference.source),
  42. generateNodeIdCode(reference.referenceType),
  43. generateExpandedNodeIdCode(reference.target))
  44. def generateReferenceTypeNodeCode(node):
  45. code = []
  46. code.append("UA_ReferenceTypeAttributes attr = UA_ReferenceTypeAttributes_default;")
  47. if node.isAbstract:
  48. code.append("attr.isAbstract = true;")
  49. if node.symmetric:
  50. code.append("attr.symmetric = true;")
  51. if node.inverseName != "":
  52. code.append("attr.inverseName = UA_LOCALIZEDTEXT(\"\", \"%s\");" % \
  53. node.inverseName)
  54. return code
  55. def generateObjectNodeCode(node):
  56. code = []
  57. code.append("UA_ObjectAttributes attr = UA_ObjectAttributes_default;")
  58. if node.eventNotifier:
  59. code.append("attr.eventNotifier = true;")
  60. return code
  61. def generateVariableNodeCode(node, nodeset, encode_binary_size):
  62. code = []
  63. codeCleanup = []
  64. codeGlobal = []
  65. code.append("UA_VariableAttributes attr = UA_VariableAttributes_default;")
  66. if node.historizing:
  67. code.append("attr.historizing = true;")
  68. code.append("attr.minimumSamplingInterval = %f;" % node.minimumSamplingInterval)
  69. code.append("attr.userAccessLevel = %d;" % node.userAccessLevel)
  70. code.append("attr.accessLevel = %d;" % node.accessLevel)
  71. # in order to be compatible with mostly OPC UA client
  72. # force valueRank = -1 for scalar VariableNode
  73. if node.valueRank == -2:
  74. node.valueRank = -1
  75. code.append("attr.valueRank = %d;" % node.valueRank)
  76. if node.valueRank > 0:
  77. code.append("attr.arrayDimensionsSize = %d;" % node.valueRank)
  78. code.append("attr.arrayDimensions = (UA_UInt32 *)UA_Array_new({}, &UA_TYPES[UA_TYPES_UINT32]);".format(node.valueRank))
  79. code.append("if (!attr.arrayDimensions) return UA_STATUSCODE_BADOUTOFMEMORY;")
  80. codeCleanup.append("UA_Array_delete(attr.arrayDimensions, {}, &UA_TYPES[UA_TYPES_UINT32]);".format(node.valueRank))
  81. if len(node.arrayDimensions) == node.valueRank:
  82. for idx, v in enumerate(node.arrayDimensions):
  83. code.append("attr.arrayDimensions[{}] = {};".format(idx, int(unicode(v))))
  84. else:
  85. for dim in range(0, node.valueRank):
  86. code.append("attr.arrayDimensions[{}] = 0;".format(dim))
  87. if node.dataType is not None:
  88. if isinstance(node.dataType, NodeId) and node.dataType.ns == 0 and node.dataType.i == 0:
  89. #BaseDataType
  90. dataTypeNode = nodeset.nodes[NodeId("i=24")]
  91. dataTypeNodeOpaque = nodeset.nodes[NodeId("i=24")]
  92. else:
  93. dataTypeNodeOpaque = nodeset.getDataTypeNode(node.dataType)
  94. dataTypeNode = nodeset.getBaseDataType(nodeset.getDataTypeNode(node.dataType))
  95. if dataTypeNode is not None:
  96. code.append("attr.dataType = %s;" % generateNodeIdCode(dataTypeNodeOpaque.id))
  97. if dataTypeNode.isEncodable():
  98. if node.value is not None:
  99. [code1, codeCleanup1, codeGlobal1] = generateValueCode(node.value, nodeset.nodes[node.id], nodeset, encode_binary_size=encode_binary_size)
  100. code += code1
  101. codeCleanup += codeCleanup1
  102. codeGlobal += codeGlobal1
  103. # #1978 Variant arrayDimensions are only required to properly decode multidimensional arrays
  104. # (valueRank >= 2) from data stored as one-dimensional array of arrayLength elements.
  105. # One-dimensional arrays are already completely defined by arraylength attribute so setting
  106. # also arrayDimensions, even if not explicitly forbidden, can confuse clients
  107. if node.valueRank > 1 and len(node.arrayDimensions) == node.valueRank:
  108. code.append("attr.value.arrayDimensionsSize = attr.arrayDimensionsSize;")
  109. code.append("attr.value.arrayDimensions = attr.arrayDimensions;")
  110. else:
  111. code += generateValueCodeDummy(dataTypeNode, nodeset.nodes[node.id], nodeset)
  112. return [code, codeCleanup, codeGlobal]
  113. def generateVariableTypeNodeCode(node, nodeset, encode_binary_size):
  114. code = []
  115. codeCleanup = []
  116. codeGlobal = []
  117. code.append("UA_VariableTypeAttributes attr = UA_VariableTypeAttributes_default;")
  118. if node.historizing:
  119. code.append("attr.historizing = true;")
  120. if node.isAbstract:
  121. code.append("attr.isAbstract = true;")
  122. code.append("attr.valueRank = (UA_Int32)%s;" % str(node.valueRank))
  123. if node.dataType is not None:
  124. if isinstance(node.dataType, NodeId) and node.dataType.ns == 0 and node.dataType.i == 0:
  125. #BaseDataType
  126. dataTypeNode = nodeset.nodes[NodeId("i=24")]
  127. else:
  128. dataTypeNode = nodeset.getBaseDataType(nodeset.getDataTypeNode(node.dataType))
  129. if dataTypeNode is not None:
  130. code.append("attr.dataType = %s;" % generateNodeIdCode(dataTypeNode.id))
  131. if dataTypeNode.isEncodable():
  132. if node.value is not None:
  133. [code1, codeCleanup1, codeGlobal1] = generateValueCode(node.value, nodeset.nodes[node.id], nodeset, encode_binary_size)
  134. code += code1
  135. codeCleanup += codeCleanup1
  136. codeGlobal += codeGlobal1
  137. else:
  138. code += generateValueCodeDummy(dataTypeNode, nodeset.nodes[node.id], nodeset)
  139. return [code, codeCleanup, codeGlobal]
  140. def generateExtensionObjectSubtypeCode(node, parent, nodeset, global_var_code, recursionDepth=0, arrayIndex=0, encode_binary_size=32000):
  141. code = [""]
  142. codeCleanup = [""]
  143. logger.debug("Building extensionObject for " + str(parent.id))
  144. logger.debug("Value " + str(node.value))
  145. logger.debug("Encoding " + str(node.encodingRule))
  146. instanceName = generateNodeValueInstanceName(node, parent, recursionDepth, arrayIndex)
  147. # If there are any ExtensionObjects instide this ExtensionObject, we need to
  148. # generate one-time-structs for them too before we can proceed;
  149. for subv in node.value:
  150. if isinstance(subv, list):
  151. logger.error("ExtensionObject contains an ExtensionObject, which is currently not encodable!")
  152. code.append("struct {")
  153. for field in node.encodingRule:
  154. ptrSym = ""
  155. # If this is an Array, this is pointer to its contents with a AliasOfFieldSize entry
  156. if field[2] != None and field[2] != 0 :
  157. code.append(" UA_Int32 " + str(field[0]) + "Size;")
  158. ptrSym = "*"
  159. if len(field[1]) == 1:
  160. code.append(" UA_" + str(field[1][0]) + " " + ptrSym + str(field[0]) + ";")
  161. else:
  162. code.append(" UA_ExtensionObject " + " " + ptrSym + str(field[0]) + ";")
  163. code.append("} " + instanceName + "_struct;")
  164. # Allocate some memory
  165. code.append("UA_ExtensionObject *" + instanceName + " = UA_ExtensionObject_new();")
  166. code.append("if (!" + instanceName + ") return UA_STATUSCODE_BADOUTOFMEMORY;")
  167. codeCleanup.append("UA_ExtensionObject_delete(" + instanceName + ");")
  168. # Assign data to the struct contents
  169. # Track the encoding rule definition to detect arrays and/or ExtensionObjects
  170. encFieldIdx = 0
  171. for subv in node.value:
  172. encField = node.encodingRule[encFieldIdx]
  173. encFieldIdx = encFieldIdx + 1
  174. logger.debug(
  175. "Encoding of field " + subv.alias + " is " + str(subv.encodingRule) + "defined by " + str(encField))
  176. # Check if this is an array
  177. if subv.valueRank is None or subv.valueRank == 0:
  178. valueName = instanceName + "_struct." + subv.alias
  179. code.append(generateNodeValueCode(valueName + " = " ,
  180. subv, instanceName,valueName, global_var_code, asIndirect=False))
  181. else:
  182. if isinstance(subv, list):
  183. # this is an array
  184. code.append(instanceName + "_struct." + subv.alias + "Size = " + str(len(subv)) + ";")
  185. code.append(
  186. "{0}_struct.{1} = (UA_{2}*) UA_malloc(sizeof(UA_{2})*{3});".format(
  187. instanceName, subv.alias, subv.__class__.__name__, str(len(subv))))
  188. code.append("if (!{0}_struct.{1}) return UA_STATUSCODE_BADOUTOFMEMORY;".format(
  189. instanceName, subv.alias))
  190. codeCleanup.append("UA_free({0}_struct.{1});".format(instanceName, subv.alias))
  191. logger.debug("Encoding included array of " + str(len(subv)) + " values.")
  192. for subvidx in range(0, len(subv)):
  193. subvv = subv[subvidx]
  194. logger.debug(" " + str(subvidx) + " " + str(subvv))
  195. valueName = instanceName + "_struct." + subv.alias + "[" + str(
  196. subvidx) + "]"
  197. code.append(generateNodeValueCode(valueName + " = ", subvv, instanceName, valueName, global_var_code))
  198. code.append("}")
  199. else:
  200. code.append(instanceName + "_struct." + subv.alias + "Size = 1;")
  201. code.append(
  202. "{0}_struct.{1} = (UA_{2}*) UA_malloc(sizeof(UA_{2}));".format(
  203. instanceName, subv.alias, subv.__class__.__name__))
  204. code.append("if (!{0}_struct.{1}) return UA_STATUSCODE_BADOUTOFMEMORY;".format(
  205. instanceName, subv.alias))
  206. codeCleanup.append("UA_free({0}_struct.{1});".format(instanceName, subv.alias))
  207. valueName = instanceName + "_struct." + subv.alias + "[0]"
  208. code.append(generateNodeValueCode(valueName + " = ",
  209. subv, instanceName, valueName, global_var_code, asIndirect=True))
  210. code.append(instanceName + "->encoding = UA_EXTENSIONOBJECT_ENCODED_BYTESTRING;")
  211. #if parent.dataType.ns == 0:
  212. binaryEncodingId = nodeset.getBinaryEncodingIdForNode(parent.dataType)
  213. code.append(
  214. instanceName + "->content.encoded.typeId = UA_NODEID_NUMERIC(" + str(binaryEncodingId.ns) + ", " +
  215. str(binaryEncodingId.i) + ");")
  216. code.append(
  217. "retVal |= UA_ByteString_allocBuffer(&" + instanceName + "->content.encoded.body, " + str(encode_binary_size) + ");")
  218. # Encode each value as a bytestring separately.
  219. code.append("UA_Byte *pos" + instanceName + " = " + instanceName + "->content.encoded.body.data;")
  220. code.append("const UA_Byte *end" + instanceName + " = &" + instanceName + "->content.encoded.body.data[" + str(encode_binary_size) + "];")
  221. encFieldIdx = 0
  222. code.append("{")
  223. for subv in node.value:
  224. # encField = node.encodingRule[encFieldIdx]
  225. encFieldIdx = encFieldIdx + 1
  226. if subv.valueRank is None or subv.valueRank == 0:
  227. code.append(
  228. "retVal |= UA_encodeBinary(&" + instanceName + "_struct." + subv.alias + ", " +
  229. getTypesArrayForValue(nodeset, subv) + ", &pos" + instanceName + ", &end" + instanceName + ", NULL, NULL);")
  230. else:
  231. if isinstance(subv, list):
  232. for subvidx in range(0, len(subv)):
  233. code.append("retVal |= UA_encodeBinary(&" + instanceName + "_struct." + subv.alias + "[" +
  234. str(subvidx) + "], " + getTypesArrayForValue(nodeset, subv) + ", &pos" +
  235. instanceName + ", &end" + instanceName + ", NULL, NULL);")
  236. else:
  237. code.append(
  238. "retVal |= UA_encodeBinary(&" + instanceName + "_struct." + subv.alias + "[0], " +
  239. getTypesArrayForValue(nodeset, subv) + ", &pos" + instanceName + ", &end" + instanceName + ", NULL, NULL);")
  240. code.append("}")
  241. # Reallocate the memory by swapping the 65k Bytestring for a new one
  242. code.append("size_t " + instanceName + "_encOffset = (uintptr_t)(" +
  243. "pos" + instanceName + "-" + instanceName + "->content.encoded.body.data);")
  244. code.append(instanceName + "->content.encoded.body.length = " + instanceName + "_encOffset;")
  245. code.append("UA_Byte *" + instanceName + "_newBody = (UA_Byte *) UA_malloc(" + instanceName + "_encOffset);")
  246. code.append("if (!" + instanceName + "_newBody) return UA_STATUSCODE_BADOUTOFMEMORY;")
  247. code.append("memcpy(" + instanceName + "_newBody, " + instanceName + "->content.encoded.body.data, " +
  248. instanceName + "_encOffset);")
  249. code.append("UA_Byte *" + instanceName + "_oldBody = " + instanceName + "->content.encoded.body.data;")
  250. code.append(instanceName + "->content.encoded.body.data = " + instanceName + "_newBody;")
  251. code.append("UA_free(" + instanceName + "_oldBody);")
  252. code.append("")
  253. return [code, codeCleanup]
  254. def generateValueCodeDummy(dataTypeNode, parentNode, nodeset):
  255. code = []
  256. valueName = generateNodeIdPrintable(parentNode) + "_variant_DataContents"
  257. typeBrowseNode = makeCIdentifier(dataTypeNode.browseName.name)
  258. if typeBrowseNode == "NumericRange":
  259. # in the stack we define a separate structure for the numeric range, but
  260. # the value itself is just a string
  261. typeBrowseNode = "String"
  262. typeArr = dataTypeNode.typesArray + "[" + dataTypeNode.typesArray + "_" + typeBrowseNode.upper() + "]"
  263. typeStr = "UA_" + typeBrowseNode
  264. if parentNode.valueRank > 0:
  265. for i in range(0, parentNode.valueRank):
  266. code.append("UA_Variant_setArray(&attr.value, NULL, (UA_Int32) " + "0, &" + typeArr + ");")
  267. elif not dataTypeNode.isAbstract:
  268. code.append("UA_STACKARRAY(" + typeStr + ", " + valueName + ", 1);")
  269. code.append("UA_init(" + valueName + ", &" + typeArr + ");")
  270. code.append("UA_Variant_setScalar(&attr.value, " + valueName + ", &" + typeArr + ");")
  271. return code
  272. def getTypesArrayForValue(nodeset, value):
  273. typeNode = nodeset.getNodeByBrowseName(value.__class__.__name__)
  274. if typeNode is None or value.isInternal:
  275. typesArray = "UA_TYPES"
  276. else:
  277. typesArray = typeNode.typesArray
  278. return "&" + typesArray + "[" + typesArray + "_" + \
  279. makeCIdentifier(value.__class__.__name__.upper()) + "]"
  280. def generateValueCode(node, parentNode, nodeset, bootstrapping=True, encode_binary_size=32000):
  281. code = []
  282. codeCleanup = []
  283. codeGlobal = []
  284. valueName = generateNodeIdPrintable(parentNode) + "_variant_DataContents"
  285. # node.value either contains a list of multiple identical BUILTINTYPES, or it
  286. # contains a single builtintype (which may be a container); choose if we need
  287. # to create an array or a single variable.
  288. # Note that some genious defined that there are arrays of size 1, which are
  289. # distinctly different then a single value, so we need to check that as well
  290. # Semantics:
  291. # -3: Scalar or 1-dim
  292. # -2: Scalar or x-dim | x>0
  293. # -1: Scalar
  294. # 0: x-dim | x>0
  295. # n: n-dim | n>0
  296. if (len(node.value) == 0):
  297. return ["", ""]
  298. if not isinstance(node.value[0], Value):
  299. return ["", ""]
  300. if parentNode.valueRank != -1 and (parentNode.valueRank >= 0
  301. or (len(node.value) > 1
  302. and (parentNode.valueRank != -2 or parentNode.valueRank != -3))):
  303. # User the following strategy for all directly mappable values a la 'UA_Type MyInt = (UA_Type) 23;'
  304. if isinstance(node.value[0], Guid):
  305. logger.warn("Don't know how to print array of GUID in node " + str(parentNode.id))
  306. elif isinstance(node.value[0], DiagnosticInfo):
  307. logger.warn("Don't know how to print array of DiagnosticInfo in node " + str(parentNode.id))
  308. elif isinstance(node.value[0], StatusCode):
  309. logger.warn("Don't know how to print array of StatusCode in node " + str(parentNode.id))
  310. else:
  311. if isinstance(node.value[0], ExtensionObject):
  312. for idx, v in enumerate(node.value):
  313. logger.debug("Building extObj array index " + str(idx))
  314. [code1, codeCleanup1] = generateExtensionObjectSubtypeCode(v, parent=parentNode, nodeset=nodeset, arrayIndex=idx,
  315. encode_binary_size=encode_binary_size, global_var_code=codeGlobal)
  316. code = code + code1
  317. codeCleanup = codeCleanup + codeCleanup1
  318. code.append("UA_" + node.value[0].__class__.__name__ + " " + valueName + "[" + str(len(node.value)) + "];")
  319. if isinstance(node.value[0], ExtensionObject):
  320. for idx, v in enumerate(node.value):
  321. logger.debug("Printing extObj array index " + str(idx))
  322. instanceName = generateNodeValueInstanceName(v, parentNode, 0, idx)
  323. code.append(generateNodeValueCode(
  324. valueName + "[" + str(idx) + "] = ",
  325. v, instanceName, valueName, codeGlobal))
  326. # code.append("UA_free(&" +valueName + "[" + str(idx) + "]);")
  327. else:
  328. for idx, v in enumerate(node.value):
  329. instanceName = generateNodeValueInstanceName(v, parentNode, 0, idx)
  330. code.append(generateNodeValueCode(
  331. valueName + "[" + str(idx) + "] = " , v, instanceName, valueName, codeGlobal))
  332. code.append("UA_Variant_setArray(&attr.value, &" + valueName +
  333. ", (UA_Int32) " + str(len(node.value)) + ", " +
  334. getTypesArrayForValue(nodeset, node.value[0]) + ");")
  335. else:
  336. # User the following strategy for all directly mappable values a la 'UA_Type MyInt = (UA_Type) 23;'
  337. if isinstance(node.value[0], Guid):
  338. logger.warn("Don't know how to print scalar GUID in node " + str(parentNode.id))
  339. elif isinstance(node.value[0], DiagnosticInfo):
  340. logger.warn("Don't know how to print scalar DiagnosticInfo in node " + str(parentNode.id))
  341. elif isinstance(node.value[0], StatusCode):
  342. logger.warn("Don't know how to print scalar StatusCode in node " + str(parentNode.id))
  343. else:
  344. # The following strategy applies to all other types, in particular strings and numerics.
  345. if isinstance(node.value[0], ExtensionObject):
  346. [code1, codeCleanup1] = generateExtensionObjectSubtypeCode(node.value[0], parent=parentNode, nodeset=nodeset,
  347. encode_binary_size=encode_binary_size, global_var_code=codeGlobal)
  348. code = code + code1
  349. codeCleanup = codeCleanup + codeCleanup1
  350. instanceName = generateNodeValueInstanceName(node.value[0], parentNode, 0, 0)
  351. if isinstance(node.value[0], ExtensionObject):
  352. code.append(generateNodeValueCode("UA_" + node.value[0].__class__.__name__ + " *" + valueName + " = " ,
  353. node.value[0], instanceName, valueName, codeGlobal, asIndirect=True ))
  354. code.append(
  355. "UA_Variant_setScalar(&attr.value, " + valueName + ", " +
  356. getTypesArrayForValue(nodeset, node.value[0]) + ");")
  357. # FIXME: There is no membership definition for extensionObjects generated in this function.
  358. # code.append("UA_" + node.value[0].__class__.__name__ + "_deleteMembers(" + valueName + ");")
  359. else:
  360. code.append("UA_" + node.value[0].__class__.__name__ + " *" + valueName + " = UA_" + node.value[
  361. 0].__class__.__name__ + "_new();")
  362. code.append("if (!" + valueName + ") return UA_STATUSCODE_BADOUTOFMEMORY;")
  363. code.append(generateNodeValueCode("*" + valueName + " = " , node.value[0], instanceName, valueName, codeGlobal, asIndirect=True))
  364. code.append(
  365. "UA_Variant_setScalar(&attr.value, " + valueName + ", " +
  366. getTypesArrayForValue(nodeset, node.value[0]) + ");")
  367. if node.value[0].__class__.__name__ == "ByteString":
  368. # The data is on the stack, not heap, so we can not delete the ByteString
  369. codeCleanup.append("{}->data = NULL;".format(valueName))
  370. codeCleanup.append("{}->length = 0;".format(valueName))
  371. codeCleanup.append("UA_{0}_delete({1});".format(
  372. node.value[0].__class__.__name__, valueName))
  373. return [code, codeCleanup, codeGlobal]
  374. def generateMethodNodeCode(node):
  375. code = []
  376. code.append("UA_MethodAttributes attr = UA_MethodAttributes_default;")
  377. if node.executable:
  378. code.append("attr.executable = true;")
  379. if node.userExecutable:
  380. code.append("attr.userExecutable = true;")
  381. return code
  382. def generateObjectTypeNodeCode(node):
  383. code = []
  384. code.append("UA_ObjectTypeAttributes attr = UA_ObjectTypeAttributes_default;")
  385. if node.isAbstract:
  386. code.append("attr.isAbstract = true;")
  387. return code
  388. def generateDataTypeNodeCode(node):
  389. code = []
  390. code.append("UA_DataTypeAttributes attr = UA_DataTypeAttributes_default;")
  391. if node.isAbstract:
  392. code.append("attr.isAbstract = true;")
  393. return code
  394. def generateViewNodeCode(node):
  395. code = []
  396. code.append("UA_ViewAttributes attr = UA_ViewAttributes_default;")
  397. if node.containsNoLoops:
  398. code.append("attr.containsNoLoops = true;")
  399. code.append("attr.eventNotifier = (UA_Byte)%s;" % str(node.eventNotifier))
  400. return code
  401. def getNodeTypeDefinition(node):
  402. for ref in node.references:
  403. # 40 = HasTypeDefinition
  404. if ref.referenceType.i == 40:
  405. return ref.target
  406. return None
  407. def generateSubtypeOfDefinitionCode(node):
  408. for ref in node.inverseReferences:
  409. # 45 = HasSubtype
  410. if ref.referenceType.i == 45:
  411. return generateNodeIdCode(ref.target)
  412. return "UA_NODEID_NULL"
  413. def generateNodeCode_begin(node, nodeset, generate_ns0, parentref, encode_binary_size, code_global):
  414. code = []
  415. codeCleanup = []
  416. code.append("UA_StatusCode retVal = UA_STATUSCODE_GOOD;")
  417. # Attributes
  418. if isinstance(node, ReferenceTypeNode):
  419. code.extend(generateReferenceTypeNodeCode(node))
  420. elif isinstance(node, ObjectNode):
  421. code.extend(generateObjectNodeCode(node))
  422. elif isinstance(node, VariableNode) and not isinstance(node, VariableTypeNode):
  423. [code1, codeCleanup1, codeGlobal1] = generateVariableNodeCode(node, nodeset, encode_binary_size)
  424. code.extend(code1)
  425. codeCleanup.extend(codeCleanup1)
  426. code_global.extend(codeGlobal1)
  427. elif isinstance(node, VariableTypeNode):
  428. [code1, codeCleanup1, codeGlobal1] = generateVariableTypeNodeCode(node, nodeset, encode_binary_size)
  429. code.extend(code1)
  430. codeCleanup.extend(codeCleanup1)
  431. code_global.extend(codeGlobal1)
  432. elif isinstance(node, MethodNode):
  433. code.extend(generateMethodNodeCode(node))
  434. elif isinstance(node, ObjectTypeNode):
  435. code.extend(generateObjectTypeNodeCode(node))
  436. elif isinstance(node, DataTypeNode):
  437. code.extend(generateDataTypeNodeCode(node))
  438. elif isinstance(node, ViewNode):
  439. code.extend(generateViewNodeCode(node))
  440. code.append("attr.displayName = " + generateLocalizedTextCode(node.displayName, alloc=False) + ";")
  441. code.append("#ifdef UA_ENABLE_NODESET_COMPILER_DESCRIPTIONS")
  442. code.append("attr.description = " + generateLocalizedTextCode(node.description, alloc=False) + ";")
  443. code.append("#endif")
  444. code.append("attr.writeMask = %d;" % node.writeMask)
  445. code.append("attr.userWriteMask = %d;" % node.userWriteMask)
  446. # AddNodes call
  447. code.append("retVal |= UA_Server_addNode_begin(server, UA_NODECLASS_{},".
  448. format(makeCIdentifier(node.__class__.__name__.upper().replace("NODE" ,""))))
  449. code.append(generateNodeIdCode(node.id) + ",")
  450. code.append(generateNodeIdCode(parentref.target) + ",")
  451. code.append(generateNodeIdCode(parentref.referenceType) + ",")
  452. code.append(generateQualifiedNameCode(node.browseName) + ",")
  453. if isinstance(node, VariableNode) or isinstance(node, ObjectNode):
  454. typeDefRef = node.popTypeDef()
  455. code.append(generateNodeIdCode(typeDefRef.target) + ",")
  456. else:
  457. code.append(" UA_NODEID_NULL,")
  458. code.append("(const UA_NodeAttributes*)&attr, &UA_TYPES[UA_TYPES_{}ATTRIBUTES],NULL, NULL);".
  459. format(makeCIdentifier(node.__class__.__name__.upper().replace("NODE" ,""))))
  460. code.extend(codeCleanup)
  461. return "\n".join(code)
  462. def generateNodeCode_finish(node):
  463. code = []
  464. if isinstance(node, MethodNode):
  465. code.append("UA_Server_addMethodNode_finish(server, ")
  466. else:
  467. code.append("UA_Server_addNode_finish(server, ")
  468. code.append(generateNodeIdCode(node.id))
  469. if isinstance(node, MethodNode):
  470. code.append(", NULL, 0, NULL, 0, NULL);")
  471. else:
  472. code.append(");")
  473. return "\n".join(code)