def readTrackParams(xmlCtx, section):
trackParams = None
if section['trackNodes'] is not None:
ctx = (xmlCtx, 'trackNodes')
trackParams = chassis_components.TrackParams(thickness=_xml.readFloat(ctx, section, 'trackThickness'), maxAmplitude=_xml.readFloat(ctx, section, 'trackNodes/maxAmplitude'), maxOffset=_xml.readFloat(ctx, section, 'trackNodes/maxOffset'), gravity=_xml.readFloat(ctx, section, 'trackNodes/gravity'))
if section['trackNodes'] is None and section['splineDesc'] is not None:
trackParams = chassis_components.TrackParams(thickness=_xml.readFloat(xmlCtx, section, 'trackThickness'), maxAmplitude=component_constants.ZERO_FLOAT, maxOffset=component_constants.ZERO_FLOAT, gravity=component_constants.ZERO_FLOAT)
return trackParams
def readTrackParams(xmlCtx, section):
"""Reads section 'trackNodes' for each chassis if it has.
:param xmlCtx: tuple(root ctx or None, path to section).
:param section: instance of DataSection.
:return: instance of TrackParams or None.
"""
trackParams = None
if section['trackNodes'] is not None:
ctx = (xmlCtx, 'trackNodes')
trackParams = chassis_components.TrackParams(
thickness=_xml.readFloat(ctx, section, 'trackThickness'),
maxAmplitude=_xml.readFloat(ctx, section,
'trackNodes/maxAmplitude'),
maxOffset=_xml.readFloat(ctx, section, 'trackNodes/maxOffset'),
gravity=_xml.readFloat(ctx, section, 'trackNodes/gravity'))
if section['trackNodes'] is None and section['splineDesc'] is not None:
trackParams = chassis_components.TrackParams(
thickness=_xml.readFloat(xmlCtx, section, 'trackThickness'),
maxAmplitude=component_constants.ZERO_FLOAT,
maxOffset=component_constants.ZERO_FLOAT,
gravity=component_constants.ZERO_FLOAT)
return trackParams
def apply(vDesc, modelDesc, modelsSet):
for key in chassis_params:
obj = modelDesc['chassis'][key]
if key == 'traces':
obj['size'] = tuple(obj['size'])
obj = cc.Traces(**obj)
elif key == 'tracks':
obj = cc.TrackMaterials(**obj)
elif key == 'wheels':
obj = cc.WheelsConfig(
groups=tuple(cc.WheelGroup(**d) for d in obj['groups']),
wheels=tuple(cc.Wheel(**d) for d in obj['wheels']))
elif key == 'groundNodes':
obj = NodesAndGroups(
nodes=tuple(cc.GroundNode(**d) for d in obj['nodes']),
groups=tuple(cc.GroundNodeGroup(**d) for d in obj['groups']),
activePostmortem=obj['activePostmortem'],
lodSettings=obj['lodSettings'])
elif key == 'trackNodes':
obj = NodesAndGroups(nodes=tuple(
cc.TrackNode(**d) for d in obj['nodes']),
groups=(),
activePostmortem=obj['activePostmortem'],
lodSettings=obj['lodSettings'])
elif key == 'splineDesc':
obj = cc.SplineConfig(
({
modelsSet:
cc.SplineSegmentModelSet(**obj['segmentModelSets'])
} or None),
**{k: v
for k, v in obj.items() if k != 'segmentModelSets'})
elif key == 'trackParams':
obj = cc.TrackParams(**obj)
elif key == 'leveredSuspension':
if obj is not None:
obj = cc.LeveredSuspensionConfig(
([cc.SuspensionLever(**d) for d in obj['levers']] or None),
**{k: v
for k, v in obj.items() if k != 'levers'})
setattr(vDesc.chassis, key, obj)
vDesc.chassis.chassisLodDistance = modelDesc['chassis'][
'chassisLodDistance']
vDesc.chassis.physicalTracks = {}
if modelDesc['chassis']['AODecals']:
AODecalsOffset = vDesc.chassis.hullPosition - Math.Vector3(
*modelDesc['chassis']['hullPosition'])
vDesc.chassis.AODecals = copy.deepcopy(
modelDesc['chassis']['AODecals'])
vDesc.chassis.AODecals[0].setElement(3, 1, AODecalsOffset.y)
exclMask = modelDesc['common']['camouflage']['exclusionMask']
vDesc.type.camouflage = Camouflage(
modelDesc['common']['camouflage']['tiling']
if exclMask else vDesc.type.camouflage.tiling, exclMask)
for partName in TankPartNames.ALL:
part = getattr(vDesc, partName)
models = part.modelsSets[modelsSet]
part.modelsSets[modelsSet] = ModelStatesPaths(
modelDesc[partName]['undamaged'], models.destroyed,
models.exploded)
part.models = part.modelsSets['default']
part.emblemSlots = tuple(modelDesc[partName]['emblemSlots'])
if partName == 'chassis':
continue
camoData = modelDesc[partName]['camouflage']
exclMask = camoData['exclusionMask']
if exclMask:
part.camouflage = Camouflage(camoData['tiling'], exclMask)
vDesc.gun.drivenJoints = modelDesc['gun']['drivenJoints']
exhaust = modelDesc['hull']['exhaust']
for effectDesc in vDesc.hull.customEffects:
if exhaust['nodes']:
effectDesc.nodes[:] = exhaust['nodes']
effectDesc._selectorDesc = g_cache._customEffects['exhaust'].get(
exhaust['pixie'], effectDesc._selectorDesc)
nationID = vDesc.type.id[0]
for partName, cacheName, cacheIDs in (('chassis', 'chassis', 'chassisIDs'),
('engine', 'engines', 'engineIDs'),
('gun', 'guns', 'gunIDs')):
soundID = modelDesc[partName].get('soundID')
if soundID:
part = getattr(vDesc, partName)
pickFrom = getattr(g_cache, cacheName)(nationID).get(
getattr(g_cache, cacheIDs)(nationID).get(soundID))
if pickFrom is not None:
if partName != 'gun':
part.sounds = pickFrom.sounds
else:
part.effects = pickFrom.effects
part.reloadEffect = pickFrom.reloadEffect