Commit f2ad2483 authored by Bryce Hepner's avatar Bryce Hepner
Browse files

This is the one that should have the change messag

parent d4e64dd5
Loading
Loading
Loading
Loading
+15 −14
Original line number Diff line number Diff line
@@ -507,37 +507,38 @@ if __name__ == "__main__":
    scenes = file_extractor(folder_name)
    images = image_extractor(scenes)
    newnamesforlater = []
    list_dic, bins = make_dictionary(images, 4, False)
    # list_dic, bins = make_dictionary(images, 4, False)
    file_sizes_new = []
    file_sizes_old = []
    # list_dic = np.load("first_dic.npy", allow_pickle="TRUE")
    list_dic = np.load("first_dic.npy", allow_pickle="TRUE")
    bins = [21,32,48]
    np.save("first_dic.npy", list_dic)
    # np.save("first_dic.npy", list_dic)
    for i in range(len(images)):
        image, new_error, diff = huffman(images[i], 4, False)
        encoded_string = encoder(new_error, list_dic, diff, bins)
        inletters = bitstring_to_bytes(encoded_string)
        # image, new_error, diff = huffman(images[i], 4, False)
        # encoded_string = encoder(new_error, list_dic, diff, bins)
        # inletters = bitstring_to_bytes(encoded_string)

        if images[i][-5:] == ".tiff":
            newname = images[i][:-5]
        else:
            newname = images[i][:-4]
        
        newnamesforlater.append(newname + "_Compressed.txt")
        with open(newname + "_Compressed.txt", 'wb') as f:
            f.write(inletters)
        # with open(newname + "_Compressed.txt", 'wb') as f:
        #     f.write(inletters)
        file_sizes_new.append((os.path.getsize(newname + "_Compressed.txt")))
        file_sizes_old.append((os.path.getsize(images[i])))
        sleep(5)
        if i % 50 == 0:
            print(i)
            sleep(20)
        # sleep(5)
        # if i % 50 == 0:
        #     print(i)
        #     sleep(20)
    
    file_sizes_new.append(os.path.getsize("first_dic.npy"))
    print(np.sum(file_sizes_new)/np.sum(file_sizes_old))
    list_dic = np.load("first_dic.npy", allow_pickle="TRUE")
    bins = [21,32,48]
    # starttime = time()
    for i,item in enumerate(newnamesforlater[2:5]):
    for i,item in enumerate(newnamesforlater[0:5]):
        print(item)
        image, new_error, diff = huffman(images[i], 4, False)
        encoded_string2 = bytes_to_bitstring(read_from_file(item))