def flatten():
clip = 0.85
#f = readImage(ffile)
#sigma = 8.0
f, g, s1, s2, r1, r2 = FakeData.seismicAndShifts2d2011A(n1, n2, 45)
f = g
d = getDeterminantsFromShifts([r1, r2])
plot(f, cmin=-clip, cmax=clip)
#plot(r1,jet)
#plot(r2,jet)
#plot(d,jet,cmin=0.65,cmax=1.35)
sigma, pmax = 1.0, 10.0
lsf = LocalSlopeFinder(sigma, pmax)
p2 = zerofloat(n1, n2)
el = zerofloat(n1, n2)
lsf.findSlopes(f, p2, el)
el = pow(el, 6)
sigma1, sigma2 = 6.0, 6.0
fl = FlattenerS(sigma1, sigma2)
#plot(el,gray)
#plot(p2,gray,-1,1)
for rotate in [0.0]:
s = fl.findShifts(rotate, p2, el)
g = fl.applyShifts(s, f)
plot(g, cmin=-clip, cmax=clip)
s1, s2 = s[0], s[1]
plot(s1, jet)
plot(s2, jet)
d = getDeterminantsFromShifts(s)
plot(d, jet, cmin=0.65, cmax=1.35)
print "average s1 =", sum(s1) / n1 / n2, "samples"
print "average s2 =", sum(s2) / n1 / n2, "samples"
def flatten():
clip = 0.85
# f = readImage(ffile)
# sigma = 8.0
f, g, s1, s2, r1, r2 = FakeData.seismicAndShifts2d2011A(n1, n2, 45)
f = g
d = getDeterminantsFromShifts([r1, r2])
plot(f, cmin=-clip, cmax=clip)
# plot(r1,jet)
# plot(r2,jet)
# plot(d,jet,cmin=0.65,cmax=1.35)
sigma, pmax = 1.0, 10.0
lsf = LocalSlopeFinder(sigma, pmax)
p2 = zerofloat(n1, n2)
el = zerofloat(n1, n2)
lsf.findSlopes(f, p2, el)
el = pow(el, 6)
sigma1, sigma2 = 6.0, 6.0
fl = FlattenerS(sigma1, sigma2)
# plot(el,gray)
# plot(p2,gray,-1,1)
for rotate in [0.0]:
s = fl.findShifts(rotate, p2, el)
g = fl.applyShifts(s, f)
plot(g, cmin=-clip, cmax=clip)
s1, s2 = s[0], s[1]
plot(s1, jet)
plot(s2, jet)
d = getDeterminantsFromShifts(s)
plot(d, jet, cmin=0.65, cmax=1.35)
print "average s1 =", sum(s1) / n1 / n2, "samples"
print "average s2 =", sum(s2) / n1 / n2, "samples"
def showFake():
clip = 0.85
f, g, s1, s2, r1, r2 = FakeData.seismicAndShifts2d2011A(n1, n2, 45)
plot(f, cmin=-clip, cmax=clip)
plot(g, cmin=-clip, cmax=clip)
plot(r1, jet)
plot(r2, jet)
d = getDeterminantsFromShifts([r1, r2])
plot(d, jet)
a = getAFromShifts([r1, r2])
plot(a, jet, cmin=-1, cmax=2)
a = applyInverseShiftsL([s1, s2], a)
plot(a, jet, cmin=-1, cmax=2)
def showFake(): clip = 0.85 f,g,s1,s2,r1,r2 = FakeData.seismicAndShifts2d2011A(n1,n2,45) plot(f,cmin=-clip,cmax=clip) plot(g,cmin=-clip,cmax=clip) plot(r1,jet) plot(r2,jet) d = getDeterminantsFromShifts([r1,r2]) plot(d,jet) a = getAFromShifts([r1,r2]) plot(a,jet,cmin=-1,cmax=2) a = applyInverseShiftsL([s1,s2],a) plot(a,jet,cmin=-1,cmax=2)
