import time
import sys
import os
import numpy as np
sys.path.append("code/utils")
import utils
#command line args as follows:
print("Command line args: model directory")
print("Uses images_dir.gt as ground truth")
print("Command line args: models_dir images_dir WITH / terminator")
print("Will try to create a groundTruth (.gt) file if possible")

dirArg = sys.argv[1]

files = ['FarCenter.jpg',
'folder_1_Spanklem.png',
'folder_2.1_VleugelMoerOplegRecht_Nieuw_geen_gaten.png',
'folder_3VleugelMoerOpleg_Rond.jpg',
'folder_4 .1_Variable_Spanklem_korg.png',
'IMG_20170904_142543.jpg',
'IMG_20170904_142553.jpg',
'IMG_20170904_142619.jpg',
'IMG_20170904_142701.jpg',
'IMG_20170904_142710.jpg',
'IMG_20170904_145514.jpg',
'IMG_20170904_145521.jpg',
'IMG_20170904_145528.jpg',
'IMG_20170904_145533.jpg',
'IMG_20170904_145541_crop1.jpg',
'IMG_20170904_145541.jpg',
'IMG_20170904_145547.jpg',
'IMG_20170904_152554.jpg',
'IMG_20170904_152559.jpg',
'IMG_20170904_152604.jpg',
'IMG_20170904_152616.jpg',
'IMG_20170904_152621.jpg',
'IMG_20170904_152627.jpg',
'IMG_20170904_152639.jpg',
'IMG_20170904_152644.jpg',
'IMG_20170904_152648.jpg',
'IMG_20170904_152654.jpg',
'IMG_20170904_163427.jpg',
'IMG_20170904_163433.jpg',
'IMG_20170904_163444.jpg',
'IMG_20170904_163504.jpg',
'IMG_20170904_163542.jpg',
'IMG_20170904_163545.jpg',
'IMG_20170904_163551.jpg',
'IMG_20170904_163557.jpg',
'IMG_20170904_163600_crop1.jpg',
'IMG_20170904_163600.jpg',
'IMG_20170904_163602.jpg',
'IMG_20170904_163613.jpg',
'IMG_20170904_163617.jpg',
'IMG_20170904_163621.jpg',
'IMG_20170904_163627.jpg',
'IMG_20170904_163631.jpg',
'IMG_20170904_163639.jpg',
'IMG_20170904_163645.jpg',
'IMG_20170904_163651.jpg',
'IMG_20170904_163657.jpg',
'IMG_20170904_163701.jpg',
'IMG_20170904_163703.jpg',
'IMG_20170904_163707.jpg',
'IMG_20170904_163710.jpg',
'IMG_20170904_163714.jpg',
'NewRedBot.jpg',
'NewRedTop.jpg',
'OldRedBot.jpg',
'OldRedTop.jpg',
'OldRusty_Crop1.jpg',
'OldRusty.jpg',
'PICT0015.JPG',
'PICT0016.JPG',
'PICT0017.JPG',
'PICT0018.JPG',
'PICT0019.JPG',
'PICT0020.JPG',
'PICT0021.JPG',
'PICT0022.JPG',
'PICT0023.JPG',
'PICT0024.JPG',
'PICT0025.JPG',
'PICT0026.JPG',
'PICT0027.JPG',
'PICT0028.JPG',
'PICT0029.JPG',
'PICT0030.JPG',
'PICT0031.JPG',
'PICT0032.JPG',
'PICT0033.JPG',
'PICT0034.JPG',
'PICT0035.JPG',
'PICT0036.JPG',
'PICT0037.JPG',
'PICT0038.JPG',
'PICT0039.JPG',
'PICT0040.JPG',
'PICT0041.JPG',
'PICT0042.JPG',
'PICT0043.JPG',
'PICT0044.JPG',
'PICT0045.JPG',
'PICT0046.JPG',
'PICT0047.JPG',
'PICT0048.JPG',
'PICT0050.JPG',
'PICT0051.JPG',
'PICT0052.JPG',
'PICT0053.JPG',
'PICT0054.JPG',
'PICT0055.JPG',
'PICT0056.JPG',
'PICT0057.JPG',
'PICT0058.JPG',
'PICT0059.JPG',
'PICT0060.JPG',
'PICT0061.JPG',
'PICT0062.JPG',
'PICT0063.JPG',
'PICT0064.JPG',
'PICT0065.JPG',
'PICT0066.JPG',
'PICT0067.JPG',
'PICT0068.JPG',
'PICT0069.JPG',
'PICT0070.JPG',
'PICT0071.JPG',
'PICT0072.JPG',
'PICT0073.JPG',
'PICT0074.JPG',
'PICT0075.JPG',
'SpanKlemHand1.jpg',
'SpanKlemHand2.jpg',
'SpanKlemTwoColor.jpg']

def sizeof_fmt(num, suffix='B'):
    for unit in ['','Ki','Mi','Gi','Ti','Pi','Ei','Zi']:
        if abs(num) < 1024.0:
            return "%3.1f%s%s" % (num, unit, suffix)
        num /= 1024.0
    return "%.1f%s%s" % (num, 'Yi', suffix)

mnparr = np.array(files)
sizes = lambda x : sizeof_fmt(os.stat(dirArg+x).st_size)

mysizes = map(sizes,files)
print(np.asmatrix([mnparr,list(map(sizes,mnparr))]).T)
