#
# Copyright 2012-2013 The libLTE Developers. See the
# COPYRIGHT file at the top-level directory of this distribution.
#
# This file is part of the libLTE library.
#
# libLTE is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as
# published by the Free Software Foundation, either version 3 of
# the License, or (at your option) any later version.
#
# libLTE is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Lesser General Public License for more details.
#
# A copy of the GNU Lesser General Public License can be found in
# the LICENSE file in the top-level directory of this distribution
# and at http://www.gnu.org/licenses/.
#



import shutil
import os

def MakeModule(m):
    dir = os.path.dirname(__file__)

    tdir = os.path.join(dir, 'template')
    ddir = tdir + m.directory_name
    print 'Creating new directory ' + ddir
    
    shutil.copytree(tdir,ddir)
    
    WriteAloeC(m,tdir + '/src/template.c',\
                    ddir + '/src/' + m.name + '.c')
    WriteAloeH(m,tdir + '/src/template.h',\
                    ddir + '/src/' + m.name + '.h')
    

def WriteInitParamsGet(m, f):
    for p in m.init_params:
        f.write('\tif (param_get_' + p['type'] + '_name("' + p['name'] + '", &' + p['variable']+ ')) {\n'
                    '\t\t' + p['variable'] + ' = ' + str(p['default']) + ';\n'\
                '\t}\n')
    
def WriteInputParamsHandlers(m, f):
    for p in m.input_params:
        f.write('pmid_t ' + p['name'] + '_id;\n')

def WriteInputParamsId(m, f):
    for p in m.input_params:
        f.write('\t' + p['name'] + '_id = param_id("' + p['name'] + '");\n')
        
def WriteInputParamsGet(m, f):
    for p in m.input_params:
        f.write('\tif (param_get_' + p['type'] + '(' + p['name'] + '_id, &' + p['variable'] + ') != 1) {\n')
        if p['default'] == None:
            f.write('\t\t' + r'moderror("Error getting parameter ' + p['name'] + '\\n");' + '\n')
            f.write('\t\treturn -1;\n\t}\n')                      
        else:
            f.write('\t\t' + r'modinfo("Parameter ' + p['name'] + ' not found. Setting to default (' + p['default'] + ')\\n");' + '\n')
            f.write('\t\t' + p['variable'] + ' = ' + p['default'] + ';\n\t}\n')
            
    
def WriteAloeC(m, input, output):
    with open(input,'r') as f:
        newlines = []
        for line in f.readlines():
            newlines.append(line.replace('-name-', m.name))
    with open(output, 'w') as f:
        for line in newlines:

            if '--input_parameters_handlers--' in line:
                WriteInputParamsHandlers(m, f)
            elif '--input_parameters_getid--' in line:
                WriteInputParamsId(m, f)
            elif '--init_parameters--' in line:
                WriteInitParamsGet(m, f)
            elif '--input_parameters--' in line:
                WriteInputParamsGet(m, f)
            else: 
                f.write(line)

def WriteAloeH(m, input, output):
    with open(input,'r') as f:
        newlines = []
        for line in f.readlines():
            if '-typeinput-' in line:
                if m.nof_inputs > 0: 
                    newlines.append(line.replace('-typeinput-', m.input_type))
            elif '-numinputs-' in line:
                newlines.append(line.replace('-numinputs-', str(m.nof_inputs)))
            elif '-sizeinput-' in line:
                if m.nof_inputs > 0: 
                    newlines.append(line.replace('-sizeinput-', m.input_size))
                else:                                        
                    newlines.append(line.replace('-sizeinput-', '0'))
            elif '-typeoutput-' in line:
                if m.nof_outputs > 0: 
                    newlines.append(line.replace('-typeoutput-', m.output_type))
            elif '-numoutputs-' in line:
                newlines.append(line.replace('-numoutputs-', str(m.nof_outputs)))
            elif '-sizeoutput-' in line:
                if m.nof_outputs > 0: 
                    newlines.append(line.replace('-sizeoutput-', m.output_size))
                else:
                    newlines.append(line.replace('-sizeoutput-', '0'))
            else:
                 newlines.append(line)
                  
    with open(output, 'w') as f:
        for line in newlines:          
              f.write(line)