mask3652days.py

# NOTE : This code is used to replicate the mask for 3652 timesteps from the mask file for one timestep.
#        This requires the absolute  file path to the maskfile with one timestep as first input.
#        The second input required is the absolute filepath to the location where the maskfile for 
#        3652 timesteps has to be stored.

import numpy as np
import netCDF4 as nc
import tkinter as tk
from tkinter import filedialog

def browse_maskfile():
        global mask_file_path
        mask_file_path = filedialog.askopenfilename(filetypes=[("NetCDF file", "*.nc"), ("All files", "*.*")])
        label_1.config(text=f"Path to shapefile : {mask_file_path}")

def finish():
    root.destroy()

root = tk.Tk()
root.title("File Browser to select a maskfile")
root.geometry("800x200")
finish_button = tk.Button(root, text="Finish selecting file", command=finish)
finish_button.pack(pady=20)
button_1 = tk.Button(root, text="Browse Mask file", command=browse_maskfile)
button_1.pack(pady=5)
label_1 = tk.Label(root, text="Mask file : ")
label_1.pack(pady=10)
root.mainloop()

mask_file = nc.Dataset(mask_file_path, 'r')


latitudes =   mask_file.variables['lat'][:]
longitudes =   mask_file.variables['lon'][:]
values =  mask_file.variables['mask_var'][:]

# Define dimensions for new file
lat_size = len(latitudes)
lon_size = len(longitudes)
time_size = 3652

# Creating the new NetCDF file 
output_file = input("Enter the file path to the NetCDF file where the mask for 3652 timestep has to be stored : ")
# Eg : r"E:\ISIMIP Climate Data\SHapefiles and netcdf file_02_12_2023\0and1_proper_india_3652.nc"
with nc.Dataset(output_file, 'w', format='NETCDF4') as nc:
    # Define the dimensions
    nc.createDimension('lat', lat_size)
    nc.createDimension('lon', lon_size)
    nc.createDimension('time', time_size)

    # Create variables
    latitudes_var = nc.createVariable('lat', 'f4', ('lat',))
    longitudes_var = nc.createVariable('lon', 'f4', ('lon',))
    time = nc.createVariable('time', 'i4', ('time',))
    mask_data = nc.createVariable('mask_data', 'f4', ('time', 'lat', 'lon'))

    # Set latitude and longitude values
    latitudes_var[:] = latitudes
    longitudes_var[:] = longitudes

    # Set time values assuming one value per day starting from 1
    time[:] = np.arange(1, time_size + 1)

    # Create a data array by replicating the values for each time step
    replicated_values = np.tile(values, (time_size, 1, 1))
    mask_data[:, :, :] = replicated_values

    mask_file.close()

print(f'NetCDF file created successfully.')