#!/usr/bin/env python3 # # SuperTuxKart - a fun racing game with go-kart # Copyright (C) 2006-2015 SuperTuxKart-Team # # This program is free software; you can redistribute it and/or # modify it under the terms of the GNU General Public License # as published by the Free Software Foundation; either version 3 # of the License, or (at your option) any later version. # # 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 General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. # This script creates the output for the AbstractCharacteristics # It takes an argument that specifies what the output of the script should be. # The output options can be seen by running this script without arguments. import sys # Input data #FIXME is wheelPosition needed? characteristics = """Suspension: stiffness, rest, travelCm, expSpringResponse(bool), maxForce Stability: rollInfluence, chassisLinearDamping, chassisAngularDamping, downwardImpulseFactor, trackConnectionAccel, smoothFlyingImpulse Turn: radius(InterpolationArray), timeResetSteer, timeFullSteer(InterpolationArray) Engine: power, maxSpeed, brakeFactor, brakeTimeIncrease, maxSpeedReverseRatio Gear: switchRatio(std::vector/floatVector), powerIncrease(std::vector/floatVector) Mass Wheels: dampingRelaxation, dampingCompression, radius, position(std::vector/floatVector) Camera: distance, forwardUpAngle, backwardUpAngle Jump: animationTime Lean: max, speed Anvil: duration, weight, speedFactor Parachute: friction, duration, durationOther, lboundFraction, uboundFraction, maxSpeed Bubblegum: duration, speedFraction, torque, fadeInTime, shieldDuration Zipper: duration, force, speedGain, maxSpeedIncrease, fadeOutTime Swatter: duration, distance, squashDuration, squashSlowdown Plunger: bandMaxLength, bandForce, bandDuration, bandSpeedIncrease, bandFadeOutTime, inFaceTime Startup: time(std::vector/floatVector), boost(std::vector/floatVector) Rescue: duration, vertOffset, height Explosion: duration, radius, invulnerabilityTime Nitro: duration, engineForce, consumption, smallContainer, bigContainer, maxSpeedIncrease, fadeOutTime, max Slipstream: duration, length, width, collectTime, useTime, addPower, minSpeed, maxSpeedIncrease, fadeOutTime""" class GroupMember: def __init__(self, name, typeC, typeStr): self.name = name if name == "value": self.getName = "" else: self.getName = name self.typeC = typeC self.typeStr = typeStr """E.g. power(std::vector/floatVector) or speed(InterpolationArray) The default type is float The name 'value' is special: Only the group name will be used to access the member but in the xml file it will be still value (because we need a name).""" def parse(content): typeC = "float" typeStr = typeC name = content.strip() pos = content.find("(") end = content.find(")", pos) if pos != -1 and end != -1: name = content[:pos].strip() pos2 = content.find("/", pos, end) if pos2 != -1: typeC = content[pos + 1:pos2].strip() typeStr = content[pos2 + 1:end].strip() else: typeC = content[pos + 1:end].strip() typeStr = typeC return GroupMember(name, typeC, typeStr) class Group: def __init__(self, baseName): self.baseName = baseName self.members = [] def addMember(self, content): self.members.append(GroupMember.parse(content)) def getBaseName(self): return self.baseName """E.g. engine: power, gears(std::vector/Gears) or mass(float) or only mass""" def parse(content): pos = content.find(":") if pos == -1: group = Group(content) group.addMember("value") return group else: group = Group(content[:pos].strip()) for m in content[pos + 1:].split(","): group.addMember(m) return group """Creates a list of words from a titlecase string""" def toList(name): result = [] cur = "" for c in name: if c.isupper() and len(cur) != 0: result.append(cur) cur = "" cur += c.lower() if len(cur) != 0: result.append(cur) return result """titleCase: true = result is titlecase false = result has underscores""" def joinSubName(group, member, titleCase): words = toList(group.baseName) + toList(member.getName) first = True if titleCase: words = [w.title() for w in words] return "".join(words) else: return "_".join(words) def main(): # Find out what to do if len(sys.argv) == 1: print("Please specify what you want to know [enum|defs|getter|getProp|getXml]") return task = sys.argv[1] groups = [Group.parse(line) for line in characteristics.split("\n")] # Find longest name to align the function bodies nameLengthTitle = 0 nameLengthUnderscore = 0 for g in groups: for m in g.members: l = len(joinSubName(g, m, True)) if l > nameLengthTitle: nameLengthTitle = l l = len(joinSubName(g, m, False)) if l > nameLengthUnderscore: nameLengthUnderscore = l # Print the results if task == "enum": for g in groups: print() print(" // {0}".format(g.getBaseName().title())) for m in g.members: print(" {0},".format(joinSubName(g, m, False).upper())) elif task == "defs": for g in groups: print() for m in g.members: nameTitle = joinSubName(g, m, True) nameUnderscore = joinSubName(g, m, False) typeC = m.typeC print(" {0} get{1}() const;". format(typeC, nameTitle, nameUnderscore)) elif task == "getter": for g in groups: for m in g.members: nameTitle = joinSubName(g, m, True) nameUnderscore = joinSubName(g, m, False) typeC = m.typeC result = "result" print("""{3} AbstractCharacteristic::get{1}() const {{ {0} result; bool is_set = false; process({2}, &result, &is_set); if (!is_set) Log::fatal("AbstractCharacteristic", "Can't get characteristic %s", getName({2}).c_str()); return {4}; }} """.format(m.typeC, nameTitle, nameUnderscore.upper(), typeC, result)) elif task == "getProp": for g in groups: for m in g.members: nameTitle = joinSubName(g, m, True) nameUnderscore = joinSubName(g, m, False) print(" case {0}:\n return TYPE_{1};". format(nameUnderscore.upper(), "_".join(toList(m.typeStr)).upper())) print("\n\n-------------------- END --------------------\n") for g in groups: for m in g.members: nameTitle = joinSubName(g, m, True) nameUnderscore = joinSubName(g, m, False).upper() print(" case {0}:\n return \"{0}\";". format(nameUnderscore)) elif task == "getXml": for g in groups: print(" if (const XMLNode *sub_node = node->getNode(\"{0}\"))\n {{". format(g.baseName.lower())) for m in g.members: nameUnderscore = joinSubName(g, m, False) nameMinus = "-".join(toList(m.name)) print(" sub_node->get(\"{0}\", &m_values[{1}]);". format(nameMinus, nameUnderscore.upper())) print(" }\n") else: print("Unknown task") #print("Constructor ****************************************") #lineLength = 4; #line = " " #for g in groups: # for n in g.subNames: # name = "m_{0} = ".format(joinSubName(g, n, False)) # l = len(name) # if lineLength + l > 80 and lineLength > 4: # print(line) # line = " " + name # lineLength = l + 4 # else: # line += name # lineLength += l #if lineLength > 4: # line += "1;" # print(line) if __name__ == '__main__': main()