From bce16bfa089cfcec870887cb19d06c240efbad97 Mon Sep 17 00:00:00 2001
From: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
Date: Tue, 30 Jun 2020 17:35:13 -0400
Subject: [PATCH] from deepmd for MultiSystems

---
 dpdata/system.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/dpdata/system.py b/dpdata/system.py
index 1816c835a..dd1af870f 100644
--- a/dpdata/system.py
+++ b/dpdata/system.py
@@ -1485,6 +1485,17 @@ def to_deepmd_npy(self, folder, set_size = 5000, prec=np.float32) :
                                  set_size = set_size,
                                  prec = prec)
 
+    def from_deepmd_raw(self, folder):
+        for dd in os.listdir(folder):
+            self.append(LabeledSystem(os.path.join(folder, dd), fmt='deepmd/raw'))
+        return self
+
+    def from_deepmd_npy(self, folder):
+        for dd in os.listdir(folder):
+            self.append(LabeledSystem(os.path.join(folder, dd), fmt='deepmd/npy'))
+        return self
+
+
 def check_System(data):
     keys={'atom_names','atom_numbs','cells','coords','orig','atom_types'}
     assert( isinstance(data,dict) )