import numpy
from tools.load import LoadMatrix
from shogun.Kernel import DirectorKernel

class DirectorLinearKernel(DirectorKernel):
    def __init__(self):
        DirectorKernel.__init__(self)
    #def kernel_function(self, idx_a, idx_b):
    #   numpy.dot(self.get_lhs(), self.get_rhs())

lm=LoadMatrix()

traindat = numpy.random.random_sample((5,5))
testdat = numpy.random.random_sample((5,5))
parameter_list=[[traindat,testdat,1.2],[traindat,testdat,1.4]]

def kernel_director_linear_modular (fm_train_real=traindat,fm_test_real=testdat,scale=1.2):

    from shogun.Features import RealFeatures
    from shogun.Kernel import LinearKernel, AvgDiagKernelNormalizer

    feats_train=RealFeatures(fm_train_real)
    feats_test=RealFeatures(fm_test_real)

    kernel=LinearKernel()
    #kernel.set_normalizer(AvgDiagKernelNormalizer(scale))
    kernel.init(feats_train, feats_train)

    dkernel=DirectorLinearKernel()
    #dkernel.set_normalizer(AvgDiagKernelNormalizer(scale))
    dkernel.init(feats_train, feats_train)

    km_train=kernel.get_kernel_matrix()
    dkm_train=dkernel.get_kernel_matrix()


    print km_train, dkm_train

    return km_train

if __name__=='__main__':
    print('DirectorLinear')
    kernel_director_linear_modular(*parameter_list[0])