#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Tue Oct 25 09:07:58 2022

@author: arrondea7b
"""

#!/usr/bin/env python
# Read data from an opendap server
import netCDF4
import time
import numpy as np
# for local windows files, note that '\t' defaults to the tab character in python, so use prefix r to indicate that it is a raw string.
baseurl = r'http://servdap.legi.grenoble-inp.fr/opendap/meige/22_TP_TMA/TP04_CORIOLIS/EXP23/im.civ-images.png.civ-images.png.civ.mproj/'
# create a dataset object
dataset = netCDF4.Dataset(baseurl + 'img_10_1.nc')

nx = len(dataset.variables['coord_x'])
ny = len(dataset.variables['coord_y'])

coord_x = np.zeros((nx))
coord_y = np.zeros((ny))

U = np.zeros((nx,ny))
V = np.zeros((nx,ny))
norm = np.zeros((nx,ny))
curl = np.zeros((nx,ny))
div = np.zeros((nx,ny))
strain = np.zeros((nx,ny))
DUDX = np.zeros((nx,ny))
DUDY = np.zeros((nx,ny))
DVDX = np.zeros((nx,ny))
DVDY = np.zeros((nx,ny))

start = time.time()
count = 0

for i in range(10,1001,1):
    
    count+=1
    
    tmp_dataset = netCDF4.Dataset(baseurl + 'img_' + str(i) + '_1.nc')
    
    coord_y += tmp_dataset['coord_y'][:]
    coord_x += tmp_dataset['coord_x']
    
    U += tmp_dataset['U']
    V += tmp_dataset['V']
    
    norm += tmp_dataset['norm']
    curl += tmp_dataset['curl']
    div += tmp_dataset['div']
    strain += tmp_dataset['strain']
    
    DUDX += tmp_dataset['DUDX']
    DUDY += tmp_dataset['DUDY']
    DVDX += tmp_dataset['DVDX']
    DVDY += tmp_dataset['DVDY']
    
end = time.time()

coord_x = coord_x/count
coord_y = coord_y/count

U = U/count
V = V/count
norm = norm/count
curl = curl/count
div = div/count
strain = strain/count
DUDX = DUDX/count
DUDY = DUDY/count
DVDX = DVDX/count
DVDY = DVDY/count

print(end-start)