Skip to content
Snippets Groups Projects
cxx_sfx_model_compute_flux.cpp 1.76 KiB
Newer Older
  • Learn to ignore specific revisions
  • 数学の武士's avatar
    数学の武士 committed
    #include <stdlib.h>
    #include <stdio.h>
    
    数学の武士's avatar
    数学の武士 committed
    #include "cxx_sfx_model_compute_flux.h"
    
    数学の武士's avatar
    数学の武士 committed
    #include "sfx_esm.h"
    #include "sfx_sheba.h"
    
    数学の武士's avatar
    数学の武士 committed
    #include <vector>
    
    // -------------------------------------------------------------------------- //
    
    void cxx_esm_compute_flux (sfxDataVecTypeC* sfx,
    
    数学の武士's avatar
    数学の武士 committed
                               meteoDataVecTypeC* meteo,
    
    数学の武士's avatar
    数学の武士 committed
                               const sfx_esm_param_C* model_param, 
    
    数学の武士's avatar
    数学の武士 committed
                               const sfx_surface_param* surface_param,
    
    数学の武士's avatar
    数学の武士 committed
                               const sfx_esm_numericsType_C* numerics,
    
    数学の武士's avatar
    数学の武士 committed
                               const sfx_phys_constants* constants,
    
    数学の武士's avatar
    数学の武士 committed
                               const int grid_size)
    
    数学の武士's avatar
    数学の武士 committed
    {
    
    数学の武士's avatar
    数学の武士 committed
    #ifdef INCLUDE_CUDA
    
    数学の武士's avatar
    数学の武士 committed
        static FluxEsm<float, MemType::CPU, MemType::CPU, MemType::GPU> F(sfx, meteo, *model_param, *surface_param, *numerics, *constants, grid_size);
        F.compute_flux();
    
    数学の武士's avatar
    数学の武士 committed
    #else
    
    数学の武士's avatar
    数学の武士 committed
        static FluxEsm<float, MemType::CPU, MemType::CPU, MemType::CPU> F(sfx, meteo, *model_param, *surface_param, *numerics, *constants, grid_size);
        F.compute_flux();
    
    数学の武士's avatar
    数学の武士 committed
    #endif
    
    数学の武士's avatar
    .  
    数学の武士 committed
    }
    
    
    void cxx_sheba_compute_flux (sfxDataVecTypeC* sfx,
    
    数学の武士's avatar
    数学の武士 committed
                               meteoDataVecTypeC* meteo,
    
    数学の武士's avatar
    数学の武士 committed
                               const sfx_sheba_param_C* model_param, 
    
    数学の武士's avatar
    数学の武士 committed
                               const sfx_surface_param* surface_param,
    
    数学の武士's avatar
    数学の武士 committed
                               const sfx_sheba_numericsType_C* numerics,
    
    数学の武士's avatar
    数学の武士 committed
                               const sfx_phys_constants* constants,
                               const int grid_size)
    
    数学の武士's avatar
    .  
    数学の武士 committed
    {
    
    #ifdef INCLUDE_CUDA
        static FluxSheba<float, MemType::CPU, MemType::CPU, MemType::GPU> F(sfx, meteo, *model_param, *surface_param, *numerics, *constants, grid_size);
        F.compute_flux();
    #else
        static FluxSheba<float, MemType::CPU, MemType::CPU, MemType::CPU> F(sfx, meteo, *model_param, *surface_param, *numerics, *constants, grid_size);
        F.compute_flux();
    #endif
    
    数学の武士's avatar
    数学の武士 committed
    }