res2cc_cmd.py

Go to the documentation of this file.

#!/usr/bin/python
# Script that compiles a set of resources into a single C++ source file. The C++ file
# offers an initResources() function, which registers the resources with the resource
# manager (class ResourceMgr)
#
# Copyright (C) 1997-2015 by Dimitri van Heesch.
#
# Permission to use, copy, modify, and distribute this software and its
# documentation under the terms of the GNU General Public License is hereby
# granted. No representations are made about the suitability of this software
# for any purpose. It is provided "as is" without express or implied warranty.
# See the GNU General Public License for more details.
#
# Documents produced by Doxygen are derivative works derived from the
# input used in their production; they are not affected by this license.
#
from __future__ import print_function
from os         import listdir, stat, walk
from os.path    import isfile, join, splitext
import sys

class File(object):
        def __init__(self,directory,subdir,fileName,mode):
                self.directory = directory
                self.subdir    = subdir
                self.fileName  = fileName
                filePath  = join(directory,subdir,fileName)
                self.fileSize  = stat(filePath).st_size
                self.bareName  = fileName.replace('.','_')
                self.inputFile = open(filePath,mode)

        def formatByte(self,byte):
                if isinstance(byte,int):
                        return "%02x" % byte
                else:
                        return format(ord(byte),'02x')

        def writeBytes(self,data,outputFile):
                bytes_per_line=16
                print("static const unsigned char %s_data[] = " % self.bareName,file=outputFile)
                print("{",file=outputFile)
                lines = [data[x:x+bytes_per_line] for x in range(0,len(data),bytes_per_line)]
                linesAsString = ',\n  '.join([', '.join(['0x'+self.formatByte(byte) for byte in line]) for line in lines])
                print('  %s' % linesAsString,file=outputFile)
                print("};",file=outputFile)
                print("const int %s_len = %d;\n" % (self.bareName,len(data)),file=outputFile)

        def convertToBytes(self,outputFile):
                lines = [x for x in self.inputFile.readlines() if not x.startswith('#')]
                w,h = (int(x) for x in lines[0].split())
                data = "".join(map(chr,[int(w>>8),int(w&0xFF),int(h>>8),int(h&0xFF)]+
                                       [int(x) for line in lines[1:] for x in line.split()]))
                self.writeBytes(data,outputFile)

        @staticmethod
        def factory(directory,subdir,fname):
                ext = splitext(fname)[1]
                if ext=='.lum':  return LumFile(directory,subdir,fname)
                if ext=='.luma': return LumaFile(directory,subdir,fname)
                if ext=='.css':  return CSSFile(directory,subdir,fname)
                return VerbatimFile(directory,subdir,fname)

class VerbatimFile(File):
        def __init__(self,directory,subdir,fileName):
                File.__init__(self,directory,subdir,fileName,"rb")
        def writeContents(self,outputFile):
                self.writeBytes(self.inputFile.read(),outputFile)
        def writeDirEntry(self,outputFile):
                print("  { \"%s\", \"%s\", %s_data, %s_len, Resource::Verbatim }," % (self.subdir,self.fileName,self.bareName,self.bareName), file=outputFile)

class CSSFile(File):
        def __init__(self,directory,subdir,fileName):
                File.__init__(self,directory,subdir,fileName,"r")
        def writeContents(self,outputFile):
                self.writeBytes(self.inputFile.read(),outputFile)
        def writeDirEntry(self,outputFile):
                print("  { \"%s\", \"%s\", %s_data, %s_len, Resource::CSS }," % (self.subdir,self.fileName,self.bareName,self.bareName), file=outputFile)

class LumFile(File):
        def __init__(self,directory,subdir,fileName):
                File.__init__(self,directory,subdir,fileName,"r")
        def writeContents(self,outputFile):
                self.convertToBytes(outputFile)
        def writeDirEntry(self,outputFile):
                print("  { \"%s\", \"%s\", %s_data, %s_len, Resource::Luminance }," % (self.subdir,self.fileName,self.bareName,self.bareName), file=outputFile)

class LumaFile(File):
        def __init__(self,directory,subdir,fileName):
                File.__init__(self,directory,subdir,fileName,"r")
        def writeContents(self,outputFile):
                self.convertToBytes(outputFile)
        def writeDirEntry(self,outputFile):
                print("  { \"%s\", \"%s\", %s_data, %s_len, Resource::LumAlpha }," % (self.subdir,self.fileName,self.bareName,self.bareName), file=outputFile)

def main():
        if len(sys.argv)<3:
                sys.exit('Usage: %s directory output_file.cpp' % sys.argv[0])
        directory = sys.argv[1]
        files = []
        for dirName, subdirList, fileList in walk(directory):
                for fname in sorted(fileList):
                        subdir = dirName[len(directory)+1:] if dirName.startswith(directory) else dirName
                        if subdir:
                                files.append(File.factory(directory,subdir,fname))
        outputFile = open(sys.argv[2],"w")
        print("#include \"resourcemgr.h\"\n",file=outputFile)
        for f in files:
                f.writeContents(outputFile)
        print("static Resource resourceDir[] =",file=outputFile)
        print("{",file=outputFile)
        for f in files:
                f.writeDirEntry(outputFile)
        print("};",file=outputFile)
        print("static int resourceDir_len = %s;" % len(files), file=outputFile)
        print("void initResources() { ResourceMgr::instance().registerResources(resourceDir,resourceDir_len); }",file=outputFile)

if __name__ == '__main__':
        main()