pyannote源码阅读（一）
pyannote源码阅读（一）
- 前言
- pyannote-core源码
- Segment类的设计和实现要点
- dataclass装饰器
- 迭代器方法
- 提供的方法
- 字符串和图形表示
- 小结
前言
读一读代码，顺便记录一下笔记。
pyannote-core源码
Segment类的设计和实现要点
from dataclasses import dataclass@dataclass(frozen=True, order=True)
class Segment:start: float = 0.0end: float = 0.0def set_precision(ndigits: Optional[int] = None):global AUTO_ROUND_TIMEglobal SEGMENT_PRECISIONdef __post_init__(self):if AUDIO_ROUND_TIME:object.__setattr__(self, 'start', int(self.start / SEGMENT_PRECISION + 0.5) * SEGMENT_PRECISION)object.__setattr__(self, 'end', int(self.end / SEGMENT_PRECISION + 0.5) * SEGMENT_PRECISION)@propertyduration(self) -> float:return self.end - self.start if self else 0.
dataclass装饰器
dataclass是从Python3.7版本开始，作为标准库中的模块被引入。
可以简化数据类的设计，在上面的代码中，直接定义属性并初始化。
start: float = 0.0end: float = 0.0
frozen=True 创建不可变类，不允许更改属性值。
创建一个Segment对象，可以这样Segment(start=1, end=5) 或者 Segment(1, 5)。
这里有一个细节，如果设置了数值经度，比如保留2位小数，但初始化时使用了三位以上的小数，要如何处理？因为使用了frozen，是无法对属性进行直接修改的。
方法是在__post_init__方法中使用如下代码：
object.__setattr__(self, 'start', int(self.start / SEGMENT_PRECISION + 0.5) * SEGMENT_PRECISION)
顾名思义，__post_init__方法在对象初始化之后调用。
- 全局变量
AUTO_ROUND_TIME 是否四舍五入
SEGMENT_PRECISION 精度
迭代器方法
def __iter__(self): -> Iterator[float]:yield self.startyield self.end
通过yield实现，从而支持
segment = Segment(start, end)
start, end = segment
提供的方法
- copy 复制一个Segment
- contains 判读一个Segment是否包含另一个Segment
- and 求两个Segment的交集
- or 求两个Segment的并集
- intersects 判断连个Segment是否相交
- overlaps 判断一个Segment是否覆盖另一个
字符串和图形表示
def __str__(self):if self:return '[%s --> %s]' % (self._str_helper(self.start), self._str_helper(self.end))
可以把Segment(1337, 1337 + 0.42)表示为:[ 00:22:17.000 --> 00:22:17.420]，very nice。
小结
这是一个非常不错的设计，在音频处理的很多场景都可以参考。