# Licensed to the Apache Software Foundation (ASF) under one # or more contributor license agreements. See the NOTICE file # distributed with this work for additional information # regarding copyright ownership. The ASF licenses this file # to you under the Apache License, Version 2.0 (the # "License"); you may not use this file except in compliance # with the License. You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, # software distributed under the License is distributed on an # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY # KIND, either express or implied. See the License for the # specific language governing permissions and limitations # under the License. """Maps object type to its constructor""" cdef list OBJECT_TYPE = [] def _register_object(int index, object cls): """register object class""" if issubclass(cls, NDArrayBase): _register_ndarray(index, cls) return global OBJECT_TYPE while len(OBJECT_TYPE) <= index: OBJECT_TYPE.append(None) OBJECT_TYPE[index] = cls cdef inline object make_ret_object(void* chandle): global OBJECT_TYPE global _CLASS_OBJECT cdef unsigned tindex cdef object cls cdef object handle object_type = OBJECT_TYPE handle = ctypes_handle(chandle) CALL(TVMObjectGetTypeIndex(chandle, &tindex)) if tindex < len(OBJECT_TYPE): cls = OBJECT_TYPE[tindex] if cls is not None: obj = cls.__new__(cls) else: obj = _CLASS_OBJECT.__new__(_CLASS_OBJECT) else: obj = _CLASS_OBJECT.__new__(_CLASS_OBJECT) (<ObjectBase>obj).chandle = chandle return obj cdef class ObjectBase: cdef void* chandle cdef inline _set_handle(self, handle): cdef unsigned long long ptr if handle is None: self.chandle = NULL else: ptr = handle.value self.chandle = <void*>(ptr) property handle: def __get__(self): if self.chandle == NULL: return None else: return ctypes_handle(self.chandle) def __set__(self, value): self._set_handle(value) def __dealloc__(self): CALL(TVMObjectFree(self.chandle)) def __init_handle_by_constructor__(self, fconstructor, *args): """Initialize the handle by calling constructor function. Parameters ---------- fconstructor : Function Constructor function. args: list of objects The arguments to the constructor Note ---- We have a special calling convention to call constructor functions. So the return handle is directly set into the Node object instead of creating a new Node. """ # avoid error raised during construction. self.chandle = NULL cdef void* chandle ConstructorCall( (<PackedFuncBase>fconstructor).chandle, kTVMObjectHandle, args, &chandle) self.chandle = chandle def same_as(self, other): """Check object identity. Parameters ---------- other : object The other object to compare against. Returns ------- result : bool The comparison result. """ if not isinstance(other, ObjectBase): return False return self.chandle == (<ObjectBase>other).chandle