diff --git a/scripts/01_aRTist_bridge.py b/scripts/01_aRTist_bridge.py index 2263754da64e47305ba9845317799d48425f2497..ff40e914eec7ff53c8f8ba007931fe6f3c0f6789 100644 --- a/scripts/01_aRTist_bridge.py +++ b/scripts/01_aRTist_bridge.py @@ -4,6 +4,11 @@ from CyXTraX.common import MeshObject import numpy as np from matplotlib import pyplot as plt +from pathlib import Path + + +FOLDER = Path(__file__) +TEMP_FOLDER = FOLDER.parent.parent / 'temp' def main(): @@ -41,13 +46,23 @@ def main(): ax1.imshow(half_rays) ax1.set_xlabel(r'$Z$ / mm') ax1.set_ylabel(r'$\alpha$ / rad') + ax1.set_title('Half Rays') + ax1.set_xticks([0, 1000, 2000]) + ax1.set_xticklabels([r'$-3141.5$', r'$0.$', r'$3141.5$']) + ax1.set_yticks([0, 1000, 2000]) + ax1.set_yticklabels([r'$-\pi$', r'$0.$', r'$\pi$']) ax2 = fig.add_subplot(122) ax2.imshow(full_rays) ax2.set_xlabel(r'$Z$ / mm') ax2.set_ylabel(r'$\alpha$ / rad') + ax2.set_title('Full Rays') + ax2.set_xticks([0, 1000, 2000]) + ax2.set_xticklabels([r'$-3141.5$', r'$0.$', r'$3141.5$']) + ax2.set_yticks([0, 1000, 2000]) + ax2.set_yticklabels([r'$-\pi$', r'$0.$', r'$\pi$']) - plt.show() + plt.savefig(TEMP_FOLDER / 'scones.png') if __name__ == '__main__': diff --git a/scripts/04_mapping.py b/scripts/04_mapping.py index c0dc75e7f61a7fe588129c5c03b18fa72cccfa03..657fe50f8c8895113acfadcfe5272adfa35cd997 100644 --- a/scripts/04_mapping.py +++ b/scripts/04_mapping.py @@ -59,7 +59,7 @@ def main(): print(f'Step {i}: Value Cylinder {values_calc[i]:2.2f} \t/\t Value Projection {mapped_values[i]:2.2f}') - fig = plt.figure(figsize=(25, 10)) + fig = plt.figure(figsize=(24, 10)) subfig1: figure.Figure subfig2: figure.Figure subfig1, subfig2 = fig.subfigures(1, 2) @@ -91,9 +91,11 @@ def main(): ax.scatter(angles_calc[0, counter], angles_calc[1, counter], c=colors[counter]) ax.set_xlabel(r'$z$ / mm') ax.set_ylabel(r'$\alpha$ / rad') - ax.set_xticks([]) - ax.set_yticks([]) - + ax.set_xticks([0, 1000, 2000]) + ax.set_xticklabels([r'$-3141.5$', r'$0.$', r'$3141.5$']) + ax.set_yticks([0, 1000, 2000]) + ax.set_yticklabels([r'$-\pi$', r'$0.$', r'$\pi$']) + counter += 1 plt.tight_layout() diff --git a/temp/mapping.png b/temp/mapping.png index cc0c91510288b5aac100e77d47e7ecf87072de28..4af6c15f22fe97cc7cea7579c2e417b86164ee86 100644 Binary files a/temp/mapping.png and b/temp/mapping.png differ