# 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)