This repository has been archived by the owner on Jul 27, 2022. It is now read-only.
/
data2d.py
91 lines (85 loc) · 2.37 KB
/
data2d.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
from struct import *
from math import *
import cmath
import scipy
from numpy import *
from scipy.signal import *
import sys
#from os import *
import os
linen=10000
outn=1500
lines=int(sys.argv[2])
fi=open(sys.argv[1]+".bin","rb")
fi.seek(2*int(sys.argv[3])*10000,os.SEEK_SET)
f2=open(sys.argv[1]+"_"+sys.argv[3]+"mag.ppm","w")
f2.write("P5\n"+str(int(outn))+" "+str(int(lines/10))+"\n"+"255\n")
#f3=open(sys.argv[1]+"real.ppm","w")
#f3.write("P5\n"+str(int(outn))+" "+str(int(lines))+"\n"+"255\n")
line=zeros(int(linen),dtype=float)
linec=zeros(int(linen),dtype=complex)
linef=zeros(int(linen),dtype=complex)
lineout=zeros(int(outn),dtype=complex)
output=zeros([int(lines),int(outn)],dtype=complex)
lineoutr=zeros(int(outn),dtype=float)
lppm=zeros(int(outn),dtype=float)
for i in range(lines):
print "ln",i,"/",lines
aa=0x8000
bb=0x8000
cc=0x8000
while((aa!=0x0000 or bb!=0x8000)):
bb=aa
aa=int(unpack("H",fi.read(2))[0]&0x8000)
for j in range(linen):
#print i,j
aaa=int(unpack("H",fi.read(2))[0])
if(aaa&0x8000==0x8000):
break
#print "break"
#snlss[j]=0
else:
line[j]=int(aaa&0x3fff)
if(int(aaa&0x3fff)<0):
print "aaa",aaa,aaa&0x3fff
#print 1
#print snlss[j]
avg=sum(line)/len(line)
print avg,min(line),max(line)
for j in range(linen):
if(line[j]<0):
#print "fuck",snlss[j]
line[j]=avg
line[j]-=avg
if(j<4000):
line[j]=0;
for j in range(linen):
linec[j]=complex(line[j],0)
linef=fft.fftshift(fft.fft(line))
lineout=linef[int(linen/2):int(linen/2)+int(outn)]
output[i]=lineout
lppm+=abs(lineout)
lineoutr=real(lineout)
#print lineout
#lineoutr=lineoutr-(sum(lineoutr)/outn)
#lineoutr=lineoutr/(0.1+max(max(lineoutr),-min(lineoutr)))
#lineoutr=lineoutr*127+128
#print "l",max(lineoutr),min(lineoutr)
#for j in range(outn):
#f3.write(pack("B",int(lineoutr[j])))
if(i%10==9):
lppm=255*abs(lppm)/max(abs(lppm))
#lppm=255*sqrt(abs(lppm))/max(sqrt(abs(lppm)))
for j in range(int(outn)):
f2.write(pack("B",int(lppm[j])))
lppm=zeros(int(outn),dtype=float)
#for j in range(1,slen-1):
#if(snlss[j]-snlss[j-1]>1000 and snlss[j]-snlss[j+1]>1000):
#print "fuck!",snlss[j-1],snlss[j],snlss[j+1]
#snlss[j]=(snlss[j+1]+snlss[j-1])/2.0
#print "fucked",snlss[j-1],snlss[j],snlss[j+1]
#print "fuck!",snlss[j-1],snlss[j],snlss[j+1]'''
#snlssi=hilbert(snlss)
fi.close()
f2.close()
save(sys.argv[1]+"_"+sys.argv[3]+"output.npy",output)