| # Copyright (c) 2017 The WebRTC project authors. All Rights Reserved. |
| # |
| # Use of this source code is governed by a BSD-style license |
| # that can be found in the LICENSE file in the root of the source |
| # tree. An additional intellectual property rights grant can be found |
| # in the file PATENTS. All contributing project authors may |
| # be found in the AUTHORS file in the root of the source tree. |
| """EvaluationScore factory class. |
| """ |
| |
| import logging |
| |
| from . import exceptions |
| from . import eval_scores |
| |
| |
| class EvaluationScoreWorkerFactory(object): |
| """Factory class used to instantiate evaluation score workers. |
| |
| The ctor gets the parametrs that are used to instatiate the evaluation score |
| workers. |
| """ |
| |
| def __init__(self, polqa_tool_bin_path, echo_metric_tool_bin_path): |
| self._score_filename_prefix = None |
| self._polqa_tool_bin_path = polqa_tool_bin_path |
| self._echo_metric_tool_bin_path = echo_metric_tool_bin_path |
| |
| def SetScoreFilenamePrefix(self, prefix): |
| self._score_filename_prefix = prefix |
| |
| def GetInstance(self, evaluation_score_class): |
| """Creates an EvaluationScore instance given a class object. |
| |
| Args: |
| evaluation_score_class: EvaluationScore class object (not an instance). |
| |
| Returns: |
| An EvaluationScore instance. |
| """ |
| if self._score_filename_prefix is None: |
| raise exceptions.InitializationException( |
| 'The score file name prefix for evaluation score workers is not set' |
| ) |
| logging.debug('factory producing a %s evaluation score', |
| evaluation_score_class) |
| |
| if evaluation_score_class == eval_scores.PolqaScore: |
| return eval_scores.PolqaScore(self._score_filename_prefix, |
| self._polqa_tool_bin_path) |
| elif evaluation_score_class == eval_scores.EchoMetric: |
| return eval_scores.EchoMetric(self._score_filename_prefix, |
| self._echo_metric_tool_bin_path) |
| else: |
| return evaluation_score_class(self._score_filename_prefix) |
