U ?hk@s4ddlZddlmZddlmZGdddeZdS)N)ChainMap)typesc@sBeZdZdZdddZddZddZd d Zd d Zd dZ dS)DataModelManagerzBManages mapping of FE types to their corresponding data model NcCs|pi|_t|_dS)z Parameters ----------- handlers: Mapping[Type, DataModel] or None Optionally provide the initial handlers mapping. N) _handlersweakrefWeakKeyDictionary_cache)selfhandlersr N/var/www/html/venv/lib/python3.8/site-packages/numba/core/datamodel/manager.py__init__ s zDataModelManager.__init__cCst|tjst||j|<dS)zNRegister the datamodel factory corresponding to a frontend-type class N) issubclassrTypeAssertionErrorr)r Z fetypeclshandlerr r r registerszDataModelManager.registercCsHz |j|WStk r YnX|jt|}|||}|j|<|S)zMReturns the corresponding datamodel given the frontend-type instance )rKeyErrorrtype)r fetypermodelr r r lookups zDataModelManager.lookupcCs ||S)zShorthand for lookup() )r)r rr r r __getitem__(szDataModelManager.__getitem__cCst|jS)z Make a copy of the manager. Use this to inherit from the default data model and specialize it for custom target. )rrcopy)r r r r r-szDataModelManager.copycCst|j|j}t|S)aCreate a new DataModelManager by chaining the handlers mapping of `other_manager` with a fresh handlers mapping. Any existing and new handlers inserted to `other_manager` will be visible to the new manager. Any handlers inserted to the new manager can override existing handlers in `other_manager` without actually mutating `other_manager`. Parameters ---------- other_manager: DataModelManager )rrr)r Z other_managerZchainedr r r chain5s zDataModelManager.chain)N) __name__ __module__ __qualname____doc__r rrrrrr r r r rs  r)r collectionsrZ numba.corerobjectrr r r r s