import os
import subprocess
import threading
import time

###  EDIT DATASET INFORMATION  ###
apix = 0.729
kv = 300
cs = 2.7
##################################


### EDIT COMPUTING INFORMATION ###
Star = 'movies.star'
Group_Size = 22
mpi = 16
threads = 4
##################################

subprocess.call(['relion_import', '--do_movies', '--i', '*.eer', '--odir', './', '--ofile', Star, '--continue', '--angpix', str(apix), '--kV', str(kv), '--Cs', str(cs)])

subprocess.run(['mpirun', '-n', str(mpi), 'relion_convert_to_tiff_mpi', '--i', Star, '--o', '.', '--j', str(threads) ,'--eer_grouping', str(Group_Size), '--compression', 'lzw', '--only_do_unfinished'])