1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
import ase.gui.ui as ui
from ase.gui.i18n import _
class Repeat:
def __init__(self, gui):
win = ui.Window(_('Repeat'), wmtype='utility')
win.add(_('Repeat atoms:'))
self.repeat = [ui.SpinBox(r, 1, 9, 1, self.change)
for r in gui.images.repeat]
win.add(self.repeat)
win.add(ui.Button(_('Set unit cell'), self.set_unit_cell))
for sb, vec in zip(self.repeat, gui.atoms.cell):
if not vec.any():
sb.active = False
self.gui = gui
def change(self):
repeat = [int(r.value) for r in self.repeat]
self.gui.images.repeat_images(repeat)
self.gui.set_frame()
def set_unit_cell(self):
self.gui.images.repeat_unit_cell()
for r in self.repeat:
r.value = 1
self.gui.set_frame()
|
