Changeset 7 for trunk/include/meshmask.h

Timestamp:

03/07/11 11:49:01 (13 years ago)

Author:

jbrlod

Message:

rst write

File:

• trunk/include/meshmask.h (modified) (9 diffs)

trunk/include/meshmask.h

@@ +1 @@
+#define nav_lon( i, j ) (nav_lon[(j)*(NX)+(i)])
+#define nav_lat( i, j ) (nav_lat[(j)*(NX)+(i)])
 #define e1u( i, j )  (e1u[(j)*(NX)+(i)])
 #define e2u( i, j )  (e2u[(j)*(NX)+(i)])
@@ -20 +22 @@
 // tous les indices commencent de 0
 //domzgr_substitute.h90 key_zco
+#   define  nav_lev( i , j , k ) nav_lev[k]
 #   define  fsdept( i , j , k )  gdept_0[k]
 #   define  fsdepw( i , j , k )  gdepw_0[k]
@@ -52 +55 @@
 #  define   fsahmv(i,j,k)   ahm0
 
+double nav_lon[NY*NX]; // from mesh_mask.nc
+double nav_lat[NY*NX]; // from mesh_mask.nc
 double e1u[NY*NX];    // from mesh_mask.nc
 double e2u[NY*NX];    // from mesh_mask.nc
@@ -60 +65 @@
 double e1f[NY*NX];    // from mesh_mask.nc
 double e2f[NY*NX];    // from mesh_mask.nc
+double nav_lev[NZ];   // from mesh_mask.nc
 double e3t_0[NZ];    // from mesh_mask.nc
 double e3w_0[NZ];    // from mesh_mask.nc
@@ -81 +87 @@
 char const    *mesh_mask_file    = PATH_NCFILES"/mesh_mask.nc";
 int mesh_mask_file_id;
+int nav_lon_id, nav_lat_id, nav_lev_id; //Utiliser juste pour info
 int e1u_id,e2u_id,e1v_id,e2v_id,e1t_id,e2t_id,e1f_id,e2f_id,e3t_0_id,e3w_0_id;
 int mbathy_id,gphit_id,gphiu_id,gphiv_id,gphif_id,gdept_0_id,gdepw_0_id;
@@ -106 +113 @@
 
 
+
 //Ecriture du netcdf
 void write_rst_global_att(int ncid); //Ecrit les attributs globaux du restart standard
-
+void voncwrite(int t,int znbval,int ynbval, int xnbval, char const *var_name,int ncid,int dimids[], YREAL vect[]);
+void soncwrite(int t,int ynbval, int xnbval, char const *var_name,int ncid,int dimids[], YREAL vect[]);
+void zoncwrite(int t,int znbval,char const *var_name,int ncid,int dimids[], YREAL vect[]);
+void zncwrite(int znbval,char const *var_name,int ncid,int dimids[], YREAL vect[]);
+void sncwrite(int ynbval, int xnbval, char const *var_name,int ncid,int dimids[], YREAL vect[]);
 void define_dim(int ncid, int dimid[]);
 //défini les dimensions en x,y,z, et t (unlimited), 
@@ -119 +131 @@
   mesh_mask_file_id=Ouvre_nc(mesh_mask_file);
 
+  nav_lon_id=Var_id(mesh_mask_file_id,"nav_lon");
+  snclire(NY,NX, mesh_mask_file_id, nav_lon_id,nav_lon);
+  
+  nav_lat_id=Var_id(mesh_mask_file_id,"nav_lat");
+  snclire(NY,NX, mesh_mask_file_id, nav_lat_id,nav_lat);
+
   e1u_id=Var_id(mesh_mask_file_id,"e1u");
   sonclire(0, NY,NX, mesh_mask_file_id, e1u_id,e1u);
@@ -181 +199 @@
   bmask_id=Var_id(mesh_mask_file_id,"bmask");
   sonclire(0, NY,NX, mesh_mask_file_id, bmask_id,bmask);
+
+  nav_lev_id=Var_id(mesh_mask_file_id,"nav_lev");
+  znclire(NZ, mesh_mask_file_id, nav_lev_id,nav_lev);
 
   gdept_0_id=Var_id(mesh_mask_file_id,"gdept_0");
@@ -358 +379 @@
 //dimid={t_id,z_id,y_id,x_id};
 { 
-
+  nc_redef(ncid);
   nc_def_dim(ncid,"x",NX,&dimid[3]);
   nc_def_dim(ncid,"y",NY,&dimid[2]);
   nc_def_dim(ncid,"z",NZ,&dimid[1]);
   nc_def_dim(ncid,"t",NC_UNLIMITED,&dimid[0]);
- 
+  nc_enddef(ncid);
  
  
 }
