# -*- coding: utf-8 -*-

import numpy as np

import math as math

import matplotlib.pylab as plt

import kcorrect as KC

from kcorrect.sdss import SDSSFilterList, SDSSPhotoZ, SDSS, SDSSKCorrect

from kcorrect.utils.cosmology import ztodm

from scipy import integrate

import matplotlib.cm as cm

import matplotlib.mlab as mlab

import pyfits as pyfits

import utils

#import sami_db

import db as db

import plot_survey as plot

#from sami_new import getDataArray as fullDataArray

#from sami_new import plotScatter as OldSchoolPlot

import matplotlib.font_manager

import csv

import readAtlas

import collections

imgDir = 'img/'

dbDir = '../db/'

dbFile = 'CALIFA.sqlite'

dataFile = 'data/Califa.csv'

morphDataFile = 'morph.dat'

observedList = 'list_observed.txt'

db_dataFile = 'db_data.txt'

#constants

c = 299792.458

pi = 3.14159265

# Cosmological parameters

H0 = 70.0 #km/s/Mpc, 1 Mpc= 3.08568025*1e19 km

tH = (3.08568025*1e19/H0)/(3600 * 365 * 24*10e9) #Hubble time in Gyr

dH = c/H0 #in Mpc

Wl = 0.728

Wm = 1 - Wl

Wh = c / H0

Wk = 1 - Wm - Wl

tH = (3.08568025*1e19/H0)/(3600 * 365 * 24*10e9) #Hubble time in Gyr

#utilities

def E_z(z): #time derivative of log(a(t)), used for integration

return 1/(np.sqrt((Wm*(1+z)**3 + Wk*(1+z)**2 + Wl)))

def comovingDistance(z):

Dc = dH * 1000* integrate.quad(E_z, 0, z)[0] #in kpc

return Dc

def angular2physical(reff, z): #return physical effective diameter of the galaxy in kpc

return (math.radians(2*reff/3600) *(comovingDistance(z)) / (1 + z)**2)

def getAbsMag(z, mag, ext):

return absmag

def convert(data):

return np.asarray(tempDATA)

def findOverlap(a, b):

a_multiset = collections.Counter(list(a))

b_multiset = collections.Counter(list(b))

return list((a_multiset & b_multiset).elements())

def main():

np.savetxt("absmag.csv", outputArray, fmt = '%i, %10.3f, %10.3f, %10.3f, %10.3e, %10.3f, %10.3e, %10.3e, %10.3e')

if __name__ == '__main__':

main()