using CSV
using Rasters
using NighttimeLights
using Plots
using DataFrames
using Dates

capex = CSV.read("./DATA/29_thermal_2015_2018_manually_checked_construction_lat_long.csv", DataFrame)

radiance_path = "/home/nighttimelights/VIIRS_ANNUAL/v21_median/"
dates = range(start = Date(2012), step = Year(1), length = length(readdir(radiance_path))) |> collect
timestamps = year.(dates)

function agg_radiance_annual(lat, long, R = 10, res = 15)
    raster1 = Raster(radiance_path * readdir(radiance_path)[1])
    bounds, m = annular_ring(raster1, R, lat, long, res) 
    radiances = [Raster(i, lazy = true)[bounds...] for i in radiance_path .* readdir(radiance_path)]
    series = RasterSeries(radiances, Ti(timestamps))
    radiance_datacube = Rasters.combine(series, Ti)       
    return [sum(skipmissing(Array(radiance_datacube[:,:,1,i] .* m))) for i in 1:length(readdir(radiance_path))]
end

##Annular rings 
## 2 - 1 Km ring 
aggs = []
for i in capex.lat_long
    lat, long = tuple(parse.(Float64, split(i, ","))...)
    push!(aggs, agg_radiance_annual(lat, long, 2,1))
end

agg_matrix =  cat(aggs..., dims = 2)'
agg_df = DataFrame(Array(agg_matrix), :auto)
rename!(agg_df, Symbol.(dates))
agg_df.lat_long = capex.lat_long

capexnl2 = leftjoin(capex, agg_df, on = :lat_long)
CSV.write("./DATA/29_thermal_2015_2018_annular_ring_2_1.csv", capexnl2)



## 4 -1 km
aggs = []
for i in capex.lat_long
    lat, long = tuple(parse.(Float64, split(i, ","))...)
    push!(aggs, agg_radiance_annual(lat, long, 4,1))
end

agg_matrix =  cat(aggs..., dims = 2)'
agg_df = DataFrame(Array(agg_matrix), :auto)
rename!(agg_df, Symbol.(dates))
agg_df.lat_long = capex.lat_long

capexnl = leftjoin(capex, agg_df, on = :lat_long)
CSV.write("./DATA/29_thermal_2015_2018_annular_ring_4_1.csv", capexnl)

## 8 - 1 km
aggs = []
for i in capex.lat_long
    lat, long = tuple(parse.(Float64, split(i, ","))...)
    push!(aggs, agg_radiance_annual(lat, long, 8,1))
end

agg_matrix =  cat(aggs..., dims = 2)'
agg_df = DataFrame(Array(agg_matrix), :auto)
rename!(agg_df, Symbol.(dates))
agg_df.lat_long = capex.lat_long

capexnl = leftjoin(capex, agg_df, on = :lat_long)
CSV.write("./DATA/29_thermal_2015_2018_annular_ring_8_1.csv", capexnl)

## 16 - 1 km
aggs = []
for i in capex.lat_long
    lat, long = tuple(parse.(Float64, split(i, ","))...)
    push!(aggs, agg_radiance_annual(lat, long, 16,1))
end

agg_matrix =  cat(aggs..., dims = 2)'
agg_df = DataFrame(Array(agg_matrix), :auto)
rename!(agg_df, Symbol.(dates))
agg_df.lat_long = capex.lat_long

capexnl = leftjoin(capex, agg_df, on = :lat_long)
CSV.write("./DATA/29_thermal_2015_2018_annular_ring_16_1.csv", capexnl)

## 20 - 1 km
aggs = []
for i in capex.lat_long
    lat, long = tuple(parse.(Float64, split(i, ","))...)
    push!(aggs, agg_radiance_annual(lat, long, 20,1))
end

agg_matrix =  cat(aggs..., dims = 2)'
agg_df = DataFrame(Array(agg_matrix), :auto)
rename!(agg_df, Symbol.(dates))
agg_df.lat_long = capex.lat_long

capexnl = leftjoin(capex, agg_df, on = :lat_long)
CSV.write("./DATA/29_thermal_2015_2018_annular_ring_20_1.csv", capexnl)

## 25 - 1 km
aggs = []
for i in capex.lat_long
    lat, long = tuple(parse.(Float64, split(i, ","))...)
    push!(aggs, agg_radiance_annual(lat, long, 25,1))
end

agg_matrix =  cat(aggs..., dims = 2)
agg_df = DataFrame(Array(agg_matrix), :auto)
rename!(agg_df, Symbol.(dates))
agg_df.lat_long = capex.lat_long

capexnl = leftjoin(capex, agg_df, on = :lat_long)
CSV.write("./DATA/29_thermal_2015_2018_annular_ring_25_1.csv", capexnl)