wip

Author: Francesco Rizzi

deprecated_to_revise/end-to-end-roms/2d_swe_customizations/__init__.py

@@ -0,0 +1,42 @@
+
+import os, subprocess
+
+def _inviscid_flux_string_to_stencil_size(stringIn):
+  if stringIn == "FirstOrder":
+    return 3
+  elif stringIn == "Weno3":
+    return 5
+  elif stringIn == "Weno5":
+    return 7
+  else:
+    sys.exit("Invalid scheme detected {}".format(scheme))
+    return None
+
+# -----------------------------------------------------------
+# fncs and constants below are mandatory 
+# -----------------------------------------------------------
+
+dimensionality = 2
+numDofsPerCell = 3
+validParameterNames = ['coriolis', 'pulsemag', 'gravity']
+
+def generate_full_mesh(pdaSrcPath, meshPath, dicIn):
+  pdaMeshDir = pdaSrcPath + "/meshing_scripts"
+
+  nx, ny = dicIn['meshSize'][0], dicIn['meshSize'][1]
+
+  # figure out the stencil size needed for the mesh
+  invFluxRec = dicIn['inviscidFluxReconstruction']
+  stencilSize = _inviscid_flux_string_to_stencil_size(invFluxRec)
+
+  print('Generating mesh {}'.format(meshPath))
+  # call script
+  owd = os.getcwd()
+  args = ("python3", pdaMeshDir + '/create_full_mesh.py',\
+          "-n", str(nx), str(ny),\
+          "--outDir", meshPath,\
+          "--bounds", "-5.0", "5.0", "-5.0", "5.0",
+          "-s", str(stencilSize))
+
+  popen  = subprocess.Popen(args, stdout=subprocess.PIPE); popen.wait()
+  output = popen.stdout.read();

deprecated_to_revise/end-to-end-roms/2d_swe_galerkin_default/CMakeLists.txt

@@ -0,0 +1,11 @@
+
+configure_file(${CMAKE_CURRENT_SOURCE_DIR}/wf.yaml wf.yaml COPYONLY)
+configure_file(${CMAKE_CURRENT_SOURCE_DIR}/plot.py plot.py COPYONLY)
+configure_file(${CMAKE_CURRENT_SOURCE_DIR}/clean.py clean.py COPYONLY)
+
+execute_process(
+  COMMAND
+  ln -sf
+  ${CMAKE_CURRENT_BINARY_DIR}/../${exename}
+  ${CMAKE_CURRENT_BINARY_DIR}
+  OUTPUT_VARIABLE OUTP)

deprecated_to_revise/end-to-end-roms/2d_swe_galerkin_default/FOM.png

@@ -0,0 +1,8 @@
+#!/usr/bin/env python3
+
+import os
+
+if __name__== "__main__":
+  os.system("rm -rf default_galerkin*")
+  os.system("rm -rf offline_rom")
+  os.system("rm -rf fom_*")

deprecated_to_revise/end-to-end-roms/2d_swe_galerkin_default/ROM_22.png

