"""
Created on Mon Dec 29 00:14:56 2014

@author: pfigueira

Editied Thur Dec 15 13:00 2016 by Jason Neal for eniric.
"""

# from eniric.IOmodule import read_2col

import numpy as np
import pandas as pd
c = 299792458  # m/s


def RVprec_calc(spectrum_file="resampled/Spectrum_M0-PHOENIX-ACES_Hband_vsini1.0_R60k_res3.txt"):
    """
    function that claculates the RV precision achievable on a spectrum
    """
    data = pd.read_table(spectrum_file, comment='#', names=["wavelength", "flux"], dtype=np.float64)
    wav, flux = data["wavelength"].values, data["flux"].values

    return [c / SqrtSumWis(wavelength, flux)]


def SqrtSumWis(wavelength, flux):
    """
    Calculation of the SquareRoot of the sum of the Wis for a spectrum
    """

    delta_F = (np.array(flux[1:]) - np.array(flux[:-1]))
    delta_l = np.array(wavelength[:-1])

    derivF_over_lambda = delta_F/delta_l

    return np.sqrt(np.sum(np.array(wavelength[:-1])**2.0 * derivF_over_lambda**2.0 / np.array(flux[:-1])))

"""
Created on Mon Dec 29 00:14:56 2014

@author: pfigueira

Editied Thur Dec 15 13:00 2016 by Jason Neal for eniric.
"""

# from eniric.IOmodule import read_2col

import numpy as np
import pandas as pd
c = 299792458  # m/s


def RVprec_calc(spectrum_file="resampled/Spectrum_M0-PHOENIX-ACES_Hband_vsini1.0_R60k_res3.txt"):
    """
    function that claculates the RV precision achievable on a spectrum
    """
    data = pd.read_table(spectrum_file, comment='#', names=["wavelength", "flux"], dtype=np.float64)
    wav, flux = data["wavelength"].values, data["flux"].values

    return [c / SqrtSumWis(wavelength, flux)]


def SqrtSumWis(wavelength, flux):
    """
    Calculation of the SquareRoot of the sum of the Wis for a spectrum
    """

    delta_F = np.diff(flux)
    delta_l = wavelength[:-1]

    derivF_over_lambda = delta_F/delta_l

    return np.sqrt(np.sum(wavelength[:-1]**2.0 * derivF_over_lambda**2.0 / flux[:-1]))

Update RVprec_calculation use numpy.diff() and remove unnecessary array calls.
Update RVprec_calculation
use numpy.diff() and remove unnecessary array calls.


Former-commit-id: 646ff0cea061feb87c08b819a47d8e9f3dd55b55