main.py

"""Project META.
by Baptiste MASSET & Norbert Feron


Usage:
    main.py enum <path> [--nbk=N --delta_max=N --mu=N]
    main.py hc <path> [--nbk=N --delta_max=N --mu=N] --max_eval=N --iter=N --move=op [-r] [--mproc]
    main.py ts <path> [--nbk=N --delta_max=N --mu=N --tabu=N] --max_eval=N --iter=N --move=op [--mproc]
    main.py sa <path> [--nbk=N --delta_max=N --mu=N] --temp=N --alpha=N --max_eval=N --iter=N --move=op [--mproc]
    main.py (-h | --help)
    main.py --version
Arguments:
    <path>  The path of the graph file
Options:
    --nbk=N             Set number of classes for graph partitionning [default: 3]
    --delta_max=N       Set the max differences of elements between classes [default: 3]
    --mu=N              Set MU value [default: 0.5]
    --temp=N            Set the start temperature for simulated annealing [default: 100]
    --alpha=N           Set the cooling coefficient for simulated annealing [default: 0.95]
    --max_eval=N        Set the max number of evaluating a solution
    --iter=N            Set the number of repeated execution (for loop)
    --tabu=N            Set the tabu queue size [default: 10]
    -r                  Restart hillclimb until max_eval is reached
    --move=op           Set the move operator for elementary move to another solution ( pNd ; swap ; sweep )
    --mproc             Running multi proc version of algorithms
    -h --help           Show this screen.
    --version           Show version.
"""
import docopt
import sys
import os
import logging
from enumerate import test_enumerate
from hillclimb import test_hillclimb
from tabusearch import test_tabusearch
from simulatedannealing import test_anneal
from multi_proc import mproc_hillclimb, mproc_tabu, mproc_anneal
from graphstructure.lectureFichier import Reader
from tools.voisinageGraphe import pick_gen, swap_gen, sweep_gen

log = logging.getLogger(__name__)

if __name__ == '__main__':
    logging.basicConfig(stream=sys.stdout, level=logging.DEBUG)

    arguments = docopt.docopt(__doc__, version='Project META v1.0')
    reader = Reader(arguments['<path>'])
    graph = reader.g
    nbk = int(arguments['--nbk'])
    delta_max = int(arguments['--delta_max'])
    mu = float(arguments['--mu'])

    logsPath = "./logs"

    if not os.path.isdir(logsPath):
        os.mkdir(logsPath)

    if arguments['enum']:
        test_enumerate.main(graph, nbk, delta_max, mu, logsPath)
    else:
        max_eval = int(arguments['--max_eval'])
        iter = int(arguments['--iter'])
        global move_operator
        if 'pNd' in arguments['--move']:
            move_operator = pick_gen
        elif 'swap' in arguments['--move']:
            move_operator = swap_gen
        elif 'sweep' in arguments['--move']:
            move_operator = sweep_gen

        if arguments['hc']:
            if arguments['--mproc']:
                mproc_hillclimb.main(graph, nbk, delta_max, mu, max_eval, iter, move_operator, logsPath, arguments['-r'])
            else:
                test_hillclimb.main(graph, nbk, delta_max, mu, max_eval, iter, move_operator,logsPath, arguments['-r'])

        elif arguments['ts']:
            tabuSize = int(arguments['--tabu'])
            if arguments['--mproc']:
                mproc_tabu.main(graph, nbk, delta_max, mu, max_eval, iter, move_operator, tabuSize, logsPath)
            else:
                test_tabusearch.main(graph, nbk, delta_max, mu, max_eval, iter, move_operator, tabuSize, logsPath)
        elif arguments['sa']:
            temp = int(arguments['--temp'])
            alpha = float(arguments['--alpha'])
            if arguments['--mproc']:
                mproc_anneal.main(graph, nbk, delta_max, mu, temp, alpha, max_eval, iter, move_operator, logsPath)
            else:
                test_anneal.main(graph, nbk, delta_max, mu, temp, alpha, max_eval, iter, move_operator, logsPath)