def setUpClass(cls): """ Setting up the test class. """ cls.input_h5 = TestUtilities.generateTestFilePath('pmi_out_0000001.h5') detector_panel = DetectorPanel( ranges={ 'fast_scan_min': 0, 'fast_scan_max': 21, 'slow_scan_min': 0, 'slow_scan_max': 21 }, pixel_size=2.2e-4 * meter, photon_response=1.0, distance_from_interaction_plane=0.13 * meter, corners={ 'x': -11, 'y': -11 }, ) cls.detector_geometry = DetectorGeometry(panels=[detector_panel]) cls.beam = PhotonBeamParameters( photon_energy=8.6e3 * electronvolt, beam_diameter_fwhm=1.0e-6 * meter, pulse_energy=1.0e-3 * joule, photon_energy_relative_bandwidth=0.001, divergence=1e-3 * radian, photon_energy_spectrum_type="SASE", )
def setUp(self): """ Setting up a test. """ self.__files_to_remove = [] self.__dirs_to_remove = [] # Construct the panel. self.__panel = DetectorPanel( dimensions=["ss", "fs"], ranges={ "fast_scan_min": 0, "fast_scan_max": 511, "slow_scan_min": 512, "slow_scan_max": 1024, }, pixel_size=2.2e-4 * meter, photon_response=1.0, distance_from_interaction_plane=0.13 * meter, distance_offset=0.0 * meter, fast_scan_xyz=None, slow_scan_xyz=None, corners={ "x": -512, "y": -256 }, saturation_adu=1e4, mask=None, good_bit_mask=None, bad_bit_mask=None, saturation_map=None, badregion_flag=False, )
def setUpClass(cls): """ Setting up the test class. """ # Setup parameters. cls.__beam_parameters = PhotonBeamParameters( photon_energy=4.96e3 * electronvolt, photon_energy_relative_bandwidth=0.01, beam_diameter_fwhm=2e-6 * meter, divergence=2e-6 * radian, pulse_energy=1e-3 * joule, photon_energy_spectrum_type='tophat') cls.__geometry = DetectorGeometry(panels=DetectorPanel( ranges={ "fast_scan_min": 0, "fast_scan_max": 63, "slow_scan_min": 0, "slow_scan_max": 63 }, pixel_size=220.0e-6 * meter, photon_response=1.0, distance_from_interaction_plane=0.1 * meter, corners={ "x": -32, "y": -32 }, saturation_adu=1e4, ))
def testBackengineGPU(self): """ Check a backengine calculation with openCL enabled. """ # Ensure cleanup. self.__dirs_to_remove.append("diffr") self.__files_to_remove.append("diffr.h5") # Clean up to make sure no old files mess things up. self.tearDown() beam_parameters = PhotonBeamParameters( photon_energy=4.96e3 * electronvolt, photon_energy_relative_bandwidth=0.01, beam_diameter_fwhm=2e-6 * meter, divergence=2e-6 * radian, pulse_energy=1e-3 * joule, photon_energy_spectrum_type='tophat') geometry = DetectorGeometry(panels=DetectorPanel( ranges={ "fast_scan_min": 0, "fast_scan_max": 63, "slow_scan_min": 0, "slow_scan_max": 63 }, pixel_size=220.0e-6 * meter, photon_response=1.0, distance_from_interaction_plane=0.1 * meter, corners={ "x": -32, "y": -32 }, saturation_adu=1e4, )) parameters = CrystFELPhotonDiffractorParameters( sample=self.__sample, beam_parameters=beam_parameters, detector_geometry=geometry, number_of_diffraction_patterns=10, use_gpu=True) # Get calculator. diffractor = CrystFELPhotonDiffractor(parameters=parameters, input_path=None, output_path='diffr') # Run backengine status = diffractor.backengine() # Check return code. self.assertEqual(status, 0) output_path = "%s" % diffractor.output_path # Check output dir was created. self.assertTrue(os.path.isdir(output_path)) # Check pattern was written. diffractor.saveH5() self.assertIn("diffr_out_0000001.h5", os.listdir(output_path))
def setUpClass(cls): detector_panel = DetectorPanel( ranges={ 'fast_scan_min': 0, 'fast_scan_max': 1023, 'slow_scan_min': 0, 'slow_scan_max': 1023 }, pixel_size=2.2e-4 * meter, photon_response=1.0, distance_from_interaction_plane=0.13 * meter, corners={ 'x': -512, 'y': 512 }, ) cls.detector_geometry = DetectorGeometry(panels=[detector_panel]) cls.beam = PhotonBeamParameters( photon_energy=8.6e3 * electronvolt, beam_diameter_fwhm=1.0e-6 * meter, pulse_energy=1.0e-3 * joule, photon_energy_relative_bandwidth=0.001, divergence=None, photon_energy_spectrum_type="SASE", )
def testBackengine(self): # Get calculator. # Ensure cleanup. self.__dirs_to_remove.append("diffr") # Setup parameters. beam_parameters = PhotonBeamParameters( photon_energy=4.96e3 * electronvolt, photon_energy_relative_bandwidth=0.01, beam_diameter_fwhm=2e-6 * meter, divergence=2e-6 * radian, pulse_energy=1e-3 * joule, photon_energy_spectrum_type='tophat') self.assertIsInstance(beam_parameters, PhotonBeamParameters) geometry = DetectorGeometry(panels=DetectorPanel( ranges={ "fast_scan_min": 0, "fast_scan_max": 63, "slow_scan_min": 0, "slow_scan_max": 63 }, pixel_size=220.0e-6 * meter, photon_response=1.0, distance_from_interaction_plane=0.1 * meter, corners={ "x": -32, "y": -32 }, saturation_adu=1e4, )) sys.stdout.flush() parameters = CrystFELPhotonDiffractorParameters( sample=self.__sample, beam_parameters=beam_parameters, detector_geometry=geometry, number_of_diffraction_patterns=10) diffractor = CrystFELPhotonDiffractor(parameters=parameters, input_path=None, output_path='diffr') # Run backengine status = diffractor.backengine() # Check return code. self.assertEqual(status, 0) # Check output dir was created. output_path = "%s" % diffractor.output_path self.assertTrue(os.path.isdir(output_path)) # Check pattern was written. self.assertIn("diffr_out-1.h5", os.listdir(output_path))
def testShapedConstructionGeometry(self): """ Testing the construction with a geometry object. """ beam_parameters = PhotonBeamParameters( photon_energy=4.96e3 * electronvolt, photon_energy_relative_bandwidth=0.01, beam_diameter_fwhm=2e-6 * meter, divergence=2e-6 * radian, pulse_energy=1e-3 * joule) geometry = DetectorGeometry(panels=DetectorPanel( ranges={ "fast_scan_min": 0, "fast_scan_max": 63, "slow_scan_min": 0, "slow_scan_max": 63 }, pixel_size=220.0e-6 * meter, photon_response=1.0, distance_from_interaction_plane=0.1 * meter, corners={ "x": -32, "y": -32 }, saturation_adu=1e4, )) # Attempt to construct an instance of the class. parameters = CrystFELPhotonDiffractorParameters( sample=self.__sample, powder=True, number_of_diffraction_patterns=10, number_of_background_photons=100, poissonize=True, suppress_fringes=True, crystal_size_min=10.0e-9 * meter, crystal_size_max=100.0e-9 * meter, uniform_rotation=False, beam_parameters=beam_parameters, detector_geometry=geometry, ) # Check all parameters are set as intended. self.assertFalse(parameters.uniform_rotation) self.assertEqual(parameters.number_of_diffraction_patterns, 10) self.assertTrue(parameters.powder) self.assertEqual(parameters.crystal_size_min.m_as(meter), 10.e-9) self.assertEqual(parameters.crystal_size_max.m_as(meter), 100.e-9) self.assertTrue(parameters.poissonize) self.assertEqual(parameters.number_of_background_photons, 100) self.assertTrue(parameters.suppress_fringes) self.assertIsInstance(parameters.beam_parameters, PhotonBeamParameters) self.assertEqual( parameters.beam_parameters.photon_energy.m_as(electronvolt), 4.96e3) self.assertEqual(parameters.detector_geometry, geometry)
def setUpClass(cls): detector_panel = DetectorPanel(ranges={'fast_scan_min': 0, 'fast_scan_max': 1023, 'slow_scan_min': 0, 'slow_scan_max': 1023}, pixel_size=2.2e-4*meter, photon_response=1.0, distance_from_interaction_plane=0.13*meter, corners={'x': -512, 'y': 512}, ) cls.detector_geometry = DetectorGeometry(panels=[detector_panel])
def testBackengineWithBeamAndGeometry(self): """ Check geom parameter logic if they are set as parameters. """ # Ensure cleanup. self.__dirs_to_remove.append("diffr") # Setup beam parameters. beam_parameters = PhotonBeamParameters( photon_energy=16.0e3 * electronvolt, photon_energy_relative_bandwidth=0.001, pulse_energy=2.0e-3 * joule, beam_diameter_fwhm=100e-9 * meter, divergence=None, photon_energy_spectrum_type="tophat", ) geometry = DetectorGeometry(panels=DetectorPanel( ranges={ "fast_scan_min": 0, "fast_scan_max": 63, "slow_scan_min": 0, "slow_scan_max": 63 }, pixel_size=220.0e-6 * meter, photon_response=1.0, distance_from_interaction_plane=0.1 * meter, corners={ "x": -32, "y": -32 }, saturation_adu=1e4, )) # Get parameters. parameters = CrystFELPhotonDiffractorParameters( sample=self.__sample, detector_geometry=geometry, beam_parameters=beam_parameters, number_of_diffraction_patterns=2, uniform_rotation=True, ) # Get calculator. diffractor = CrystFELPhotonDiffractor(parameters=parameters, input_path=None, output_path='diffr') # Run backengine status = diffractor.backengine() # Check return code. self.assertEqual(status, 0)
def setUp(self): """ Setting up a test. """ self.__panel0 = DetectorPanel( ranges={ "fast_scan_min": 0, "fast_scan_max": 511, "slow_scan_min": 512, "slow_scan_max": 1023, }, pixel_size=2.2e-4 * meter, photon_response=1.0, distance_from_interaction_plane=0.13 * meter, distance_offset=0.0 * meter, fast_scan_xyz=None, slow_scan_xyz=None, corners={ "x": -512, "y": -256 }, saturation_adu=1e4, mask=None, good_bit_mask=None, bad_bit_mask=None, saturation_map=None, badregion_flag=False, ) # Copy-construct a second panel. self.__panel1 = self.__panel0( ranges={ 'fast_scan_min': self.__panel0.ranges['fast_scan_min'], 'fast_scan_max': self.__panel0.ranges['fast_scan_max'], 'slow_scan_min': self.__panel0.ranges['slow_scan_max'] + 1, 'slow_scan_max': 2 * self.__panel0.ranges['slow_scan_max'] - self.__panel0.ranges['slow_scan_min'] + 1 }, corners={ 'x': self.__panel0.corners['x'], 'y': self.__panel0.corners['y'] + 512 }, ) self.__files_to_remove = [] self.__dirs_to_remove = []
def plot_diffr_vs_detector(self): """ Compare patterns before and after detector sim. """ # Cleanup. #self.__files_to_remove.append('5mzd.pdb') #self.__files_to_remove.append('diffr.h5') #self.__dirs_to_remove.append('diffr') # Avoid crash due to multiple instances of G4RunManager del self._detector # Setup detector geometry. detector_panel = DetectorPanel( ranges={ 'fast_scan_min': 0, 'fast_scan_max': 511, 'slow_scan_min': 0, 'slow_scan_max': 511 }, pixel_size=2.2e-4 * Units.meter, photon_response=1.0, distance_from_interaction_plane=0.13 * Units.meter, corners={ 'x': -256, 'y': -256 }, ) detector_geometry = DetectorGeometry(panels=[detector_panel]) # Setup photon beam. beam = PhotonBeamParameters( photon_energy=4.96e3 * Units.electronvolt, beam_diameter_fwhm=1.0e-6 * Units.meter, pulse_energy=1.0e-3 * Units.joule, photon_energy_relative_bandwidth=0.001, divergence=1e-3 * Units.radian, photon_energy_spectrum_type="SASE", ) # Setup and run the diffraction sim. diffraction_parameters = SingFELPhotonDiffractorParameters( uniform_rotation=None, calculate_Compton=False, number_of_diffraction_patterns=1, detector_geometry=detector_geometry, beam_parameters=beam, sample="5mzd.pdb", forced_mpi_command='mpirun -np 1', ) photon_diffractor = SingFELPhotonDiffractor( parameters=diffraction_parameters, output_path='diffr', ) photon_diffractor.backengine() photon_diffractor.saveH5() analysis1 = DiffractionAnalysis(photon_diffractor.output_path, pattern_indices=[1], poissonize=True) analysis1.plotPattern( operation=None, logscale=False, ) parameters = XCSITPhotonDetectorParameters( detector_type="AGIPDSPB", patterns=[0], ) detector = XCSITPhotonDetector( parameters=parameters, input_path="diffr.h5", output_path="detector_out.h5", ) detector._readH5() detector.backengine() detector.saveH5() # Weak test Check we have photons in the signal. pattern = h5py.File("detector_out.h5", 'r')['data/0000001/data'].value analysis2 = DiffractionAnalysis(detector.output_path, pattern_indices=[1], poissonize=True) analysis2.plotPattern( operation=None, logscale=False, ) mpl.pyplot.show()
if os.path.isdir(d): shutil.rmtree(d) for f in files_to_remove: if os.path.isfile(f): os.remove(f) # Detector setup p0 = DetectorPanel(ranges={ 'fast_scan_min': 0, 'fast_scan_max': 99, 'slow_scan_min': 0, 'slow_scan_max': 99 }, pixel_size=560e-6 * meter, photon_response=1.0, distance_from_interaction_plane=0.13 * meter, corners={ 'x': -49.5, 'y': -49.5 }, fast_scan_xyz='1.0x', slow_scan_xyz='1.0y') detector_geometry = DetectorGeometry(panels=[p0]) beam = PhotonBeamParameters( photon_energy=6.0e3 * electronvolt, beam_diameter_fwhm=226.0e-9 * meter, #sqrt(250*160/pi)*2 pulse_energy=4.0e-3 * joule, )
def testDefaultConstruction(self): """ Testing the default construction. """ # Attempt to construct an instance of the class. panel = DetectorPanel() self.assertIsInstance(panel, DetectorPanel)
def testSaveH5(self): """ Check that saveh5() creates correct filenames. """ # Ensure cleanup. self.__dirs_to_remove.append("diffr") self.__files_to_remove.append("diffr.h5") # Setup beam parameters. beam_parameters = PhotonBeamParameters( photon_energy=5e3 * electronvolt, pulse_energy=2e-3 * joule, photon_energy_relative_bandwidth=1e-3, photon_energy_spectrum_type="tophat", beam_diameter_fwhm=3e-6 * meter, ) geometry = DetectorGeometry(panels=DetectorPanel( ranges={ "fast_scan_min": 0, "fast_scan_max": 63, "slow_scan_min": 0, "slow_scan_max": 63 }, pixel_size=220.0e-6 * meter, photon_response=1.0, distance_from_interaction_plane=0.1 * meter, corners={ "x": -32, "y": -32 }, saturation_adu=1e4, )) # Get parameters. parameters = CrystFELPhotonDiffractorParameters( sample=self.__sample, detector_geometry=geometry, beam_parameters=beam_parameters, number_of_diffraction_patterns=2, uniform_rotation=True, ) # Get calculator. diffractor = CrystFELPhotonDiffractor(parameters=parameters, input_path=None, output_path='diffr') # Run backengine status = diffractor.backengine() # Check return code. self.assertEqual(status, 0) # Check output dir was created. self.assertTrue(os.path.isdir(diffractor.output_path)) # Save correctly. diffractor.saveH5() # Check output file was created. self.assertTrue(os.path.isfile(diffractor.output_path)) # Check pattern was written. self.assertIn("diffr_out_0000001.h5", os.listdir("diffr")) self.assertIn("diffr_out_0000002.h5", os.listdir("diffr")) # Open linked h5 file. with h5py.File(diffractor.output_path, 'r') as h5: self.assertIn("data", list(h5.keys())) self.assertIn("0000001", list(h5["data"].keys())) self.assertIn("0000002", list(h5["data"].keys())) self.assertIn("data", list(h5["data/0000001"].keys())) self.assertIn("data", list(h5["data/0000002"].keys())) self.assertIn("params", list(h5.keys())) self.assertIn("beam", list(h5["params"].keys())) self.assertIn("photonEnergy", list(h5["params/beam"].keys())) self.assertIn("focusArea", list(h5["params/beam"].keys())) # Check metafile was created. self.assertIn( os.path.split(diffractor.output_path)[-1], os.listdir(os.path.dirname(diffractor.output_path)))