Colorize compilation order

[jgpallero]

It would be nice that, as old colormake.pl, colormake.py could colorize the compilation order too. The feature implementation is trivial and only a new (pattern,color scheme) is needed. The former colormake oly supports the gcc family compiler so I maintain a fork of colormake.pl in which lots of another compiler names are included (https://bitbucket.org/jgpallero/colormake).

I have made some tests with colormake.py and the new proposed feature and works well. I have added a new line to the patterns vector:

(r'gcc', col_magenta+col_brighten)

It works, but not always. When the compilation order it broken in the makefile as, for example

gcc -c -Wall
file.c -o file.o

The line after '' character is not colorized. And I don't know how fix this bug

Thanks

[andrzej-bieniek]

I wonder if we could collect all lines with '' as a last character (until we get line without it), then pass it as a one line to line_get_color function, then apply this color to all collected lines. Would that work?

How colormake.pl does it?

[jgpallero]

Hello:

I've added support to colormake to detect a list of compilers in order to colorize the compilation order and parse correctly multiline compilation orders.
I have modified too the code in order to detect more general warning and error messages emited by other compilers than gcc.

Here, the code:

#!/usr/bin/env python

'''
colormake
---------

A script to color the output of make,
ripping-off colormake.pl by Bjarni R.
Einarsson, with some small changes and
improvements.

Daniel C. Jones <dcjones@cs.washington.edu>
2011.06.01.20.30
J. L. G. Pallero <jgpallero@gmail.com>
2012.10.18.12.10

'''

import sys
import re

# Some useful color codes.

col_black =        '\033[30m'
col_red =          '\033[31m'
col_green =        '\033[32m'
col_yellow =       '\033[33m'
col_blue =         '\033[34m'
col_magenta =      '\033[35m'
col_cyan =         '\033[36m'
col_ltgray =       '\033[37m'

col_norm =         '\033[00m'
col_background =   '\033[07m'
col_brighten =     '\033[01m'
col_underline =    '\033[04m'
col_blink =        '\033[05m'

# Colors to print

col_compilation      = col_magenta
col_compilation_last = col_yellow
col_warning          = col_green   + col_brighten
col_error            = col_red     + col_brighten
col_filename         = col_blue    + col_brighten
col_make             = col_cyan
col_makeall          = col_cyan    + col_brighten
col_silent           = col_norm    + col_brighten
col_default          = col_ltgray
col_normal           = col_norm




# This script works simply by matching regular expressions and wrapping the
# matches in ANSI terminal codes. 'patterns' is an array of (pattern, prefix)
# pairs. If a pattern is matched, the prefix is inserted before the line.

# list of compilers (regular expressions)
compilers = (r'^(g|llvm-g|path|open|uh|t|p|nv|nw|ow|tendra|m|plain)?cc',
             r'^(i|sun|pg|l)?cc',
             r'^(open|uh|path|sun|pg)?CC',
             r'^(g|clang|c|llvm-g|llvm-c|xlc|gxlc)\+\+',
             r'^(mkoctfile|clang|ack|vc|c89|c99|icpc|xlc|gxlc|xlC|gxlC)',
             r'^(g|f|sunf|pgf|fort)77',
             r'^(f|openf|uhf|pathf|sunf|pgf|xlf)90',
             r'^(g|f|openf|uhf|pathf|sunf|pgf|xlf)95',
             r'^(gfortran|pgfortran|xlf|xlf2003,f2003|ifort)',
             r'^(ar|ld|cpp|mcxx|plaincxx)',
             r'^((libtool: )(compile:|link:))',
             r'^(i386-|i486-|i586-|i686-|x86_64-)'
            )

# common patterns
filename_pat = r'[A-z0-9_\/\. ]+'
gcc_msg_pat  = filename_pat + r':( In | At |\d+:?\d+:)'

patterns = [
            # make messages
            (r'^make\[\d+\]:',  col_make),
            (r'^Making all in', col_makeall),

            # gcc warnings, errors, etc.
            (r'warning:',            col_warning),
            (r'error:',              col_error),
            (r'cannot find',         col_error),
            (r'compilation aborted', col_error),
            (gcc_msg_pat,            col_filename),

            # normal
            (r'', col_default)
           ]

patterns = [(re.compile(pat[0]),pat[1]) for pat in patterns]

def line_add_color(line):
    line_out = ''
    line_compilation = False

    for i in compilers:
        pat = re.compile(i)
        if pat.match(line):
            line_out += col_compilation
            line_compilation = True
            break

    if line_compilation==False:
        for (pat, form) in patterns:
            if pat.search(line):
                line_out += form
                break

    line_out += line
    line_out += col_normal
    return line_out

def line_add_color_compilation(line,last):
    if last==True:
        return col_compilation + line + col_normal
    else:
        return col_compilation_last + line + col_normal


if __name__ == '__main__':
    #identifier
    previous_comp_order = 0
    #scanning lines
    for line in iter(sys.stdin.readline, ''):
        #suppress possible newline at the end of line
        line = line.rstrip()
        #check if the end is the character '\'
        if (len(line)>0) and (line[-1]==chr(92)):
            last_char = True
        else:
            last_char = False
        #check if the previous line was a compiler invocation
        if previous_comp_order==True:
            #force color to compiler invocation
            line_color = line_add_color_compilation(line,last_char)
        else:
            #color
            line_color = line_add_color(line)
        #update last character identifier
        previous_comp_order = last_char
        #newline
        line_color += '\n'
        #writing line
        sys.stdout.write(line_color)