Thesis-OptimizationModules/macop/utils/progress.py

"""Utils progress `macop` module when verbose enable
"""
import sys


class Colors:
    """Macop color representation
    """
    ENDC = '\033[m'
    GREEN = '\033[32m'
    GREY = '\033[90m'


def macop_text(algo, msg):
    """Display Macop message to user interface

    Args:
        algo: {Algorithm} -- current algorithm instance
        msg: {str} -- message to display
    """
    if algo._verbose:
        print(Colors.GREEN + 'M' + Colors.ENDC + Colors.GREY + 'acop' \
        + Colors.ENDC + Colors.GREEN + ' :: ' + Colors.ENDC \
        + Colors.GREY + msg + Colors.ENDC)


def macop_line(algo):
    """Macop split line

    Args:
        algo: {Algorithm} -- current algorithm instance
    """

    if not algo._verbose:
        return

    line = ''

    for i in range(41):

        if i % 2 == 0:
            line += Colors.GREEN + '----' + Colors.ENDC
        else:
            line += Colors.GREY + '----' + Colors.ENDC

    print(line)


def macop_progress(algo, evaluations, max):
    """Progress line of macop

    Args:
        algo: {Algorithm} -- current algorithm instance
        evaluations: {int} -- current number of evaluations
        max: {int} -- max number of expected evaluations
    """
    if not algo._verbose:
        return

    barWidth = 156

    progress = evaluations / float(max)

    output_str = Colors.GREEN + '[' + Colors.ENDC
    pos = int(barWidth * progress)
    for i in range(barWidth):
        if i < pos:
            output_str = output_str + Colors.GREY + '=' + Colors.ENDC
        elif i == pos:
            output_str = output_str + Colors.GREEN + '>' + Colors.ENDC
        else:
            output_str = output_str + Colors.GREY + ' ' + Colors.ENDC

    output_str = output_str + Colors.GREEN + '] ' + Colors.ENDC + str(
        int(progress * 100.0)) + "%\r"
    print(output_str)
    sys.stdout.write("\033[F")

    # go to line
    if progress >= 1.:
        print()
        macop_line(algo)