+
+void write_rst_var(int t,   int ncid, int dimids[]) {
+YREAL vect[NZ*NY*NX];
+ 
+//nav_lon
+ for(int j=0;j<NY;j++)
+      for(int i=0;i<NX;i++){
+        vect[(j)*(NX)+(i)]=nav_lon( i, j );
+      }
+ sncwrite(NY,NX,"nav_lon",ncid,dimids,vect);
+ 
+ //nav_lat
+ for(int j=0;j<NY;j++)
+   for(int i=0;i<NX;i++){
+     vect[(j)*(NX)+(i)]=nav_lat( i, j );
+   }
+ sncwrite(NY,NX,"nav_lat",ncid,dimids,vect);
+ 
+ //nav_lev
+ for(int k=0;k<NZ;k++){
+  vect[k]=nav_lev[k];  
+}
+zncwrite(NZ,"nav_lev",ncid,dimids,vect);
+
+//gcx
+for(int j=0;j<NY;j++)
+   for(int i=0;i<NX;i++){
+     vect[(j)*(NX)+(i)]=YS_gcx2(0,i,j,t);
+   }
+ soncwrite(0,NY,NX,"gcx",ncid,dimids,vect);
+ 
+//gcxb
+for(int j=0;j<NY;j++)
+   for(int i=0;i<NX;i++){
+     vect[(j)*(NX)+(i)]=YS_gcx2(0,i,j,t-1);
+   }
+ soncwrite(0,NY,NX,"gcxb",ncid,dimids,vect);
+ 
+//sshb
+for(int j=0;j<NY;j++)
+   for(int i=0;i<NX;i++){
+     vect[(j)*(NX)+(i)]=YS_sshb(0,i,j,t);
+   }
+ soncwrite(0,NY,NX,"sshb",ncid,dimids,vect);
+ 
+//sshn
+for(int j=0;j<NY;j++)
+   for(int i=0;i<NX;i++){
+     vect[(j)*(NX)+(i)]=YS_sshn(0,i,j,t);
+   }
+ soncwrite(0,NY,NX,"sshn",ncid,dimids,vect);
+ 
+ //ub
+ for(int k=0;k<NZ;k++)
+   for(int j=0;j<NY;j++)
+     for(int i=0;i<NX;i++){
+       vect[(k)*(NY*NX)+(j)*(NX)+(i)]=YS_ub(0,i,j,k,t);
+     }
+ voncwrite(0,NZ,NY,NX,"ub",ncid,dimids,vect);
+
+//vb
+ for(int k=0;k<NZ;k++)
+   for(int j=0;j<NY;j++)
+     for(int i=0;i<NX;i++){
+       vect[(k)*(NY*NX)+(j)*(NX)+(i)]=YS_vb(0,i,j,k,t);
+     }
+ voncwrite(0,NZ,NY,NX,"vb",ncid,dimids,vect);
+
+//tb
+ for(int k=0;k<NZ;k++)
+   for(int j=0;j<NY;j++)
+     for(int i=0;i<NX;i++){
+       vect[(k)*(NY*NX)+(j)*(NX)+(i)]=YS_tb(0,i,j,k,t);
+     }
+ voncwrite(0,NZ,NY,NX,"tb",ncid,dimids,vect);
+
+//sb
+ for(int k=0;k<NZ;k++)
+   for(int j=0;j<NY;j++)
+     for(int i=0;i<NX;i++){
+       vect[(k)*(NY*NX)+(j)*(NX)+(i)]=YS_sb(0,i,j,k,t);
+     }
+ voncwrite(0,NZ,NY,NX,"sb",ncid,dimids,vect);
+
+//rotb
+ for(int k=0;k<NZ;k++)
+   for(int j=0;j<NY;j++)
+     for(int i=0;i<NX;i++){
+       vect[(k)*(NY*NX)+(j)*(NX)+(i)]=YS_rotn(0,i,j,k,t-1);
+     }
+ voncwrite(0,NZ,NY,NX,"rotb",ncid,dimids,vect);
+
+//hdivb
+ for(int k=0;k<NZ;k++)
+   for(int j=0;j<NY;j++)
+     for(int i=0;i<NX;i++){
+       vect[(k)*(NY*NX)+(j)*(NX)+(i)]=YS_hdivn(0,i,j,k,t-1);
+     }
+ voncwrite(0,NZ,NY,NX,"hdivb",ncid,dimids,vect);
+
+//un
+ for(int k=0;k<NZ;k++)
+   for(int j=0;j<NY;j++)
+     for(int i=0;i<NX;i++){
+       vect[(k)*(NY*NX)+(j)*(NX)+(i)]=YS_ua(0,i,j,k,t);
+     }
+ voncwrite(0,NZ,NY,NX,"un",ncid,dimids,vect);
+
+//vn
+ for(int k=0;k<NZ;k++)
+   for(int j=0;j<NY;j++)
+     for(int i=0;i<NX;i++){
+       vect[(k)*(NY*NX)+(j)*(NX)+(i)]=YS_va(0,i,j,k,t);
+     }
+ voncwrite(0,NZ,NY,NX,"vn",ncid,dimids,vect);
+
+//tn
+ for(int k=0;k<NZ;k++)
+   for(int j=0;j<NY;j++)
+     for(int i=0;i<NX;i++){
+       vect[(k)*(NY*NX)+(j)*(NX)+(i)]=YS_ta(0,i,j,k,t);
+     }
+ voncwrite(0,NZ,NY,NX,"tn",ncid,dimids,vect);
+
+//sn
+ for(int k=0;k<NZ;k++)
+   for(int j=0;j<NY;j++)
+     for(int i=0;i<NX;i++){
+       vect[(k)*(NY*NX)+(j)*(NX)+(i)]=YS_sa(0,i,j,k,t);
+     }
+ voncwrite(0,NZ,NY,NX,"sn",ncid,dimids,vect);
+
+//rotn
+ for(int k=0;k<NZ;k++)
+   for(int j=0;j<NY;j++)
+     for(int i=0;i<NX;i++){
+       vect[(k)*(NY*NX)+(j)*(NX)+(i)]=YS_rotn(0,i,j,k,t);
+     }
+ voncwrite(0,NZ,NY,NX,"rotn",ncid,dimids,vect);
+
+//hdivn
+ for(int k=0;k<NZ;k++)
+   for(int j=0;j<NY;j++)
+     for(int i=0;i<NX;i++){
+       vect[(k)*(NY*NX)+(j)*(NX)+(i)]=YS_hdivn(0,i,j,k,t);
+     }
+ voncwrite(0,NZ,NY,NX,"hdivn",ncid,dimids,vect);
+
+
+}
+
+
 
 void write_rst_global_att(int ncid) {