#radial velocity correction H_beta = 4861 H_gamma = 4341 H_delta = 4102 H_epsilon = 3970 H_alpha = 6562 #H_zeta = 3889 dw = 160 minid = np.array([]) rvset = np.array([]) H_wave = [H_beta, H_gamma, H_delta, H_epsilon, H_alpha] toff = SRP.find_line(star_norm_lamb_2, flux2, H_wave[0], dw) rvset = np.append(rvset, (toff)) for w in H_wave[1:]: toff2 = SRP.find_line(star_norm_lamb_2, flux2, w + (toff - H_beta), dw / 2) rvset = np.append(rvset, (toff2)) #if np.abs(rvset[0]) < np.abs(rvset[1]): # H_wave = H_wave[1:] # rvset = rvset[1:] print H_wave, rvset param = np.polyfit(rvset, H_wave, 1) func = np.poly1d(param)
elif (np.size(bid) == 3): mtype = "triplet" #sys.exit() ids = np.where((rvcorr_lamb < w + dw) & (rvcorr_lamb > w - dw)) number = np.shape(ids)[1] #weights = np.sqrt(np.abs( np.arange(number)-np.floor(number/2) )+1)/number #need to get fancy to renormalize the metal lines so that gaussian fit works linewave = rvcorr_lamb[ids] lineflux = flux[ids] #recenter around nearby line toff = SRP.find_line(linewave, lineflux, w, 4) - w w2 = w + toff ids = np.where((rvcorr_lamb < (w2 + dw)) & (rvcorr_lamb > (w2 - dw))) linewave = rvcorr_lamb[ids] lineflux = flux[ids] #midwave = np.int(np.shape(linewave)[0]/2) #res = np.abs(linewave[midwave]-linewave[midwave+1]) tab_p_prime = lineflux - np.roll((lineflux), 3) tab_m_prime = np.roll((lineflux), -3) - lineflux tab_p_prime2 = lineflux - np.roll((lineflux), 7) tab_m_prime2 = np.roll((lineflux), -7) - lineflux