|
18 | 18 |
|
19 | 19 | import copy |
20 | 20 | import re |
| 21 | + |
21 | 22 | import numpy |
22 | 23 |
|
23 | | -from diffpy.pdfgui.control.pdfstructure import PDFStructure |
24 | 24 | from diffpy.pdfgui.control.constraint import Constraint |
25 | | -from diffpy.pdfgui.control.parameter import Parameter |
26 | 25 | from diffpy.pdfgui.control.controlerrors import ControlTypeError, ControlValueError |
| 26 | +from diffpy.pdfgui.control.parameter import Parameter |
| 27 | +from diffpy.pdfgui.control.pdfstructure import PDFStructure |
27 | 28 | from diffpy.structure import Atom |
28 | 29 |
|
29 | 30 |
|
@@ -309,12 +310,7 @@ def expandSuperCell(self, mno): |
309 | 310 | # back to business |
310 | 311 | acd = self._popAtomConstraints() |
311 | 312 | mnofloats = numpy.array(mno[:3], dtype=float) |
312 | | - ijklist = [ |
313 | | - (i, j, k) |
314 | | - for i in range(mno[0]) |
315 | | - for j in range(mno[1]) |
316 | | - for k in range(mno[2]) |
317 | | - ] |
| 313 | + ijklist = [(i, j, k) for i in range(mno[0]) for j in range(mno[1]) for k in range(mno[2])] |
318 | 314 | # build a list of new atoms |
319 | 315 | newatoms = [] |
320 | 316 | for a in self.initial: |
@@ -389,9 +385,7 @@ def getSpaceGroupList(self): |
389 | 385 | existing_names[sg.short_name] = True |
390 | 386 | # sort by International Tables number, stay compatible with 2.3 |
391 | 387 | n_sg = [(sg.number % 1000, sg) for sg in unique_named_list] |
392 | | - n_sg = sorted( |
393 | | - n_sg, key=lambda x: x[0] |
394 | | - ) # sort by the first element of tuple. |
| 388 | + n_sg = sorted(n_sg, key=lambda x: x[0]) # sort by the first element of tuple. |
395 | 389 | FitStructure.sorted_standard_space_groups = [sg for n, sg in n_sg] |
396 | 390 | sglist = list(FitStructure.sorted_standard_space_groups) |
397 | 391 | if self.custom_spacegroup: |
@@ -438,9 +432,7 @@ def expandAsymmetricUnit(self, spacegroup, indices, sgoffset=[0, 0, 0]): |
438 | 432 | coreatoms = [self.initial[i] for i in ruindices] |
439 | 433 | corepos = [a.xyz for a in coreatoms] |
440 | 434 | coreUijs = [a.U for a in coreatoms] |
441 | | - eau = ExpandAsymmetricUnit( |
442 | | - spacegroup, corepos, coreUijs, sgoffset=sgoffset, eps=self.symposeps |
443 | | - ) |
| 435 | + eau = ExpandAsymmetricUnit(spacegroup, corepos, coreUijs, sgoffset=sgoffset, eps=self.symposeps) |
444 | 436 | # build a nested list of new atoms: |
445 | 437 | newatoms = [] |
446 | 438 | for i in range(len(coreatoms)): |
@@ -469,9 +461,7 @@ def expandAsymmetricUnit(self, spacegroup, indices, sgoffset=[0, 0, 0]): |
469 | 461 | self._restoreAtomConstraints(acd) |
470 | 462 | return |
471 | 463 |
|
472 | | - def applySymmetryConstraints( |
473 | | - self, spacegroup, indices, posflag, Uijflag, sgoffset=[0, 0, 0] |
474 | | - ): |
| 464 | + def applySymmetryConstraints(self, spacegroup, indices, posflag, Uijflag, sgoffset=[0, 0, 0]): |
475 | 465 | """Generate symmetry constraints for positions and thermal factors. |
476 | 466 | Both positions and thermal factors may get corrected to reflect |
477 | 467 | space group symmetry. Old positional and thermal constraints get |
@@ -511,9 +501,7 @@ def applySymmetryConstraints( |
511 | 501 | selatoms = [self.initial[i] for i in uindices] |
512 | 502 | selpos = [a.xyz for a in selatoms] |
513 | 503 | selUijs = [a.U for a in selatoms] |
514 | | - symcon = SymmetryConstraints( |
515 | | - spacegroup, selpos, selUijs, sgoffset=sgoffset, eps=self.symposeps |
516 | | - ) |
| 504 | + symcon = SymmetryConstraints(spacegroup, selpos, selUijs, sgoffset=sgoffset, eps=self.symposeps) |
517 | 505 | # deal with positions |
518 | 506 | if posflag: |
519 | 507 | # fix positions: |
@@ -762,8 +750,8 @@ def load(self, z, subpath): |
762 | 750 | subpath -- path to its own storage within project file |
763 | 751 | """ |
764 | 752 | # subpath = projname/fitname/structure/myname/ |
765 | | - from diffpy.pdfgui.utils import asunicode |
766 | 753 | from diffpy.pdfgui.control.pdfguicontrol import CtrlUnpickler |
| 754 | + from diffpy.pdfgui.utils import asunicode |
767 | 755 |
|
768 | 756 | subs = subpath.split("/") |
769 | 757 | rootDict = z.fileTree[subs[0]][subs[1]][subs[2]][subs[3]] |
|
0 commit comments