You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
107 lines
2.2 KiB
107 lines
2.2 KiB
5 years ago
|
#!/usr/bin/env python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
"""
|
||
|
|
Quantities: Size
|
||
|
|
"""
|
||
|
|
import re
|
||
|
|
from abc import abstractmethod
|
||
|
|
|
||
|
|
import six
|
||
|
|
|
||
|
|
from ..common import seps
|
||
|
|
|
||
|
|
|
||
|
|
class Quantity(object):
|
||
|
|
"""
|
||
|
|
Represent a quantity object with magnitude and units.
|
||
|
|
"""
|
||
|
|
|
||
|
|
parser_re = re.compile(r'(?P<magnitude>\d+(?:[.]\d+)?)(?P<units>[^\d]+)?')
|
||
|
|
|
||
|
|
def __init__(self, magnitude, units):
|
||
|
|
self.magnitude = magnitude
|
||
|
|
self.units = units
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
@abstractmethod
|
||
|
|
def parse_units(cls, value):
|
||
|
|
"""
|
||
|
|
Parse a string to a proper unit notation.
|
||
|
|
"""
|
||
|
|
raise NotImplementedError
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def fromstring(cls, string):
|
||
|
|
"""
|
||
|
|
Parse the string into a quantity object.
|
||
|
|
:param string:
|
||
|
|
:return:
|
||
|
|
"""
|
||
|
|
values = cls.parser_re.match(string).groupdict()
|
||
|
|
try:
|
||
|
|
magnitude = int(values['magnitude'])
|
||
|
|
except ValueError:
|
||
|
|
magnitude = float(values['magnitude'])
|
||
|
|
units = cls.parse_units(values['units'])
|
||
|
|
|
||
|
|
return cls(magnitude, units)
|
||
|
|
|
||
|
|
def __hash__(self):
|
||
|
|
return hash(str(self))
|
||
|
|
|
||
|
|
def __eq__(self, other):
|
||
|
|
if isinstance(other, six.string_types):
|
||
|
|
return str(self) == other
|
||
|
|
if not isinstance(other, self.__class__):
|
||
|
|
return NotImplemented
|
||
|
|
return self.magnitude == other.magnitude and self.units == other.units
|
||
|
|
|
||
|
|
def __ne__(self, other):
|
||
|
|
return not self == other
|
||
|
|
|
||
|
|
def __repr__(self):
|
||
|
|
return '<{0} [{1}]>'.format(self.__class__.__name__, self)
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
return '{0}{1}'.format(self.magnitude, self.units)
|
||
|
|
|
||
|
|
|
||
|
|
class Size(Quantity):
|
||
|
|
"""
|
||
|
|
Represent size.
|
||
|
|
|
||
|
|
e.g.: 1.1GB, 300MB
|
||
|
|
"""
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def parse_units(cls, value):
|
||
|
|
return value.strip(seps).upper()
|
||
|
|
|
||
|
|
|
||
|
|
class BitRate(Quantity):
|
||
|
|
"""
|
||
|
|
Represent bit rate.
|
||
|
|
|
||
|
|
e.g.: 320Kbps, 1.5Mbps
|
||
|
|
"""
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def parse_units(cls, value):
|
||
|
|
value = value.strip(seps).capitalize()
|
||
|
|
for token in ('bits', 'bit'):
|
||
|
|
value = value.replace(token, 'bps')
|
||
|
|
|
||
|
|
return value
|
||
|
|
|
||
|
|
|
||
|
|
class FrameRate(Quantity):
|
||
|
|
"""
|
||
|
|
Represent frame rate.
|
||
|
|
|
||
|
|
e.g.: 24fps, 60fps
|
||
|
|
"""
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def parse_units(cls, value):
|
||
|
|
return 'fps'
|