main.py

from __future__ import print_function, division

import sys

import numpy as np
from PIL import Image

from config.Config import Config
from core.CTScanSimulator import CTScanSimulator
from core.SinogramConverter import SinogramConverter


def do_scan():
    sinogram, radius = generate_sinogram()
    img = Image.fromarray(np.uint8(sinogram * 255), 'L')
    img.save(Config.config["outputSinogram"])
    result = convert_sinogram(sinogram, radius)
    img2 = Image.fromarray(np.uint8(result * 255), 'L')
    img2.save(Config.config["outputCtScan"])


def generate_sinogram():
    simulator = CTScanSimulator(image_name=sys.argv[1])
    return simulator.simulate()


def convert_sinogram(sinogram, radius):
    converter = SinogramConverter(sinogram, radius)
    return converter.convert()


if __name__ == '__main__':
    do_scan()