#@C Copyright Notice #@C ================ #@C This file is part of cpswTreeGUI. It is subject to the license terms in the #@C LICENSE.txt file found in the top-level directory of this distribution and at #@C #@C https://confluence.slac.stanford.edu/display/ppareg/LICENSE.html. #@C #@C No part of cpswTreeGUI, including this file, may be copied, modified, propagated, or #@C distributed except according to the terms contained in the LICENSE.txt file. import yaml_cpp import pycpsw class Fixup(pycpsw.YamlFixup): def __init__(self, ipAddr=None, disableStreams=False, srpV2=False, useTcp=False, disableDepack=False, portMaps=None, disableComm=False, justLoadYaml=False, srpTimeoutUS=None, rssiBridge=None): pycpsw.YamlFixup.__init__(self) self._disableStreams = disableStreams self._srpV2 = srpV2 self._useTcp = useTcp self._ipAddr = ipAddr self._disableDepack = disableDepack self._portMaps = portMaps self._noComm = disableComm self._justLoadYaml = justLoadYaml self._srpTimeout = srpTimeoutUS self._rssiBridge = rssiBridge if None != self._rssiBridge: # CPSW already maps all udp to tcp self._useTcp = False if self._noComm: self._srpV2 = False self._useTcp = False self._ipAddr = None self._disableDepack = False self._portMaps = None self._rssiBridge = None self._portMap = None if ( self._portMaps == None): self._portMaps = [] def getJustLoadYaml(self): return self._justLoadYaml def ok(self,node): return node != None and node.IsDefined() and not node.IsNull() def edit(self, node, l, editPass): if editPass == 0: disableStreams = self._disableStreams srpV2 = self._srpV2 useTcp = self._useTcp ipAddr = self._ipAddr disableDepack = self._disableDepack rssiBridge = self._rssiBridge else: disableStreams = False srpV2 = False useTcp = False ipAddr = None disableDepack = False rssiBridge = None if len(self._portMaps) == 0: portMap = None else: portMap = self._portMaps[editPass] cl = node["class"] if self.ok(cl) and cl.Scalar() == "NetIODev": if self._noComm or self._justLoadYaml: cl.set( "NullDev" ) if None != ipAddr: ip = self.find(node, "ipAddr") if not self.ok(ip): raise RuntimeError("No ipAddr node found") print("Fixing IP address; setting to {}\n".format(ipAddr)) ip.set( ipAddr ) if None != rssiBridge: node["rssiBridge"]=rssiBridge if (not useTcp and not disableStreams and not srpV2 and not disableDepack and portMap == None): if editPass > 0: raise RuntimeError("FOO") return children = node["children"] if not self.ok(children): raise RuntimeError("No NetIODev children found") for childit in children: #print("Checking {}".format(childit.first.Scalar())) child = childit.second at = self.find(child,"at") isSrp = False isStrm = False isSrp2Remap = False if ( disableStreams or srpV2 or disableDepack or useTcp or portMap != None ): got = self.findWithPath(at, "SRP/protocolVersion") if None != got: (srpProt, path, parent) = got path.insert(0, "at") path.insert(0, childit.first.Scalar()) if srpProt.Scalar() == "SRP_UDP_NONE": isStrm = True if disableStreams: print("SRP Node found: {} {} -- disabling\n".format( "/".join(path), srpProt.Scalar())) child["instantiate"] = "False" else: isSrp = True if srpV2 and srpProt.Scalar() == "SRP_UDP_V3": print("SRP Node found: {} {} -- changing to SRP_UDP_V2\n".format( "/".join(path), srpProt.Scalar())) srpProt.set("SRP_UDP_V2") isSrp2Remap = True if disableDepack: got = self.findWithPath(at, "depack") if None != got: (node, path, parent) = got path.insert(0, "at") path.insert(0, childit.first.Scalar()) print("depack found: {} -- disabling\n".format( "/".join(path))) node["instantiate"] = "False" got = self.findWithPath(at, "TDESTMux") if None != got: (node, path, parent) = got path.insert(0, "at") path.insert(0, childit.first.Scalar()) print("TDESTMux found: {} -- disabling\n".format( "/".join(path))) node["instantiate"] = "False" if (isSrp2Remap or useTcp) and None != self._srpTimeout: print("Patching SRP timeout (fixup to TCP or SRP2): {}".format(self._srpTimeout)) parent["timeoutUS"] = self._srpTimeout else: #print("Non-SRP Node found: {}\n".format( "/".join(l) )) pass if useTcp or portMap != None: got = self.findWithPath(at, "UDP") if None == got: got = self.findWithPath(at, "TCP") if None != got: (proto, path, parent) = got path.insert(0, "at") path.insert(0, childit.first.Scalar()) prt = self.find(proto, "port") else: prt = None if self.ok(prt): if portMap != None: origPort = int( prt.getAs(), 0 ) if portMap[1] == 0: if origPort == portMap[0]: print("Keeping port {:d}".format(origPort)) else: print("Disabling port {:d}".format(origPort)) child["instantiate"] = "False" else: if origPort == portMap[0]: print("Changing port {:d} to {:d}".format(origPort, portMap[1])) prt.set("{:d}".format(portMap[1])) if useTcp: for x in parent: if x.first.Scalar() == "UDP": rssimsg = "" for rssi in parent: if rssi.first.Scalar() == "RSSI": rssimsg = "(with RSSI)" break print("UDP{} Node found: {} -- changing to TCP (port {})\n".format( rssimsg, "/".join(path), prt.getAs() )) x.first.set("TCP") break else: #print("Non-UDP Node found: {}\n".format( "/".join(l) )) pass def trav(self, node, l=[], editPass=0): if node.IsMap(): self.edit(node, l, editPass) for it in node: l.append(it.first.Scalar()) self.trav(it.second, l, editPass) l.pop() def find(self, node, name): res = self.findWithPath(node, name) if None == res: return None node = res[0] return node def findWithPath(self, node, name): l = [] for part in name.split("/"): res = [] self.findNod(node, part, [], res) if len(res) == 0: return None lvl = 0 while len(res) > 1: for it in res: (n, l, par) = it eln = l[lvl] if eln != "<<": break if eln == part: return (n, l, par) if eln == "<<": # only merge keys; drop one level pass else: # filter merge nodes for i in range(len(res)-1,-1,-1): if res[i][1][lvl] == "<<": res.pop(i) elif res[i][1][lvl] != eln: raise RuntimeError("find: name mismatch -- scope probably too wide") # drop one level lvl = lvl + 1 node = res[0][0] l.extend(res[0][1]) return (node, l, res[0][2]) def findNod(self, node, name, l, res): if node.IsMap(): fnd = node[name] if self.ok(fnd): nl = list(l) nl.append( name ) res.append( (fnd, nl, node) ) for it in node: l.append( it.first.Scalar() ) self.findNod(it.second, name, l, res) l.pop() def fixup(self, node): self(node, None) def __call__(self, node, top): if ( None != self._ipAddr or self._useTcp or self._disableStreams or self._srpV2 or self._disableDepack or len(self._portMaps) > 0 or self._noComm or self._justLoadYaml or None != self._rssiBridge ): to = len(self._portMaps) if to == 0 or self._noComm: to = 1 for editPass in range(0,to): self.trav(node, [], editPass)