def findFile(namestart, parser, prefix='.', dir_name='.'):
for f in os.listdir(prefix):
fn, ext = os.path.splitext(f)
if ext == ".cbwb" and fn.startswith(namestart):
index = int(fn.strip(namestart))
print "Found",f,"and setting index to", index
if not parser is None:
parser.index = index + 1
return readObject(prefix+'/'+f)
return Node(directory=dir_name)
if __name__ == "__main__":
# Read in Data
namecols = [0]
timecols = [2]
valcols = [9]
parser = Parser("AllPhoneData_pruned.csv",namecols,timecols,valcols,normalize=True)
#t = readObject('test.cbwb')
t = findFile('test_', parser)
old_index = parser.index
while not parser.atEnd():
addInc(t, parser.getNext())
if parser.index >= old_index * 2:
t.saveObject('test_'+str(parser.index)+'.cbwb', remove='test_'+str(old_index)+'.cbwb')
old_index = parser.index
t.viz.plotClusters('time', Instance.properties[0], depth=2)
t.saveObject('test.cbwb')
