Пример #1
0
def getNodIDs(inp_fileName):
        '''
Function that returns the node id's of the co-ordinates as a list.
        '''
        nsidData=[]
        getNSID=sncfheaders.showNSID(inp_fileName)
        startNSID=int(getNSID)
        numOfNodes=sncfheaders.getNumberOfNodes(inp_fileName)
        for itr in range(0,numOfNodes):
                nsidData.append(itr+startNSID)
        return nsidData
Пример #2
0
def writeCoordinates (inp_fileName,out_fileName):
        '''
A follow up routine for getCoordinates() method. Reads the co-ordinates from sncf file inp_fileName Writes the co-ordinates to an XML file
out_fileName, that contains node information.
        '''
        # Since sncfheaders.showNSID() returns the NSID in string literal, it would be wise to convert it to integer to able to operate on it.
        # An empty list that makes space for data to write to XML files
        xmlBuff=[]
        # Initial setup
        nsidData=[] # Container for rest of NSID's
        getNSID=sncfheaders.showNSID(inp_fileName)
        startNSID=int(getNSID)
        # Get n number of nodes
        numOfNodes=sncfheaders.getNumberOfNodes(inp_fileName)
        # Append data into nsidData
                # Has NSIDs for the rest of the nodes.
        for itr in range(0,numOfNodes):
                nsidData.append(itr+startNSID)
        # Store the data of coordinates in coordData
        coordData=getCoordinates(inp_fileName)
        ##       Structure design of XML and replace variables with XML data.
        ##       XML initial tags
        nL = sncfheaders.printNew()
        xmlBegin='''<?xml version="1.0" encoding="UTF-8"?>
<nodes xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
 xsi:noNamespaceSchemaLocation="http://sumo.sf.net/xsd/nodes_file.xsd">'''+sncfheaders.printNew()
        xmlEnd='</nodes>'
        xmlLineBegin='	<node id=\"'
        xmlLineEnd='/>'
        xmlX='x=\"'
        xmlY='y=\"'
        ## For the fact that 'number' of nodes equal the number of NSIDs, generate the exact 'number' of intermediate XML tags.
        fileHandle=open(out_fileName,'w')
        fileHandle.write(xmlBegin)
        for itr in range(0,numOfNodes):
                xmlLine=xmlLineBegin+str (nsidData[itr])+'\"'+' '+xmlX+str(coordData[itr].split(',')[0])+'\"'+' '+xmlY+str(coordData[itr].split(',')[1].split('\n')[0])+'\"'+' '+xmlLineEnd+nL 
                fileHandle.write(xmlLine)
                xmlBuff.append(xmlLine)
                ## Keep the line below for diagnostics!
        ## print xmlBuff 
        fileHandle.write(xmlEnd)
        fileHandle.close()
        baseName=os.path.basename (out_fileName)
        extName=os.path.splitext(baseName)[1]
        if extName=='.xml':
                print'''
[writeCoordinates says]: The file %s has been written to the location %s.
''' %(out_fileName,os.getcwd())
        else:
                print'''
[writeCoordinates says]: The file %s should have an extension *.nod.xml. Please rename it manually.
''' %(out_fileName)