Exemplo n.º 1
0
	def cr_text_dynamic(self, mt_out_path, cr_out_path, waves, n_waves, image, n_cols):
	
		self.writeMessage("cr_text_dynamic")
			
		wt = self.wave_thresh
		
		srange_min = float(self.shortRangeStartEntry.get())
		srange_max = float(self.shortRangeStopEntry.get())
		lrange_min = float(self.longRangeStartEntry.get())
		lrange_max = float(self.longRangeStopEntry.get())
		kernel = int(self.kernelEntry.get())
		
		if kernel % 2 == 0:
			self.writeMessage("Error. Kernel size must be odd.")
			return
		
		#
		# validate
		#
		if (srange_max - srange_min < wt) or (lrange_min - srange_max < wt) or (lrange_max - lrange_min < wt):
			self.writeMessage("Error. Bad wavelength range.")
			return
		
		ss0 = self.shortRangeZeroIntVar.get()
		ss1 = self.shortRangeMinIntVar.get()
		ss2 = self.shortRangeMaxIntVar.get()
		ls0 = self.longRangeZeroIntVar.get()
		ls1 = self.longRangeMinIntVar.get()
		ls2 = self.longRangeMaxIntVar.get()
		
		if ss0 == 0 and ss1 == 0 and ss2 == 0:
			self.writeMessage("Error. Must select short wavelength strategy.")
			return
			
		if ls0 == 0 and ls1 == 0 and ls2 == 0:
			self.writeMessage("Error. Must select short wavelength strategy.")
			return
		
		s_strat = 0
		if ss1 == 1: s_strat = 1
		if ss2 == 1: s_strat = 2
		
		l_strat = 0
		if ls1 == 1: l_strat = 1
		if ls2 == 1: l_strat = 2
		
		(cr_width, low, high) = \
		self.compute_cr_band_extents(waves, srange_min, lrange_max)
		
		cr_waves = waves[low:high+1]
		
		#
		# run
		#
		self.writeMessage("\tcr_text_setup")
		CR_module.cr_text_setup(cr_waves, cr_width, mt_out_path, cr_out_path)
		
		
		self.writeMessage("Processing:")
		
		for col in range(n_cols):
			
			#self.writeMessage("\tcolumn " + str(col+1) + "/" + str(n_cols))
			spectrum = [list[col] for list in image]
			cr_spect = spectrum[low:high+1]
			
			(low_i, high_i) =\
			CR_module.cr_dynamic_range(\
				spectrum,\
				waves,\
				n_waves,\
				s_strat,\
				l_strat,\
				srange_min,\
				srange_max,\
				lrange_min,\
				lrange_max,\
				kernel)
			
			if low_i < low or low_i > high:
				self.writeMessage("Error: low_i to high or low:" + str(low_i))
				return;
			if high_i < low or high_i > high:
				self.writeMessage("Error: high_i to high or low:" + str(high_i))
				return;
		
			offset  = low_i - low
			f_width = high_i - low_i + 1
			
			self.writeMessage("\t" + str(col+1) + "/" + str(n_cols) + "[" + str(low) + "," + str(low_i) + "," + str(high_i) + "," + str(high) + "]")
			
			#pass in the exact spectra
			CR_module.cr_text_dynamic(\
				cr_spect,\
				cr_waves,\
				cr_width,
				f_width,
				offset)
			
		CR_module.cr_text_teardown()
		self.writeMessage("Done!")