from Rstfr2 import rstfr
import mat4py
import numpy as np
import matplotlib.pyplot as plt
import base64
import io
import matplotlib.image as mpimg

data = mat4py.loadmat('examples/ACRG.mat') # seismic data
sig = np.array([data['t'], data['r'], data['z']])
b64dataM, b64datam, b64dataext, b64datarej = rstfr(sig, "s_stft", "rayleigh", 100, 400, 0.1, 0.12, 0.25, 0.3, 0.26, 0.23)

f = open("examples/outputRstfrB64_S_STFTMajor.txt", "w")
f.write(b64dataM)
f.close()
f1 = open("examples/outputRstfrB64_S_STFTMinor.txt", "w")
f1.write(b64datam)
f1.close()
f2 = open("examples/outputRstfrB64_S_STFTextraction.txt", "w")
f2.write(b64dataext)
f2.close()
f3 = open("examples/outputRstfrB64_S_STFTrejection.txt", "w")
f3.write(b64datarej)
f3.close()



imageM = base64.b64decode(b64dataM)
imageM = io.BytesIO(imageM)
imageM = mpimg.imread(imageM, format='jpg')
plt.imshow(imageM, interpolation='nearest')
plt.show()
plt.close()

imagem = base64.b64decode(b64datam)
imagem = io.BytesIO(imagem)
imagem = mpimg.imread(imagem, format='jpg')
plt.imshow(imagem, interpolation='nearest')
plt.show()

imageM = base64.b64decode(b64dataext)
imageM = io.BytesIO(imageM)
imageM = mpimg.imread(imageM, format='jpg')
plt.imshow(imageM, interpolation='nearest')
plt.show()
plt.close()

imagem = base64.b64decode(b64datarej)
imagem = io.BytesIO(imagem)
imagem = mpimg.imread(imagem, format='jpg')
plt.imshow(imagem, interpolation='nearest')
plt.show()
"""
f = open("examples/outputRstfrB64_S_STFTextraction.txt", "r")
f2 = open("examples/outputRstfrB64_S_STFTrejection.txt", "r")
imagem = base64.b64decode(f.read())
imagem = io.BytesIO(imagem)
imagem = mpimg.imread(imagem, format='jpg')
plt.imshow(imagem, interpolation='nearest')
plt.show()

imagem = base64.b64decode(f2.read())
imagem = io.BytesIO(imagem)
imagem = mpimg.imread(imagem, format='jpg')
plt.imshow(imagem, interpolation='nearest')
plt.show()
"""
from Rstfr2 import rstfr
import mat4py
import numpy as np
import matplotlib.pyplot as plt
import base64
import io
import matplotlib.image as mpimg

data = mat4py.loadmat('examples/ACRG.mat') # seismic data
sig = np.array([data['t'], data['r'], data['z']])
b64dataM, b64datam, b64dataext, b64datarej = rstfr(sig, "s_stft", "rayleigh", 100, 400, 0.1, 0.12, 0.25, 0.3, 0.26, 0.23)

f = open("examples/outputRstfrB64_S_STFTMajor.txt", "w")
f.write(b64dataM)
f.close()
f1 = open("examples/outputRstfrB64_S_STFTMinor.txt", "w")
f1.write(b64datam)
f1.close()
f2 = open("examples/outputRstfrB64_S_STFTextraction.txt", "w")
f2.write(b64dataext)
f2.close()
f3 = open("examples/outputRstfrB64_S_STFTrejection.txt", "w")
f3.write(b64datarej)
f3.close()



imageM = base64.b64decode(b64dataM)
imageM = io.BytesIO(imageM)
imageM = mpimg.imread(imageM, format='jpg')
plt.imshow(imageM, interpolation='nearest')
plt.show()
plt.close()

imagem = base64.b64decode(b64datam)
imagem = io.BytesIO(imagem)
imagem = mpimg.imread(imagem, format='jpg')
plt.imshow(imagem, interpolation='nearest')
plt.show()

imageM = base64.b64decode(b64dataext)
imageM = io.BytesIO(imageM)
imageM = mpimg.imread(imageM, format='jpg')
plt.imshow(imageM, interpolation='nearest')
plt.show()
plt.close()

imagem = base64.b64decode(b64datarej)
imagem = io.BytesIO(imagem)
imagem = mpimg.imread(imagem, format='jpg')
plt.imshow(imagem, interpolation='nearest')
plt.show()
"""
f = open("examples/outputRstfrB64_S_STFTextraction.txt", "r")
f2 = open("examples/outputRstfrB64_S_STFTrejection.txt", "r")
imagem = base64.b64decode(f.read())
imagem = io.BytesIO(imagem)
imagem = mpimg.imread(imagem, format='jpg')
plt.imshow(imagem, interpolation='nearest')
plt.show()

imagem = base64.b64decode(f2.read())
imagem = io.BytesIO(imagem)
imagem = mpimg.imread(imagem, format='jpg')
plt.imshow(imagem, interpolation='nearest')
plt.show()
"""
# seismic data f = open("examples/outputRstfrB64_S_STFTextraction.txt", "r") f2 = open("examples/outputRstfrB64_S_STFTrejection.txt", "r") imagem = base64.b64decode(f.read()) imagem = io.BytesIO(imagem) imagem = mpimg.imread(imagem, format='jpg') plt.imshow(imagem, interpolation='nearest') plt.show()  imagem = base64.b64decode(f2.read()) imagem = io.BytesIO(imagem) imagem = mpimg.imread(imagem, format='jpg') plt.imshow(imagem, interpolation='nearest') plt.show()