In [1]:
from kbmodpy import kbmod as kb
from trajectoryFiltering import *
import numpy as np
import random as rd
import math
import matplotlib.pyplot as plt
import os
%matplotlib inline

In [2]:
def p_im(image):
    fig = plt.figure(figsize=(12,12))
    plt.imshow(image , origin='lower',  vmin=-50, vmax=100)#cmap=plt.cm.Greys_r,
    plt.xlabel('X Pixels')
    plt.ylabel('Y Pixels')
    plt.colorbar()

In [3]:
im_count = 10
p = kb.psf(1.3)

In [4]:
images = [kb.layered_image(str(n), 350, 350, 10.0, 100.0, n/10) for n in range(im_count)]

In [5]:
for i in range(im_count):
    images[i].add_object(150+3*i, 135+3.5*i, 180.0, p)

In [6]:
p_im(images[0].science())



In [7]:
stack = kb.image_stack(images)

In [8]:
search = kb.stack_search(stack, p)

In [9]:
%time
search.gpu(180, 100, 0.0, 1.0, 25.0, 50.0, 4)
search.get_results(0,20)


CPU times: user 0 ns, sys: 0 ns, total: 0 ns
Wall time: 8.58 µs
Out[9]:
[lh: 10.089127 flux: 150.650848 x: 150 y: 136 x_v: 29.905968 y_v: 33.289570 obs_count: 10,
 lh: 10.089127 flux: 150.650848 x: 150 y: 136 x_v: 30.426653 y_v: 33.493004 obs_count: 10,
 lh: 10.089127 flux: 150.650848 x: 150 y: 136 x_v: 30.258549 y_v: 33.307961 obs_count: 10,
 lh: 10.089127 flux: 150.650848 x: 150 y: 136 x_v: 30.443127 y_v: 33.139343 obs_count: 10,
 lh: 9.770747 flux: 145.896805 x: 150 y: 137 x_v: 30.124802 y_v: 32.070179 obs_count: 10,
 lh: 9.770747 flux: 145.896805 x: 150 y: 137 x_v: 29.953638 y_v: 31.887962 obs_count: 10,
 lh: 9.770747 flux: 145.896805 x: 150 y: 137 x_v: 30.474674 y_v: 32.083588 obs_count: 10,
 lh: 9.770747 flux: 145.896805 x: 150 y: 137 x_v: 30.302502 y_v: 31.902327 obs_count: 10,
 lh: 9.689487 flux: 144.683426 x: 150 y: 135 x_v: 29.707096 y_v: 35.773430 obs_count: 10,
 lh: 9.689487 flux: 144.683426 x: 150 y: 135 x_v: 30.066160 y_v: 35.799282 obs_count: 10,
 lh: 9.689487 flux: 144.683426 x: 150 y: 135 x_v: 30.426424 y_v: 35.822239 obs_count: 10,
 lh: 9.689487 flux: 144.683426 x: 150 y: 135 x_v: 30.264582 y_v: 35.631695 obs_count: 10,
 lh: 9.684531 flux: 144.609421 x: 151 y: 136 x_v: 29.369249 y_v: 33.431980 obs_count: 10,
 lh: 9.622617 flux: 143.684921 x: 151 y: 136 x_v: 29.019114 y_v: 33.405891 obs_count: 10,
 lh: 9.610224 flux: 143.499878 x: 151 y: 136 x_v: 28.507282 y_v: 33.187305 obs_count: 10,
 lh: 9.610224 flux: 143.499878 x: 151 y: 136 x_v: 28.670181 y_v: 33.376949 obs_count: 10,
 lh: 9.570250 flux: 142.902985 x: 149 y: 136 x_v: 32.571037 y_v: 33.536396 obs_count: 10,
 lh: 9.570250 flux: 142.902985 x: 149 y: 136 x_v: 32.756847 y_v: 33.354931 obs_count: 10,
 lh: 9.570250 flux: 142.902985 x: 149 y: 136 x_v: 32.581676 y_v: 33.176563 obs_count: 10,
 lh: 9.493246 flux: 141.753159 x: 149 y: 136 x_v: 31.864689 y_v: 33.175919 obs_count: 10]

In [10]:
%time
search.region_search(30, 35, 20, 6, 4)


CPU times: user 0 ns, sys: 0 ns, total: 0 ns
Wall time: 12.9 µs
Out[10]:
[ix: 150 iy: 136 fx: 177 fy: 166 depth: 0 obs_count: 10 lh: 10.089128 flux 150.650848,
 ix: 150 iy: 136 fx: 178 fy: 166 depth: 0 obs_count: 10 lh: 9.856699 flux 147.180222,
 ix: 150 iy: 136 fx: 177 fy: 167 depth: 0 obs_count: 10 lh: 9.848759 flux 147.061676,
 ix: 150 iy: 135 fx: 177 fy: 167 depth: 0 obs_count: 10 lh: 9.579309 flux 143.038239,
 ix: 151 iy: 136 fx: 177 fy: 166 depth: 0 obs_count: 10 lh: 9.575986 flux 142.988617,
 ix: 150 iy: 137 fx: 177 fy: 166 depth: 0 obs_count: 10 lh: 9.533123 flux 142.348602,
 ix: 150 iy: 136 fx: 178 fy: 167 depth: 0 obs_count: 10 lh: 9.501141 flux 141.871033,
 ix: 150 iy: 135 fx: 177 fy: 166 depth: 0 obs_count: 10 lh: 9.473289 flux 141.455154,
 ix: 149 iy: 136 fx: 178 fy: 166 depth: 0 obs_count: 10 lh: 9.360481 flux 139.770706,
 ix: 150 iy: 137 fx: 177 fy: 167 depth: 0 obs_count: 10 lh: 9.357248 flux 139.722427,
 ix: 151 iy: 136 fx: 178 fy: 166 depth: 0 obs_count: 10 lh: 9.343557 flux 139.517990,
 ix: 149 iy: 136 fx: 177 fy: 166 depth: 0 obs_count: 10 lh: 9.326967 flux 139.270279,
 ix: 150 iy: 135 fx: 178 fy: 166 depth: 0 obs_count: 10 lh: 9.323347 flux 139.216217,
 ix: 151 iy: 136 fx: 177 fy: 167 depth: 0 obs_count: 10 lh: 9.200114 flux 137.376099,
 ix: 150 iy: 137 fx: 178 fy: 166 depth: 0 obs_count: 10 lh: 9.176478 flux 137.023178,
 ix: 150 iy: 135 fx: 178 fy: 167 depth: 0 obs_count: 10 lh: 9.168327 flux 136.901459,
 ix: 150 iy: 137 fx: 177 fy: 165 depth: 0 obs_count: 10 lh: 9.135387 flux 136.409607,
 ix: 149 iy: 136 fx: 177 fy: 167 depth: 0 obs_count: 10 lh: 9.122036 flux 136.210251,
 ix: 151 iy: 136 fx: 176 fy: 166 depth: 0 obs_count: 10 lh: 9.084455 flux 135.649078,
 ix: 151 iy: 136 fx: 176 fy: 167 depth: 0 obs_count: 10 lh: 9.060242 flux 135.287521]

In [ ]:


In [ ]:


In [ ]:


In [ ]:


In [ ]: