counter_level.py

from utils import rescale_ctf_volume, apply_ctf, rotate
from file_handler import read_mrc, write_mrc

import os, sys

current_dir = os.getcwd()
split_path = current_dir.split("/")
tomoID = split_path[-2]

vol = read_mrc(sys.argv[1])
ctf = read_mrc(sys.argv[2])

rescaled_ctf = rescale_ctf_volume(ctf,ctf.shape,vol.shape)
ctf_vol = apply_ctf(vol,rescaled_ctf)
write_mrc(ctf_vol,'sim'+tomoID+'_side.mrc')

vol_disk = rotate(vol,[0,90,0])
ctf_vol = apply_ctf(vol_disk,rescaled_ctf)
write_mrc(ctf_vol,'sim'+tomoID+'_disk.mrc')
vol_tilt = rotate(vol,[0,45,45])
ctf_vol = apply_ctf(vol_tilt,rescaled_ctf)
write_mrc(ctf_vol,'sim'+tomoID+'_tilt.mrc')