Add support for gradients in LEOS

LLNL · Jul 14, 2023 · 527090e · 527090e
1 parent adea44b
commit 527090e
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/pyranda/pyrandaLEOS.py b/pyranda/pyrandaLEOS.py
@@ -113,13 +113,13 @@ def temperature(self,matID,rho,ie):
 
 
 
-    def addMaterial(self,EOS_number,function_names=['Pt','Et','Cs'],matID=None):
+    def addMaterial(self,EOS_number,function_names=['Pt','Et','Cs'],matID=None,dxdy=False):
 
         # Assign a name/ID for this material
         if not matID:
             matID = EOS_number
 
-        self.materials[matID] = self.__setupLEOS__(EOS_number,function_names)
+        self.materials[matID] = self.__setupLEOS__(EOS_number,function_names,dxdy=dxdy)
 
 
 
@@ -129,7 +129,10 @@ def __setupLEOS__(self,EOS_File,fnames,dxdy=False):
 
         # Set options
         lopts = leospy.LEOS_LookupOptions()
-        lopts.calculateFunction(True).calculateDFDX(False).calculateDFDY(False)
+        if not dxdy:
+            lopts.calculateFunction(True).calculateDFDX(False).calculateDFDY(False)
+        else :
+            lopts.calculateFunction(True).calculateDFDX(True).calculateDFDY(True)
 
         fopts = leospy.LEOS_FunctionOptions()
         fopts.interpolation(leospy.BICUBIC)    # Bi-cubic