#!/usr/bin/env python
#
##############################################################################
#
# MODULE:       Minimum Redundancy Maximum Relevance Feature Selection
#
# AUTHOR(S):    Steven Pawley
#
##############################################################################
#%module
#% description: Perform Minimum Redundancy Maximum Relevance Feature Selection on a GRASS Attribute Table 
#%end

#%option G_OPT_V_INPUT
#% description: Vector features
#% key: table
#% required : yes
#%end

#%option G_OPT_V_FIELD
#% key: layer
#% required : yes
#%end

#%option
#% description: Discretization threshold
#% key: threshold
#% type: double
#% answer: 1.0
#% required : no
#% guisection: Options
#%end

#%option
#% description: Number of features (attributes)
#% key: nfeatures
#% type: integer
#% answer: 50
#% required : yes
#% guisection: Options
#%end

#%option
#% description: Maximum number of samples
#% key: nsamples
#% type: integer
#% answer: 1000
#% required : yes
#% guisection: Options
#%end

#%option
#% description: Maximum number of variables/attributes
#% key: maxvar
#% type: integer
#% answer: 10000
#% required : yes
#% guisection: Options
#%end

#%option
#% description: Feature selection method
#% key: method
#% type: string
#% options: MID,MIQ 
#% answer: MID
#% required : yes
#% guisection: Options
#%end

import sys
import os
import subprocess
import shutil

import grass.script as grass
import tempfile
import atexit
import os.path
  
# env = grass.gisenv()
# gisdbase = env['GISDBASE']
# location = env['LOCATION_NAME']
# mapset = env['MAPSET']
# path = os.path.join(gisdbase, location, mapset, 'sqlite.db')

tmpdir = tempfile.mkdtemp()
tmptable = "mrmrdat.csv"

def cleanup():
    shutil.rmtree(tmpdir)
    return 0

def main():
    table = options['table']
    layer = options['layer']
    threshold = options['threshold']
    nfeatures = options['nfeatures']
    maxvar = options['maxvar']
    nsamples = options['nsamples']
    method = options['method']

    os.chdir(tmpdir)

    grass.run_command("v.out.ogr",
                      input = table,
                      layer= layer,
                      type = 'auto',
                      output = tmpdir + '/' + tmptable,
                      format = 'CSV',
                      flags = 's')

    mrmrcmd = 'mrmr -i ' + tmptable +' -m ' + method + ' -t ' + threshold + ' -n ' + nfeatures + ' -s ' + nsamples + ' -v ' + maxvar

    subprocess.call(mrmrcmd, shell=True)

    return 0

if __name__ == "__main__":
    options, flags = grass.parser()
    atexit.register(cleanup)
    sys.exit(main())