# coding: utf-8 # In[64]: import struct import matplotlib.pyplot as plt import math from py_rmatools_v03 import rma import re from datetime import datetime, timedelta import pandas as pd import numpy as np plt.rcParams.update({'figure.max_open_warning': 0}) # In[65]: meshFilename = 'pok005.rm1' channelWidth = 100 RMAfilenames = ['POA034'] #If RMA11 constNum = [1] #If RMA11 3D #Options = [4,8,1,'Max'] #Option available 'Max' , an integer correspond to the layer number to extract the results, if 2D use options = [1] - 'Percentile' (percentile only work for RMA11 FOR now) Options = [8] percentiles = [0.5,0.8,0.9,0.95,1] NodeLayersFile = 'NodeLayers005.txt' #the node layer file was created by getting the nodes number from the RMA Outfile (just run RMA11 for a few steps) Nnodes = 6674 #Number of nodes (including mid side node) - thisn umber also correspond to the last node number in the nodelayers file # In[66]: def isElementOneD(nodelist1): if len(nodelist1) == 3: return True return False def isElementSquare(nodelist1): if len(nodelist1) == 8: return True return False def isElementTriangle(nodelist1): if len(nodelist1) == 6: return True return False def square2Triangle(ElementNum): nodelist1 = ElementDict[ElementNum] if isElementSquare(nodelist1): ElementDict[ElementNum] = [nodelist1[0], nodelist1[1], nodelist1[7]] ElementList.append(max(ElementList) + 1) ElementDict[ElementList[-1]]= [nodelist1[1], nodelist1[2], nodelist1[3]] ElementList.append(max(ElementList) + 1) ElementDict[ElementList[-1]]= [nodelist1[1], nodelist1[3], nodelist1[7]] ElementList.append(max(ElementList) + 1) ElementDict[ElementList[-1]]= [nodelist1[3], nodelist1[4], nodelist1[5]] ElementList.append(max(ElementList) + 1) ElementDict[ElementList[-1]]= [nodelist1[3], nodelist1[5], nodelist1[7]] ElementList.append(max(ElementList) + 1) ElementDict[ElementList[-1]]= [nodelist1[5], nodelist1[6], nodelist1[7]] def triangle2Triangles(ElementNum): nodelist1 = ElementDict[ElementNum] if isElementTriangle(nodelist1): ElementDict[ElementNum] = [nodelist1[0], nodelist1[1], nodelist1[5]] ElementList.append(max(ElementList) + 1) ElementDict[ElementList[-1]]= [nodelist1[1], nodelist1[2], nodelist1[3]] ElementList.append(max(ElementList) + 1) ElementDict[ElementList[-1]]= [nodelist1[1], nodelist1[3], nodelist1[5]] ElementList.append(max(ElementList) + 1) ElementDict[ElementList[-1]]= [nodelist1[3], nodelist1[4], nodelist1[5]] def oneD2triangle(ElementNum): if isElementOneD(ElementDict[ElementNum]): nAe = ElementDict[ElementNum][0] #nAe Node A existing nBe = ElementDict[ElementNum][1] if not nAe in node1Dduplicate: node1Dduplicate[nAe] = [] if not nBe in node1Dduplicate: node1Dduplicate[nBe] = [] xA = nodeDict[nAe][0] xB = nodeDict[nBe][0] yA = nodeDict[nAe][1] yB = nodeDict[nBe][1] normalVec = [-(yB - yA),(xB - xA)] dist = math.sqrt(normalVec[0]**2 + normalVec[1]**2) normalVec[0] = normalVec[0] / dist normalVec[1] = normalVec[1] / dist xA2 = xA + channelWidth * normalVec[0] xB2 = xB + channelWidth * normalVec[0] yA2 = yA + channelWidth * normalVec[1] yB2 = yB + channelWidth * normalVec[1] nA = max(NodeList) + 1 nB = max(NodeList) + 2 node1Dduplicate[nAe].append(nA) node1Dduplicate[nBe].append(nB) node2nodevalue[nA] = nAe node2nodevalue[nB] = nBe NodeList.append(nA) NodeList.append(nB) nodeDict[nA] = [xA2, yA2, -1.01] nodeDict[nB] = [xB2, yB2, -1.01] newEle = max(ElementList) + 1 ElementList .append(newEle) ElementDict[ElementNum] = [nAe, nA, nBe] ElementDict[newEle] = [nA, nB, nBe] def RMA2toSerafin(): f = open('{}.slf'.format(RMAfilename), 'wb') f.write(struct.pack(">l",80)) str='{0: >80}'.format('SERAFIN ') f.write(str.encode('ascii')) f.write(struct.pack(">l",80)) f.write(struct.pack(">l",8)) f.write(struct.pack(">l",len(constName))) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",8)) for c in constName: f.write(struct.pack(">l",32)) str='{0: <32}'.format(c) f.write(str.encode('ascii')) f.write(struct.pack(">l",32)) f.write(struct.pack(">l",40)) f.write(struct.pack(">l",1)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",1)) f.write(struct.pack(">l",40)) f.write(struct.pack(">l",24)) f.write(struct.pack(">l",R.year)) f.write(struct.pack(">l",1)) f.write(struct.pack(">l",1)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",24)) f.write(struct.pack(">l",16)) f.write(struct.pack(">l",len(ElementList))) f.write(struct.pack(">l",len(NodeList))) f.write(struct.pack(">l",3)) f.write(struct.pack(">l",1)) f.write(struct.pack(">l",16)) f.write(struct.pack(">l",len(ElementList)*3*4)) for el in ElementList: for nd in ElementDict[el]: f.write(struct.pack(">l",nodeOrdered[nd])) f.write(struct.pack(">l",len(ElementList)*3*4)) f.write(struct.pack(">l",len(NodeList))) for i in range(0,len(NodeList)): f.write(struct.pack(">l",0)) f.write(struct.pack(">l",len(NodeList))) f.write(struct.pack(">l",len(NodeList)*4)) for key, value in nodeDict.items(): f.write(struct.pack(">f",value[0])) f.write(struct.pack(">l",len(NodeList)*4)) f.write(struct.pack(">l",len(NodeList)*4)) for key, value in nodeDict.items(): f.write(struct.pack(">f",value[1])) f.write(struct.pack(">l",len(NodeList)*4)) while R.next(): f.write(struct.pack(">l",4)) f.write(struct.pack(">f",R.time * 3600)) f.write(struct.pack(">l",4)) f.write(struct.pack(">l",len(NodeList)*4)) for key, value in nodeDict.items(): if key in node2nodevalue.keys(): f.write(struct.pack(">f",R.xvel[node2nodevalue[key]])) else: f.write(struct.pack(">f",R.xvel[key])) f.write(struct.pack(">l",len(NodeList)*4)) f.write(struct.pack(">l",len(NodeList)*4)) for key, value in nodeDict.items(): if key in node2nodevalue.keys(): f.write(struct.pack(">f",R.yvel[node2nodevalue[key]])) else: f.write(struct.pack(">f",R.yvel[key])) f.write(struct.pack(">l",len(NodeList)*4)) f.write(struct.pack(">l",len(NodeList)*4)) for key, value in nodeDict.items(): if key in node2nodevalue.keys(): f.write(struct.pack(">f",R.depth[node2nodevalue[key]])) else: f.write(struct.pack(">f",R.depth[key])) f.write(struct.pack(">l",len(NodeList)*4)) f.write(struct.pack(">l",len(NodeList)*4)) for key, value in nodeDict.items(): if key in node2nodevalue.keys(): f.write(struct.pack(">f",R.elevation[node2nodevalue[key]])) else: f.write(struct.pack(">f",R.elevation[key])) f.write(struct.pack(">l",len(NodeList)*4)) f.close() def RMA11toSerafin(option=1): f = open('{}_{}.slf'.format(run,option), 'wb') f.write(struct.pack(">l",80)) strtemp='{0: >80}'.format('SERAFIN ') f.write(strtemp.encode('ascii')) f.write(struct.pack(">l",80)) f.write(struct.pack(">l",8)) f.write(struct.pack(">l",len(constName))) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",8)) for idx,c in enumerate(constName): f.write(struct.pack(">l",32)) strtemp='{0: <32}'.format(str(idx)+c) f.write(strtemp.encode('ascii')) f.write(struct.pack(">l",32)) f.write(struct.pack(">l",40)) f.write(struct.pack(">l",1)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",1)) f.write(struct.pack(">l",40)) f.write(struct.pack(">l",24)) f.write(struct.pack(">l",startDate.year)) f.write(struct.pack(">l",startDate.month)) f.write(struct.pack(">l",startDate.day)) f.write(struct.pack(">l",startDate.hour)) f.write(struct.pack(">l",startDate.minute)) f.write(struct.pack(">l",startDate.second)) f.write(struct.pack(">l",24)) f.write(struct.pack(">l",16)) f.write(struct.pack(">l",len(ElementList))) f.write(struct.pack(">l",len(NodeList))) f.write(struct.pack(">l",3)) f.write(struct.pack(">l",1)) f.write(struct.pack(">l",16)) f.write(struct.pack(">l",len(ElementList)*3*4)) for el in ElementList: for nd in ElementDict[el]: f.write(struct.pack(">l",nodeOrdered[nd])) f.write(struct.pack(">l",len(ElementList)*3*4)) f.write(struct.pack(">l",len(NodeList))) for i in range(0,len(NodeList)): f.write(struct.pack(">l",0)) f.write(struct.pack(">l",len(NodeList))) f.write(struct.pack(">l",len(NodeList)*4)) for key, value in nodeDict.items(): f.write(struct.pack(">f",value[0])) f.write(struct.pack(">l",len(NodeList)*4)) f.write(struct.pack(">l",len(NodeList)*4)) for key, value in nodeDict.items(): f.write(struct.pack(">f",value[1])) f.write(struct.pack(">l",len(NodeList)*4)) while R.next(): #for i in range(3): currentDate = datetime(R.year,1,1) + timedelta(hours = R.time) timeCurrentStep = currentDate - startDate f.write(struct.pack(">l",4)) f.write(struct.pack(">f",timeCurrentStep.total_seconds())) f.write(struct.pack(">l",4)) f.write(struct.pack(">l",len(NodeList)*4)) for key, value in nodeDict.items(): writeConst('X-VEL',key,f) f.write(struct.pack(">l",len(NodeList)*4)) f.write(struct.pack(">l",len(NodeList)*4)) for key, value in nodeDict.items(): writeConst('Y-VEL',key,f) f.write(struct.pack(">l",len(NodeList)*4)) f.write(struct.pack(">l",len(NodeList)*4)) for key, value in nodeDict.items(): writeConst('DEPTH',key,f) f.write(struct.pack(">l",len(NodeList)*4)) f.write(struct.pack(">l",len(NodeList)*4)) for key, value in nodeDict.items(): writeConst('FREE SURFACE',key,f) f.write(struct.pack(">l",len(NodeList)*4)) for c in constNum: f.write(struct.pack(">l",len(NodeList)*4)) for key, value in nodeDict.items(): writeConst(c,key,f) f.write(struct.pack(">l",len(NodeList)*4)) f.close() def RMA10toSerafin(option=1): f = open('{}_{}.slf'.format(run,option), 'wb') f.write(struct.pack(">l",80)) strtemp='{0: >80}'.format('SERAFIN ') f.write(strtemp.encode('ascii')) f.write(struct.pack(">l",80)) f.write(struct.pack(">l",8)) f.write(struct.pack(">l",len(constName))) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",8)) for idx,c in enumerate(constName): f.write(struct.pack(">l",32)) strtemp='{0: <32}'.format(str(idx)+c) f.write(strtemp.encode('ascii')) f.write(struct.pack(">l",32)) f.write(struct.pack(">l",40)) f.write(struct.pack(">l",1)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",1)) f.write(struct.pack(">l",40)) f.write(struct.pack(">l",24)) f.write(struct.pack(">l",startDate.year)) f.write(struct.pack(">l",startDate.month)) f.write(struct.pack(">l",startDate.day)) f.write(struct.pack(">l",startDate.hour)) f.write(struct.pack(">l",startDate.minute)) f.write(struct.pack(">l",startDate.second)) f.write(struct.pack(">l",24)) f.write(struct.pack(">l",16)) f.write(struct.pack(">l",len(ElementList))) f.write(struct.pack(">l",len(NodeList))) f.write(struct.pack(">l",3)) f.write(struct.pack(">l",1)) f.write(struct.pack(">l",16)) f.write(struct.pack(">l",len(ElementList)*3*4)) for el in ElementList: for nd in ElementDict[el]: f.write(struct.pack(">l",nodeOrdered[nd])) f.write(struct.pack(">l",len(ElementList)*3*4)) f.write(struct.pack(">l",len(NodeList))) for i in range(0,len(NodeList)): f.write(struct.pack(">l",0)) f.write(struct.pack(">l",len(NodeList))) f.write(struct.pack(">l",len(NodeList)*4)) for key, value in nodeDict.items(): f.write(struct.pack(">f",value[0])) f.write(struct.pack(">l",len(NodeList)*4)) f.write(struct.pack(">l",len(NodeList)*4)) for key, value in nodeDict.items(): f.write(struct.pack(">f",value[1])) f.write(struct.pack(">l",len(NodeList)*4)) while R.next(): #for i in range(3): currentDate = datetime(R.year,1,1) + timedelta(hours = R.time) timeCurrentStep = currentDate - startDate f.write(struct.pack(">l",4)) f.write(struct.pack(">f",timeCurrentStep.total_seconds())) f.write(struct.pack(">l",4)) # f.write(struct.pack(">l",len(NodeList)*4)) # for key, value in nodeDict.items(): # writeConst('X-VEL',key,f) # f.write(struct.pack(">l",len(NodeList)*4)) for c in constName: f.write(struct.pack(">l",len(NodeList)*4)) data = [None] * len(nodeDict) idx = 0 for key, value in nodeDict.items(): data[idx] = writeConst(c,key) idx += 1 f.write(b''.join(data)) f.write(struct.pack(">l",len(NodeList)*4)) f.close() def writeConst(param,key): #get the surface node number # ['X-VEL','Y-VEL','DEPTH','FREE SURFACE','Constituent 1'......] if param == 'X-VEL': tempR = R.xvel elif param == 'Y-VEL': tempR = R.yvel elif param == 'Z-VEL': tempR = R.zvel elif param == 'DEPTH': tempR = R.depth elif param == 'FREE SURFACE': tempR = R.elevation elif param == 'TEMPERATURE': tempR = R.temperature elif param == 'SALINITY': tempR = R.salinity elif param == 'SSED': tempR = R.sussed else: tempR = R.constit[c] if key in node2nodevalue.keys(): key = node2nodevalue[key] #based on the surface node number and option selected calculate the value if option == 'Max': tempArr = [] for n in nodelayer[key]: tempArr.append(tempR[n]) tempVal1 = max(tempArr) elif isinstance(option, int): temppos = (option - 1 ) * 2 temppos2 = min([temppos,len(nodelayer[key])-1]) tempVal1 = tempR[nodelayer[key][temppos2]] else: tempVal1 = tempR[key] #save the value in the selaphin file #f.write(struct.pack(">f",tempVal1)) return struct.pack(">f",tempVal1) def writePCTL(c,pctl): f = open('{}_{}_{}.slf'.format(run,c,option), 'wb') f.write(struct.pack(">l",80)) strtemp='{0: >80}'.format('SERAFIN ') f.write(strtemp.encode('ascii')) f.write(struct.pack(">l",80)) f.write(struct.pack(">l",8)) f.write(struct.pack(">l",len(pctl))) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",8)) for idx,p in enumerate(pctl): f.write(struct.pack(">l",32)) strtemp='{0: <32}'.format(str(idx)+ 'pct ' + str(p)) f.write(strtemp.encode('ascii')) f.write(struct.pack(">l",32)) f.write(struct.pack(">l",40)) f.write(struct.pack(">l",1)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",0)) f.write(struct.pack(">l",1)) f.write(struct.pack(">l",40)) f.write(struct.pack(">l",24)) f.write(struct.pack(">l",startDate.year)) f.write(struct.pack(">l",startDate.month)) f.write(struct.pack(">l",startDate.day)) f.write(struct.pack(">l",startDate.hour)) f.write(struct.pack(">l",startDate.minute)) f.write(struct.pack(">l",startDate.second)) f.write(struct.pack(">l",24)) f.write(struct.pack(">l",16)) f.write(struct.pack(">l",len(ElementList))) f.write(struct.pack(">l",len(NodeList))) f.write(struct.pack(">l",3)) f.write(struct.pack(">l",1)) f.write(struct.pack(">l",16)) f.write(struct.pack(">l",len(ElementList)*3*4)) for el in ElementList: for nd in ElementDict[el]: f.write(struct.pack(">l",nodeOrdered[nd])) f.write(struct.pack(">l",len(ElementList)*3*4)) f.write(struct.pack(">l",len(NodeList))) for i in range(0,len(NodeList)): f.write(struct.pack(">l",0)) f.write(struct.pack(">l",len(NodeList))) f.write(struct.pack(">l",len(NodeList)*4)) for key, value in nodeDict.items(): f.write(struct.pack(">f",value[0])) f.write(struct.pack(">l",len(NodeList)*4)) f.write(struct.pack(">l",len(NodeList)*4)) for key, value in nodeDict.items(): f.write(struct.pack(">f",value[1])) f.write(struct.pack(">l",len(NodeList)*4)) columnName = [node2nodevalue[key] if key in node2nodevalue.keys() else key for key, value in nodeDict.items()] global dfAll dfAll = {} for key, value in nodeDict.items(): if key in node2nodevalue.keys(): key = node2nodevalue[key] dfAll[key] = np.array([]) cpt = 0 currentDate = datetime(R.year,1,1) + timedelta(hours = R.time) timeCurrentStep = currentDate - startDate f.write(struct.pack(">l",4)) f.write(struct.pack(">f",timeCurrentStep.total_seconds())) f.write(struct.pack(">l",4)) while R.next(): tempR = R.constit[c] for key, value in nodeDict.items(): if key in node2nodevalue.keys(): key = node2nodevalue[key] tempArr = [] for n in nodelayer[key]: tempArr.append(tempR[n]) tempVal1 = max(tempArr) dfAll[key] = np.append(dfAll[key],tempVal1) cpt = cpt + 1 for p in pctl: f.write(struct.pack(">l",len(NodeList)*4)) for key, value in nodeDict.items(): if key in node2nodevalue.keys(): key = node2nodevalue[key] f.write(struct.pack(">f",np.quantile(dfAll[key],p))) f.write(struct.pack(">l",len(NodeList)*4)) def readNodeLayers(fname,Nnodes): fn = open(fname) lines = fn.readlines() surfaceNode = int(lines[0]) nodelayer[surfaceNode] = [] for l in lines: l = int(l) if (l != surfaceNode) & (l < Nnodes + 1): surfaceNode = l nodelayer[surfaceNode] = [] nodelayer[surfaceNode].append(l) # In[67]: #Read mesh file and extract node (except mid node) and elements - plus convert 1D element to 2D for vizualisation nodelayer = {} NodeList = [] ElementList = [] ElementDict = {} nodeDict = {} node1Dduplicate = {} #Original Number: List of Duplicates node2nodevalue = {} #link between the node number and the node value to use #(e.g. if node 10 is a 1D node: 10 is not duplicate so {1:1}, #but node 2050 (duplicate of 10) (1D to 2D) the value of the duplicated #node will be the same as the original so we might have {2050: 10}) with open(meshFilename) as f: line = f.readline() line = f.readline() line = f.readline() line = f.readline() cpt = 1 while line and line != ' 9999\n': #temp = line.split() temp = re.findall('.....',line) ElementDict[int(temp[0])] = [int(temp[i]) for i in range(1,9) if int(temp[i]) != 0 and int(temp[9]) < 100] ElementList.append(int(temp[0])) line = f.readline() for key, value in ElementDict.items(): NodeList.extend(value) NodeList = list(set(NodeList)) line = f.readline() while line and line != ' 9999\n': formatFix = (10,16,20,14,10,10) temp = [] for form in formatFix: temp.append(line[:form]) line = line[form:] #line.split() if int(temp[0]) in NodeList: nodeDict[int(temp[0])] = [float(temp[1]),float(temp[2]),float(temp[3])] line = f.readline() print('Processing triangles') ElementList2 = ElementList[:] for e in ElementList2: print(e) if isElementTriangle(ElementDict[e]): triangle2Triangles(e) elif isElementOneD(ElementDict[e]): oneD2triangle(e) else: square2Triangle(e) for key in list(ElementDict): #Remove Special Element 902..... if len(ElementDict[key]) != 3: print(key, ElementDict[key]) ElementDict.pop(key) ElementList.remove(key) nodeOrdered = {} cpt = 1 for key, value in nodeDict.items(): nodeOrdered[key] = cpt cpt +=1 # # Open and Read First Step of the RMA File and Save a Serafin # In[72]: for run in RMAfilenames: for option in Options: RMAfilename = '{}'.format(run) R=rma() R.open(RMAfilename) R.next() startDate = datetime(R.year,1,1) + timedelta(hours = R.time) if R.type==b'RMA11 ': constName = [] readNodeLayers(NodeLayersFile,Nnodes) constName = ['X-VEL','Y-VEL','DEPTH','FREE SURFACE'] for c in constNum: constName.append(R.constit_name[c].decode("utf-8")) print(c) if option == 'Percentile': for c in constNum: writePCTL(c,percentiles) else: RMA11toSerafin(option) if R.type=='RMA10 ': constName = [] readNodeLayers(NodeLayersFile,Nnodes) constName = ['X-VEL','Y-VEL','Z-VEL','FREE SURFACE','SALINITY','TEMPERATURE'] if option == 'Percentile': for c in constNum: writePCTL(c,percentiles) else: RMA10toSerafin(option) if R.type==b'RMA2 ': constName = ['X-VEL','Y-VEL','DEPTH','FREE SURFACE'] RMA2toSerafin()