import rasterio
from constants import *
import matplotlib.pyplot as plt
from map_generator import predict_end_to_end
import pandas as pd

MARS=GEODATA / 'planets' / 'Mars_MGS_MOLA_DEM_mosaic_global_463m.tif'
elevation = rasterio.open(ELEVATION)
elevation_data = elevation.read(1)


data = {
    'longitude': [],
    'latitude': [],
    'elevation': [],
    'distance_to_water': []
}

for longitude in range(-179, 179):
    print('-', end='')
    for latitude in range(-89, 89):
        data['longitude'].append(longitude)
        data['latitude'].append(latitude)

        elev = elevation_data[elevation.index(longitude, latitude)]
        data['elevation'].append(elev)
        data['distance_to_water'] = 999
        print('+', end='')

    print('')

df = pd.DataFrame(data)

predict_end_to_end(df)
plt.show()