sync_compiler_translation.py

Go to the documentation of this file.

#!/usr/bin/env python

#   Aseba - an event-based framework for distributed robot control
#   Copyright (C) 2007--2011:
#           Stephane Magnenat <stephane at magnenat dot net>
#           (http://stephane.magnenat.net)
#           and other contributors, see authors.txt for details
#
#   This program is free software: you can redistribute it and/or modify
#   it under the terms of the GNU Lesser General Public License as published
#   by the Free Software Foundation, version 3 of the License.
#
#   This program is distributed in the hope that it will be useful,
#   but WITHOUT ANY WARRANTY; without even the implied warranty of
#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#   GNU Lesser General Public License for more details.
#
#   You should have received a copy of the GNU Lesser General Public License
#   along with this program. If not, see <http://www.gnu.org/licenses/>.

# System lib
import os
import os.path
import sys
from string import Template
import re
import subprocess

INPUT_DIRECTORY = '../../compiler'
INPUT_FILE = 'errors.cpp'

OUTPUT_CODE_FILE = 'errors_code.h'
OUTPUT_CODE_DIRECTORY = INPUT_DIRECTORY

OUTPUT_QT_FILE = 'CompilerTranslator.cpp'
OUTPUT_QT_DIRECTORY = '../../studio/translations'

OUTPUT_TRANSLATION_DIRECTORY = '../../studio'
translation_files = list()
translation_files.append(['french', 'fr', 'compiler_fr.ts'])
translation_files.append(['german', 'de', 'compiler_de.ts'])
translation_files.append(['spanish', 'es', 'compiler_es.ts'])
translation_files.append(['italian', 'it', 'compiler_it.ts'])

comment_regexp = re.compile(r'\A\s*// (.*)')
error_regexp = re.compile(r'error_map\[(.*?)\]\s*=\s*L"(.*?)";')

output_code_file = \
"""
#ifndef __ERRORS_H
#define __ERRORS_H

/*
This file was automatically generated. Do not modify it,
or your changes will be lost!
You can manually run the script 'sync_translation.py'
to generate a new version, based on errors.cpp
*/

// define error codes
namespace Aseba
{
        enum ErrorCode
        {
                // ${start_comment}
                ${id_0} = 0,
${id_others}
                ERROR_END
        };
};

#endif // __ERRORS_H
"""
output_code_file_template = Template(output_code_file)

output_qt_file = \
"""
#include "CompilerTranslator.h"
#include "../../compiler/errors_code.h"

#include <CompilerTranslator.moc>

namespace Aseba
{
        CompilerTranslator::CompilerTranslator()
        {

        }

        const std::wstring CompilerTranslator::translate(ErrorCode error)
        {
                QString msg;

                switch (error)
                {
${elements}
                        default:
                                msg = tr("Unknown error");
                }

                return msg.toStdWString();
        }
};
"""
output_qt_file_template = Template(output_qt_file)

output_qt_element = \
"""
                        case ${error_id}:
                                msg = tr("${error_msg}");
                                break;
"""
output_qt_element_template = Template(output_qt_element)

# process input file
filename = os.path.join(INPUT_DIRECTORY, INPUT_FILE)
print "Reading " + filename
try:
    fh = file(filename)
except:
    print >> sys.stderr, "Invalid file " + filename
    exit(1)

case_vector = ""
count = 0
id_0 = None
start_comment = ""
id_others = ""

print "Processing..."
for line in fh:
    # a comment?
    match = comment_regexp.search(line)
    if match:
        # add it to the error codes file
        if not id_0:
            start_comment = match.group(1)
        else:
            id_others += "              // " + match.group(1) + "\n"
        continue

    match = error_regexp.search(line)
    if match:
        count = count + 1
        code = match.group(1)
        msg = match.group(2)
        # error codes file
        if not id_0:
            id_0 = code
        else:
            id_others += "              " + code + ",\n"
        # qt file
        case_vector += output_qt_element_template.substitute(error_id=code, error_msg=msg)

fh.close()
print "Found " + str(count) + " string(s)"

# writing errors_code.h
result = output_code_file_template.substitute(start_comment=start_comment, id_0=id_0, id_others=id_others)
filename = os.path.join(OUTPUT_CODE_DIRECTORY, OUTPUT_CODE_FILE)
print "Writing to " + filename
try:
    fh = file(filename, 'w')
except:
    print >> sys.stderr, "Invalid file " + filename
    exit(1)

fh.write(result)
fh.close()

# writing the qt file
result = output_qt_file_template.substitute(elements=case_vector)
filename = os.path.join(OUTPUT_QT_DIRECTORY, OUTPUT_QT_FILE)
print "Writing to " + filename
try:
    fh = file(filename, 'w')
except:
    print >> sys.stderr, "Invalid file " + filename
    exit(1)

fh.write(result)
fh.close()
print "Done!"

# update translation files
for translation in translation_files:
    translation_name = translation[0]
    translation_code = translation[1]
    translation_file = translation[2]
    answer = raw_input("Do you want to update " + translation_name + " [" + translation_code + "] translation file (" + translation_file + ")? [y/n] ")
    if answer == 'y' or answer == 'Y':
        print "Updating " + translation_file + "..."
        file1 = os.path.join(OUTPUT_QT_DIRECTORY, OUTPUT_QT_FILE)
        file2 = os.path.join(OUTPUT_TRANSLATION_DIRECTORY, translation_file)
        cmd = "lupdate-qt4 " + file1 + " -ts " + file2
        print "> " + cmd
        subprocess.call(cmd, shell=True)
    else:
        print "Skipping..."