@@ -0,0 +1,74 @@
+#!/usr/bin/env python3
+
+import matplotlib.pyplot as plt
+from matplotlib import cm
+import numpy as np
+import re, os
+
+def extractMeshSize(ns):
+  reg = re.compile(r''+ns+'.+')
+  file1 = open('./fom_mesh/info.dat', 'r')
+  strings = re.search(reg, file1.read())
+  file1.close()
+  assert(strings)
+  return int(strings.group().split()[1])
+
+def plotStateErrorOverTime(romDirs):
+  # plot state error over time
+  fig = plt.figure(1)
+  for dirIt in romDirs:
+    # extract how many modes
+    modeCount = np.loadtxt(dirIt + "/rom_dofs_count.txt")
+    print(modeCount)
+    # load time errors
+    errors = np.loadtxt(dirIt + "/errors_in_time.txt")
+    plt.plot(errors[:,0], '-', label="modeCount = " + str(modeCount))
+
+  plt.ylim((0,0.1))
+  plt.legend()
+
+def plotHeight(romDirs):
+  nx = extractMeshSize('nx')
+  ny = extractMeshSize('ny')
+  fomTotDofs = nx*ny*3
+
+  meshCoords = np.loadtxt('./fom_mesh/coordinates.dat', dtype=float)
+  x, y = meshCoords[:,1], meshCoords[:,2]
+  x = np.reshape(x, (ny,nx))
+  y = np.reshape(y, (ny,nx))
+
+  fig = plt.figure(1)
+  ax = fig.add_subplot(1, 1, 1, projection='3d')
+  fomD = np.fromfile("fom_test_runid_0/state_snapshots.bin")
+  nt = int(np.size(fomD)/fomTotDofs)
+  print("fomTest: nt = ", nt)
+  fomD = np.reshape(fomD, (nt, fomTotDofs))
+  h = fomD[-1,0::3]
+  bounds = [np.min(h), np.max(h)]
+  fomD = np.reshape(h, (ny,nx))
+  surf = ax.plot_surface(x, y, fomD, cmap=cm.jet, \
+                          vmin=bounds[0], vmax=bounds[1])
+  ax.set_title("FOM", fontsize=20)
+  ax.set_zlim((0.98, 1.06))
+  fig.savefig("FOM.png", format="png", bbox_inches='tight', dpi=300)
+
+  for i in [0,1]:
+    fig  = plt.figure(i+2)
+    ax   = fig.add_subplot(1, 1, 1, projection='3d')
+    K    = np.loadtxt(romDirs[i] + "/rom_dofs_count.txt", dtype=int)
+    romD = np.loadtxt(romDirs[i]+"/reconstructed.txt")
+    romD = np.reshape(romD[0::3,-1], (ny,nx))
+    surf = ax.plot_surface(x, y, romD, cmap=cm.jet, \
+                           vmin=bounds[0], vmax=bounds[1])
+    ax.set_title("ROM: # modes = " + str(K), fontsize=20)
+    ax.set_zlim((0.98, 1.06))
+    fig.savefig("ROM_"+str(K)+".png", format="png", bbox_inches='tight', dpi=300)
+
+  plt.show()
+
+if __name__== "__main__":
+  # find all rom RUNs
+  romDirs = [d for d in os.listdir(".") if "default_galerkin" in d]
+  assert len(romDirs)==2, "for this demo, should only have two runs"
+
+  plotHeight(romDirs)

deprecated_to_revise/end-to-end-roms/2d_swe_galerkin_default/ROM_55.png

@@ -0,0 +1,35 @@
+problem: 2d_swe
+
+parameterSpace:
+  names: ['coriolis', 'pulsemag', 'gravity']
+  trainPoints:
+    0: [-1.0, 0.125, 9.8]
+    1: [ 0.0, 0.125, 9.8]
+  testPoints:
+    0: [-0.5, 0.125, 9.8]
+
+fom:
+  meshSize: [65, 65]
+  inviscidFluxReconstruction: "Weno3"
+  odeScheme: "RK4"
+  timeStepSize: 0.005
+  train:
+    finalTime: 5.0
+    stateSamplingFreq: 10
+    rhsSamplingFreq: 5
+  test:
+    finalTime: 5.0
+    stateSamplingFreq: 10
+    rhsSamplingFreq: 100
+
+offlineRom:
+  pod:
+    stateData:
+      useTrainingRuns: all
+    rhsData:
+      useTrainingRuns: all
+
+onlineRom:
+  algorithm: defaultGalerkin
+  podTruncation:
+    energyBased: [99.999, 99.99999]

deprecated_to_revise/end-to-end-roms/2d_swe_galerkin_default/clean.py

@@ -0,0 +1,11 @@
+
+configure_file(${CMAKE_CURRENT_SOURCE_DIR}/wf.yaml wf.yaml COPYONLY)
+configure_file(${CMAKE_CURRENT_SOURCE_DIR}/plot.py plot.py COPYONLY)
+configure_file(${CMAKE_CURRENT_SOURCE_DIR}/clean.py clean.py COPYONLY)
+
+execute_process(
+  COMMAND
+  ln -sf
+  ${CMAKE_CURRENT_BINARY_DIR}/../${exename}
+  ${CMAKE_CURRENT_BINARY_DIR}
+  OUTPUT_VARIABLE OUTP)