Example #1
0
 def makeModel(self, tt, period, incl, t, dor, ror, ldm_coeff1, ldm_coeff2):
     """
     -t2z
     I have modified this function so that the midpoint of every transit is 0
     
     find lmdk from the star 
     """
     # use t2z to make z
     z = t2z(tt, period, incl, t, dor)
     return occultquad(z, ror, [ldm_coeff1, ldm_coeff2])
def transit_model(x, hjd, airmass, fix=[], SPOT_DATA=False):

    spotlong = x[6]
    spotlat = x[7]
    spotsize = abs(x[8])
    spotflux = abs(x[9])

    if spotsize > 20:
        spotsize = 20

    if spotflux > 1.0:
        spotflux = 1.0

    print SPOT_DATA

    print x

    p0 = x[0]

    #p0 = abs(x[0])
    #if p0 > 0.25:
    #p0 = 0.25

    gamma = x[10:12]

    if len(fix) > 0:
        x = fix

    #if abs(gamma[0]) > 1.0:
    #gamma[0] = 2.0*(gamma[0]/abs(gamma[0]))

    #if abs(gamma[0] + gamma[1]) >= 1.0:
    #gamma[1] = 1.0 - gamma[0]

    airmass_2 = x[1]
    norm_1 = x[2]
    norm_2 = x[3]
    t_11 = x[4]
    t_21 = x[5]

    poly_p = x[12:]

    poly_p_1 = x[12:12 + len(poly_p) / 2]
    poly_p_2 = x[12 + len(poly_p) / 2:]

    #poly_p_2[1] = poly_p_1[1]
    #poly_p_2[2] = poly_p_1[2]

    per = 1.7497798
    inc = 85.35
    ars = 7.38
    #  gamma = [0,0]
    tt = -1.25135820e-04

    tt += 2456892.54327

    t_10 = 1.0
    t_20 = 1.0

    #print airmass_2
    #print norm_1, norm_2
    #print t_1, t_2

    print gamma

    divider = 2456895

    poly_m_1 = poly_model(poly_p_1,
                          hjd[hjd < divider] - min(hjd[hjd < divider]))
    poly_m_2 = poly_model(poly_p_2,
                          hjd[hjd > divider] - min(hjd[hjd > divider]))

    poly_t_1 = poly_model([t_10, t_11],
                          hjd[hjd < divider] - min(hjd[hjd < divider]))
    poly_t_2 = poly_model([t_20, t_21],
                          hjd[hjd > divider] - min(hjd[hjd > divider]))

    t_dep = array(list(poly_t_1) + list(poly_t_2))

    extinction = array(list(poly_m_1) + list(poly_m_2))

    poly_m = 1.0 + 10.0**(0.4 *
                          (airmass + airmass_2 * airmass**2.0) * extinction)

    z = transit.t2z(tt, per, inc, hjd, ars)
    #model = transit.occultquad(z, p0, gamma)
    #  model = transit.occultnonlin(z, p0, gamma)

    radiustotal = (1.0 + p0) / ars

    phase_offset = 0

    INPUT = [
        p0, radiustotal, gamma[0], gamma[1], inc, phase_offset, spotlong,
        spotlat, spotsize, spotflux
    ]

    if SPOT_DATA == True:
        plotting = False
        model = fprism((hjd - tt) / (per), INPUT, SPOT_DATA, plotting=plotting)
        if plotting == True:
            quit()
    else:
        model = transit.occultquad(z, p0, gamma)

    #plot(hjd,model2,'b-')
    #plot(hjd,model,'ro')
    #show()

    detrended = t_dep * model / poly_m

    for i in range(0, len(hjd)):
        if hjd[i] < divider:
            detrended[i] = detrended[i] * norm_1
        else:
            detrended[i] = detrended[i] * norm_2


#  detrended = detrended*sum(model)/sum(detrended)

    return detrended, z
def transit_model(x,hjd,airmass,fix=[]):

  p0 = abs(x[0])
  gamma = x[5:7]

  if len(fix) > 0:
    x = fix

  tt = x[1]
  per = x[2]
  inc = x[3]
  ars = x[4]
  airmass_2 = x[7]
  norm_1 = x[8]
  norm_2 = x[9]
  t_10 = x[10] 
  t_11 = x[11]
  t_20 = x[12]
  t_21 = x[13]
  poly_p = x[14:]

  poly_p_1 = x[14:14+len(poly_p)/2]
  poly_p_2 = x[14+len(poly_p)/2:]

  #poly_p_2[1] = poly_p_1[1]
  #poly_p_2[2] = poly_p_1[2]

  per = 1.7497798
  inc = 85.35
  ars = 7.38
