%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%
% Create and fill CROCO clim and bry files with OGCM data 
% from mercator for a forecast run
%
% On crocotools_param.m, available datasets:
%
%   mercator -> 3 options for mercator_type parameter:
%
%          3 -->  1/12 deg Mercator global forecast  (See Section 8.)
%                   (daily forecast from 00h UTC up to 10 days,
%                                     updated daily at 12h UTC)
%          4 -->  1/24 deg Mercator Mediterranean analysis/forecast (See Section 8.)
%                   (daily forecast from 00h UTC up to 9 days,
%                                      updated daily at 16h UTC)
%          5 -->      the same than 4 but with detiding postprocessing on current and ssh
%
%        Note for 3/4/5:   (now -> 10days              : P1-P10)
%                          (now -1day                  : P0)
%                          (7days from now -> now-1day : Near Real Time analysis)
%                          (more than 15days from now  : Best analysis)
%
%  Further Information:
%  http://www.croco-ocean.org
%
%  This file is part of CROCOTOOLS
%
%  CROCOTOOLS is free software; you can redistribute it and/or modify
%  it under the terms of the GNU General Public License as published
%  by the Free Software Foundation; either version 2 of the License,
%  or (at your option) any later version.
%
%  CROCOTOOLS is distributed in the hope that it will be useful, but
%  WITHOUT ANY WARRANTY; without even the implied warranty of
%  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
%  GNU General Public License for more details.
%
%  You should have received a copy of the GNU General Public License
%  along with this program; if not, write to the Free Software
%  Foundation, Inc., 59 Temple Place, Suite 330, Boston,
%  MA  02111-1307  USA
%
%  Copyright (c) 2006 by Pierrick Penven
%  e-mail:Pierrick.Penven@ird.fr
%
%  Updated    8-Sep-2006 by Pierrick Penven
%  Updated   20-Aug-2008 by Matthieu Caillaud & P. Marchesiello
%  Updated   12-Feb-2016 by P. Marchesiello
%  Updated   14-Oct-2020 by P. Marchesiello
%  Updated   18-Mar-2024 by G. Cambon
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
clear all
close all
tic
%%%%%%%%%%%%%%%%%%%%% USERS DEFINED VARIABLES %%%%%%%%%%%%%%%%%%%%%%%%
%
% Common parameters
%
crocotools_param; 
%
% Specific to forecast
%
makeplot = 0;
%
% Get date
%
rundate_str=date;
date_run=datenum(rundate_str);
rundate=date_run-datenum(Yorig,1,1);
date_frcst=rundate;
for i=1:hdays+1
  time1(i)=date_run-(hdays+1-i);
end
if hdays ~= 0
  % date_deb = rundate - hindcast days : example 2024-02-28T00:00:00Z-PT2D
  % for a validity date 2 days before run 20240228 00hUTC ...
  date_deb=strcat(datestr(time1(1+hdays),'yyyy-mm-ddT00:00:00') ,['-PT',num2str(hdays),'D']);
else
  date_deb=datestr(time1(1+hdays),'yyyy-mm-ddT00:00:00Z');
end
%
% Set generic OGCM file name
%
FRCST_prefix=['mercator_'];
OGCM_name=[FRCST_dir,FRCST_prefix,num2str(rundate),'.cdf'];
%
%
%  MERCATOR : see oct_get_file_python_mercator.m
%
raw_mercator_name=[FRCST_dir,'raw_mercator_',date_deb,'.nc'];
%
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
% end of user input  parameters
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%
if level==0
  nc_suffix='.nc';
else
  nc_suffix=['.nc.',num2str(level)];
  grdname=[grdname,'.',num2str(level)];
end
%
% Get the model grid
%
ncid = netcdf.open(grdname, 'NC_NOWRITE');
lon=netcdf.getVar(ncid, netcdf.inqVarID(ncid, 'lon_rho'));
lat=netcdf.getVar(ncid, netcdf.inqVarID(ncid, 'lat_rho'));
angle=netcdf.getVar(ncid, netcdf.inqVarID(ncid, 'angle'));
h=netcdf.getVar(ncid, netcdf.inqVarID(ncid, 'h'));
pm=netcdf.getVar(ncid, netcdf.inqVarID(ncid, 'pm'));
pn=netcdf.getVar(ncid, netcdf.inqVarID(ncid, 'pn'));
rmask=netcdf.getVar(ncid, netcdf.inqVarID(ncid, 'mask_rho'));
netcdf.close(ncid);
%---------------------------------------------------------------
% Extract data from the Internet
%---------------------------------------------------------------
if Download_data
  %
  % Get model limits
  %
  lonmin=min(min(lon)); 
  lonmax=max(max(lon));
  latmin=min(min(lat));
  latmax=max(max(lat));
  %
  % Download data (matlab routine for download depends on OGCM)
  %
  %
  % Use of Copernicusmarine client (python)
  %  -> script oct_download_mercator_frcst_python.m
  %
  disp('Download data...')
  oct_download_mercator_frcst_python(pathCMC,user,password,mercator_type, ...
                                 raw_mercator_name,product_id, ...
                                 hdays,fdays, ...
                                 lonmin,lonmax,latmin,latmax,hmax, ...
                                 FRCST_dir,FRCST_prefix,date_frcst,Yorig);
