# 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.
"""TestDataGenerator factory class.
"""

import logging

from . import exceptions
from . import test_data_generation


class TestDataGeneratorFactory(object):
    """Factory class used to create test data generators.

  Usage: Create a factory passing parameters to the ctor with which the
  generators will be produced.
  """

    def __init__(self, aechen_ir_database_path, noise_tracks_path,
                 copy_with_identity):
        """Ctor.

    Args:
      aechen_ir_database_path: Path to the Aechen Impulse Response database.
      noise_tracks_path: Path to the noise tracks to add.
      copy_with_identity: Flag indicating whether the identity generator has to
                          make copies of the clean speech input files.
    """
        self._output_directory_prefix = None
        self._aechen_ir_database_path = aechen_ir_database_path
        self._noise_tracks_path = noise_tracks_path
        self._copy_with_identity = copy_with_identity

    def SetOutputDirectoryPrefix(self, prefix):
        self._output_directory_prefix = prefix

    def GetInstance(self, test_data_generators_class):
        """Creates an TestDataGenerator instance given a class object.

    Args:
      test_data_generators_class: TestDataGenerator class object (not an
                                  instance).

    Returns:
      TestDataGenerator instance.
    """
        if self._output_directory_prefix is None:
            raise exceptions.InitializationException(
                'The output directory prefix for test data generators is not set'
            )
        logging.debug('factory producing %s', test_data_generators_class)

        if test_data_generators_class == (
                test_data_generation.IdentityTestDataGenerator):
            return test_data_generation.IdentityTestDataGenerator(
                self._output_directory_prefix, self._copy_with_identity)
        elif test_data_generators_class == (
                test_data_generation.ReverberationTestDataGenerator):
            return test_data_generation.ReverberationTestDataGenerator(
                self._output_directory_prefix, self._aechen_ir_database_path)
        elif test_data_generators_class == (
                test_data_generation.AdditiveNoiseTestDataGenerator):
            return test_data_generation.AdditiveNoiseTestDataGenerator(
                self._output_directory_prefix, self._noise_tracks_path)
        else:
            return test_data_generators_class(self._output_directory_prefix)