def readTrackNodes(xmlCtx, section): if section['trackNodes'] is None: return component_constants.EMPTY_TUPLE else: defElasticity = _xml.readFloat(xmlCtx, section, 'trackNodes/elasticity', 1500.0) defDamping = _xml.readFloat(xmlCtx, section, 'trackNodes/damping', 1.0) defForwardElastK = _xml.readFloat(xmlCtx, section, 'trackNodes/forwardElastK', 1.0) defBackwardElastK = _xml.readFloat(xmlCtx, section, 'trackNodes/backwardElastK', 1.0) defOffset = _xml.readFloat(xmlCtx, section, 'trackNodes/offset', 0.0) trackNodes = [] xmlCtx = (xmlCtx, 'trackNodes') for sname, subsection in _xml.getChildren(xmlCtx, section, 'trackNodes'): if sname == 'node': ctx = (xmlCtx, 'trackNodes/node') name = _xml.readStringOrNone(ctx, subsection, 'leftSibling') if name is not None: leftNodeName = intern(name) else: leftNodeName = None name = _xml.readStringOrNone(ctx, subsection, 'rightSibling') if name is not None: rightNodeName = intern(name) else: rightNodeName = None trackNode = chassis_components.TrackNode(name=intern(_xml.readNonEmptyString(ctx, subsection, 'name')), isLeft=_xml.readBool(ctx, subsection, 'isLeft'), initialOffset=_xml.readFloat(ctx, subsection, 'offset', defOffset), leftNodeName=leftNodeName, rightNodeName=rightNodeName, damping=_xml.readFloat(ctx, subsection, 'damping', defDamping), elasticity=_xml.readFloat(ctx, subsection, 'elasticity', defElasticity), forwardElasticityCoeff=_xml.readFloat(ctx, subsection, 'forwardElastK', defForwardElastK), backwardElasticityCoeff=_xml.readFloat(ctx, subsection, 'backwardElastK', defBackwardElastK)) trackNodes.append(trackNode) return tuple(trackNodes)
def readTrackNodes(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: tuple containing TrackNode items. """ if section['trackNodes'] is None: return component_constants.EMPTY_TUPLE else: xmlCtx = (xmlCtx, 'trackNodes') trackNodes = [] defElasticity = section.readFloat('trackNodes/elasticity', 1500.0) defDamping = section.readFloat('trackNodes/damping', 1.0) defForwardElastK = section.readFloat('trackNodes/forwardElastK', 1.0) defBackwardElastK = section.readFloat('trackNodes/backwardElastK', 1.0) defOffset = section.readFloat('trackNodes/offset') for sname, subsection in _xml.getChildren(xmlCtx, section, 'trackNodes'): if sname == 'node': ctx = (xmlCtx, 'trackNodes/node') trackNode = chassis_components.TrackNode( name=_xml.readNonEmptyString(ctx, subsection, 'name'), isLeft=_xml.readBool(ctx, subsection, 'isLeft'), initialOffset=subsection.readFloat('offset', defOffset), leftNodeName=_xml.readStringOrNone(ctx, subsection, 'leftSibling'), rightNodeName=_xml.readStringOrNone( ctx, subsection, 'rightSibling'), damping=subsection.readFloat('damping', defDamping), elasticity=subsection.readFloat('elasticity', defElasticity), forwardElasticityCoeff=subsection.readFloat( 'forwardElastK', defForwardElastK), backwardElasticityCoeff=subsection.readFloat( 'backwardElastK', defBackwardElastK)) trackNodes.append(trackNode) return tuple(trackNodes)
def apply(vDesc, modelDesc, modelsSet): for key in chassis_params: obj = copy.deepcopy(modelDesc['chassis'][key]) if key == 'traces': obj['size'] = tuple(obj['size']) obj = cc.Traces(**obj) elif key == 'tracks': obj = cc.TrackBasicParams(**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 == 'trackSplineParams': obj = cc.TrackSplineParams(**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, None, None) 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, None, None) 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