from enum import Enum, unique from itertools import groupby from typing import Dict, Iterable, Tuple from ereuse_devicehub.resources.action.models import BenchmarkDataStorage, BenchmarkProcessor, \ BenchmarkProcessorSysbench, RateComputer from ereuse_devicehub.resources.action.rate.rate import BaseRate from ereuse_devicehub.resources.device.models import Computer, DataStorage, Processor, \ RamModule class RateAlgorithm(BaseRate): """The algorithm that generates the Rate v1.0. Rate v1.0 rates only computers, counting their processor, ram, data storage, appearance and functionality. This rate is only triggered by a Snapshot from Workbench that has a VisualTest. The algorithm is as follows: 1. Specialized subclasses of :class:`BaseRate` compute a rating for each component. To perform this, each class normalizes first the characteristics and benchmarks of the components between 0 and 1, and then they merge the values with specific formulas for each components to get a resulting score. The classes are: * :class:`ProcessorRate`, using cores, speed, and ``BenchmarkProcessor``. * :class:`RamRate`, using the total of RAM size and speed. * :class:`DataStorageRate`, using the total of disk capacity, and ``BenchmarkDataStorage``. 2. Merge the components individual rates into a single rate for all components, using a weighted harmonic mean of 50% for the processor rating, 20% for the data storage rating, and 30% for the RAM rating. 3. Merge the rate for the components with the appearance and functionality from :class:`VisualTest`. ``Final Rate = Components Rate + Functionality Rate + Appearance Rate``. The value is between 0 and 4.7, included. """ @unique class Appearance(Enum): Z = 0.5 A = 0.3 B = 0 C = -0.2 D = -0.5 E = -1.0 @unique class Functionality(Enum): A = 0.4 B = -0.5 C = -0.75 D = -1 def __init__(self) -> None: super().__init__() self.RATES = { # composition: type: (field, compute class) Processor.t: ('processor', ProcessorRate()), RamModule.t: ('ram', RamRate()), DataStorage.t: ('data_storage', DataStorageRate()) } # type: Dict[str, Tuple[str, BaseRate]] def compute(self, device: Computer) -> RateComputer: """Generates a new :class:`ereuse_devicehub.resources.action.models.RateComputer` for the passed-in device. Do not call directly this class, but use :meth:`ereuse_devicehub.resources.action.models.RateComputer.compute`, which then calls this. """ assert isinstance(device, Computer), 'Can only rate computers' rate = RateComputer() rate.processor = rate.data_storage = rate.ram = 1 # Init # Group cpus, rams, storage and compute their rate # Treat the same way with HardDrive and SolidStateDrive like (DataStorage) clause = lambda x: DataStorage.t if isinstance(x, DataStorage) else x.t c = (c for c in device.components if clause(c) in set(self.RATES.keys())) for type, components in groupby(sorted(c, key=clause), key=clause): if type == Processor.t: # ProcessorRate.compute expects only 1 processor components = next(components) field, rate_cls = self.RATES[type] result = rate_cls.compute(components) if result: setattr(rate, field, result) rate_components = self.harmonic_mean_rates(rate.processor, rate.data_storage, rate.ram) rate.rating = rate_components device.actions_one.add(rate) assert 0 <= rate.rating return rate class ProcessorRate(BaseRate): """Calculate a ProcessorRate of all Processor devices.""" # processor.xMin, processor.xMax PROCESSOR_NORM = 3196.17, 17503.81 DEFAULT_CORES = 1 DEFAULT_SPEED = 1.6 DEFAULT_SCORE = 4000 def compute(self, processor: Processor): """Compute processor rate We assume always exists a Benchmark Processor. Obs: cores and speed are possible NULL value :return: result is a rate (score) of Processor characteristics """ cores = processor.cores or self.DEFAULT_CORES speed = processor.speed or self.DEFAULT_SPEED benchmark_cpu = next( e for e in reversed(processor.actions) if isinstance(e, BenchmarkProcessor) and not isinstance(e, BenchmarkProcessorSysbench) ) benchmark_cpu = benchmark_cpu.rate or self.DEFAULT_SCORE # STEP: Fusion components processor_rate = (benchmark_cpu + speed * 2000 * cores) / 2 # STEP: Normalize values processor_norm = max(self.norm(processor_rate, *self.PROCESSOR_NORM), 0) # STEP: Compute rate/score from every component # Calculate processor_rate if processor_norm >= self.CEXP: processor_rate = self.rate_exp(processor_norm) if self.CLIN <= processor_norm < self.CLOG: processor_rate = self.rate_lin(processor_norm) if processor_norm >= self.CLOG: processor_rate = self.rate_log(processor_norm) return processor_rate class RamRate(BaseRate): """Calculate a RamRate of all RamModule devices.""" # ram.size.xMin; ram.size.xMax SIZE_NORM = 256, 8192 RAM_SPEED_NORM = 133, 1333 # ram.speed.factor RAM_SPEED_FACTOR = 3.7 # ram.size.weight; ram.speed.weight; RAM_WEIGHTS = 0.7, 0.3 def compute(self, ram_devices: Iterable[RamModule]): """If ram speed or ram size, we assume default values before declared. :return: result is a rate (score) of all RamModule components """ size = 0.0 speed = 0.0 # STEP: Filtering, data cleaning and merging of component parts for ram in ram_devices: _size = ram.size or 0 size += _size if ram.speed: speed += (ram.speed or 0) * _size else: speed += (_size / self.RAM_SPEED_FACTOR) * _size # STEP: Fusion components # To guarantee that there will be no 0/0 if size: speed /= size # STEP: Normalize values size_norm = max(self.norm(size, *self.SIZE_NORM), 0) ram_speed_norm = max(self.norm(speed, *self.RAM_SPEED_NORM), 0) # STEP: Compute rate/score from every component # Calculate size_rate if self.CEXP <= size_norm < self.CLIN: size_rate = self.rate_exp(size_norm) if self.CLIN <= size_norm < self.CLOG: size_rate = self.rate_lin(size_norm) if size_norm >= self.CLOG: size_rate = self.rate_log(size_norm) # Calculate ram_speed_rate if self.CEXP <= ram_speed_norm < self.CLIN: ram_speed_rate = self.rate_exp(ram_speed_norm) if self.CLIN <= ram_speed_norm < self.CLOG: ram_speed_rate = self.rate_lin(ram_speed_norm) if ram_speed_norm >= self.CLOG: ram_speed_rate = self.rate_log(ram_speed_norm) # STEP: Fusion Characteristics return self.harmonic_mean(self.RAM_WEIGHTS, rates=(size_rate, ram_speed_rate)) class DataStorageRate(BaseRate): """Calculate the rate of all DataStorage devices.""" # drive.size.xMin; drive.size.xMax SIZE_NORM = 4, 265000 READ_SPEED_NORM = 2.7, 109.5 WRITE_SPEED_NORM = 2, 27.35 # drive.size.weight; drive.readingSpeed.weight; drive.writingSpeed.weight; DATA_STORAGE_WEIGHTS = 0.5, 0.25, 0.25 def compute(self, data_storage_devices: Iterable[DataStorage]): """Obs: size != NULL and 0 value & read_speed and write_speed != NULL :return: result is a rate (score) of all DataStorage devices """ size = 0 read_speed = 0 write_speed = 0 # STEP: Filtering, data cleaning and merging of component parts for storage in data_storage_devices: # We assume all hdd snapshots have BenchmarkDataStorage benchmark = storage.last_action_of(BenchmarkDataStorage) # prevent NULL values _size = storage.size or 0 size += _size read_speed += benchmark.read_speed * _size write_speed += benchmark.write_speed * _size # STEP: Fusion components # Check almost one storage have size, try catch exception 0/0 if size: read_speed /= size write_speed /= size # STEP: Normalize values size_norm = max(self.norm(size, *self.SIZE_NORM), 0) read_speed_norm = max(self.norm(read_speed, *self.READ_SPEED_NORM), 0) write_speed_norm = max(self.norm(write_speed, *self.WRITE_SPEED_NORM), 0) # STEP: Compute rate/score from every component # Calculate size_rate if size_norm >= self.CLOG: size_rate = self.rate_log(size_norm) elif self.CLIN <= size_norm < self.CLOG: size_rate = self.rate_lin(size_norm) elif self.CEXP <= size_norm < self.CLIN: size_rate = self.rate_exp(size_norm) # Calculate read_speed_rate if read_speed_norm >= self.CLOG: read_speed_rate = self.rate_log(read_speed_norm) elif self.CLIN <= read_speed_norm < self.CLOG: read_speed_rate = self.rate_lin(read_speed_norm) elif self.CEXP <= read_speed_norm < self.CLIN: read_speed_rate = self.rate_exp(read_speed_norm) # write_speed_rate if write_speed_norm >= self.CLOG: write_speed_rate = self.rate_log(write_speed_norm) elif self.CLIN <= write_speed_norm < self.CLOG: write_speed_rate = self.rate_lin(write_speed_norm) elif self.CEXP <= write_speed_norm < self.CLIN: write_speed_rate = self.rate_exp(write_speed_norm) # STEP: Fusion Characteristics return self.harmonic_mean(self.DATA_STORAGE_WEIGHTS, rates=(size_rate, read_speed_rate, write_speed_rate)) rate_algorithm = RateAlgorithm() class CannotRate(Exception): pass