o 8ήc5ã@s(ddlZddlmZGdd„deƒZdS)éN)Útypesc@s8eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd S) ÚDataModelManagerzBManages mapping of FE types to their corresponding data model cCsi|_t ¡|_dS)N)Ú _handlersÚweakrefÚWeakKeyDictionaryÚ_cache)Úself©r úC/tmp/pip-target-vg8gfxp4/lib/python/numba/core/datamodel/manager.pyÚ__init__ szDataModelManager.__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)rrr 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Údmmr r r r&s zDataModelManager.copyN) Ú__name__Ú __module__Ú __qualname__Ú__doc__r rrrrr r r r rs r)rÚ numba.corerÚobjectrr r r r Ús