# PyTransit: fast and easy exoplanet transit modelling in Python.
# Copyright (C) 2010-2019 Hannu Parviainen
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
from numpy import array, ones_like, zeros_like, diff, arange
from scipy.interpolate import interp1d
class Filter:
def __init__(self, name):
self.name = name
def __call__(self, wl):
return NotImplementedError
[docs]class ClearFilter(Filter):
"""Constant unity transmission.
"""
def __call__(self, wl):
return ones_like(wl).astype("d")
[docs]class BoxcarFilter(Filter):
"""Filter with a transmission of 1 inside the minimum and maximum wavelengths and 0 outside.
"""
def __init__(self, name, wl_min, wl_max):
"""
Parameters
----------
:param name: passband name
:param wl_min: minimum wavelength
:param wl_max: maximum wavelength
"""
super().__init__(name)
self.wl_min = wl_min
self.wl_max = wl_max
def __call__(self, wl):
w = zeros_like(wl)
w[(wl > self.wl_min) & (wl < self.wl_max)] = 1.
return w
[docs]class TabulatedFilter(Filter):
"""Interpolated tabulated filter.
"""
def __init__(self, name, wl, tm):
"""
Parameters
----------
name : string passband name
wl : array_like a list of wavelengths
tm : array_like a list of transmission values
"""
super().__init__(name)
self.wl = array(wl)
self.tm = array(tm)
assert self.wl.size == self.tm.size, "The wavelength and transmission arrays must be of same size"
assert all(diff(self.wl) > 0.), "Wavelength array must be monotonously increasing"
assert all((self.tm >= 0.0) & (self.tm <= 1.0)), "Transmission must always be between 0.0 and 1.0"
self._ip = interp1d(self.wl, self.tm, kind='cubic')
def __call__(self, wl):
return self._ip(wl)
sdss_g = BoxcarFilter("g'", 400, 550) #: SDSS G filter
sdss_r = BoxcarFilter("r'", 570, 690)
sdss_i = BoxcarFilter("i'", 710, 790)
sdss_z = BoxcarFilter("z'", 810, 900)
kepler = TabulatedFilter('kepler',
arange(350, 960, 25),
array([0.000, 0.001, 0.000, 0.056, 0.465, 0.536, 0.624, 0.663,
0.681, 0.715, 0.713, 0.696, 0.670, 0.649, 0.616, 0.574,
0.541, 0.490, 0.468, 0.400, 0.332, 0.279, 0.020, 0.000,
0.000]))
__all__ = 'Filter TabulatedFilter BoxcarFilter sdss_g sdss_r sdss_i sdss_z kepler'.split()