#  gamma = [0,0]
  tt = -1.25135820e-04

  tt += 2456892.54327

  gamma[0] = abs(gamma[0])
  gamma[1] = abs(gamma[1])
  
  if gamma[1] > gamma[0]:
    gamma[1] = gamma[0]

  #print airmass_2
  #print norm_1, norm_2
  #print t_1, t_2

  z = transit.t2z(tt, per, inc, hjd, ars)

  divider = 2456895

  poly_m_1 = poly_model(poly_p_1,hjd[hjd < divider] - min(hjd[hjd < divider]))
  poly_m_2 = poly_model(poly_p_2,hjd[hjd > divider] - min(hjd[hjd > divider]))

  poly_t_1 = poly_model([t_10,t_11],hjd[hjd < divider] - min(hjd[hjd < divider]))
  poly_t_2 = poly_model([t_20,t_21],hjd[hjd > divider] - min(hjd[hjd > divider]))

  t_dep = array(list(poly_t_1) + list(poly_t_2))

  extinction = array(list(poly_m_1) + list(poly_m_2))

  poly_m = 1.0 + 10.0**(0.4*(airmass + airmass_2*airmass**2.0)*extinction) 
  
  print x
  model = transit.occultquad(z, p0, gamma)

  detrended = t_dep*model / poly_m

  for i in range(0,len(hjd)):
    if hjd[i] < divider:
      detrended[i] = detrended[i]*norm_1
    else:
      detrended[i] = detrended[i]*norm_2

#  detrended = detrended*sum(model)/sum(detrended)

  return detrended, z
def transit_model(x,hjd,airmass,fix=[],SPOT_DATA=False):

  spotlong = x[6]
  spotlat = x[7]
  spotsize = abs(x[8])
  spotflux = abs(x[9])

  if spotsize > 20:
    spotsize = 20

  if spotflux > 1.0:
    spotflux = 1.0

  print SPOT_DATA

  print x

  p0 = x[0]

  #p0 = abs(x[0])
  #if p0 > 0.25:
    #p0 = 0.25

  gamma = x[10:12]

  if len(fix) > 0:
    x = fix

  #if abs(gamma[0]) > 1.0:
    #gamma[0] = 2.0*(gamma[0]/abs(gamma[0]))

  #if abs(gamma[0] + gamma[1]) >= 1.0:
    #gamma[1] = 1.0 - gamma[0]


  airmass_2 = x[1]
  norm_1 = x[2]
  norm_2 = x[3]
  t_11 = x[4]
  t_21 = x[5]

  poly_p = x[12:]

  poly_p_1 = x[12:12+len(poly_p)/2]
  poly_p_2 = x[12+len(poly_p)/2:]

  #poly_p_2[1] = poly_p_1[1]
  #poly_p_2[2] = poly_p_1[2]

  per = 1.7497798
  inc = 85.35
  ars = 7.38
#  gamma = [0,0]
  tt = -1.25135820e-04

  tt += 2456892.54327

  t_10 = 1.0
  t_20 = 1.0


  #print airmass_2
  #print norm_1, norm_2
  #print t_1, t_2

  print gamma

  divider = 2456895

  poly_m_1 = poly_model(poly_p_1,hjd[hjd < divider] - min(hjd[hjd < divider]))
  poly_m_2 = poly_model(poly_p_2,hjd[hjd > divider] - min(hjd[hjd > divider]))

  poly_t_1 = poly_model([t_10,t_11],hjd[hjd < divider] - min(hjd[hjd < divider]))
  poly_t_2 = poly_model([t_20,t_21],hjd[hjd > divider] - min(hjd[hjd > divider]))

  t_dep = array(list(poly_t_1) + list(poly_t_2))

  extinction = array(list(poly_m_1) + list(poly_m_2))

  poly_m = 1.0 + 10.0**(0.4*(airmass + airmass_2*airmass**2.0)*extinction) 
  
  z = transit.t2z(tt, per, inc, hjd, ars)
  #model = transit.occultquad(z, p0, gamma)
#  model = transit.occultnonlin(z, p0, gamma)

  radiustotal = (1.0 + p0)/ars

  phase_offset = 0

  INPUT = [p0,radiustotal,gamma[0],gamma[1],inc,phase_offset,spotlong,spotlat,spotsize,spotflux]

  if SPOT_DATA == True:
    plotting = False
    model = fprism((hjd-tt)/(per),INPUT,SPOT_DATA,plotting=plotting)
    if plotting == True:
      quit()
  else:
    model = transit.occultquad(z, p0, gamma)

  #plot(hjd,model2,'b-')
  #plot(hjd,model,'ro')
  #show()

  detrended = t_dep*model / poly_m

  for i in range(0,len(hjd)):
    if hjd[i] < divider:
      detrended[i] = detrended[i]*norm_1
    else:
      detrended[i] = detrended[i]*norm_2

#  detrended = detrended*sum(model)/sum(detrended)

  return detrended, z