def logP2(p,value=0.):
lp = 0.
for i in range(len(imgs)):
if i == 0:
x0,y0 = 0,0
else:
x0 = pars[0].value
y0 = pars[1].value
print 'dx,dy = ', x0, y0
#print x0,y0
image = imgs[i]
sigma = sigs[i]
psf = PSFs[i]
lp += lensModel2.lensFit(None,image,sigma,gals,lenses,srcs,xc+x0,yc+y0,OVRS,verbose=False,psf=psf,mask=mask,csub=1,showAmps=False)
#print lp
return lp
def logP(value=0.,p=pars):
lp = 0.
fits = []
for i in range(len(imgs)):
if i == 0:
x0,y0 = 0,0
else:
x0 = pars[0].value
y0 = pars[1].value
#print x0,y0
image = imgs[i]
sigma = sigs[i]
psf = PSFs[i]
lp += lensModel.lensFit(None,image,sigma,gals,lenses,srcs,xc+x0,yc+y0,OVRS,
verbose=False,psf=psf,csub=1)
fits.append(lensModel2.lensFit(None,image,sigma,gals,lenses,srcs,xc+x0,yc+y0,OVRS,verbose=False,psf=psf,csub=1,showAmps=True))
return lp,fits
import lensModel2
ims = []
models = []
sfit = []
for i in range(len(imgs)):
image = imgs[i]
sigma = sigs[i]
psf = PSFs[i]
if i == 0:
x0,y0 = 0,0
else:
x0,y0 = det['xoffset'][-1], det['yoffset'][-1] # xoffset, yoffset #
print x0,y0
model = lensModel2.lensFit(coeff,image,sigma,gals,lenses,srcs,xc+x0,yc+y0,OVRS,noResid=True,psf=psf,verbose=True,getModel=True,showAmps=True) # return the model decomposed into the separate galaxy and source components
sfit.append([model[2],model[3]])
ims = []
tims = np.zeros(imgs[0].shape)
for i in range(len(srcs)):
src = srcs[i]
im = src.pixeval(xc,yc) * sfit[0][i]
ims.append(im)
tims +=im
pl.figure()
pl.imshow(im,origin='lower',interpolation='nearest')
pl.colorbar()
pl.figure()
ims = []
models = []
for i in range(len(imgs)):
image = imgs[i]
sigma = sigs[i]
psf = PSFs[i]
print psf.shape, sigma.shape,image.shape
if i == 0:
x0,y0 = 0,0
else:
x0,y0 = dic['xoffset'][ii][0], dic['yoffset'][ii][0]
im = lensModel.lensFit(None,image,sigma,gals,lenses,srcs,xc+x0,yc+y0,OVRS,psf=psf,verbose=True)
im = lensModel.lensFit(coeff,image,sigma,gals,lenses,srcs,xc+x0,yc+y0,OVRS,noResid=True,psf=psf,verbose=True) # return model
model = lensModel.lensFit(coeff,image,sigma,gals,lenses,srcs,xc+x0,yc+y0,OVRS,noResid=True,psf=psf,verbose=True,getModel=True,showAmps=True) # return the model decomposed into the separate galaxy and source components
print lensModel2.lensFit(coeff,image,sigma,gals,lenses,srcs,xc+x0,yc+y0,OVRS,noResid=True,psf=psf,verbose=True,getModel=True,showAmps=True)
ims.append(im)
models.append(model)
colours = ['F606W', 'F814W']
for i in range(len(imgs)):
image = imgs[i]
im = ims[i]
model = models[i]
sigma = sigs[i]
#pyfits.PrimaryHDU(model).writeto('/data/ljo31/Lens/J1347/components_uniform'+str(colours[i])+str(X)+'.fits',clobber=True)
#pyfits.PrimaryHDU(im).writeto('/data/ljo31/Lens/J1347/model_uniform'+str(colours[i])+str(X)+'.fits',clobber=True)
#pyfits.PrimaryHDU(image-im).writeto('/data/ljo31/Lens/J1347/resid_uniform'+str(colours[i])+str(X)+'.fits',clobber=True)
#f = open('/data/ljo31/Lens/J1347/coeff'+str(X),'wb')
#cPickle.dump(coeff,f,2)
#f.close()
