Merge pull request #50 from LLNL/ContinuationSolver

Initial implementation of continuation solve

pyscripts/rundt.py

@@ -22,6 +22,18 @@ def __init__(self, *args, n_stor = False, **kwargs):
         # TODO: Add restore/recover from timeslice
         # TODO: Add plot timeslice directly
         # NOTE: No -> Utilize direct I/O from file instead
+        self.numvar = bbb.numvar
+        try:
+            self.nx
+        except:
+            self.nx = com.nx
+        try:
+            self.ny
+        except:
+            self.ny = com.ny
+        self.ixpt1 = com.ixpt1[0]
+        self.ixpt2 = com.ixpt2[0]
+        self.iysptrx = com.iysptrx
         self.equationkey = array([b'te', b'ti', b'phi', b'up', b'ni', b'ng', 
             b'tg'])
 
@@ -49,7 +61,15 @@ def itroub(self):
             bbb.idxu, bbb.idxn, bbb.idxg, bbb.idxtg]
         equationsdescription = [ 'Electron energy', 'Ion energy', 'Potential',
             'Ion momentum', 'Ion density', 'Gas density', 'Gas temperature']
-
+        # Assert required dict is present
+        try:
+            self.classvars
+        except:
+            self.classvars = {}
+            for cvar in ['itrouble', 'troubleeq', 'internaleq', 
+                    'internalspecies', 'troubleindex', 'dtrealfail',
+                    'ylfail', 'yldotsfscalfail']:
+                self.classvars[cvar] = []
         # Find the fortran index of the troublemaking equation
         self.classvars['itrouble'].append(deepcopy(argmax(abs(bbb.yldot*\
             bbb.sfscal)[:bbb.neq])+1))
@@ -120,7 +140,7 @@ def store_timeslice(self):
         for var in ['ni', 'ng', 'up', 'te', 'ti', 'tg', 'phi', 'dt_tot']:
             self.classvars['slice_{}'.format(var)].append(deepcopy(getattr(bbb, var)))
 
-
+            
     def save_intermediate(self, savename):
         from uedge.hdf5 import hdf5_save
         from os.path import exists
@@ -906,7 +926,6 @@ def dostaticiter():
                 return True
             return False
 
-
         def staticiter():
             """ Tries to reduce initial fnrm by iterating at constant var """
             from uedge import bbb
@@ -1140,6 +1159,10 @@ def dtsolve(dtdeltafac):
         bbb.ftol = ftol
         bbb.incpset = incpset
         bbb.itermx = itermx
+        # TODO: Resolve how to run continuation solver w/ mmaxu
+        # TODO: Add coding for autodetecting when avg nfe approx mmaxu
+        #       and upate jacobian then
+
 #        bbb.ismmaxuc = 0
 #        bbb.mmaxu = 70
         if (bbb.iterm == 1) and (bbb.ijactot > 0):