Andrei tomut commited on
Commit
7459d43
1 Parent(s): 25a8539
Files changed (1) hide show
  1. backend/latmathcher/pipeline.py +3 -2
backend/latmathcher/pipeline.py CHANGED
@@ -98,7 +98,7 @@ class PipelineLatMatch:
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,
@@ -126,7 +126,8 @@ class PipelineLatMatch:
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:
132
  atomic_numbers.append(atomic_number.number)
 
98
 
99
  return self.superA_xyz, self.superB_xyz
100
 
101
+ def get_new_structure(self,dist):
102
 
103
  structure = {
104
  "lattice_vectors": self.sc_vec3,
 
126
  for element in superB_xyz:
127
  structure["host_guest"].append("guest")
128
  symbol = element[0]
129
+ dd=element[1].tolist()+dist
130
+ positions.append(dd)
131
  atomic_number = getattr(periodictable, symbol)
132
  if atomic_number is not None:
133
  atomic_numbers.append(atomic_number.number)