#! /usr/bin/python

import sys
import os

import helperFunctions

###

def selectActiveLearning(method, configFile=None):

  sys.path.append(os.path.abspath(os.path.dirname(__file__)) + '/activeLearning')

  if (method == 'LinGP') or (method == 'LinGPemoc'):
    import activeLearningLinGPemoc
    return activeLearningLinGPemoc.Classifier(configFile=configFile)

  elif (method == 'LinGPapprox') or (method == 'LinGPemocApprox'):
    import activeLearningLinGPemocApprox
    return activeLearningLinGPemocApprox.Classifier(configFile=configFile)

  ###

  elif (method == 'GPLinK') or (method == 'GPLinKemoc'):
    import activeLearningGPLinKemoc
    return activeLearningGPLinKemoc.Classifier(configFile=configFile)

  ###

  elif method == 'GPGenKemoc':
    import activeLearningGPGenKemoc
    return activeLearningGPGenKemoc.Classifier(configFile=configFile)

  ###

  elif (method == 'LinGPwali') or (method == 'WlinGP1vs2'):
    import activeLearningWlinGP1vs2
    return activeLearningWlinGP1vs2.Classifier(configFile=configFile)

  ###

  else:
    raise Exception('Unknown method %s!'%method)

###

def selectReweighter(method, configFile=None):

  sys.path.append(os.path.abspath(os.path.dirname(__file__)) + '/reweighting')

  if method == 'None':
    import reweightNone
    return reweightNone.Reweighter(configFile=configFile)

  elif method == 'GPLinK':
    import reweightGPLinK
    return reweightGPLinK.Reweighter(configFile=configFile)

  elif method == 'GPGenK':
    import reweightGPGenK
    return reweightGPGenK.Reweighter(configFile=configFile)

  elif method == 'LinGP':
    import reweightLinGP
    return reweightLinGP.Reweighter(configFile=configFile)

  elif (method == 'LinGPwali') or (method == 'WlinGP'):
    import reweightWlinGP
    return reweightWlinGP.Reweighter(configFile=configFile)

  else:
    raise Exception('Unknown method %s!'%method)