Skip to content

Commit

Permalink
add another wavelet transform script
Browse files Browse the repository at this point in the history
  • Loading branch information
iizukak committed Nov 1, 2015
1 parent fb47db5 commit 50a9652
Showing 1 changed file with 61 additions and 0 deletions.
61 changes: 61 additions & 0 deletions src/wavelet_converter2.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,61 @@
import pywt
import numpy
import csv

# setting CSV data directory
DATA_DIR = "data/"

# setting file
import argparse
parser = argparse.ArgumentParser(description = "")
parser.add_argument('--target', required=True)
args = parser.parse_args()
t = args.target
targetPath = DATA_DIR + t + ".csv"
outputPath = DATA_DIR + t + "_wavelet.csv"

# setting input file
targetFile = open(targetPath, "r")
csvReader = csv.reader(targetFile)
print("DEBUG: TARGET FILE PATH:", targetPath)

outputFile = open(outputPath, "w")
csvWriter = csv.writer(outputFile, lineterminator='\n')
print("DEBUG: OUTPUT FILE PATH:", outputPath)

dateList = []
waveletList = []
rawList = []
counter = 0
for row in csvReader:
date, value = row[0], int(row[1])
dateList.append(date)
waveletList.append(value)

# after wavelet transform, value downsize to half size
if counter % 2 == 0:
rawList.append(value)
counter += 1

waveletList = pywt.dwt(waveletList, "db3")[0]
s = sum(waveletList)
offset = 500 - (s / len(waveletList))
print("DEBUG: OFFFSET:", offset)
waveletList = waveletList + offset


outList = []
for d, r, w in zip(dateList, rawList, waveletList):
print(w - r)
outList.append((d, r, w))

# write csv headers
csvWriter.writerow(["timestamp", "raw_value", "wavelet_value"])
csvWriter.writerow(["datetime", "int", "float"])
csvWriter.writerow(["T", "", ""])

# write main data
csvWriter.writerows(outList)

targetFile.close()
outputFile.close()

0 comments on commit 50a9652

Please sign in to comment.