Source code for relion.convert.convert30

# **************************************************************************
# *
# * Authors:     J.M. de la Rosa Trevin ( [1]
# *              Grigory Sharov ( [2]
# *
# * [1] SciLifeLab, Stockholm University
# * [2] MRC Laboratory of Molecular Biology, MRC-LMB
# *
# * This program is free software; you can redistribute it and/or modify
# * it under the terms of the GNU General Public License as published by
# * the Free Software Foundation; either version 3 of the License, or
# * (at your option) any later version.
# *
# * This program is distributed in the hope that it will be useful,
# * but WITHOUT ANY WARRANTY; without even the implied warranty of
# * GNU General Public License for more details.
# *
# * You should have received a copy of the GNU General Public License
# * along with this program; if not, write to the Free Software
# * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
# * 02111-1307  USA
# *
# *  All comments concerning this program package may be sent to the
# *  e-mail address ''
# *
# **************************************************************************

import numpy as np

import pwem
import pwem.convert.transformations as tfs

from ..constants import *
from .convert_base import WriterBase, ReaderBase
from .convert_deprecated import readSetOfParticles

[docs]class Writer(WriterBase): """ Helper class to convert from Scipion SetOfImages subclasses with star file format previous to Relion>3.1, but providing the same interface as the new Writer class. """
[docs] def writeSetOfMovies(self, moviesIterable, starFile): self._writeSetOfMoviesOrMics(moviesIterable, starFile, 'movies', 'rlnMicrographMovieName')
[docs] def writeSetOfMicrographs(self, micsIterable, starFile): self._writeSetOfMoviesOrMics(micsIterable, starFile, 'micrographs', 'rlnMicrographName')
[docs] def writeSetOfParticles(self, partsSet, starFile, **kwargs): # FIXME: Remove deprecated import from .convert_deprecated import _writeSetOfParticles _writeSetOfParticles(partsSet, starFile, **kwargs)
def _writeSetOfMoviesOrMics(self, imgIterable, starFile, tableName, imgLabelName): """ Internal function to write either movies or micrographs star files. Input can be any iterable of these type of images (e.g set, list, etc). """ # Process the first item and create the table based # on the generated columns self._imgLabelName = imgLabelName self._prefix = tableName[:3] micRow = OrderedDict() micRow[imgLabelName] = '' # Just to add label, proper value later iterMics = iter(imgIterable) mic = next(iterMics) self._setCtf = mic.hasCTF() self._micToRow(mic, micRow) micsTable = self._createTableFromDict(micRow) while mic is not None: micRow[imgLabelName] = self._convert(mic) self._micToRow(mic, micRow) micsTable.addRow(**micRow) mic = next(iterMics, None) with open(starFile, 'w') as f: f.write("# Star file generated with Scipion\n") micsTable.writeStar(f, tableName=tableName) def _micToRow(self, mic, row): WriterBase._micToRow(self, mic, row) # Set CTF values if self._setCtf: self._ctfToRow(mic.getCTF(), row) # Add now the Acquisition labels acq = mic.getAcquisition() row.update({ 'rlnVoltage': acq.getVoltage(), 'rlnSphericalAberration': acq.getSphericalAberration(), 'rlnAmplitudeContrast': acq.getAmplitudeContrast() })
[docs]class Reader(ReaderBase): ALIGNMENT_LABELS = [ "rlnOriginX", "rlnOriginY", "rlnOriginZ", "rlnAngleRot", "rlnAngleTilt", "rlnAnglePsi", ]
[docs] def readSetOfParticles(self, starFile, partsSet, **kwargs): """ Convert a star file into a set of particles. Params: starFile: the filename of the star file partsSet: output particles set Keyword Arguments: blockName: The name of the data block (default particles) alignType: removeDisabled: """ readSetOfParticles(starFile, partsSet, **kwargs)
[docs] def setParticleTransform(self, particle, row): """ Set the transform values from the row. """ if ((self._alignType == pwem.ALIGN_NONE) or not row.hasAnyColumn(self.ALIGNMENT_LABELS)): self.setParticleTransform = self.__setParticleTransformNone else: # Ensure the Transform object exists self._angles = np.zeros(3) self._shifts = np.zeros(3) particle.setTransform(pwem.objects.Transform()) if self._alignType == pwem.ALIGN_2D: self.setParticleTransform = self.__setParticleTransform2D elif self._alignType == pwem.ALIGN_PROJ: self.setParticleTransform = self.__setParticleTransformProj else: raise TypeError("Unexpected alignment type: %s" % self._alignType) # Call again the modified function self.setParticleTransform(particle, row)
def __setParticleTransformNone(self, particle, row): particle.setTransform(None) def __setParticleTransform2D(self, particle, row): angles = self._angles shifts = self._shifts def _get(label): return float(getattr(row, label, 0.)) shifts[0] = _get('rlnOriginX') shifts[1] = _get('rlnOriginY') angles[2] = -_get('rlnAnglePsi') radAngles = -np.deg2rad(angles) M = tfs.euler_matrix(radAngles[0], radAngles[1], radAngles[2], 'szyz') M[:3, 3] = shifts[:3] particle.getTransform().setMatrix(M) def __setParticleTransformProj(self, particle, row): angles = self._angles shifts = self._shifts def _get(label): return float(getattr(row, label, 0.)) shifts[0] = _get('rlnOriginX') shifts[1] = _get('rlnOriginY') shifts[2] = _get('rlnOriginZ') angles[0] = _get('rlnAngleRot') angles[1] = _get('rlnAngleTilt') angles[2] = _get('rlnAnglePsi') radAngles = -np.deg2rad(angles) # TODO: jmrt: Maybe we should test performance and consider if keeping # TODO: the matrix and not creating one everytime will make things faster M = tfs.euler_matrix(radAngles[0], radAngles[1], radAngles[2], 'szyz') M[:3, 3] = -shifts[:3] M = np.linalg.inv(M) particle.getTransform().setMatrix(M)