def test_get_tilt_series_label(self):
self.assertEqual(util.get_tilt_series_label(0), 'a')
self.assertEqual(util.get_tilt_series_label(2), 'c')
self.assertEqual(util.get_tilt_series_label(24), 'y')
self.assertEqual(util.get_tilt_series_label(25), 'z')
self.assertEqual(util.get_tilt_series_label(26), 'ba')
self.assertEqual(util.get_tilt_series_label(27), 'bb')
self.assertEqual(util.get_tilt_series_label(51), 'bz')
self.assertEqual(util.get_tilt_series_label(52), 'ca')
self.assertEqual(util.get_tilt_series_label(180), 'gy')
def _put_all_tilts_into_result_dir(self, dirlist):
resultdir = self._get_result_dir()
if not os.path.isdir(resultdir):
os.makedirs(resultdir)
counter = 0
for path in dirlist:
tiltname = util.get_tilt_series_label(counter)
# copy clip mrc file
shutil.copy(
os.path.join(path, self._projectionclipmrc),
os.path.join(resultdir, self._mrcname + tiltname + TiltSeriesCreator.PREALI_EXT),
)
# copy clip fid file
shutil.copy(
os.path.join(path, self._projectionclipfid),
os.path.join(resultdir, self._mrcname + tiltname + TiltSeriesCreator.FID_EXT),
)
# copy rawtilt if it exists
rawtlt = os.path.join(path, self._rawtlt)
if os.path.isfile(rawtlt):
destrawtlt = os.path.join(resultdir, self._mrcname + tiltname + TiltSeriesCreator.RAW_TLT_EXT)
shutil.copy(rawtlt, destrawtlt)
else:
if destrawtlt is not None:
shutil.copy(
destrawtlt, os.path.join(resultdir, self._mrcname + tiltname + TiltSeriesCreator.RAW_TLT_EXT)
)
counter += 1
