ZDoom

Code: Select all

import sys, time, pydoc, io, traceback, os.path
import console, wads
from zdoom.textcolors import TEXTCOLOR_BLUE, TEXTCOLOR_RED
from zdoom.constants import BASEWAD
from zdoom.console_setup import ConsolePrinter

help = pydoc.help

# Helper functions
def BaseFileSearch (file, ext = None, lookfirstinprogdir = False):
    if lookfirstinprogdir:
        wad = os.path.join (sys.progdir, file)
        if os.path.exists (wad):
            return wad

    if os.path.exists (file):
        return wad

    # TODO: Config file searching

    if '.' not in file and ext:
        return BaseFileSearch (file + "." + ext, None, lookfirstinprogdir)

    return None

if True:
    console.print ("Performing console setup...\n")
    
    sys.stdin  = ConsolePrinter ()
    sys.stdout = ConsolePrinter (TEXTCOLOR_BLUE)
    sys.stderr = ConsolePrinter (TEXTCOLOR_RED)
    
    console.print ("Checking command-line arguments...\n")

    # Will need to use a list to take care of this

    console.print ("Loading {}...\n".format (BASEWAD))
    basewad = wads.ResourceFile (BaseFileSearch (BASEWAD, None, True))
    basewad_lumpdir = basewad.lumpnames ()
    # FIXME: This reports less than there actually are because it returns the 8-letter names, not the full names.
    # Make a function to return an alternate directory with full names. It should return None for wads, since they don't have full names.
    print ("  {} lumps\n".format (len (basewad_lumpdir)))

    iwad_info = basewad.readlump (basewad_lumpdir["IWADINFO"])

    # Make a scanner. Shouldn't be too hard with Python's string manipulation, of course.
    #print (iwad_info.decode ("cp1252", "ignore"))