o Û¾ikã@s4ddlZddlmZddlmZGdd„deƒZdS)éN)ÚChainMap)Útypesc@sBeZdZdZddd„Zdd„Zdd„Zd d „Zd d „Zd d„Z 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)Ú _handlersÚweakrefÚWeakKeyDictionaryÚ_cache)ÚselfÚhandlers©r úP/home/ubuntu/.local/lib/python3.10/site-packages/numba/core/datamodel/manager.pyÚ__init__ s zDataModelManager.__init__cCst|tjƒsJ‚||j|<dS)zNRegister the datamodel factory corresponding to a frontend-type class N)Ú issubclassrÚTyper)r Ú fetypeclsÚhandlerr r r ÚregisterszDataModelManager.registercCsFz|j|WStyYnw|jt|ƒ}|||ƒ}|j|<|S)zMReturns the corresponding datamodel given the frontend-type instance )rÚKeyErrorrÚtype)r ÚfetyperÚmodelr r r Úlookups  ÿzDataModelManager.lookupcCs | |¡S)zShorthand for lookup() )r)r rr r r Ú __getitem__(s zDataModelManager.__getitem__cCst|j ¡ƒS)z’ Make a copy of the manager. Use this to inherit from the default data model and specialize it for custom target. )rrÚcopy)r r r r r-szDataModelManager.copycCst|j|jƒ}t|ƒS)aÕCreate 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 Ú other_managerÚchainedr r r Úchain5s zDataModelManager.chain)N) Ú__name__Ú __module__Ú __qualname__Ú__doc__r rrrrrr r r r rs    r)rÚ collectionsrÚ numba.corerÚobjectrr r r r Ús