Пример #1
0
import sys
import os
import re

import zxbpplex
import zxbasmpplex
import ply.yacc as yacc

from zxbpplex import tokens
from common import OPTIONS
from output import msg, warning, error, CURRENT_FILE
from prepro import DefinesTable, ID, MacroCall, Arg, ArgList
from prepro.exceptions import PreprocError

OPTIONS.add_option_if_not_defined('Sinclair', bool, False)

OUTPUT = ''
INCLUDED = {}    # Already included files (with lines)
SPACES = re.compile(r'[ \t]+')

# Set to BASIC or ASM depending on the Lexer context
# e.g. for .ASM files should be set to zxbppasmlex.Lexer()
# Use setMode('ASM' or 'BASIC') to change this FLAG
LEXER = zxbpplex.Lexer()

# CURRENT working directory for this cpp

precedence = (
    ('left', 'DUMMY'),
    ('left', 'EQ', 'NE', 'LT', 'LE', 'GT', 'GE'),
Пример #2
0
def main(argv):
    ''' Entry point when executed from command line.
    You can use zxb.py as a module with import, and this
    function won't be executed.
    '''
    global FLAG_use_BASIC, FLAG_autorun
    global FILE_input, FILE_output, FILE_output_ext, OPTIONS_asm

    OPTIONS.add_option_if_not_defined('memoryCheck', bool, False)
    OPTIONS.add_option_if_not_defined('strictBool', bool, False)
    OPTIONS.add_option_if_not_defined('arrayCheck', bool, False)
    OPTIONS.add_option_if_not_defined('array_base', int, 0)
    OPTIONS.add_option_if_not_defined('string_base', int, 0)
    OPTIONS.add_option_if_not_defined('enableBreak', bool, False)
    OPTIONS.add_option_if_not_defined('emmitBackend', bool, False)
    OPTIONS.add_option_if_not_defined('arch', str, 'zx48k')
    OPTIONS.add_option_if_not_defined('__DEFINES', dict, {})
    OPTIONS.add_option_if_not_defined('explicit', bool, False)

    # ------------------------------------------------------------
    # Command line parsing
    # ------------------------------------------------------------

    parser = OptionParser(usage='Usage: %prog <input file> [options]',
                          version='%prog ' + VERSION)

    parser.add_option(
        "-d",
        "--debug",
        action="count",
        dest="debug",
        default=OPTIONS.Debug.value,
        help=
        "Enable verbosity/debugging output. Additional -d increase verbosity/debug level."
    )

    parser.add_option("-O",
                      "--optimize",
                      type="int",
                      dest="optimization_level",
                      help="Sets optimization level. 0 = None",
                      default=DEFAULT_OPTIMIZATION_LEVEL)

    parser.add_option(
        "-o",
        "--output",
        type="string",
        dest="output_file",
        help="Sets output file. Default is input filename with .bin extension",
        default=None)

    parser.add_option("-T",
                      "--tzx",
                      action="store_true",
                      dest="tzx",
                      default=False,
                      help="Sets output format to tzx (default is .bin)")

    parser.add_option("-t",
                      "--tap",
                      action="store_true",
                      dest="tap",
                      default=False,
                      help="Sets output format to tap (default is .bin)")

    parser.add_option(
        "-B",
        "--BASIC",
        action="store_true",
        dest="basic",
        default=False,
        help=
        "Creates a BASIC loader which load the rest of the CODE. Requires -T ot -t"
    )

    parser.add_option("-a",
                      "--autorun",
                      action="store_true",
                      dest="autorun",
                      default=False,
                      help="Sets the program to be run once loaded")

    parser.add_option("-A",
                      "--asm",
                      action="store_true",
                      dest="asm",
                      default=False,
                      help="Sets output format to asm")

    parser.add_option("-S",
                      "--org",
                      type="int",
                      dest="org",
                      help="Start of machine code. By default %i" %
                      OPTIONS.org.value,
                      default=OPTIONS.org.value)

    parser.add_option(
        "-e",
        "--errmsg",
        type="string",
        dest="stderr",
        default=OPTIONS.StdErrFileName.value,
        help="Error messages file (standard error console by default)")

    parser.add_option("--array-base",
                      type="int",
                      dest="array_base",
                      default=OPTIONS.array_base.value,
                      help="Default lower index for arrays (0 by default)")

    parser.add_option("--string-base",
                      type="int",
                      dest="string_base",
                      default=OPTIONS.string_base.value,
                      help="Default lower index for strings (0 by default)")

    parser.add_option(
        "-Z",
        "--sinclair",
        action="store_true",
        dest="sinclair",
        default=False,
        help=
        "Enable by default some more original ZX Spectrum Sinclair BASIC features: ATTR, SCREEN$, POINT"
    )

    parser.add_option("-H",
                      "--heap-size",
                      type="int",
                      dest="heap_size",
                      default=OPTIONS.heap_size.value,
                      help="Sets heap size in bytes (default %i bytes)" %
                      OPTIONS.heap_size.value)

    parser.add_option("--debug-memory",
                      action="store_true",
                      dest="debug_memory",
                      default=False,
                      help="Enables out-of-memory debug")

    parser.add_option("--debug-array",
                      action="store_true",
                      dest="debug_array",
                      default=False,
                      help="Enables array boundary checking")

    parser.add_option("--strict-bool",
                      action="store_true",
                      dest="strict_bool",
                      default=False,
                      help="Enforce boolean values to be 0 or 1")

    parser.add_option("--enable-break",
                      action="store_true",
                      dest="enable_break",
                      default=False,
                      help="Enables program execution BREAK detection")

    parser.add_option("-E",
                      "--emmit-backend",
                      action="store_true",
                      dest="emmit_backend",
                      default=False,
                      help="Emmits backend code instead of ASM or binary")

    parser.add_option(
        "--explicit",
        action="store_true",
        dest="explicit",
        default=False,
        help="Requires all variables and functions to be declared before used")

    parser.add_option(
        "-D",
        "--define",
        type="str",
        dest="defines",
        action="append",
        help="Defines de given macro. Eg. -D MYDEBUG or -D NAME=Value")

    (options, args) = parser.parse_args()

    if len(args) != 1:
        parser.error("missing input file. (Try -h)")
        return 3

    # ------------------------------------------------------------
    # Setting of internal parameters according to command line
    # ------------------------------------------------------------

    OPTIONS.Debug.value = options.debug
    asmparse.FLAG_optimize = OPTIONS.optimization.value = options.optimization_level
    asmparse.FILE_output = OPTIONS.outputFileName.value = FILE_output = options.output_file
    asmparse.FILE_stderr = OPTIONS.StdErrFileName.value = options.stderr
    OPTIONS.array_base.value = options.array_base
    OPTIONS.string_base.value = options.string_base
    OPTIONS.Sinclair.value = options.sinclair
    OPTIONS.org.value = options.org
    OPTIONS.heap_size.value = options.heap_size
    OPTIONS.memoryCheck.value = options.debug_memory
    OPTIONS.strictBool.value = options.strict_bool or OPTIONS.Sinclair.value
    OPTIONS.arrayCheck.value = options.debug_array
    OPTIONS.emmitBackend.value = options.emmit_backend
    OPTIONS.enableBreak.value = options.enable_break
    OPTIONS.explicit.value = options.explicit

    if options.defines:
        for i in options.defines:
            name, val = tuple(i.split('=', 1))
            OPTIONS.__DEFINES.value[name] = val
            zxbpp.ID_TABLE.define(name, lineno=0)

    if OPTIONS.Sinclair.value:
        OPTIONS.array_base.value = 1
        OPTIONS.string_base.value = 1
        OPTIONS.strictBool.value = True

    debug.ENABLED = OPTIONS.Debug.value

    if int(options.tzx) + int(options.tap) + int(options.asm) + int(
            options.emmit_backend) > 1:
        parser.error(
            "Options --tap, --tzx, --emmit-backend and --asm are excluyent")
        return 3

    asmparse.FLAG_use_BASIC = options.basic
    backend.FLAG_autostart = asmparse.FLAG_autorun = options.autorun

    if asmparse.FLAG_use_BASIC and not options.tzx and not options.tap:
        parser.error(
            'Option --BASIC and --autorun requires --tzx or tap format')
        return 4

    if options.tzx:
        FILE_output_ext = 'tzx'

    elif options.tap:
        FILE_output_ext = 'tap'

    elif options.asm:
        FILE_output_ext = 'asm'

    elif options.emmit_backend:
        FILE_output_ext = 'ic'

    if not os.path.exists(args[0]):
        parser.error("No such file or directory: '%s'" % args[0])
        return 2

    if OPTIONS.memoryCheck.value:
        OPTIONS.__DEFINES.value['__MEMORY_CHECK__'] = ''
        zxbpp.ID_TABLE.define('__MEMORY_CHECK__', lineno=0)

    if OPTIONS.arrayCheck.value:
        OPTIONS.__DEFINES.value['__CHECK_ARRAY_BOUNDARY__'] = ''
        zxbpp.ID_TABLE.define('__CHECK_ARRAY_BOUNDARY__', lineno=0)

    zxbpp.main(args)
    asmparse.FILE_output_ext = FILE_output_ext
    input = zxbpp.OUTPUT
    asmparse.FILE_input = FILE_input = zxbparser.FILENAME = os.path.basename(
        args[0])

    if FILE_output is None:
        OPTIONS.outputFileName.value = FILE_output = os.path.splitext(
            os.path.basename(FILE_input))[0] + '.' + FILE_output_ext
        asmparse.FILE_output = FILE_output

    if OPTIONS.StdErrFileName.value is not None:
        FILE_stderr = asmparse.FILE_stderr = OPTIONS.StdErrFileName.value
        OPTIONS.stderr.value = open(FILE_stderr, 'wt')

    zxbparser.parser.parse(input,
                           lexer=zxblex.lexer,
                           tracking=True,
                           debug=(OPTIONS.Debug.value > 2))

    if gl.has_errors:
        return 1  # Exit with errors

    zxbtrad.traverse(zxbparser.ast)  # This will fill MEMORY with code
    zxbtrad.traverse(
        zxbtrad.FUNCTIONS)  # This will fill MEMORY with pending functions
    zxbtrad.emmit_strings()

    if OPTIONS.emmitBackend.value:
        output_file = open(FILE_output, 'wt')
        for quad in zxbtrad.dumpMemory(MEMORY):
            output_file.write(str(quad) + '\n')

        MEMORY[:] = []  # Empties memory
        zxbtrad.traverse(
            zxbparser.data_ast
        )  # This will fill MEMORY with global declared variables

        for quad in zxbtrad.dumpMemory(MEMORY):
            output_file.write(str(quad) + '\n')

        output_file.close()
        return 0

    # Join all lines into a single string and ensures an INTRO at end of file
    asm_output = backend.emmit(MEMORY)
    from optimizer import optimize
    asm_output = optimize(asm_output) + '\n'

    # Now put user asm blocks back
    from backend import ASMS
    asm_output = asm_output.split('\n')

    for i in range(len(asm_output)):
        tmp = ASMS.get(asm_output[i], None)
        if tmp is not None:
            asm_output[i] = '\n'.join(tmp)

    asm_output = '\n'.join(asm_output)

    # Now filter them against the preprocessor again
    zxbpp.setMode('asm')
    zxbpp.OUTPUT = ''
    zxbpp.filter(asm_output, args[0])
    # Now output the result
    asm_output = zxbpp.OUTPUT.split('\n')
    get_inits(asm_output)  # Find out remaining inits
    MEMORY[:] = []
    zxbtrad.traverse(zxbparser.data_ast
                     )  # This will fill MEMORY with global declared variables

    tmp = [x for x in backend.emmit(MEMORY) if x.strip()[0] != '#']
    asm_output += tmp
    asm_output = backend.emmit_start() + asm_output
    asm_output += backend.emmit_end(asm_output)

    if options.asm:  # Only output assembler file
        output_file = open(FILE_output, 'wt')
        output(asm_output, output_file)
        output_file.close()
    else:
        from StringIO import StringIO

        fout = StringIO()
        output(asm_output, fout)
        asmparse.assemble(fout.getvalue())
        fout.close()
        asmparse.generate_binary(FILE_output, FILE_output_ext)

    return 0  # Exit success
Пример #3
0
def main(argv):
    ''' Entry point when executed from command line.
    You can use zxb.py as a module with import, and this
    function won't be executed.
    '''
    global FLAG_use_BASIC, FLAG_autorun
    global FILE_input, FILE_output, FILE_output_ext, OPTIONS_asm

    OPTIONS.add_option_if_not_defined('memoryCheck', bool, False)
    OPTIONS.add_option_if_not_defined('strictBool', bool, False)
    OPTIONS.add_option_if_not_defined('arrayCheck', bool, False)
    OPTIONS.add_option_if_not_defined('array_base', int, 0)
    OPTIONS.add_option_if_not_defined('string_base', int, 0)
    OPTIONS.add_option_if_not_defined('enableBreak', bool, False)
    OPTIONS.add_option_if_not_defined('emmitBackend', bool, False)
    OPTIONS.add_option_if_not_defined('arch', str, 'zx48k')
    OPTIONS.add_option_if_not_defined('__DEFINES', dict, {})
    OPTIONS.add_option_if_not_defined('explicit', bool, False)

    # ------------------------------------------------------------
    # Command line parsing
    # ------------------------------------------------------------

    parser = OptionParser(usage='Usage: %prog <input file> [options]',
                          version = '%prog ' + VERSION)

    parser.add_option("-d", "--debug",
            action="count", dest="debug", default=OPTIONS.Debug.value,
            help="Enable verbosity/debugging output. Additional -d increase verbosity/debug level.")

    parser.add_option("-O", "--optimize", type="int", dest="optimization_level",
            help="Sets optimization level. 0 = None", default=DEFAULT_OPTIMIZATION_LEVEL)

    parser.add_option("-o", "--output", type="string", dest="output_file",
            help="Sets output file. Default is input filename with .bin extension", default=None)

    parser.add_option("-T", "--tzx", action="store_true", dest="tzx", default=False,
            help="Sets output format to tzx (default is .bin)")

    parser.add_option("-t", "--tap", action="store_true", dest="tap", default=False,
            help="Sets output format to tap (default is .bin)")

    parser.add_option("-B", "--BASIC", action="store_true", dest="basic", default=False,
            help="Creates a BASIC loader which load the rest of the CODE. Requires -T ot -t")

    parser.add_option("-a", "--autorun", action="store_true", dest="autorun", default=False,
            help="Sets the program to be run once loaded")

    parser.add_option("-A", "--asm", action="store_true", dest="asm", default=False,
            help="Sets output format to asm")

    parser.add_option("-S", "--org", type="int", dest="org",
            help="Start of machine code. By default %i" % OPTIONS.org.value, default=OPTIONS.org.value)

    parser.add_option("-e", "--errmsg", type="string", dest="stderr", default=OPTIONS.StdErrFileName.value,
            help="Error messages file (standard error console by default)")

    parser.add_option("--array-base", type="int", dest="array_base", default=OPTIONS.array_base.value,
            help="Default lower index for arrays (0 by default)")

    parser.add_option("--string-base", type="int", dest="string_base", default=OPTIONS.string_base.value,
            help="Default lower index for strings (0 by default)")

    parser.add_option("-Z", "--sinclair", action="store_true", dest="sinclair", default=False,
            help="Enable by default some more original ZX Spectrum Sinclair BASIC features: ATTR, SCREEN$, POINT")

    parser.add_option("-H", "--heap-size", type="int", dest="heap_size", default=OPTIONS.heap_size.value,
            help="Sets heap size in bytes (default %i bytes)" % OPTIONS.heap_size.value)

    parser.add_option("--debug-memory", action="store_true", dest="debug_memory", default=False,
            help="Enables out-of-memory debug")

    parser.add_option("--debug-array", action="store_true", dest="debug_array", default=False,
            help="Enables array boundary checking")

    parser.add_option("--strict-bool", action="store_true", dest="strict_bool", default=False,
            help="Enforce boolean values to be 0 or 1")

    parser.add_option("--enable-break", action="store_true", dest="enable_break", default=False,
            help="Enables program execution BREAK detection")

    parser.add_option("-E", "--emmit-backend", action="store_true", dest="emmit_backend", default=False,
            help="Emmits backend code instead of ASM or binary")

    parser.add_option("--explicit", action="store_true", dest="explicit", default=False,
            help="Requires all variables and functions to be declared before used")

    parser.add_option("-D", "--define", type="str", dest="defines", action="append",
            help="Defines de given macro. Eg. -D MYDEBUG or -D NAME=Value")

    (options, args) = parser.parse_args()

    if len(args) != 1:
        parser.error("missing input file. (Try -h)")
        return 3

    # ------------------------------------------------------------
    # Setting of internal parameters according to command line
    # ------------------------------------------------------------

    OPTIONS.Debug.value = options.debug
    asmparse.FLAG_optimize = OPTIONS.optimization.value = options.optimization_level
    asmparse.FILE_output = OPTIONS.outputFileName.value = FILE_output = options.output_file
    asmparse.FILE_stderr = OPTIONS.StdErrFileName.value = options.stderr
    OPTIONS.array_base.value = options.array_base
    OPTIONS.string_base.value = options.string_base
    OPTIONS.Sinclair.value = options.sinclair
    OPTIONS.org.value = options.org
    OPTIONS.heap_size.value = options.heap_size
    OPTIONS.memoryCheck.value = options.debug_memory
    OPTIONS.strictBool.value = options.strict_bool or OPTIONS.Sinclair.value
    OPTIONS.arrayCheck.value = options.debug_array
    OPTIONS.emmitBackend.value = options.emmit_backend
    OPTIONS.enableBreak.value = options.enable_break
    OPTIONS.explicit.value = options.explicit

    if options.defines:
        for i in options.defines:
            name, val = tuple(i.split('=', 1))
            OPTIONS.__DEFINES.value[name] = val
            zxbpp.ID_TABLE.define(name, lineno = 0)

    if OPTIONS.Sinclair.value:
        OPTIONS.array_base.value = 1
        OPTIONS.string_base.value = 1
        OPTIONS.strictBool.value = True

    debug.ENABLED = OPTIONS.Debug.value

    if int(options.tzx) + int(options.tap) + int(options.asm) + int(options.emmit_backend) > 1:
        parser.error("Options --tap, --tzx, --emmit-backend and --asm are excluyent")
        return 3

    asmparse.FLAG_use_BASIC = options.basic
    backend.FLAG_autostart = asmparse.FLAG_autorun = options.autorun

    if asmparse.FLAG_use_BASIC and not options.tzx and not options.tap:
        parser.error('Option --BASIC and --autorun requires --tzx or tap format')
        return 4

    if options.tzx:
        FILE_output_ext = 'tzx'

    elif options.tap:
        FILE_output_ext = 'tap'

    elif options.asm:
        FILE_output_ext = 'asm'

    elif options.emmit_backend:
        FILE_output_ext = 'ic'

    if not os.path.exists(args[0]):
        parser.error("No such file or directory: '%s'" % args[0])
        return 2

    if OPTIONS.memoryCheck.value:
        OPTIONS.__DEFINES.value['__MEMORY_CHECK__'] = ''
        zxbpp.ID_TABLE.define('__MEMORY_CHECK__', lineno = 0)

    if OPTIONS.arrayCheck.value:
        OPTIONS.__DEFINES.value['__CHECK_ARRAY_BOUNDARY__'] = ''
        zxbpp.ID_TABLE.define('__CHECK_ARRAY_BOUNDARY__', lineno = 0)

    zxbpp.main(args)
    asmparse.FILE_output_ext = FILE_output_ext
    input = zxbpp.OUTPUT
    asmparse.FILE_input = FILE_input = zxbparser.FILENAME = os.path.basename(args[0])

    if FILE_output is None:
        OPTIONS.outputFileName.value = FILE_output = os.path.splitext(os.path.basename(FILE_input))[0] + '.' + FILE_output_ext
        asmparse.FILE_output = FILE_output

    if OPTIONS.StdErrFileName.value is not None:
        FILE_stderr = asmparse.FILE_stderr = OPTIONS.StdErrFileName.value
        OPTIONS.stderr.value = open(FILE_stderr, 'wt')

    zxbparser.parser.parse(input, lexer = zxblex.lexer, tracking = True, debug = (OPTIONS.Debug.value > 2))

    if gl.has_errors:
        return 1 # Exit with errors

    zxbtrad.traverse(zxbparser.ast) # This will fill MEMORY with code
    zxbtrad.traverse(zxbtrad.FUNCTIONS) # This will fill MEMORY with pending functions
    zxbtrad.emmit_strings()

    if OPTIONS.emmitBackend.value:
        output_file = open(FILE_output, 'wt')
        for quad in zxbtrad.dumpMemory(MEMORY):
            output_file.write(str(quad) + '\n')

        MEMORY[:] = [] # Empties memory
        zxbtrad.traverse(zxbparser.data_ast) # This will fill MEMORY with global declared variables

        for quad in zxbtrad.dumpMemory(MEMORY):
            output_file.write(str(quad) + '\n')

        output_file.close()
        return 0

    # Join all lines into a single string and ensures an INTRO at end of file
    asm_output = backend.emmit(MEMORY) 
    from optimizer import optimize
    asm_output = optimize(asm_output) + '\n'

    # Now put user asm blocks back
    from backend import ASMS
    asm_output = asm_output.split('\n')

    for i in range(len(asm_output)):
        tmp = ASMS.get(asm_output[i], None)
        if tmp is not None:
            asm_output[i] = '\n'.join(tmp)

    asm_output = '\n'.join(asm_output)

    # Now filter them against the preprocessor again
    zxbpp.setMode('asm')
    zxbpp.OUTPUT = ''
    zxbpp.filter(asm_output, args[0])
    # Now output the result
    asm_output = zxbpp.OUTPUT.split('\n')
    get_inits(asm_output) # Find out remaining inits
    MEMORY[:] = []
    zxbtrad.traverse(zxbparser.data_ast) # This will fill MEMORY with global declared variables

    tmp = [x for x in backend.emmit(MEMORY) if x.strip()[0] != '#']
    asm_output += tmp
    asm_output = backend.emmit_start() + asm_output
    asm_output += backend.emmit_end(asm_output)

    if options.asm: # Only output assembler file
        output_file = open(FILE_output, 'wt')
        output(asm_output, output_file)
        output_file.close()
    else:
        from StringIO import StringIO

        fout = StringIO()
        output(asm_output, fout)
        asmparse.assemble(fout.getvalue())
        fout.close()
        asmparse.generate_binary(FILE_output, FILE_output_ext)

    return 0 # Exit success
Пример #4
0
import sys, os
import re
from optparse import OptionParser

import asmparse
from asmparse import Asm, Expr, Container
import zxbpp

from common import OPTIONS

# Release version
VERSION = '1.4'

# Create default options
OPTIONS.add_option_if_not_defined('optimization', int, 0)

# Create option parser
o_parser = OptionParser(usage='Usage: %prog <input file> [options]',
            version = '%prog ' + VERSION)

o_parser.add_option("-d", "--debug",
        action="store_true", dest="debug", default=OPTIONS.Debug.value,
        help="Enable verbosity/debugging output")

o_parser.add_option("-O", "--optimize", type="int", dest="optimization_level",
        help="Sets optimization level. 0 = None", default=asmparse.FLAG_optimize)

o_parser.add_option("-o", "--output", type="string", dest="output_file",
        help="Sets output file. Default is input filename with .bin extension", default=None)
Пример #5
0
import sys, os
import re
from optparse import OptionParser

import asmparse
from asmparse import Asm, Expr, Container
import zxbpp

from common import OPTIONS

# Release version
VERSION = '1.4'

# Create default options
OPTIONS.add_option_if_not_defined('optimization', int, 0)

# Create option parser
o_parser = OptionParser(usage='Usage: %prog <input file> [options]',
                        version='%prog ' + VERSION)

o_parser.add_option("-d",
                    "--debug",
                    action="store_true",
                    dest="debug",
                    default=OPTIONS.Debug.value,
                    help="Enable verbosity/debugging output")

o_parser.add_option("-O",
                    "--optimize",
                    type="int",