support for reading and writing neqsim objects to xml (#212)

* feat: added support for reading and writing neqsim objects to xml * refact: clean up imports --------- Co-authored-by: Åsmund Våge Fannemel <[email protected]>
equinor · Nov 16, 2023 · a3feabe · a3feabe
1 parent 3387328
commit a3feabe
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 2 deletions.
diff --git a/neqsim/__init__.py b/neqsim/__init__.py
@@ -4,7 +4,7 @@
 It uses the Jpype module for bridging python and Java.
 """
 
-from .neqsimpython import *
+from neqsim.neqsimpython import jNeqSim, jpype
 
 def methods(checkClass):
     methods = checkClass.getClass().getMethods()
@@ -23,6 +23,20 @@ def has_tabulate():
 
 
 def setDatabase(connectionString):
-    from neqsim.neqsimpython import jNeqSim
     jNeqSim.util.database.NeqSimDataBase.setConnectionString(connectionString)
     jNeqSim.util.database.NeqSimDataBase.setCreateTemporaryTables(True)
+
+def save_xml(javaobject, filename):
+    xstream = jpype.JPackage('com.thoughtworks.xstream')
+    streamer = xstream.XStream()
+    xml = streamer.toXML(javaobject)
+    print(xml, file=open(filename, 'w'))
+    return xml
+
+def open_xml(filename):
+    xstream = jpype.JPackage('com.thoughtworks.xstream')
+    streamer = xstream.XStream()
+    streamer.addPermission(xstream.security.AnyTypePermission.ANY)
+    str = open(filename, 'r').read()
+    neqsimobj = streamer.fromXML(str)
+    return neqsimobj
diff --git a/tests/test_NeqSim.py b/tests/test_NeqSim.py
@@ -4,6 +4,7 @@
 
 @author: ESOL
 """
+import os
 from neqsim.neqsimpython import jNeqSim
 
 
@@ -231,3 +232,18 @@ def test_fullOffshoreProcess():
     exportCompressor.setOutletPressure(200.0, 'bara')
     exportGas = stream(exportCompressor.getOutStream())
     runProcess()
+
+def testwriteandopen(tmp_path):
+    import neqsim
+    from neqsim.thermo import createfluid
+
+    temp_dir = tmp_path / "test"
+    os.mkdir(temp_dir)
+
+    temp_file =  temp_dir / "name.xml"  
+
+    fluid1 = createfluid('dry gas')
+    neqsim.save_xml(fluid1, temp_file)
+    fluid2 = neqsim.open_xml(temp_file)
+
+    assert fluid1.getTemperature() == fluid2.getTemperature()