def test_integral(self, get_rsr_version, download_rsr, load, isfile, exists): """Test the calculation of the integral of the spectral responses""" load.return_code = None download_rsr.return_code = None isfile.return_code = True exists.return_code = True get_rsr_version.return_code = RSR_DATA_VERSION with patch('pyspectral.rsr_reader.get_config', return_value=TEST_CONFIG): test_rsr = RelativeSpectralResponse('EOS-Aqua', 'modis') test_rsr.rsr = TEST_RSR2 res = test_rsr.integral('20') self.assertAlmostEqual(res['det-1'], 0.185634, 6)
def test_integral(self, get_rsr_version, download_rsr, load, isfile, exists): """Test the calculation of the integral of the spectral responses""" load.return_code = None download_rsr.return_code = None isfile.return_code = True exists.return_code = True get_rsr_version.return_code = RSR_DATA_VERSION with patch('pyspectral.rsr_reader.get_config', return_value=TEST_CONFIG): test_rsr = RelativeSpectralResponse('EOS-Aqua', 'modis') test_rsr.rsr = TEST_RSR2 res = test_rsr.integral('20') self.assertAlmostEqual(res['det-1'], 0.185634, 6)
def _get_rsr(self): """Get the relative spectral responses. Get the relative spectral responses from file, find the bandname, and convert to the requested wave-spave (wavelength or wave number) """ sensor = RelativeSpectralResponse(self.platform_name, self.instrument) if self.wavespace == WAVE_NUMBER: LOG.debug("Converting to wavenumber...") self.rsr, info = convert2wavenumber(sensor.rsr) else: self.rsr = sensor.rsr info = {'unit': sensor.unit, 'si_scale': sensor.si_scale} self._wave_unit = info['unit'] self._wave_si_scale = info['si_scale'] if isinstance(self.band, str): self.bandname = BANDNAMES.get(self.instrument, BANDNAMES['generic']).get( self.band, self.band) elif isinstance(self.band, Number): self.bandwavelength = self.band self.bandname = get_bandname_from_wavelength( self.instrument, self.band, self.rsr) self.wavelength_or_wavenumber = ( self.rsr[self.bandname][self.detector][self.wavespace] * self._wave_si_scale) self.response = self.rsr[self.bandname][self.detector]['response'] # Get the integral of the spectral response curve: self.rsr_integral = np.trapz(self.response, self.wavelength_or_wavenumber)
def get_effective_wavelength(self, bandname): """Get the effective wavelength with Rayleigh scattering in mind""" try: rsr = RelativeSpectralResponse(self.platform_name, self.sensor) except(IOError, OSError): LOG.exception( "No spectral responses for this platform and sensor: %s %s", self.platform_name, self.sensor) if isinstance(bandname, (float, integer_types)): LOG.warning( "Effective wavelength is set to the requested band wavelength = %f", bandname) return bandname return None if isinstance(bandname, str): bandname = BANDNAMES.get(self.sensor, BANDNAMES['generic']).get(bandname, bandname) elif isinstance(bandname, (float, integer_types)): if not(0.4 < bandname < 0.8): raise BandFrequencyOutOfRange( 'Requested band frequency should be between 0.4 and 0.8 microns!') bandname = get_bandname_from_wavelength(self.sensor, bandname, rsr.rsr) wvl, resp = rsr.rsr[bandname][ 'det-1']['wavelength'], rsr.rsr[bandname]['det-1']['response'] cwvl = get_central_wave(wvl, resp, weight=1. / wvl**4) LOG.debug("Band name: %s Effective wavelength: %f", bandname, cwvl) return cwvl
def test_rsr_reponse(self, download_rsr, load, isfile, exists): """Test the RelativeSpectralResponse class initialisation""" load.return_code = None download_rsr.return_code = None isfile.return_code = True exists.return_code = True with patch('pyspectral.rsr_reader.get_config', return_value=TEST_CONFIG): with self.assertRaises(AttributeError): test_rsr = RelativeSpectralResponse('GOES-16') test_rsr = RelativeSpectralResponse(instrument='ABI') test_rsr = RelativeSpectralResponse('GOES-16', 'AbI') self.assertEqual(test_rsr.platform_name, 'GOES-16') self.assertEqual(test_rsr.instrument, 'abi') test_rsr = RelativeSpectralResponse('GOES-16', 'ABI') self.assertEqual(test_rsr.instrument, 'abi') with patch('pyspectral.rsr_reader.get_config', return_value=TEST_CONFIG): test_rsr = RelativeSpectralResponse('GOES-16', 'abi') self.assertEqual(test_rsr.platform_name, 'GOES-16') self.assertEqual(test_rsr.instrument, 'abi') self.assertEqual(test_rsr.filename, os.path.join(TEST_RSR_DIR, 'rsr_abi_GOES-16.h5')) with patch('pyspectral.rsr_reader.get_config', return_value=TEST_CONFIG): test_rsr = RelativeSpectralResponse( filename=os.path.join(TEST_RSR_DIR, 'rsr_abi_GOES-16.h5')) self.assertEqual(test_rsr.filename, os.path.join(TEST_RSR_DIR, 'rsr_abi_GOES-16.h5'))
def get_rsr(self): """Get all spectral responses for the sensor""" from pyspectral.utils import convert2wavenumber from pyspectral.rsr_reader import RelativeSpectralResponse sensor = RelativeSpectralResponse(self.platform, self.satnumber, self.instrument) LOG.debug("Wavenumber? " + str(self.wavespace)) if self.wavespace == WAVE_NUMBER: LOG.debug("Converting to wavenumber...") self.rsr, info = convert2wavenumber(sensor.rsr) else: self.rsr = sensor.rsr info = {'unit': sensor.unit, 'si_scale': sensor.si_scale} self._wave_unit = info['unit'] self._wave_si_scale = info['si_scale']
def main(arguments): """Retrieve wavelength range based on user-supplied arguments.""" platform = arguments.platform_name sensor = arguments.sensor threshold = arguments.minimum_response rsr = RelativeSpectralResponse(platform, sensor) if arguments.bandname: bands = [arguments.bandname] else: bands = rsr.band_names det = arguments.detector for bname in bands: wvls = get_wave_range(rsr.rsr[bname][det], threshold) print(f'name: {bname}') print( f' wavelength: [{wvls[0]:5.3f}, {wvls[1]:5.3f}, {wvls[2]:5.3f}]')
def test_convert(self, get_rsr_version, download_rsr, load, isfile, exists): """Test the conversion method""" load.return_code = None download_rsr.return_code = None isfile.return_code = True exists.return_code = True get_rsr_version.return_code = RSR_DATA_VERSION with patch('pyspectral.rsr_reader.get_config', return_value=TEST_CONFIG): test_rsr = RelativeSpectralResponse('EOS-Aqua', 'modis') test_rsr.rsr = TEST_RSR test_rsr.convert() self.assertAlmostEqual(test_rsr.rsr['20']['det-1']['central_wavenumber'], 2647.397, 3) self.assertTrue(np.allclose(test_rsr.rsr['20']['det-1'][WAVE_NUMBER], RESULT_WVN_RSR, 5)) self.assertEqual(test_rsr._wavespace, WAVE_NUMBER) with self.assertRaises(NotImplementedError): test_rsr.convert()
def test_convert(self, get_rsr_version, download_rsr, load, isfile, exists): """Test the conversion method""" load.return_code = None download_rsr.return_code = None isfile.return_code = True exists.return_code = True get_rsr_version.return_code = RSR_DATA_VERSION with patch('pyspectral.rsr_reader.get_config', return_value=TEST_CONFIG): test_rsr = RelativeSpectralResponse('EOS-Aqua', 'modis') test_rsr.rsr = TEST_RSR test_rsr.convert() self.assertAlmostEqual(test_rsr.rsr['20']['det-1']['central_wavenumber'], 2647.397, 3) self.assertTrue(np.allclose(test_rsr.rsr['20']['det-1'][WAVE_NUMBER], RESULT_WVN_RSR, 5)) self.assertEqual(test_rsr._wavespace, WAVE_NUMBER) with self.assertRaises(NotImplementedError): test_rsr.convert()
band = args.bandname elif args.wavelength: req_wvl = args.wavelength figscale = 1.0 if wvlmin: figscale = (wvlmax - wvlmin) / 4. figsize = (figscale * 1. + 10, figscale * 0.5 + 5) plt.figure(figsize=figsize) something2plot = False for platform in platform_names: for sensor in sensors: try: rsr = RelativeSpectralResponse(platform, sensor) except IOError: # LOG.exception('Failed getting the rsr data for platform %s ' + # 'and sensor %s', platform, sensor) rsr = None else: break if not rsr: continue something2plot = True LOG.debug("Platform name %s and Sensor: %s", str(rsr.platform_name), str(rsr.instrument)) if req_wvl: bands = get_bandname_from_wavelength(sensor, req_wvl, rsr.rsr, 0.25, multiple_bands=True)
sza = -62.64299 saa = 165.6918 storm_lat = -3.2609048 storm_lon = 163.2608 storm_time = datetime(2018, 12, 29, 13, 30, tzinfo=pytz.utc) # Below are pre-defined BT values for the storm coordinates and time given above. # Also included are the wavelength ranges required for the plotting later on. # This threshold is used to define min/max wavelength rsr_thresh = 0.15 # NOAA-20 / VIIRS first rsr = RelativeSpectralResponse('NOAA-20', 'viirs') viirsi_data = { 'I04': [208.000] + get_wave_range(rsr.rsr['I4']['det-1'], rsr_thresh), 'I05': [161.955] + get_wave_range(rsr.rsr['I5']['det-1'], rsr_thresh) } viirsm_data = { 'M12': [203.000] + get_wave_range(rsr.rsr['M12']['det-1'], rsr_thresh), 'M13': [191.920] + get_wave_range(rsr.rsr['M13']['det-1'], rsr_thresh), 'M14': [165.503] + get_wave_range(rsr.rsr['M14']['det-1'], rsr_thresh), 'M15': [163.733] + get_wave_range(rsr.rsr['M15']['det-1'], rsr_thresh), 'M16': [165.274] + get_wave_range(rsr.rsr['M16']['det-1'], rsr_thresh) } # Now GOES-17 / ABI rsr = RelativeSpectralResponse('GOES-17', 'abi')
print("... read responce functions") from pyspectral.near_infrared_reflectance import Calculator from pyspectral.solar import (SolarIrradianceSpectrum, TOTAL_IRRADIANCE_SPECTRUM_2000ASTM) solar_irr = SolarIrradianceSpectrum(TOTAL_IRRADIANCE_SPECTRUM_2000ASTM, dlambda=0.0005) #from pyspectral.seviri_rsr import load #seviri = load() #rsr = {'wavelength': seviri['IR3.9']['wavelength'], # 'response': seviri['IR3.9']['met10']['95']} #sflux = solar_irr.solar_flux_over_band(rsr) solar_irr = SolarIrradianceSpectrum(TOTAL_IRRADIANCE_SPECTRUM_2000ASTM, dlambda=0.0005) from pyspectral.rsr_reader import RelativeSpectralResponse seviri = RelativeSpectralResponse('Meteosat-10', 'seviri') sflux = solar_irr.inband_solarflux(seviri.rsr['IR3.9']) ch39 = local_data['IR_039'] ch11 = local_data['IR_108'] ch13 = local_data['IR_134'] lonlats = ch39.area.get_lonlats() from pyorbital.astronomy import sun_zenith_angle sunz = sun_zenith_angle(tslot, lonlats[0], lonlats[1]) print("... create look-up-table") #refl37 = Calculator(rsr) refl37 = Calculator('Meteosat-9', 'seviri', 'IR3.9', solar_flux=sflux) #refl37.make_tb2rad_lut('/tmp/seviri_37_tb2rad_lut.npz') # new syntax ->
_DEFAULT_TIME_FORMAT = '%Y-%m-%d %H:%M:%S' #: Default log format _DEFAULT_LOG_FORMAT = '[%(levelname)s: %(asctime)s : %(name)s] %(message)s' LOG = logging.getLogger('test_ahi') handler = logging.StreamHandler(sys.stderr) formatter = logging.Formatter(fmt=_DEFAULT_LOG_FORMAT, datefmt=_DEFAULT_TIME_FORMAT) handler.setFormatter(formatter) handler.setLevel(logging.DEBUG) LOG.setLevel(logging.DEBUG) logging.getLogger("").setLevel(logging.DEBUG) LOG.addHandler(handler) ahi = RelativeSpectralResponse('Himawari-8', 'ahi') solar_irr = SolarIrradianceSpectrum(TOTAL_IRRADIANCE_SPECTRUM_2000ASTM, dlambda=0.005) sflux = solar_irr.inband_solarflux(ahi.rsr['ch7']) LOG.info("Solar flux over Band: " + str(sflux)) # sunz = [80., 80.5] # tb7 = [288.0, 390.0] # tb14 = [282.0, 272.0] # tb16 = [275.0, 269.0] sunz = np.random.rand(5500, 550) * 120. tb7 = np.random.rand(5500, 550) * 120. + 260.0 tb14 = np.random.rand(5500, 550) * 30. + 260.0 tb16 = np.random.rand(5500, 550) * 30. + 250.0