io_scene_fbx/json2fbx.py

   1 #!/usr/bin/env python3
   2 # SPDX-License-Identifier: GPL-2.0-or-later
   3
   4 # Script copyright (C) 2014 Blender Foundation
   5
   6 """
   7 Usage
   8 =====
   9
  10    json2fbx [FILES]...
  11
  12 This script will write a binary FBX file for each JSON argument given.
  13
  14
  15 Input
  16 ======
  17
  18 The JSON data is formatted into a list of nested lists of 4 items:
  19
  20    ``[id, [data, ...], "data_types", [subtree, ...]]``
  21
  22 Where each list may be empty, and the items in
  23 the subtree are formatted the same way.
  24
  25 data_types is a string, aligned with data that spesifies a type
  26 for each property.
  27
  28 The types are as follows:
  29
  30 * 'Y': - INT16
  31 * 'C': - BOOL
  32 * 'I': - INT32
  33 * 'F': - FLOAT32
  34 * 'D': - FLOAT64
  35 * 'L': - INT64
  36 * 'R': - BYTES
  37 * 'S': - STRING
  38 * 'f': - FLOAT32_ARRAY
  39 * 'i': - INT32_ARRAY
  40 * 'd': - FLOAT64_ARRAY
  41 * 'l': - INT64_ARRAY
  42 * 'b': - BOOL ARRAY
  43 * 'c': - BYTE ARRAY
  44
  45 Note that key:value pairs aren't used since the id's are not
  46 ensured to be unique.
  47 """
  48
  49
  50 def elem_empty(elem, name):
  51     import encode_bin
  52     sub_elem = encode_bin.FBXElem(name)
  53     if elem is not None:
  54         elem.elems.append(sub_elem)
  55     return sub_elem
  56
  57
  58 def parse_json_rec(fbx_root, json_node):
  59     name, data, data_types, children = json_node
  60     ver = 0
  61
  62     assert(len(data_types) == len(data))
  63
  64     e = elem_empty(fbx_root, name.encode())
  65     for d, dt in zip(data, data_types):
  66         if dt == "C":
  67             e.add_bool(d)
  68         elif dt == "Y":
  69             e.add_int16(d)
  70         elif dt == "I":
  71             e.add_int32(d)
  72         elif dt == "L":
  73             e.add_int64(d)
  74         elif dt == "F":
  75             e.add_float32(d)
  76         elif dt == "D":
  77             e.add_float64(d)
  78         elif dt == "R":
  79             d = eval('b"""' + d + '"""')
  80             e.add_bytes(d)
  81         elif dt == "S":
  82             d = d.encode().replace(b"::", b"\x00\x01")
  83             e.add_string(d)
  84         elif dt == "i":
  85             e.add_int32_array(d)
  86         elif dt == "l":
  87             e.add_int64_array(d)
  88         elif dt == "f":
  89             e.add_float32_array(d)
  90         elif dt == "d":
  91             e.add_float64_array(d)
  92         elif dt == "b":
  93             e.add_bool_array(d)
  94         elif dt == "c":
  95             e.add_byte_array(d)
  96
  97     if name == "FBXVersion":
  98         assert(data_types == "I")
  99         ver = int(data[0])
 100
 101     for child in children:
 102         _ver = parse_json_rec(e, child)
 103         if _ver:
 104             ver = _ver
 105
 106     return ver
 107
 108
 109 def parse_json(json_root):
 110     root = elem_empty(None, b"")
 111     ver = 0
 112
 113     for n in json_root:
 114         _ver = parse_json_rec(root, n)
 115         if _ver:
 116             ver = _ver
 117
 118     return root, ver
 119
 120
 121 def json2fbx(fn):
 122     import os
 123     import json
 124
 125     import encode_bin
 126
 127     fn_fbx = "%s.fbx" % os.path.splitext(fn)[0]
 128     print("Writing: %r " % fn_fbx, end="")
 129     json_root = []
 130     with open(fn) as f_json:
 131         json_root = json.load(f_json)
 132     fbx_root, fbx_version = parse_json(json_root)
 133     print("(Version %d) ..." % fbx_version)
 134     encode_bin.write(fn_fbx, fbx_root, fbx_version)
 135
 136
 137 # ----------------------------------------------------------------------------
 138 # Command Line
 139
 140 def main():
 141     import sys
 142
 143     if "--help" in sys.argv:
 144         print(__doc__)
 145         return
 146
 147     for arg in sys.argv[1:]:
 148         try:
 149             json2fbx(arg)
 150         except:
 151             print("Failed to convert %r, error:" % arg)
 152
 153             import traceback
 154             traceback.print_exc()
 155
 156
 157 if __name__ == "__main__":
 158     main()