B   ›\hã@s8ddlmZmZddlZddlmZGdd„deƒZdS)é)Úprint_functionÚabsolute_importN)Ú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 ú6lib/python3.7/site-packages/numba/datamodel/manager.pyÚ__init__ szDataModelManager.__init__cCst|tjƒst‚||j|<dS)zNRegister the datamodel factory corresponding to a frontend-type class N)Ú issubclassrZTypeÚAssertionErrorr)r Z fetypeclsÚhandlerr r r ÚregisterszDataModelManager.registercCsFy |j|Stk rYnX|jt|ƒ}|||ƒ}|j|<|S)zMReturns the corresponding datamodel given the frontend-type instance )r ÚKeyErrorrÚtype)r ÚfetyperZmodelr r r Úlookups zDataModelManager.lookupcCs | |¡S)zShorthand for lookup() )r)r rr r r Ú __getitem__#szDataModelManager.__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 Zdmmr r r r(s zDataModelManager.copyN) Ú__name__Ú __module__Ú __qualname__Ú__doc__r rrrrr r r r rs  r)Z __future__rrrZnumbarÚobjectrr r r r Ús