51 lines
1.5 KiB
Python
51 lines
1.5 KiB
Python
# ev-to-dicom
|
|
# Copyright © 2023 Lee Yingtong Li
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero 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
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
import os
|
|
import sys
|
|
|
|
from .ev_to_dcm import ev_files_to_dcm_file
|
|
|
|
if len(sys.argv) < 2:
|
|
print('Usage: {} [output directory]'.format(sys.argv[0]))
|
|
sys.exit(0)
|
|
|
|
outdir = sys.argv[1]
|
|
if not os.path.exists(outdir):
|
|
print('Output directory does not exist')
|
|
sys.exit(1)
|
|
|
|
# Get series data
|
|
|
|
series_number = int(os.path.split(os.getcwd())[1].split('_')[0]) + 1
|
|
series_uid = os.path.split(os.getcwd())[1].split('_')[1]
|
|
study_uid = os.path.split(os.path.split(os.getcwd())[0])[1]
|
|
|
|
output_filename = '{:04d}_0000.dcm'.format(series_number - 1)
|
|
|
|
# Convert images
|
|
|
|
input_filenames = []
|
|
|
|
for filename in os.listdir('.'):
|
|
if filename.endswith('.bin'):
|
|
print(filename)
|
|
input_filenames.append(filename)
|
|
|
|
input_filenames.sort()
|
|
|
|
ev_files_to_dcm_file(study_uid, series_uid, series_number, 1, input_filenames, os.path.join(outdir, output_filename))
|