U
    Vc
	                     @   sF   d Z ddlmZ ddlmZ dZdZejjZddddZ	d	d
 Z
dS )z,Internal information about the audio plugin.    )summary_pb2)plugin_data_pb2audioN)converted_to_tensorc                C   s4   t jt||d}tj| |tjjt| dd}|S )zuCreate a `SummaryMetadata` proto for audio plugin data.

    Returns:
      A `SummaryMetadata` protobuf object.
    )versionencodingr   )Zplugin_namecontent)display_nameZsummary_descriptionZplugin_data)r   AudioPluginDataPROTO_VERSIONr   ZSummaryMetadataZ
PluginDataPLUGIN_NAMEZSerializeToString)r	   descriptionr   r   r   metadata r   F/tmp/pip-unpacked-wheel-g8kmtpbc/tensorboard/plugins/audio/metadata.pycreate_summary_metadata   s     r   c                 C   s0   t | tstdtj| }|jdkr,|S |S )zParse summary metadata to a Python object.

    Arguments:
      content: The `content` field of a `SummaryMetadata` proto
        corresponding to the audio plugin.

    Returns:
      An `AudioPluginData` protobuf object.
    zContent type must be bytesr   )
isinstancebytes	TypeErrorr   r
   Z
FromStringr   )r   resultr   r   r   parse_plugin_metadata6   s    


r   )__doc__Ztensorboard.compat.protor   Ztensorboard.plugins.audior   r   r   r
   Encodingr   r   r   r   r   r   <module>   s   