end
%---------------------------------------------------------------
% Get OGCM grid 
%---------------------------------------------------------------
ncid = netcdf.open(OGCM_name, 'NC_NOWRITE');
lonT=netcdf.getVar(ncid, netcdf.inqVarID(ncid, 'lonT'));
latT=netcdf.getVar(ncid, netcdf.inqVarID(ncid, 'latT'));
lonU=netcdf.getVar(ncid, netcdf.inqVarID(ncid, 'lonU'));
latU=netcdf.getVar(ncid, netcdf.inqVarID(ncid, 'latU'));
lonV=netcdf.getVar(ncid, netcdf.inqVarID(ncid, 'lonV'));
latV=netcdf.getVar(ncid, netcdf.inqVarID(ncid, 'latV'));
Z=-netcdf.getVar(ncid, netcdf.inqVarID(ncid, 'depth'));
NZ=length(Z);
NZ=NZ-rmdepth;
Z=Z(1:NZ);
%---------------------------------------------------------------
% Get time array 
%---------------------------------------------------------------
OGCM_time=netcdf.getVar(ncid, netcdf.inqVarID(ncid, 'time'));
time_cycle=0;
delta=1; % >1 if subsampling
trange=[1:delta:length(OGCM_time)];
time=zeros(length(trange),1);
for i=1:length(trange)
  time(i)=OGCM_time(trange(i));
end
netcdf.close(ncid);
%---------------------------------------------------------------
% Initial file 
%---------------------------------------------------------------
if makeini==1
  %
  ininame=[ini_prefix,num2str(date_frcst),nc_suffix];
  disp(['-> Create an initial file for date ',num2str(date_deb);])
  oct_create_inifile(ininame,grdname,CROCO_title,...
                 theta_s,theta_b,hc,N,...
                 date_frcst-hdays,'clobber',vtransform); % starts at 00:00
  ncid_ini = netcdf.open(ininame, 'NC_WRITE');
 
  oct_interp_OGCM_frcst(OGCM_name,Roa,interp_method,...
              lonU,latU,lonV,latV,lonT,latT,Z,1,...
              ncid_ini,[],lon,lat,angle,h,pm,pn,rmask,...
              1,vtransform,obc)
  netcdf.close(ncid_ini);
  if hdays ~= 0
    if (isoctave == 0)
      eval(['!cp ',ininame,' ',ini_prefix,'hct_',num2str(date_frcst),nc_suffix])
    else
      eval(['cp ',ininame,' ',ini_prefix,'hct_',num2str(date_frcst),nc_suffix])
    end
  end 
end
%---------------------------------------------------------------
% Clim and Bry files
%---------------------------------------------------------------
if makeclim==1 | makebry==1
  if makebry==1
    bryname=[bry_prefix,num2str(date_frcst),nc_suffix];
    disp([' '])
    disp(['-> Create bry files from ',num2str(date_deb);])
    oct_create_bryfile(bryname,grdname,CROCO_title,[1 1 1 1],...
                   theta_s,theta_b,hc,N,...
                   time,time_cycle,'clobber',vtransform);
    ncid_bry = netcdf.open(bryname, 'NC_WRITE');
  else
    ncid_bry=[];
  end
  if makeclim==1
    clmname=[clm_prefix,num2str(date_frcst),nc_suffix];
    disp(['-> Create clim files from ',num2str(date_deb);])
    oct_create_climfile(clmname,grdname,CROCO_title,...
                    theta_s,theta_b,hc,N,...
                    time,time_cycle,'clobber',vtransform);
    ncid_clm = netcdf.open(clmname, 'NC_WRITE');
  else
    ncid_clm=[];
  end
%---------------------------------------------------------------
% Perform interpolations for all selected records
%---------------------------------------------------------------
  for tndx=1:length(time)
    disp([' Time step : ',num2str(tndx),' of ',num2str(length(time)),' :'])
    oct_interp_OGCM_frcst(OGCM_name,Roa,interp_method,...
                      lonU,latU,lonV,latV,lonT,latT,Z,trange(tndx),...
                      ncid_clm,ncid_bry,lon,lat,angle,h,pm,pn,rmask, ...
                      tndx,vtransform,obc)
  end
%
% Close CROCO files
%
  if ~isempty(ncid_clm)
    netcdf.close(ncid_clm);
  end
  if ~isempty(ncid_bry)
    netcdf.close(ncid_bry);
  end
%
end
%---------------------------------------------------------------
% Make a few plots
%---------------------------------------------------------------
if makeplot==1
  disp(' ')
  disp(' Make a few plots...')
  if makeini==1
    ininame=[ini_prefix,num2str(date_frcst),nc_suffix];
    figure
    oct_test_clim(ininame,grdname,'temp',1,coastfileplot)
    figure
    oct_test_clim(ininame,grdname,'salt',1,coastfileplot)
  end
  if makeclim==1
    clmname=[clm_prefix,num2str(date_frcst),nc_suffix];
    figure
    oct_test_clim(clmname,grdname,'temp',1,coastfileplot)
    figure
    oct_test_clim(clmname,grdname,'salt',1,coastfileplot)
  end
  if makebry==1
    bryname=[bry_prefix,num2str(frcst),nc_suffix];
    figure
    oct_test_bry(bryname,grdname,'temp',1,obc)
    figure
    oct_test_bry(bryname,grdname,'salt',1,obc)
    figure
    oct_test_bry(bryname,grdname,'u',1,obc)
    figure
    oct_test_bry(bryname,grdname,'v',1,obc)
  end
end

toc