n = 21
import initialise as i
numberofhours, mincount, reconstructiongridwidth, orientation, eff, flux, area, solidangle, selectionefficiency = i.run(
)
import sys
sys.path.append('/afs/desy.de/user/s/steinrob/Documents/DESY/positioning/')
import argparse, math, random, time
import csv
import numpy as np
import simulate as s
import batchprocessing as bp
import batchreconstruction as br
sourcedata = "executedefault" + str(n)
processdata = "executeprocess" + str(n)
reconstructdata = "executereconstructed" + str(n)
detectedflux = float(flux) * float(area) * float(solidangle) * float(
selectionefficiency)
rateperhour = detectedflux * 60 * 60
n = int(rateperhour * float(numberofhours))
print time.asctime(
time.localtime()
), "Cosmic Ray Iron Flux is", flux, "Simulated Area is", area, "Field of View is", solidangle, "Detected Flux is", detectedflux
print time.asctime(
time.localtime()
n = 15
import initialise as i
numberofhours, mincount, reconstructiongridwidth, orientation, eff, flux, area, solidangle, selectionefficiency = (
i.run()
)
import sys
sys.path.append("/afs/desy.de/user/s/steinrob/Documents/DESY/positioning/")
import argparse, math, random, time
import csv
import numpy as np
import simulate as s
import batchprocessing as bp
import batchreconstruction as br
sourcedata = "executedefault" + str(n)
processdata = "executeprocess" + str(n)
reconstructdata = "executereconstructed" + str(n)
detectedflux = float(flux) * float(area) * float(solidangle) * float(selectionefficiency)
rateperhour = detectedflux * 60 * 60
n = int(rateperhour * float(numberofhours))
print time.asctime(
time.localtime()
), "Cosmic Ray Iron Flux is", flux, "Simulated Area is", area, "Field of View is", solidangle, "Detected Flux is", detectedflux
print time.asctime(
#!/bin/env python
n = 4
import initialise as i
numberofhours, mincount, reconstructiongridwidth, orientation, eff, flux, area, solidangle, selectionefficiency = i.run()
import sys
sys.path.append('/afs/desy.de/user/s/steinrob/Documents/DESY/positioning/')
import argparse, math, random, time
import csv
import numpy as np
import simulate as s
import batchprocessing as bp
import batchreconstruction as br
sourcedata="executedefault" + str(n)
processdata="executeprocess" + str(n)
reconstructdata="executereconstructed" + str(n)
detectedflux = float(flux)*float(area)*float(solidangle)*float(selectionefficiency)
rateperhour = detectedflux * 60 * 60
n = int(rateperhour*float(numberofhours))
print time.asctime(time.localtime()),"Cosmic Ray Iron Flux is", flux, "Simulated Area is", area, "Field of View is", solidangle, "Detected Flux is", detectedflux
print time.asctime(time.localtime()),"Rate per hour", rateperhour, "Simulated Hours", numberofhours, "Simulated Events", n
with open("/afs/desy.de/user/s/steinrob/Documents/DESY/positioning/orientations/"+ orientation +".csv", 'rb') as csvfile:
reader = csv.reader(csvfile, delimiter=',', quotechar='|')
#!/bin/env python
import initialise as i
numberofhours, mincount, gridwidth, layout, raweff, flux, area, solidangle, selectionefficiency, hmacceptance = i.run()
import sys
sys.path.append('/nfs/astrop/d6/rstein/Hamburg-Cosmic-Rays/positioning/')
import argparse, math, random, time
import csv
import numpy as np
import simulate as s
import batchprocessing as bp
import batchreconstruction as br
from classes import *
parser = argparse.ArgumentParser()
parser.add_argument("-n", "--number", default=1)
cfg = parser.parse_args()
sourcedata="executedefault" + str(cfg.number)
processdata="executeprocess" + str(cfg.number)
reconstructdata="executereconstructed" + str(cfg.number)
detectedflux = float(flux)*float(area)*float(solidangle)*float(selectionefficiency)
rateperhour = detectedflux * 60 * 60
n = int(rateperhour*float(numberofhours))
print time.asctime(time.localtime()),"Cosmic Ray Iron Flux is", flux, "Simulated Area is", area, "Field of View is", solidangle, "Detected Flux is", detectedflux
#!/bin/env python
import initialise as i
numberofhours, mincount, gridwidth, layout, raweff, flux, area, solidangle, selectionefficiency, hmacceptance = i.run(
)
import sys
sys.path.append('/nfs/astrop/d6/rstein/Hamburg-Cosmic-Rays/positioning/')
import argparse, math, random, time
import csv
import numpy as np
import simulate as s
import batchprocessing as bp
import batchreconstruction as br
from classes import *
parser = argparse.ArgumentParser()
parser.add_argument("-n", "--number", default=1)
cfg = parser.parse_args()
sourcedata = "executedefault" + str(cfg.number)
processdata = "executeprocess" + str(cfg.number)
reconstructdata = "executereconstructed" + str(cfg.number)
detectedflux = float(flux) * float(area) * float(solidangle) * float(
selectionefficiency)
rateperhour = detectedflux * 60 * 60
n = int(rateperhour * float(numberofhours))
import inpkfd import importlib import initialise import finalise path = importlib.import_module(inpkfd.path) # Imports the path specified in inpkfd. initialise.run() path.run() finalise.run()
def main():
# The following value keeps track of whether phi_0 has been defined yet for the transformation. After the first
# iteration this variable is set to true, which skips the define_phi_0 function.
occurrence_phi_0_definition = False
# These lists are collected for checking by pytest.
all_initial_pixel_value_lists = []
all_subpixel_lists = []
all_filter_attenuation_correction_lists = []
all_sample_attenuation_correction_lists = []
all_thomson_attenuation_correction_lists = []
all_final_pixel_values_lists = []
all_image_bin_lists = []
for i in range(len(ip.image_files)):
######################################
# This section performs a transformation into theta-phi space.
im, width, height, pix, initial_pixel_values, initial_total_pixel_value = \
initialise.run(
ip.image_files[i], ip.debug, i)
pix, subpixel_list, width, height = \
make_subpixels.run(
pix, width, height, ip.num_width_subpixels, ip.num_height_subpixels, ip.debug)
central_pixel, width_mm_per_pixel, height_mm_per_pixel, norm_view_x, norm_view_y, \
vector_origin_to_central_pixel, unit_vector_source_to_origin, adjust_to_centre_of_pixel = \
work_out_common_results.run(
width, height, ip.x_scale[i], ip.y_scale[i], ip.view_x[i], ip.view_y[i], ip.offset[i], ip.normal[i],
ip.source_position[i], ip.debug)
if occurrence_phi_0_definition is False:
unit_phi_plane_normal, phi_0_vector = define_phi_0.run(
ip.source_position[i], ip.phi_0_definer, ip.debug)
occurrence_phi_0_definition = True
else:
unit_phi_plane_normal = unit_phi_plane_normal
phi_0_vector = phi_0_vector
filter_angles_list_deg, thomson_angles_list_deg, gsqr, pixel_value, phi = \
calc_theta_phi.run(
width, height, pix, ip.phi_limit, ip.gsqr_limit, ip.wavelength, ip.a_lattice, ip.normal[i], norm_view_x,
norm_view_y, central_pixel, width_mm_per_pixel, height_mm_per_pixel, vector_origin_to_central_pixel,
unit_vector_source_to_origin, adjust_to_centre_of_pixel, unit_phi_plane_normal, phi_0_vector, ip.debug)
##########################################
# The following functions apply corrections for thomson polarisation, filter attenuation, and the sample absorption.
pixel_value, filter_attenuation_correction_factor_list = \
compensate_for_filters.run(
filter_angles_list_deg, pixel_value, ip.filter_attenuation_length[i], ip.filter_thickness[i], ip.debug)
# The thomson_angles_list_deg is the angle from source -> origin -> pixel, and is the same list of angles required by
# the sample attenuation correction.
pixel_value, sample_attenuation_correction_factor_list = \
compensate_for_sample_attenuation.run(
thomson_angles_list_deg, pixel_value, ip.source_position[i], ip.sample_normal[i])
pixel_value, thomson_attenuation_correction_factor_list = \
compensate_for_thomson_factor.run(
thomson_angles_list_deg, pixel_value, ip.debug)
##########################################
# This section makes the images of masks for filter and thomson corrections.
make_mask_image.run(
"filter_angles_mask.png", "degrees", filter_angles_list_deg, width,
height, [min(filter_angles_list_deg),
max(filter_angles_list_deg)], ip.debug)
make_mask_image.run(
"thomson_angles_polarisation_mask.png", "degrees",
thomson_angles_list_deg, width, height,
[min(thomson_angles_list_deg),
max(thomson_angles_list_deg)], ip.debug)
for j in range(len(ip.filter_thickness)):
make_mask_image.run(
"filter_" + str(j) + "_attenuation_factor_mask.png",
"attenuation_factor",
filter_attenuation_correction_factor_list[j], width, height, [
min(filter_attenuation_correction_factor_list[j]),
max(filter_attenuation_correction_factor_list[j])
], ip.debug)
make_mask_image.run(
"thomsons_attenuation_factor_mask.png", "attenuation_factor",
thomson_attenuation_correction_factor_list, width, height, [
min(thomson_attenuation_correction_factor_list),
max(thomson_attenuation_correction_factor_list)
], ip.debug)
############################################
# This section bins the pixels according to their gsqr-phi values calculated above.
gsqr_bin_list, phi_bin_list, image_bin_list = \
make_bins_for_theta_phi.run(
ip.num_gsqr_bins, ip.num_phi_bins, gsqr, phi, ip.debug)
image_bin_list = \
populate_theta_phi_bins.run(
gsqr, phi, pixel_value, gsqr_bin_list, phi_bin_list, image_bin_list, ip.debug)
image_bin_list = \
sum_pixel_values_in_bins.run(
image_bin_list, pixel_value, ip.debug)
gsqr_integrated_intensity = \
sum_along_theta.run(
image_bin_list, gsqr_bin_list, phi_bin_list, ip.debug)
plot_integrated_intensity.run(gsqr_bin_list, gsqr_integrated_intensity,
ip.name_plot_integrated_intensity,
ip.debug)
write_data_to_file.run("integrated_intensity_vs_gsqr.dat",
gsqr_bin_list, gsqr_integrated_intensity,
ip.debug)
build_theta_phi_image.run('normalised_gsqr_vs_phi.tif', image_bin_list,
ip.num_gsqr_bins, ip.num_phi_bins,
gsqr_bin_list, phi_bin_list, True, ip.debug)
build_theta_phi_image.run('preserved_gsqr_vs_phi.tif', image_bin_list,
ip.num_gsqr_bins, ip.num_phi_bins,
gsqr_bin_list, phi_bin_list, False, ip.debug)
check_image_total_pixel_value.run(initial_total_pixel_value,
'preserved_gsqr_vs_phi.tif',
ip.debug)
if ip.plot is True:
plt.scatter(gsqr, pixel_value, 1)
plt.xlabel("$G^2$")
plt.ylabel("PSL")
plt.savefig('gsqr_vs_intensity.png')
sort_data_files.run(ip.image_files[i], i)
all_initial_pixel_value_lists.append(initial_pixel_values)
all_subpixel_lists.append(subpixel_list)
all_filter_attenuation_correction_lists.append(
filter_attenuation_correction_factor_list)
print "Finished with " + ip.image_files[i] + "\n"
return all_initial_pixel_value_lists, all_subpixel_lists, all_filter_attenuation_correction_lists
