Can create PNG images that GIMP can convert to the unity raw format

unity_heightmap.py

@@ -2,6 +2,8 @@ import numpy as np
 import matplotlib.pyplot as plt
 from scipy.io import loadmat
 from PIL import Image
+from PIL import features
+#import libtiff
 import cv2
 
 #Loads matlab matrices created by feature_set.py as numpy arrays
@@ -34,6 +36,9 @@ def make_heightmap(elev,savedir):
 
     name = savedir + "kentland_heightmap.raw"
 
+
+    #Makes sure libtiff is working right
+    print(features.check('libtiff'))
     #fig = plt.figure()
     #plt.imshow(elev)
     #plt.show()
@@ -44,19 +49,32 @@ def make_heightmap(elev,savedir):
     #elev_gray = np.ushort(elev)
     #elev_gray = np.uint16(elev)
 
+
+    #From array expects uint8
+
+
     fig = plt.figure()
-    plt.imshow(elev_gray)
-    plt.show()
+    #plt.imshow(elev_gray)
+    #plt.savefig('./unity_data/test.png',pad_inches=0)
+    #plt.show()
 
 
-    f = open(name,'w')
+    #f = open(name,'w')
 
 
     #Convert to raw format image
-    img = Image.fromarray(elev_gray)
+    elev_int8 = (elev_gray*255).astype(np.uint8)
+    img = Image.fromarray(elev_int8)
+    img.save('./unity_data/kentland_gimptest.png')
+
+
     img = img.convert('L')
+    img.save('./unity_data/kentland_gray.raw', 'TIFF', compression='raw')
+    #img = img.convert('1')
     #img = img.convert('I;16')
-    img.save(name, 'PNG')  #Should I add L or B to the end (endian)
+
+    img = img.convert('1')
+    img.save(name, 'TIFF', compression='tiff_raw_16')  #want tiff_raw_16
     img.show()
     #cv2.imshow('Kentland Heightmap',elev_gray)
     #cv2.waitKey(0)