From dd6eb7e7795f7cbf9c4a195a36726fc8134b2ccb Mon Sep 17 00:00:00 2001
From: Lizzzka007 <gashchuk2011@mail.ru>
Date: Tue, 19 Dec 2023 18:21:35 +0300
Subject: [PATCH] start add sheba
---
...x_compute_func.cuh => sfx_compute_esm.cuh} | 2 +-
..._flux_compute_func.h => sfx_compute_esm.h} | 2 +-
includeCXX/sfx_compute_sheba.h | 14 +
includeCXX/{sfx_flux.h => sfx_esm.h} | 6 +-
includeCXX/sfx_sheba.h | 43 +++
...lux_compute_func.cu => sfx_compute_esm.cu} | 8 +-
srcCXX/sfx_call_class_func.cpp | 6 +-
...x_compute_func.cpp => sfx_compute_esm.cpp} | 8 +-
srcCXX/sfx_compute_sheba.cpp | 112 ++++++++
srcCXX/{sfx_flux.cpp => sfx_esm.cpp} | 52 ++--
srcCXX/sfx_sheba.cpp | 256 ++++++++++++++++++
11 files changed, 467 insertions(+), 42 deletions(-)
rename includeCU/{sfx_flux_compute_func.cuh => sfx_compute_esm.cuh} (81%)
rename includeCXX/{sfx_flux_compute_func.h => sfx_compute_esm.h} (81%)
create mode 100644 includeCXX/sfx_compute_sheba.h
rename includeCXX/{sfx_flux.h => sfx_esm.h} (97%)
create mode 100644 includeCXX/sfx_sheba.h
rename srcCU/{sfx_flux_compute_func.cu => sfx_compute_esm.cu} (96%)
rename srcCXX/{sfx_flux_compute_func.cpp => sfx_compute_esm.cpp} (95%)
create mode 100644 srcCXX/sfx_compute_sheba.cpp
rename srcCXX/{sfx_flux.cpp => sfx_esm.cpp} (78%)
create mode 100644 srcCXX/sfx_sheba.cpp
diff --git a/includeCU/sfx_flux_compute_func.cuh b/includeCU/sfx_compute_esm.cuh
similarity index 81%
rename from includeCU/sfx_flux_compute_func.cuh
rename to includeCU/sfx_compute_esm.cuh
index 2628d5d..df2f949 100644
--- a/includeCU/sfx_flux_compute_func.cuh
+++ b/includeCU/sfx_compute_esm.cuh
@@ -1,7 +1,7 @@
#pragma once
template<typename T>
-void compute_flux_gpu(T *zeta_, T *Rib_, T *Re_, T *B_, T *z0_m_, T *z0_t_, T *Rib_conv_lim_, T *Cm_, T *Ct_, T *Km_, T *Pr_t_inv_,
+void compute_flux_esm_gpu(T *zeta_, T *Rib_, T *Re_, T *B_, T *z0_m_, T *z0_t_, T *Rib_conv_lim_, T *Cm_, T *Ct_, T *Km_, T *Pr_t_inv_,
const T *U_, const T *dT_, const T *Tsemi_, const T *dQ_, const T *h_, const T *in_z0_m_,
const T kappa, const T Pr_t_0_inv, const T Pr_t_inf_inv,
const T alpha_m, const T alpha_h, const T alpha_h_fix,
diff --git a/includeCXX/sfx_flux_compute_func.h b/includeCXX/sfx_compute_esm.h
similarity index 81%
rename from includeCXX/sfx_flux_compute_func.h
rename to includeCXX/sfx_compute_esm.h
index 2cfa0f4..1d7564e 100644
--- a/includeCXX/sfx_flux_compute_func.h
+++ b/includeCXX/sfx_compute_esm.h
@@ -1,7 +1,7 @@
#pragma once
template<typename T>
-void compute_flux_cpu(T *zeta_, T *Rib_, T *Re_, T *B_, T *z0_m_, T *z0_t_, T *Rib_conv_lim_, T *Cm_, T *Ct_, T *Km_, T *Pr_t_inv_,
+void compute_flux_esm_cpu(T *zeta_, T *Rib_, T *Re_, T *B_, T *z0_m_, T *z0_t_, T *Rib_conv_lim_, T *Cm_, T *Ct_, T *Km_, T *Pr_t_inv_,
const T *U_, const T *dT_, const T *Tsemi_, const T *dQ_, const T *h_, const T *in_z0_m_,
const T kappa, const T Pr_t_0_inv, const T Pr_t_inf_inv,
const T alpha_m, const T alpha_h, const T alpha_h_fix,
diff --git a/includeCXX/sfx_compute_sheba.h b/includeCXX/sfx_compute_sheba.h
new file mode 100644
index 0000000..92516a2
--- /dev/null
+++ b/includeCXX/sfx_compute_sheba.h
@@ -0,0 +1,14 @@
+#pragma once
+
+template<typename T>
+void compute_flux_sheba_cpu(T *zeta_, T *Rib_, T *Re_, T *B_, T *z0_m_, T *z0_t_, T *Rib_conv_lim_, T *Cm_, T *Ct_, T *Km_, T *Pr_t_inv_,
+ const T *U_, const T *dT_, const T *Tsemi_, const T *dQ_, const T *h_, const T *in_z0_m_,
+ const T kappa, const T Pr_t_0_inv, const T Pr_t_inf_inv,
+ const T alpha_m, const T alpha_h, const T alpha_h_fix,
+ const T beta_m, const T beta_h, const T Rib_max, const T Re_rough_min,
+ const T B1_rough, const T B2_rough,
+ const T B_max_land, const T B_max_ocean, const T B_max_lake,
+ const T gamma_c, const T Re_visc_min,
+ const T Pr_m, const T nu_air, const T g,
+ const int maxiters_charnock, const int maxiters_convection,
+ const int grid_size);
\ No newline at end of file
diff --git a/includeCXX/sfx_flux.h b/includeCXX/sfx_esm.h
similarity index 97%
rename from includeCXX/sfx_flux.h
rename to includeCXX/sfx_esm.h
index 418d54b..5f47957 100644
--- a/includeCXX/sfx_flux.h
+++ b/includeCXX/sfx_esm.h
@@ -3,7 +3,7 @@
#include <cstddef>
template<typename T, MemType memIn, MemType memOut, MemType RunMem >
-class Flux
+class FluxEsm
{
private:
T *U, *dT, *Tsemi, *dQ, *h, *in_z0_m;
@@ -23,7 +23,7 @@ private:
bool ifAllocated;
size_t allocated_size;
public:
- Flux();
+ FluxEsm();
void set_params(const int grid_size, const T kappa, const T Pr_t_0_inv, const T Pr_t_inf_inv,
const T alpha_m, const T alpha_h, const T alpha_h_fix,
const T beta_m, const T beta_h, const T Rib_max, const T Re_rough_min,
@@ -31,7 +31,7 @@ public:
const T B_max_land, const T B_max_ocean, const T B_max_lake,
const T gamma_c, const T Re_visc_min,
const T Pr_m, const T nu_air, const T g);
- ~Flux();
+ ~FluxEsm();
void compute_flux(T *zeta_, T *Rib_, T *Re_, T *B_, T *z0_m_, T *z0_t_, T *Rib_conv_lim_, T *Cm_, T *Ct_, T *Km_, T *Pr_t_inv_,
T *U_, T *dT_, T *Tsemi_, T *dQ_, T *h_, T *in_z0_m_,
diff --git a/includeCXX/sfx_sheba.h b/includeCXX/sfx_sheba.h
new file mode 100644
index 0000000..3b2cfd3
--- /dev/null
+++ b/includeCXX/sfx_sheba.h
@@ -0,0 +1,43 @@
+#pragma once
+#include "sfx_template_parameters.h"
+#include <cstddef>
+
+template<typename T, MemType memIn, MemType memOut, MemType RunMem >
+class FluxSheba
+{
+private:
+ T *U, *dT, *Tsemi, *dQ, *h, *in_z0_m;
+ T *zeta, *Rib, *Re, *B, *z0_m, *z0_t, *Rib_conv_lim, *Cm, *Ct, *Km, *Pr_t_inv;
+
+ // T kappa, Pr_t_0_inv, Pr_t_inf_inv,
+ // alpha_m, alpha_h, alpha_h_fix,
+ // beta_m, beta_h,
+ // Rib_max, Re_rough_min,
+ // B1_rough, B2_rough,
+ // B_max_land, B_max_ocean, B_max_lake,
+ // gamma_c,
+ // Re_visc_min,
+ // Pr_m, nu_air, g;
+
+ int grid_size;
+ bool ifAllocated;
+ size_t allocated_size;
+public:
+ FluxSheba();
+ void set_params(const int grid_size, const T kappa, const T Pr_t_0_inv, const T Pr_t_inf_inv,
+ const T alpha_m, const T alpha_h, const T alpha_h_fix,
+ const T beta_m, const T beta_h, const T Rib_max, const T Re_rough_min,
+ const T B1_rough, const T B2_rough,
+ const T B_max_land, const T B_max_ocean, const T B_max_lake,
+ const T gamma_c, const T Re_visc_min,
+ const T Pr_m, const T nu_air, const T g);
+ ~Flux();
+
+ void compute_flux(T *zeta_, T *Rib_, T *Re_, T *B_, T *z0_m_, T *z0_t_, T *Rib_conv_lim_, T *Cm_, T *Ct_, T *Km_, T *Pr_t_inv_,
+ T *U_, T *dT_, T *Tsemi_, T *dQ_, T *h_, T *in_z0_m_,
+ const int maxiters_charnock, const int maxiters_convection);
+
+private:
+ void set_data( T *zeta_, T *Rib_, T *Re_, T *B_, T *z0_m_, T *z0_t_, T *Rib_conv_lim_, T *Cm_, T *Ct_, T *Km_, T *Pr_t_inv_,
+ T *U_, T *dT_, T *Tsemi_, T *dQ_, T *h_, T *in_z0_m_);
+};
\ No newline at end of file
diff --git a/srcCU/sfx_flux_compute_func.cu b/srcCU/sfx_compute_esm.cu
similarity index 96%
rename from srcCU/sfx_flux_compute_func.cu
rename to srcCU/sfx_compute_esm.cu
index ff912b9..b9c4e2c 100644
--- a/srcCU/sfx_flux_compute_func.cu
+++ b/srcCU/sfx_compute_esm.cu
@@ -1,6 +1,6 @@
#include <cmath>
#include <iostream>
-#include "../includeCU/sfx_flux_compute_func.cuh"
+#include "../includeCU/sfx_compute_esm.cuh"
template<typename T>
__device__ void get_charnock_roughness(T &z0_m, T &u_dyn0,
@@ -237,7 +237,7 @@ template __device__ void get_psi_semi_convection(double &psi_m, double &psi_h, d
const int maxiters);
template<typename T>
-__global__ void compute_flux(T *zeta_, T *Rib_, T *Re_, T *B_, T *z0_m_, T *z0_t_, T *Rib_conv_lim_, T *Cm_, T *Ct_, T *Km_, T *Pr_t_inv_,
+__global__ void compute_flux_esm_gpu(T *zeta_, T *Rib_, T *Re_, T *B_, T *z0_m_, T *z0_t_, T *Rib_conv_lim_, T *Cm_, T *Ct_, T *Km_, T *Pr_t_inv_,
const T *U_, const T *dT_, const T *Tsemi_, const T *dQ_, const T *h_, const T *in_z0_m_,
const T kappa, const T Pr_t_0_inv, const T Pr_t_inf_inv,
const T alpha_m, const T alpha_h, const T alpha_h_fix,
@@ -351,7 +351,7 @@ __global__ void compute_flux(T *zeta_, T *Rib_, T *Re_, T *B_, T *z0_m_, T *z0_t
}
}
-template __global__ void compute_flux(float *zeta_, float *Rib_, float *Re_, float *B_, float *z0_m_, float *z0_t_, float *Rib_conv_lim_, float *Cm_, float *Ct_, float *Km_, float *Pr_t_inv_,
+template __global__ void compute_flux_esm_gpu(float *zeta_, float *Rib_, float *Re_, float *B_, float *z0_m_, float *z0_t_, float *Rib_conv_lim_, float *Cm_, float *Ct_, float *Km_, float *Pr_t_inv_,
const float *U, const float *dt, const float *T_semi, const float *dq, const float *H, const float *in_z0_m,
const float kappa, const float Pr_t_0_inv, const float Pr_t_inf_inv,
const float alpha_m, const float alpha_h, const float alpha_h_fix,
@@ -362,7 +362,7 @@ template __global__ void compute_flux(float *zeta_, float *Rib_, float *Re_, flo
const float Pr_m, const float nu_air, const float g,
const int maxiters_charnock, const int maxiters_convection,
const int grid_size);
-template __global__ void compute_flux(double *zeta_, double *Rib_, double *Re_, double *B_, double *z0_m_, double *z0_t_, double *Rib_conv_lim_, double *Cm_, double *Ct_, double *Km_, double *Pr_t_inv_,
+template __global__ void compute_flux_esm_gpu(double *zeta_, double *Rib_, double *Re_, double *B_, double *z0_m_, double *z0_t_, double *Rib_conv_lim_, double *Cm_, double *Ct_, double *Km_, double *Pr_t_inv_,
const double *U, const double *dt, const double *T_semi, const double *dq, const double *H, const double *in_z0_m,
const double kappa, const double Pr_t_0_inv, const double Pr_t_inf_inv,
const double alpha_m, const double alpha_h, const double alpha_h_fix,
diff --git a/srcCXX/sfx_call_class_func.cpp b/srcCXX/sfx_call_class_func.cpp
index 9d852e8..2090d62 100644
--- a/srcCXX/sfx_call_class_func.cpp
+++ b/srcCXX/sfx_call_class_func.cpp
@@ -1,7 +1,7 @@
#include <stdlib.h>
#include <stdio.h>
#include "../includeCXX/sfx_call_class_func.h"
-#include "../includeCXX/sfx_flux.h"
+#include "../includeCXX/sfx_esm.h"
#include <vector>
// -------------------------------------------------------------------------- //
@@ -18,9 +18,9 @@ void surf_flux_CXX (float *zeta_, float *Rib_, float *Re_, float *B_, float *z0_
const int grid_size)
{
#ifdef INCLUDE_CUDA
- static Flux<float, MemType::CPU, MemType::CPU, MemType::GPU> F;
+ static FluxEsm<float, MemType::CPU, MemType::CPU, MemType::GPU> F;
#else
- static Flux<float, MemType::CPU, MemType::CPU, MemType::CPU> F;
+ static FluxEsm<float, MemType::CPU, MemType::CPU, MemType::CPU> F;
#endif
F.set_params(grid_size, kappa, Pr_t_0_inv, Pr_t_inf_inv,
diff --git a/srcCXX/sfx_flux_compute_func.cpp b/srcCXX/sfx_compute_esm.cpp
similarity index 95%
rename from srcCXX/sfx_flux_compute_func.cpp
rename to srcCXX/sfx_compute_esm.cpp
index 1e6bebb..b50f49d 100644
--- a/srcCXX/sfx_flux_compute_func.cpp
+++ b/srcCXX/sfx_compute_esm.cpp
@@ -1,6 +1,6 @@
#include <cmath>
#include <iostream>
-#include "../includeCXX/sfx_flux_compute_func.h"
+#include "../includeCXX/sfx_compute_esm.h"
template<typename T>
void get_charnock_roughness(T &z0_m, T &u_dyn0,
@@ -237,7 +237,7 @@ template void get_psi_semi_convection(double &psi_m, double &psi_h, double &zeta
const int maxiters);
template<typename T>
-void compute_flux_cpu(T *zeta_, T *Rib_, T *Re_, T *B_, T *z0_m_, T *z0_t_, T *Rib_conv_lim_, T *Cm_, T *Ct_, T *Km_, T *Pr_t_inv_,
+void compute_flux_esm_cpu(T *zeta_, T *Rib_, T *Re_, T *B_, T *z0_m_, T *z0_t_, T *Rib_conv_lim_, T *Cm_, T *Ct_, T *Km_, T *Pr_t_inv_,
const T *U_, const T *dT_, const T *Tsemi_, const T *dQ_, const T *h_, const T *in_z0_m_,
const T kappa, const T Pr_t_0_inv, const T Pr_t_inf_inv,
const T alpha_m, const T alpha_h, const T alpha_h_fix,
@@ -349,7 +349,7 @@ void compute_flux_cpu(T *zeta_, T *Rib_, T *Re_, T *B_, T *z0_m_, T *z0_t_, T *R
}
}
-template void compute_flux_cpu(float *zeta_, float *Rib_, float *Re_, float *B_, float *z0_m_, float *z0_t_, float *Rib_conv_lim_, float *Cm_, float *Ct_, float *Km_, float *Pr_t_inv_,
+template void compute_flux_esm_cpu(float *zeta_, float *Rib_, float *Re_, float *B_, float *z0_m_, float *z0_t_, float *Rib_conv_lim_, float *Cm_, float *Ct_, float *Km_, float *Pr_t_inv_,
const float *U, const float *dt, const float *T_semi, const float *dq, const float *H, const float *in_z0_m,
const float kappa, const float Pr_t_0_inv, const float Pr_t_inf_inv,
const float alpha_m, const float alpha_h, const float alpha_h_fix,
@@ -360,7 +360,7 @@ template void compute_flux_cpu(float *zeta_, float *Rib_, float *Re_, float *B_,
const float Pr_m, const float nu_air, const float g,
const int maxiters_charnock, const int maxiters_convection,
const int grid_size);
-template void compute_flux_cpu(double *zeta_, double *Rib_, double *Re_, double *B_, double *z0_m_, double *z0_t_, double *Rib_conv_lim_, double *Cm_, double *Ct_, double *Km_, double *Pr_t_inv_,
+template void compute_flux_esm_cpu(double *zeta_, double *Rib_, double *Re_, double *B_, double *z0_m_, double *z0_t_, double *Rib_conv_lim_, double *Cm_, double *Ct_, double *Km_, double *Pr_t_inv_,
const double *U, const double *dt, const double *T_semi, const double *dq, const double *H, const double *in_z0_m,
const double kappa, const double Pr_t_0_inv, const double Pr_t_inf_inv,
const double alpha_m, const double alpha_h, const double alpha_h_fix,
diff --git a/srcCXX/sfx_compute_sheba.cpp b/srcCXX/sfx_compute_sheba.cpp
new file mode 100644
index 0000000..ad420c0
--- /dev/null
+++ b/srcCXX/sfx_compute_sheba.cpp
@@ -0,0 +1,112 @@
+#include <cmath>
+#include <iostream>
+#include "../includeCXX/sfx_compute_sheba.h"
+
+template<typename T>
+void compute_flux_sheba_cpu(T *zeta_, T *Rib_, T *Re_, T *B_, T *z0_m_, T *z0_t_, T *Rib_conv_lim_, T *Cm_, T *Ct_, T *Km_, T *Pr_t_inv_,
+ const T *U_, const T *dT_, const T *Tsemi_, const T *dQ_, const T *h_, const T *in_z0_m_,
+ const T kappa, const T Pr_t_0_inv, const T Pr_t_inf_inv,
+ const T alpha_m, const T alpha_h, const T alpha_h_fix,
+ const T beta_m, const T beta_h, const T Rib_max, const T Re_rough_min,
+ const T B1_rough, const T B2_rough,
+ const T B_max_land, const T B_max_ocean, const T B_max_lake,
+ const T gamma_c, const T Re_visc_min,
+ const T Pr_m, const T nu_air, const T g,
+ const int maxiters_charnock, const int maxiters_convection,
+ const int grid_size)
+{
+ T h, U, dT, Tsemi, dQ, z0_m;
+ T z0_t, B, h0_m, h0_t, u_dyn0, Re,
+ zeta, Rib, Udyn, Tdyn, Qdyn, phi_m, phi_h,
+ Km, Pr_t_inv, Cm, Ct;
+
+ int surface_type;
+
+ for (int step = 0; step < grid_size; step++)
+ {
+ U = U_[step];
+ Tsemi = Tsemi_[step];
+ dT = dT_[step];
+ dQ = dQ_[step];
+ h = h_[step];
+ z0_m = in_z0_m_[step];
+
+ if (z0_m < 0.0) surface_type = 0;
+ else surface_type = 1;
+
+ if (surface_type == surface_ocean)
+ {
+ get_charnock_roughness(z0_m, u_dyn0, U, h, numerics%maxiters_charnock);
+ h0_m = h / z0_m;
+ }
+ if (surface_type == surface_land)
+ {
+ h0_m = h / z0_m;
+ u_dyn0 = U * kappa / log(h0_m);
+ }
+
+ Re = u_dyn0 * z0_m / nu_air;
+ get_thermal_roughness(z0_t, B, z0_m, Re, surface_type);
+
+ // --- define relative height [thermal]
+ h0_t = h / z0_t;
+
+ // --- define Ri-bulk
+ Rib = (g / Tsemi) * h * (dT + 0.61e0 * Tsemi * dQ) / (U*U);
+
+ // --- get the fluxes
+ // ----------------------------------------------------------------------------
+ get_dynamic_scales(Udyn, Tdyn, Qdyn, zeta, U, Tsemi, dT, dQ, h, z0_m, z0_t, (g / Tsemi), 10);
+ // ----------------------------------------------------------------------------
+
+ get_phi(phi_m, phi_h, zeta);
+ // ----------------------------------------------------------------------------
+
+ // --- define transfer coeff. (momentum) & (heat)
+ Cm = 0.0;
+ if (U > 0.0)
+ Cm = Udyn / U;
+ Ct = 0.0;
+ if (fabs(dT) > 0.0)
+ Ct = Tdyn / dT;
+
+ // --- define eddy viscosity & inverse Prandtl number
+ Km = kappa * Cm * U * h / phi_m;
+ Pr_t_inv = phi_m / phi_h;
+
+ zeta_[step] = zeta;
+ Rib_[step] = Rib;
+ Re_[step] = Re;
+ B_[step] = B;
+ z0_m_[step] = z0_m;
+ z0_t_[step] = z0_t;
+ Rib_conv_lim_[step] = Rib_conv_lim;
+ Cm_[step] = Cm;
+ Ct_[step] = Ct;
+ Km_[step] = Km;
+ Pr_t_inv_[step] = Pr_t_inv;
+ }
+}
+
+template void compute_flux_sheba_cpu(float *zeta_, float *Rib_, float *Re_, float *B_, float *z0_m_, float *z0_t_, float *Rib_conv_lim_, float *Cm_, float *Ct_, float *Km_, float *Pr_t_inv_,
+ const float *U, const float *dt, const float *T_semi, const float *dq, const float *H, const float *in_z0_m,
+ const float kappa, const float Pr_t_0_inv, const float Pr_t_inf_inv,
+ const float alpha_m, const float alpha_h, const float alpha_h_fix,
+ const float beta_m, const float beta_h, const float Rib_max, const float Re_rough_min,
+ const float B1_rough, const float B2_rough,
+ const float B_max_land, const float B_max_ocean, const float B_max_lake,
+ const float gamma_c, const float Re_visc_min,
+ const float Pr_m, const float nu_air, const float g,
+ const int maxiters_charnock, const int maxiters_convection,
+ const int grid_size);
+template void compute_flux_sheba_cpu(double *zeta_, double *Rib_, double *Re_, double *B_, double *z0_m_, double *z0_t_, double *Rib_conv_lim_, double *Cm_, double *Ct_, double *Km_, double *Pr_t_inv_,
+ const double *U, const double *dt, const double *T_semi, const double *dq, const double *H, const double *in_z0_m,
+ const double kappa, const double Pr_t_0_inv, const double Pr_t_inf_inv,
+ const double alpha_m, const double alpha_h, const double alpha_h_fix,
+ const double beta_m, const double beta_h, const double Rib_max, const double Re_rough_min,
+ const double B1_rough, const double B2_rough,
+ const double B_max_land, const double B_max_ocean, const double B_max_lake,
+ const double gamma_c, const double Re_visc_min,
+ const double Pr_m, const double nu_air, const double g,
+ const int maxiters_charnock, const int maxiters_convection,
+ const int grid_size);
\ No newline at end of file
diff --git a/srcCXX/sfx_flux.cpp b/srcCXX/sfx_esm.cpp
similarity index 78%
rename from srcCXX/sfx_flux.cpp
rename to srcCXX/sfx_esm.cpp
index 7ed8604..5564c28 100644
--- a/srcCXX/sfx_flux.cpp
+++ b/srcCXX/sfx_esm.cpp
@@ -1,16 +1,16 @@
#include <iostream>
-#include "../includeCXX/sfx_flux.h"
-#include "../includeCXX/sfx_flux_compute_func.h"
+#include "../includeCXX/sfx_esm.h"
+#include "../includeCXX/sfx_compute_esm.h"
#ifdef INCLUDE_CUDA
- #include "../includeCU/sfx_flux_compute_func.cuh"
+ #include "../includeCU/sfx_compute_esm.cuh"
#include "../includeCU/sfx_memory_processing.cuh"
#endif
#include "../includeCXX/sfx_memory_processing.h"
template<typename T, MemType memIn, MemType memOut, MemType RunMem >
-Flux<T, memIn, memOut, RunMem>::Flux()
+FluxEsm<T, memIn, memOut, RunMem>::FluxEsm()
{
kappa = 0, Pr_t_0_inv = 0, Pr_t_inf_inv = 0,
alpha_m = 0, alpha_h = 0, alpha_h_fix = 0,
@@ -29,7 +29,7 @@ Flux<T, memIn, memOut, RunMem>::Flux()
}
template<typename T, MemType memIn, MemType memOut, MemType RunMem >
-void Flux<T, memIn, memOut, RunMem>::set_params(const int grid_size_, const T kappa_, const T Pr_t_0_inv_, const T Pr_t_inf_inv_,
+void FluxEsm<T, memIn, memOut, RunMem>::set_params(const int grid_size_, const T kappa_, const T Pr_t_0_inv_, const T Pr_t_inf_inv_,
const T alpha_m_, const T alpha_h_, const T alpha_h_fix_,
const T beta_m_, const T beta_h_, const T Rib_max_, const T Re_rough_min_,
const T B1_rough_, const T B2_rough_,
@@ -97,7 +97,7 @@ void Flux<T, memIn, memOut, RunMem>::set_params(const int grid_size_, const T ka
}
template<typename T, MemType memIn, MemType memOut, MemType RunMem >
-Flux<T, memIn, memOut, RunMem>::~Flux()
+FluxEsm<T, memIn, memOut, RunMem>::~FluxEsm()
{
kappa = 0, Pr_t_0_inv = 0, Pr_t_inf_inv = 0,
alpha_m = 0, alpha_h = 0, alpha_h_fix = 0,
@@ -142,7 +142,7 @@ Flux<T, memIn, memOut, RunMem>::~Flux()
}
template<typename T, MemType memIn, MemType memOut, MemType RunMem >
-void Flux<T, memIn, memOut, RunMem>::set_data(T *zeta_, T *Rib_, T *Re_, T *B_, T *z0_m_, T *z0_t_, T *Rib_conv_lim_, T *Cm_, T *Ct_, T *Km_, T *Pr_t_inv_,
+void FluxEsm<T, memIn, memOut, RunMem>::set_data(T *zeta_, T *Rib_, T *Re_, T *B_, T *z0_m_, T *z0_t_, T *Rib_conv_lim_, T *Cm_, T *Ct_, T *Km_, T *Pr_t_inv_,
T *U_, T *dT_, T *Tsemi_, T *dQ_, T *h_, T *in_z0_m_)
{
if(RunMem == memIn)
@@ -183,14 +183,14 @@ void Flux<T, memIn, memOut, RunMem>::set_data(T *zeta_, T *Rib_, T *Re_, T *B_,
}
template<typename T, MemType memIn, MemType memOut, MemType RunMem >
-void Flux<T, memIn, memOut, RunMem>::compute_flux(T *zeta_, T *Rib_, T *Re_, T *B_, T *z0_m_, T *z0_t_, T *Rib_conv_lim_, T *Cm_, T *Ct_, T *Km_, T *Pr_t_inv_,
+void FluxEsm<T, memIn, memOut, RunMem>::compute_FluxEsm(T *zeta_, T *Rib_, T *Re_, T *B_, T *z0_m_, T *z0_t_, T *Rib_conv_lim_, T *Cm_, T *Ct_, T *Km_, T *Pr_t_inv_,
T *U_, T *dT_, T *Tsemi_, T *dQ_, T *h_, T *in_z0_m_,
const int maxiters_charnock, const int maxiters_convection)
{
set_data(zeta_, Rib_, Re_, B_, z0_m_, z0_t_, Rib_conv_lim_, Cm_, Ct_, Km_, Pr_t_inv_,
U_, dT_, Tsemi_, dQ_, h_, in_z0_m_);
- if(RunMem == MemType::CPU) compute_flux_cpu(zeta, Rib, Re, B, z0_m, z0_t, Rib_conv_lim, Cm, Ct, Km, Pr_t_inv,
+ if(RunMem == MemType::CPU) compute_flux_esm_cpu(zeta, Rib, Re, B, z0_m, z0_t, Rib_conv_lim, Cm, Ct, Km, Pr_t_inv,
U, dT, Tsemi, dQ, h, in_z0_m,
kappa, Pr_t_0_inv, Pr_t_inf_inv,
alpha_m, alpha_h, alpha_h_fix,
@@ -203,7 +203,7 @@ void Flux<T, memIn, memOut, RunMem>::compute_flux(T *zeta_, T *Rib_, T *Re_, T *
grid_size);
#ifdef INCLUDE_CUDA
- else compute_flux_gpu(zeta, Rib, Re, B, z0_m, z0_t, Rib_conv_lim, Cm, Ct, Km, Pr_t_inv,
+ else compute_flux_esm_gpu(zeta, Rib, Re, B, z0_m, z0_t, Rib_conv_lim, Cm, Ct, Km, Pr_t_inv,
U, dT, Tsemi, dQ, h, in_z0_m,
kappa, Pr_t_0_inv, Pr_t_inf_inv,
alpha_m, alpha_h, alpha_h_fix,
@@ -234,23 +234,23 @@ void Flux<T, memIn, memOut, RunMem>::compute_flux(T *zeta_, T *Rib_, T *Re_, T *
}
}
-template class Flux<float, MemType::CPU, MemType::CPU, MemType::CPU>;
-template class Flux<double, MemType::CPU, MemType::CPU, MemType::CPU>;
+template class FluxEsm<float, MemType::CPU, MemType::CPU, MemType::CPU>;
+template class FluxEsm<double, MemType::CPU, MemType::CPU, MemType::CPU>;
#ifdef INCLUDE_CUDA
- template class Flux<float, MemType::GPU, MemType::GPU, MemType::GPU>;
- template class Flux<float, MemType::GPU, MemType::GPU, MemType::CPU>;
- template class Flux<float, MemType::GPU, MemType::CPU, MemType::GPU>;
- template class Flux<float, MemType::CPU, MemType::GPU, MemType::GPU>;
- template class Flux<float, MemType::CPU, MemType::CPU, MemType::GPU>;
- template class Flux<float, MemType::CPU, MemType::GPU, MemType::CPU>;
- template class Flux<float, MemType::GPU, MemType::CPU, MemType::CPU>;
+ template class FluxEsm<float, MemType::GPU, MemType::GPU, MemType::GPU>;
+ template class FluxEsm<float, MemType::GPU, MemType::GPU, MemType::CPU>;
+ template class FluxEsm<float, MemType::GPU, MemType::CPU, MemType::GPU>;
+ template class FluxEsm<float, MemType::CPU, MemType::GPU, MemType::GPU>;
+ template class FluxEsm<float, MemType::CPU, MemType::CPU, MemType::GPU>;
+ template class FluxEsm<float, MemType::CPU, MemType::GPU, MemType::CPU>;
+ template class FluxEsm<float, MemType::GPU, MemType::CPU, MemType::CPU>;
- template class Flux<double, MemType::GPU, MemType::GPU, MemType::GPU>;
- template class Flux<double, MemType::GPU, MemType::GPU, MemType::CPU>;
- template class Flux<double, MemType::GPU, MemType::CPU, MemType::GPU>;
- template class Flux<double, MemType::CPU, MemType::GPU, MemType::GPU>;
- template class Flux<double, MemType::CPU, MemType::CPU, MemType::GPU>;
- template class Flux<double, MemType::CPU, MemType::GPU, MemType::CPU>;
- template class Flux<double, MemType::GPU, MemType::CPU, MemType::CPU>;
+ template class FluxEsm<double, MemType::GPU, MemType::GPU, MemType::GPU>;
+ template class FluxEsm<double, MemType::GPU, MemType::GPU, MemType::CPU>;
+ template class FluxEsm<double, MemType::GPU, MemType::CPU, MemType::GPU>;
+ template class FluxEsm<double, MemType::CPU, MemType::GPU, MemType::GPU>;
+ template class FluxEsm<double, MemType::CPU, MemType::CPU, MemType::GPU>;
+ template class FluxEsm<double, MemType::CPU, MemType::GPU, MemType::CPU>;
+ template class FluxEsm<double, MemType::GPU, MemType::CPU, MemType::CPU>;
#endif
\ No newline at end of file
diff --git a/srcCXX/sfx_sheba.cpp b/srcCXX/sfx_sheba.cpp
new file mode 100644
index 0000000..5edfa73
--- /dev/null
+++ b/srcCXX/sfx_sheba.cpp
@@ -0,0 +1,256 @@
+#include <iostream>
+
+#include "../includeCXX/sfx_sheba.h"
+#include "../includeCXX/sfx_compute_sheba.h"
+#ifdef INCLUDE_CUDA
+ #include "../includeCU/sfx_compute_sheba.cuh"
+ #include "../includeCU/sfx_memory_processing.cuh"
+#endif
+
+#include "../includeCXX/sfx_memory_processing.h"
+
+template<typename T, MemType memIn, MemType memOut, MemType RunMem >
+FluxSheba<T, memIn, memOut, RunMem>::FluxSheba()
+{
+ kappa = 0, Pr_t_0_inv = 0, Pr_t_inf_inv = 0,
+ alpha_m = 0, alpha_h = 0, alpha_h_fix = 0,
+ beta_m = 0, beta_h = 0,
+ Rib_max = 0, Re_rough_min = 0,
+ B1_rough = 0, B2_rough = 0,
+ B_max_land = 0, B_max_ocean = 0, B_max_lake = 0,
+ gamma_c = 0,
+ Re_visc_min = 0,
+ Pr_m = 0, nu_air = 0, g = 0;
+
+ grid_size = 0;
+
+ ifAllocated = false;
+ allocated_size = 0;
+}
+
+template<typename T, MemType memIn, MemType memOut, MemType RunMem >
+void FluxSheba<T, memIn, memOut, RunMem>::set_params(const int grid_size_, const T kappa_, const T Pr_t_0_inv_, const T Pr_t_inf_inv_,
+ const T alpha_m_, const T alpha_h_, const T alpha_h_fix_,
+ const T beta_m_, const T beta_h_, const T Rib_max_, const T Re_rough_min_,
+ const T B1_rough_, const T B2_rough_,
+ const T B_max_land_, const T B_max_ocean_, const T B_max_lake_,
+ const T gamma_c_, const T Re_visc_min_,
+ const T Pr_m_, const T nu_air_, const T g_)
+{
+ kappa = kappa_, Pr_t_0_inv = Pr_t_0_inv_, Pr_t_inf_inv = Pr_t_inf_inv_,
+ alpha_m = alpha_m_, alpha_h = alpha_h_, alpha_h_fix = alpha_h_fix_,
+ beta_m = beta_m_, beta_h = beta_h_,
+ Rib_max = Rib_max_, Re_rough_min = Re_rough_min_, B_max_lake = B_max_lake_,
+ B1_rough = B1_rough_, B2_rough = B2_rough_,
+ B_max_land = B_max_land_, B_max_ocean = B_max_ocean_,
+ gamma_c = gamma_c_,
+ Re_visc_min = Re_visc_min_,
+ Pr_m = Pr_m_, nu_air = nu_air_, g = g_;
+
+ grid_size = grid_size_;
+
+ if(RunMem != memIn)
+ {
+ const size_t new_size = grid_size * sizeof(T);
+
+ allocated_size = 0;
+ memproc::realloc<RunMem>((void *&)(U), allocated_size, new_size);
+ allocated_size = 0;
+ memproc::realloc<RunMem>((void *&)(dT), allocated_size, new_size);
+ allocated_size = 0;
+ memproc::realloc<RunMem>((void *&)(Tsemi), allocated_size, new_size);
+ allocated_size = 0;
+ memproc::realloc<RunMem>((void *&)(dQ), allocated_size, new_size);
+ allocated_size = 0;
+ memproc::realloc<RunMem>((void *&)(h), allocated_size, new_size);
+ allocated_size = 0;
+ memproc::realloc<RunMem>((void *&)(in_z0_m), allocated_size, new_size);
+ }
+ if(RunMem != memOut)
+ {
+ const size_t new_size = grid_size * sizeof(T);
+ allocated_size = 0;
+ memproc::realloc<RunMem>((void *&)(zeta), allocated_size, new_size);
+ allocated_size = 0;
+ memproc::realloc<RunMem>((void *&)(Rib), allocated_size, new_size);
+ allocated_size = 0;
+ memproc::realloc<RunMem>((void *&)(Re), allocated_size, new_size);
+ allocated_size = 0;
+ memproc::realloc<RunMem>((void *&)(Rib_conv_lim), allocated_size, new_size);
+ allocated_size = 0;
+ memproc::realloc<RunMem>((void *&)(z0_m), allocated_size, new_size);
+ allocated_size = 0;
+ memproc::realloc<RunMem>((void *&)(z0_t), allocated_size, new_size);
+ allocated_size = 0;
+ memproc::realloc<RunMem>((void *&)(B), allocated_size, new_size);
+ allocated_size = 0;
+ memproc::realloc<RunMem>((void *&)(Cm), allocated_size, new_size);
+ allocated_size = 0;
+ memproc::realloc<RunMem>((void *&)(Ct), allocated_size, new_size);
+ allocated_size = 0;
+ memproc::realloc<RunMem>((void *&)(Km), allocated_size, new_size);
+ allocated_size = 0;
+ memproc::realloc<RunMem>((void *&)(Pr_t_inv), allocated_size, new_size);
+
+ ifAllocated = true;
+ }
+}
+
+template<typename T, MemType memIn, MemType memOut, MemType RunMem >
+FluxSheba<T, memIn, memOut, RunMem>::~FluxSheba()
+{
+ kappa = 0, Pr_t_0_inv = 0, Pr_t_inf_inv = 0,
+ alpha_m = 0, alpha_h = 0, alpha_h_fix = 0,
+ beta_m = 0, beta_h = 0,
+ Rib_max = 0, Re_rough_min = 0,
+ B1_rough = 0, B2_rough = 0,
+ B_max_land = 0, B_max_ocean = 0, B_max_lake = 0,
+ gamma_c = 0,
+ Re_visc_min = 0,
+ Pr_m = 0, nu_air = 0, g = 0;
+
+ grid_size = 0;
+
+ if(ifAllocated == true)
+ {
+ if(RunMem != memIn)
+ {
+ memproc::dealloc<RunMem>((void*&)U);
+ memproc::dealloc<RunMem>((void*&)dT);
+ memproc::dealloc<RunMem>((void*&)Tsemi);
+ memproc::dealloc<RunMem>((void*&)dQ);
+ memproc::dealloc<RunMem>((void*&)h);
+ }
+ if(RunMem != memOut)
+ {
+ memproc::dealloc<RunMem>((void*&)zeta);
+ memproc::dealloc<RunMem>((void*&)Rib);
+ memproc::dealloc<RunMem>((void*&)Re);
+ memproc::dealloc<RunMem>((void*&)Rib_conv_lim);
+ memproc::dealloc<RunMem>((void*&)z0_m);
+ memproc::dealloc<RunMem>((void*&)z0_t);
+ memproc::dealloc<RunMem>((void*&)B);
+ memproc::dealloc<RunMem>((void*&)Cm);
+ memproc::dealloc<RunMem>((void*&)Ct);
+ memproc::dealloc<RunMem>((void*&)Km);
+ memproc::dealloc<RunMem>((void*&)Pr_t_inv);
+ }
+
+ ifAllocated = false;
+ allocated_size = 0;
+ }
+}
+
+template<typename T, MemType memIn, MemType memOut, MemType RunMem >
+void FluxSheba<T, memIn, memOut, RunMem>::set_data(T *zeta_, T *Rib_, T *Re_, T *B_, T *z0_m_, T *z0_t_, T *Rib_conv_lim_, T *Cm_, T *Ct_, T *Km_, T *Pr_t_inv_,
+ T *U_, T *dT_, T *Tsemi_, T *dQ_, T *h_, T *in_z0_m_)
+{
+ if(RunMem == memIn)
+ {
+ U = U_;
+ dT = dT_;
+ Tsemi = Tsemi_;
+ dQ = dQ_;
+ h = h_;
+ in_z0_m = in_z0_m_;
+ }
+ else
+ {
+ const size_t new_size = grid_size * sizeof(T);
+
+ memproc::memcopy<RunMem, memIn>(U, U_, new_size);
+ memproc::memcopy<RunMem, memIn>(dT, dT_, new_size);
+ memproc::memcopy<RunMem, memIn>(Tsemi, Tsemi_, new_size);
+ memproc::memcopy<RunMem, memIn>(dQ, dQ_, new_size);
+ memproc::memcopy<RunMem, memIn>(h, h_, new_size);
+ memproc::memcopy<RunMem, memIn>(in_z0_m, in_z0_m_, new_size);
+ }
+
+ if(RunMem == memOut)
+ {
+ zeta = zeta_;
+ Rib = Rib_;
+ Re = Re_;
+ Rib_conv_lim = Rib_conv_lim_;
+ z0_m = z0_m_;
+ z0_t = z0_t_;
+ B = B_;
+ Cm = Cm_;
+ Ct = Ct_;
+ Km = Km_;
+ Pr_t_inv = Pr_t_inv_;
+ }
+}
+
+template<typename T, MemType memIn, MemType memOut, MemType RunMem >
+void FluxSheba<T, memIn, memOut, RunMem>::compute_FluxSheba(T *zeta_, T *Rib_, T *Re_, T *B_, T *z0_m_, T *z0_t_, T *Rib_conv_lim_, T *Cm_, T *Ct_, T *Km_, T *Pr_t_inv_,
+ T *U_, T *dT_, T *Tsemi_, T *dQ_, T *h_, T *in_z0_m_,
+ const int maxiters_charnock, const int maxiters_convection)
+{
+ set_data(zeta_, Rib_, Re_, B_, z0_m_, z0_t_, Rib_conv_lim_, Cm_, Ct_, Km_, Pr_t_inv_,
+ U_, dT_, Tsemi_, dQ_, h_, in_z0_m_);
+
+ if(RunMem == MemType::CPU) compute_FluxSheba_cpu(zeta, Rib, Re, B, z0_m, z0_t, Rib_conv_lim, Cm, Ct, Km, Pr_t_inv,
+ U, dT, Tsemi, dQ, h, in_z0_m,
+ kappa, Pr_t_0_inv, Pr_t_inf_inv,
+ alpha_m, alpha_h, alpha_h_fix,
+ beta_m, beta_h, Rib_max, Re_rough_min,
+ B1_rough, B2_rough,
+ B_max_land, B_max_ocean, B_max_lake,
+ gamma_c, Re_visc_min,
+ Pr_m, nu_air, g,
+ maxiters_charnock, maxiters_convection,
+ grid_size);
+
+#ifdef INCLUDE_CUDA
+ else compute_FluxSheba_gpu(zeta, Rib, Re, B, z0_m, z0_t, Rib_conv_lim, Cm, Ct, Km, Pr_t_inv,
+ U, dT, Tsemi, dQ, h, in_z0_m,
+ kappa, Pr_t_0_inv, Pr_t_inf_inv,
+ alpha_m, alpha_h, alpha_h_fix,
+ beta_m, beta_h, Rib_max, Re_rough_min,
+ B1_rough, B2_rough,
+ B_max_land, B_max_ocean, B_max_lake,
+ gamma_c, Re_visc_min,
+ Pr_m, nu_air, g,
+ maxiters_charnock, maxiters_convection,
+ grid_size);
+#endif
+
+ if(RunMem != memOut)
+ {
+ const size_t new_size = grid_size * sizeof(T);
+
+ memproc::memcopy<memOut, RunMem>(zeta_, zeta, new_size);
+ memproc::memcopy<memOut, RunMem>(Rib_, Rib, new_size);
+ memproc::memcopy<memOut, RunMem>(Re_, Re, new_size);
+ memproc::memcopy<memOut, RunMem>(Rib_conv_lim_, Rib_conv_lim, new_size);
+ memproc::memcopy<memOut, RunMem>(z0_m_, z0_m, new_size);
+ memproc::memcopy<memOut, RunMem>(z0_t_, z0_t, new_size);
+ memproc::memcopy<memOut, RunMem>(B_, B, new_size);
+ memproc::memcopy<memOut, RunMem>(Cm_, Cm, new_size);
+ memproc::memcopy<memOut, RunMem>(Ct_, Ct, new_size);
+ memproc::memcopy<memOut, RunMem>(Km_, Km, new_size);
+ memproc::memcopy<memOut, RunMem>(Pr_t_inv_, Pr_t_inv, new_size);
+ }
+}
+
+template class FluxSheba<float, MemType::CPU, MemType::CPU, MemType::CPU>;
+template class FluxSheba<double, MemType::CPU, MemType::CPU, MemType::CPU>;
+
+#ifdef INCLUDE_CUDA
+ template class FluxSheba<float, MemType::GPU, MemType::GPU, MemType::GPU>;
+ template class FluxSheba<float, MemType::GPU, MemType::GPU, MemType::CPU>;
+ template class FluxSheba<float, MemType::GPU, MemType::CPU, MemType::GPU>;
+ template class FluxSheba<float, MemType::CPU, MemType::GPU, MemType::GPU>;
+ template class FluxSheba<float, MemType::CPU, MemType::CPU, MemType::GPU>;
+ template class FluxSheba<float, MemType::CPU, MemType::GPU, MemType::CPU>;
+ template class FluxSheba<float, MemType::GPU, MemType::CPU, MemType::CPU>;
+
+ template class FluxSheba<double, MemType::GPU, MemType::GPU, MemType::GPU>;
+ template class FluxSheba<double, MemType::GPU, MemType::GPU, MemType::CPU>;
+ template class FluxSheba<double, MemType::GPU, MemType::CPU, MemType::GPU>;
+ template class FluxSheba<double, MemType::CPU, MemType::GPU, MemType::GPU>;
+ template class FluxSheba<double, MemType::CPU, MemType::CPU, MemType::GPU>;
+ template class FluxSheba<double, MemType::CPU, MemType::GPU, MemType::CPU>;
+ template class FluxSheba<double, MemType::GPU, MemType::CPU, MemType::CPU>;
+#endif
\ No newline at end of file
--
GitLab