mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-21 10:13:39 +00:00
87 lines
2.7 KiB
Python
87 lines
2.7 KiB
Python
# this is based on jsarray.py
|
|
|
|
from ..base import *
|
|
try:
|
|
import numpy
|
|
except:
|
|
pass
|
|
|
|
|
|
@Js
|
|
def Uint16Array():
|
|
TypedArray = (PyJsInt8Array, PyJsUint8Array, PyJsUint8ClampedArray,
|
|
PyJsInt16Array, PyJsUint16Array, PyJsInt32Array,
|
|
PyJsUint32Array, PyJsFloat32Array, PyJsFloat64Array)
|
|
a = arguments[0]
|
|
if isinstance(a, PyJsNumber): # length
|
|
length = a.to_uint32()
|
|
if length != a.value:
|
|
raise MakeError('RangeError', 'Invalid array length')
|
|
temp = Js(numpy.full(length, 0, dtype=numpy.uint16))
|
|
temp.put('length', a)
|
|
return temp
|
|
elif isinstance(a, PyJsString): # object (string)
|
|
temp = Js(numpy.array(list(a.value), dtype=numpy.uint16))
|
|
temp.put('length', Js(len(list(a.value))))
|
|
return temp
|
|
elif isinstance(a, PyJsArray) or isinstance(a, TypedArray) or isinstance(
|
|
a, PyJsArrayBuffer): # object (Array, TypedArray)
|
|
array = a.to_list()
|
|
array = [(int(item.value) if item.value != None else 0)
|
|
for item in array]
|
|
temp = Js(numpy.array(array, dtype=numpy.uint16))
|
|
temp.put('length', Js(len(array)))
|
|
return temp
|
|
elif isinstance(a, PyObjectWrapper): # object (ArrayBuffer, etc)
|
|
if len(a.obj) % 2 != 0:
|
|
raise MakeError(
|
|
'RangeError',
|
|
'Byte length of Uint16Array should be a multiple of 2')
|
|
if len(arguments) > 1:
|
|
offset = int(arguments[1].value)
|
|
if offset % 2 != 0:
|
|
raise MakeError(
|
|
'RangeError',
|
|
'Start offset of Uint16Array should be a multiple of 2')
|
|
else:
|
|
offset = 0
|
|
if len(arguments) > 2:
|
|
length = int(arguments[2].value)
|
|
else:
|
|
length = int((len(a.obj) - offset) / 2)
|
|
array = numpy.frombuffer(
|
|
a.obj, dtype=numpy.uint16, count=length, offset=offset)
|
|
temp = Js(array)
|
|
temp.put('length', Js(length))
|
|
temp.buff = array
|
|
return temp
|
|
temp = Js(numpy.full(0, 0, dtype=numpy.uint16))
|
|
temp.put('length', Js(0))
|
|
return temp
|
|
|
|
|
|
Uint16Array.create = Uint16Array
|
|
Uint16Array.own['length']['value'] = Js(3)
|
|
|
|
Uint16Array.define_own_property(
|
|
'prototype', {
|
|
'value': Uint16ArrayPrototype,
|
|
'enumerable': False,
|
|
'writable': False,
|
|
'configurable': False
|
|
})
|
|
|
|
Uint16ArrayPrototype.define_own_property(
|
|
'constructor', {
|
|
'value': Uint16Array,
|
|
'enumerable': False,
|
|
'writable': True,
|
|
'configurable': True
|
|
})
|
|
|
|
Uint16ArrayPrototype.define_own_property('BYTES_PER_ELEMENT', {
|
|
'value': Js(2),
|
|
'enumerable': False,
|
|
'writable': False,
|
|
'configurable': False
|
|
})
|