o p i- @sddlZddlmZddlmZgZGdddZGdddZeaea d e d e fd d Z d dZ dddZ edddZdS)N)deepcopy)signature_safe_contextmanagerc@s:eZdZdZd ddZddZddZd d Zd d ZdS)UniqueNameGeneratorz Generate unique name with prefix. Args: prefix(str): The generated name prefix. All generated name will be started with this prefix. NcCs"tt|_|dur d}||_dS)N) collections defaultdictintidsprefix)selfr r ^/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/paddle/base/unique_name.py__init__ s  zUniqueNameGenerator.__init__cCs ||SN)generate)r keyr r r__call__&s zUniqueNameGenerator.__call__cCs4|j|}|j|d7<|jd|t|gS)z Generate unique names with prefix Args: key(str): The key of return string. Returns(str): A unique string with the prefix r_)r r joinstr)r rtmpr r rr)s zUniqueNameGenerator.generatecCs*ddlm}m}|r|S||SNr)_dygraph_tracerin_dygraph_mode) frameworkrr_generate_unique_namer)r rrrr r rgenerate_with_ignorable_key6s  z/UniqueNameGenerator.generate_with_ignorable_keycCst|j}t|j|_|Sr)rr rr )r retr r rclone>s  zUniqueNameGenerator.cloner) __name__ __module__ __qualname____doc__rrrrrr r r rrs  rc@s eZdZdZddZddZdS)DygraphParameterNameCheckerz6 Check whether the name of parameter is used. cCs t|_dSr)set _name_set)r r r rrIs z$DygraphParameterNameChecker.__init__cCs||jvrdS|j|dS)z Check whether the name is used. If not used, insert into the _name_set. Args: name(str): The name of parameter to check. Returns(bool): If the name is in name_set, return True; Otherwise, return False. TF)r&add)r namer r rrLs  z$DygraphParameterNameChecker.__call__N)r r!r"r#rrr r r rr$Ds r$rreturncCst|S)aZ Generate unique name with prefix key. Currently, Paddle distinguishes the names of the same key by numbering it from zero. For example, when key=fc, it continuously generates fc_0, fc_1, fc_2, etc. Args: key(str): The prefix of generated name. Returns: str: A unique string with the prefix key. Examples: .. code-block:: python >>> import paddle >>> name1 = paddle.utils.unique_name.generate('fc') >>> name2 = paddle.utils.unique_name.generate('fc') >>> print(name1, name2) fc_0 fc_1 ) generator)rr r rrbsrcCs(ddlm}m}|r|St|Sr)rrrrr*)rrrr r rrs rcCs>t}t}|dur tan|a|durta||fS|a||fS)a/ Switch the namespace of in current context to a new namespace. Though :code:`switch()` and :code:`guard()` can both change namespace, :code:`guard()` is recommended since it can manage the context better together with :code:`with` statement. Args: new_generator(UniqueNameGenerator, optional): A new UniqueNameGenerator, not required normally. Default is None, which means switch to a new anonymous namespace. new_para_name_checker(DygraphParameterNameChecker, optional): A new DygraphParameterNameChecker, not required normally. Default is None, which means switch to a new parameter name checker. Returns: UniqueNameGenerator: The previous UniqueNameGenerator. DygraphParameterNameChecker: The previous DygraphParameterNameChecker Examples: .. code-block:: python >>> import paddle >>> name1 = paddle.utils.unique_name.generate('fc') >>> name2 = paddle.utils.unique_name.generate('fc') >>> print(name1, name2) fc_0 fc_1 >>> pre_generator, pre_dygraph_name_checker = paddle.utils.unique_name.switch() # switch to a new anonymous namespace. >>> name2 = paddle.utils.unique_name.generate('fc') >>> print(name2) fc_0 >>> paddle.utils.unique_name.switch(pre_generator, pre_dygraph_name_checker) # switch back to pre_generator. >>> name3 = paddle.utils.unique_name.generate('fc') >>> print(name3) fc_2 N)r*dygraph_parameter_name_checkerrr$) new_generatorZnew_para_name_checker old_generatorold_para_name_checkerr r rswitchs(r/c cs\t|tr t|}n t|trt|}t|\}}z dVWt||dSt||w)a# Change the namespace of unique name with :code:`with` statement. After calling it, a new namespace in the context of :code:`with` will be created, and it will number names from zero again when calling :code:`generate()` with same key. Args: new_generator(str|bytes, optional): New name of global namespace. Note that str in Python2 was split into str and bytes in Python3, so here are two types. Default is None. If not None, new_generator will be added into the prefix of unique name generated by :code:`generate()`. Returns: None. Examples: .. code-block:: python >>> import paddle >>> with paddle.utils.unique_name.guard(): ... name_1 = paddle.utils.unique_name.generate('fc') >>> with paddle.utils.unique_name.guard(): ... name_2 = paddle.utils.unique_name.generate('fc') >>> print(name_1, name_2) fc_0 fc_0 >>> with paddle.utils.unique_name.guard('A'): ... name_1 = paddle.utils.unique_name.generate('fc') >>> with paddle.utils.unique_name.guard('B'): ... name_2 = paddle.utils.unique_name.generate('fc') >>> print(name_1, name_2) Afc_0 Bfc_0 N) isinstancerrbytesdecoder/)r,r-r.r r rguards #    r3)NNr)rcopyrwrapped_decoratorr__all__rr$r+r*rrrr/r3r r r rs  -+ 7