tif_index_geojson.py

import pandas as pd
import json

tomnod = pd.read_csv("/Users/Andrew/tomnod_complete-both-columns.csv")
tifRange = pd.read_csv("/Users/Andrew/DSSG/hyak_files/andrew_scripts/tifRange-tiles-run-1.csv", header=None,
                       names=['tif_id', 'minxy','maxxy'])


def process_tup(tup):
    return [float(ele) for ele in (tup.strip('()').split(','))]


for i in range(len(tifRange)):
    tifRange.at[i,'catalog_id'] = (tifRange.iloc[i]['tif_id'].split(sep = "_")[1])
    tifRange.at[i, 'minxy'] = process_tup(tifRange.iloc[i]["minxy"])
    tifRange.at[i, 'maxxy'] = process_tup(tifRange.iloc[i]["maxxy"])


for i in range(len(tomnod)):
    x = tomnod.iloc[i]['label']
    if x == "Flooded / Damaged Building":
        tomnod.at[i,'type_id'] = '1'
    elif x == "Flooded / Blocked Road":
        tomnod.at[i,'type_id'] = '2'
    elif x == "Trash Heap":
        tomnod.at[i,'type_id'] = '3'
    elif x == "Blocked Bridge":
        tomnod.at[i,'type_id'] = '4'


tomnod['tif_id'] = ""
for index_tomnod, row_tomnod in tomnod.iterrows():
    if index_tomnod % 1 == 0:
        print('tomnod row: ', index_tomnod)
    tifRange_temp = tifRange[tifRange.catalog_id == row_tomnod['complete_catalog_id']]

    for index_tif, row_tif in tifRange_temp.iterrows():
        if row_tif['minxy'][0] <= row_tomnod['tomnod_x'] <= row_tif['maxxy'][0] \
                and row_tif['minxy'][1] <= row_tomnod['tomnod_y'] <= row_tif['maxxy'][1]:
            if tomnod.at[index_tomnod, 'tif_id'] == "":
                tomnod.at[index_tomnod, 'tif_id'] = row_tif["tif_id"]
                print ('yaaas')
            elif tomnod.at[index_tomnod, 'tif_id'] != "":
                tomnod = tomnod.append(tomnod.iloc[index_tomnod], ignore_index=True)
                tomnod.at[index_tomnod, 'tif_id'] = row_tif["tif_id"]


tomnod_out = tomnod[['label', 'type_id', 'complete_catalog_id', 'tif_id', 'tomnod_x', 'tomnod_y']]
tomnod_out.columns = ["label", "TYPE_ID", "CAT_ID", "IMAGE_ID", "tomnod_x", "tomnod_y"]


def df_to_geojson(df, properties, lat='latitude', lon='longitude'):
    geojson = {'type':'FeatureCollection', 'features':[]}
    for _, row in df.iterrows():
        feature = {'type':'Feature',
                   'properties':{},
                   'geometry':{'type':'Point',
                               'coordinates':[]}}
        feature['geometry']['coordinates'] = [row[lon],row[lat]]
        for prop in properties:
            feature['properties'][prop] = row[prop]
        geojson['features'].append(feature)
    return geojson


cols = list(tomnod_out.columns)
geojson = df_to_geojson(tomnod_out, cols, lat = 'tomnod_y', lon = 'tomnod_x')

output_filename = 'coordinateandtif.geojson'
with open(output_filename, 'w') as output_file:
    json.dump(geojson, output_file, indent=2)