Andrei tomut commited on
Commit
25a8539
1 Parent(s): 87f2929
backend/latmathcher/__pycache__/pipeline.cpython-310.pyc DELETED
Binary file (6.49 kB)
 
backend/latmathcher/__pycache__/pipeline.cpython-311.pyc DELETED
Binary file (12.5 kB)
 
backend/latmathcher/pipeline.py CHANGED
@@ -14,6 +14,7 @@ class PipelineLatMatch:
14
  self.Blat3D = Blat3D
15
  self.Alat = self.Alat3D[:2, :2][:2, :2]
16
  self.Blat = self.Blat3D[:2, :2][:2, :2]
 
17
  self.dz = dz
18
 
19
  if sc_vec is not None:
@@ -97,13 +98,14 @@ class PipelineLatMatch:
97
 
98
  return self.superA_xyz, self.superB_xyz
99
 
100
- def get_new_structure(self, inter_distance=[0,0,0]):
101
 
102
- structure = {"atoms": [],
103
  "lattice_vectors": self.sc_vec3,
104
  "pbc": [True, True, False],
105
  "positions": [],
106
- "host_guest": [], }
 
107
 
108
  superA_xyz, superB_xyz = self.compute_super_atoms()
109
 
@@ -124,8 +126,6 @@ class PipelineLatMatch:
124
  for element in superB_xyz:
125
  structure["host_guest"].append("guest")
126
  symbol = element[0]
127
- pos=element[1].tolist()+inter_distance
128
-
129
  positions.append(element[1].tolist())
130
  atomic_number = getattr(periodictable, symbol)
131
  if atomic_number is not None:
 
14
  self.Blat3D = Blat3D
15
  self.Alat = self.Alat3D[:2, :2][:2, :2]
16
  self.Blat = self.Blat3D[:2, :2][:2, :2]
17
+
18
  self.dz = dz
19
 
20
  if sc_vec is not None:
 
98
 
99
  return self.superA_xyz, self.superB_xyz
100
 
101
+ def get_new_structure(self):
102
 
103
+ structure = {
104
  "lattice_vectors": self.sc_vec3,
105
  "pbc": [True, True, False],
106
  "positions": [],
107
+ "host_guest": [],
108
+ "atoms": [], }
109
 
110
  superA_xyz, superB_xyz = self.compute_super_atoms()
111
 
 
126
  for element in superB_xyz:
127
  structure["host_guest"].append("guest")
128
  symbol = element[0]
 
 
129
  positions.append(element[1].tolist())
130
  atomic_number = getattr(periodictable, symbol)
131
  if atomic_number is not None: