-
Notifications
You must be signed in to change notification settings - Fork 31
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
a2ea377
commit 0a9400f
Showing
4 changed files
with
187 additions
and
12 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,5 +1,52 @@ | ||
| # Increment post processing | ||
| ecbuild_add_executable( TARGET fv3jedi_fv3inc.x | ||
| SOURCES fv3jedi_fv3inc.cc fv3jedi_fv3inc.h) | ||
| target_compile_features( fv3jedi_fv3inc.x PUBLIC cxx_std_17) | ||
| target_link_libraries( fv3jedi_fv3inc.x PUBLIC NetCDF::NetCDF_CXX oops fv3jedi) | ||
| # ------------------------------------------------------------------------- # | ||
| # Application for the GDAS system # | ||
| # ------------------------------------------------------------------------- # | ||
|
|
||
| # Check for minimim cmake requirement | ||
| cmake_minimum_required( VERSION 3.20 FATAL_ERROR ) | ||
|
|
||
| find_package(ecbuild 3.5 REQUIRED HINTS ${CMAKE_CURRENT_SOURCE_DIR} ${CMAKE_CURRENT_SOURCE_DIR}/../ecbuild) | ||
|
|
||
| project(GDASApp VERSION 1.0.0 LANGUAGES C CXX Fortran ) | ||
| # include_directories(${CMAKE_SOURCE_DIR}) | ||
| include_directories(${CMAKE_CURRENT_SOURCE_DIR}) | ||
|
|
||
| include(GNUInstallDirs) | ||
| enable_testing() | ||
|
|
||
| # Build type. | ||
| if(NOT CMAKE_BUILD_TYPE MATCHES "^(Debug|Release|RelWithDebInfo|MinSizeRel)$") | ||
| message(STATUS "Setting build type to 'Release' as none was specified.") | ||
| set(CMAKE_BUILD_TYPE | ||
| "Release" | ||
| CACHE STRING "Choose the type of build." FORCE) | ||
| set_property(CACHE CMAKE_BUILD_TYPE PROPERTY STRINGS "Debug" "Release" | ||
| "MinSizeRel" "RelWithDebInfo") | ||
| endif() | ||
|
|
||
| # Find dependencies. | ||
| find_package(Python3 REQUIRED COMPONENTS Interpreter) | ||
|
|
||
| # Include ecbuild_bundle macro | ||
| include( ecbuild_bundle ) | ||
|
|
||
| # Enable MPI | ||
| set( ENABLE_MPI ON CACHE BOOL "Compile with MPI" ) | ||
|
|
||
| # Handle user options. | ||
| option(BUILD_GDASBUNDLE "Build GDAS Bundle" ON) | ||
| option(CLONE_JCSDADATA "Clone JCSDA test data repositories" OFF) | ||
| option(WORKFLOW_TESTS "Include global-workflow dependent tests" OFF) | ||
|
|
||
| # Build GDAS-managed JEDI executables | ||
| if( BUILD_GDASBUNDLE ) | ||
| find_package( fv3jedi REQUIRED ) | ||
| find_package( soca REQUIRED ) | ||
| add_subdirectory( mains ) | ||
| endif() | ||
|
|
||
| # Install utility scripts. | ||
| add_subdirectory(ush) | ||
|
|
||
| # Include testing. | ||
| add_subdirectory(test) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,8 @@ | ||
| #include "fv3jedi_recenter.h" | ||
| #include "oops/runs/Run.h" | ||
|
|
||
| int main(int argc, char ** argv) { | ||
| oops::Run run(argc, argv); | ||
| gdasapp::recenter recenter; | ||
| return run.execute(recenter); | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,118 @@ | ||
| #pragma once | ||
|
|
||
| #include <iostream> | ||
| #include <memory> | ||
| #include <string> | ||
| #include <vector> | ||
|
|
||
| #include "eckit/config/LocalConfiguration.h" | ||
|
|
||
| #include "fv3jedi/Geometry/Geometry.h" | ||
| #include "fv3jedi/Increment/Increment.h" | ||
| #include "fv3jedi/State/State.h" | ||
|
|
||
| #include "oops/mpi/mpi.h" | ||
| #include "oops/runs/Application.h" | ||
| #include "oops/util/ConfigFunctions.h" | ||
| #include "oops/util/DateTime.h" | ||
| #include "oops/util/Duration.h" | ||
| #include "oops/util/Logger.h" | ||
|
|
||
| namespace gdasapp { | ||
|
|
||
| // Main application class | ||
| class recenter : public oops::Application { | ||
| public: | ||
| explicit recenter(const eckit::mpi::Comm & comm = oops::mpi::world()) | ||
| : Application(comm) {} | ||
| static const std::string classname() {return "gdasapp::recenter";} | ||
|
|
||
| int execute(const eckit::Configuration & fullConfig, bool validate) const { | ||
| // Get analysis parameters | ||
| std::vector<std::string> fcstHours; | ||
| std::string windowBeginStr; | ||
| fullConfig.get("forecast hours", fcstHours); | ||
| fullConfig.get("window begin", windowBeginStr); | ||
| const util::DateTime windowBegin(windowBeginStr); | ||
| const oops::Variables stateVars(fullConfig, "state variables"); | ||
| const oops::Variables incrVars(fullConfig, "increment variables"); | ||
|
|
||
| // Get geometry configurations | ||
| const eckit::LocalConfiguration bkgGeomConfig(fullConfig, "background geometry"); | ||
| const eckit::LocalConfiguration incrGeomConfig(fullConfig, "increment geometry"); | ||
| const eckit::LocalConfiguration anlEnsMeanGeomConfig(fullConfig, \ | ||
| "ensemble mean analysis geometry"); | ||
| const eckit::LocalConfiguration incrCorGeomConfig(fullConfig, \ | ||
| "correction increment geometry"); | ||
|
|
||
| // Setup geometries | ||
| const fv3jedi::Geometry incrGeom(incrGeomConfig, this->getComm()); | ||
| const fv3jedi::Geometry bkgGeom(bkgGeomConfig, this->getComm()); | ||
| const fv3jedi::Geometry anlEnsMeanGeom(anlEnsMeanGeomConfig, this->getComm()); | ||
| const fv3jedi::Geometry incrCorGeom(incrCorGeomConfig, this->getComm()); | ||
|
|
||
| // Get additions configuration | ||
| int nhrs = fcstHours.size(); | ||
| std::vector<eckit::LocalConfiguration> additionsConfig; | ||
| eckit::LocalConfiguration additionsFromTemplateConfig(fullConfig, \ | ||
| "additions from template"); | ||
| eckit::LocalConfiguration templateConfig(additionsFromTemplateConfig, "template"); | ||
| std::string pattern; | ||
| additionsFromTemplateConfig.get("pattern", pattern); | ||
|
|
||
| for ( int ihrs = 0; ihrs < nhrs; ihrs++ ) { | ||
| eckit::LocalConfiguration thisAdditionsConfig(templateConfig); | ||
| util::seekAndReplace(thisAdditionsConfig, pattern, fcstHours[ihrs]); | ||
| additionsConfig.push_back(thisAdditionsConfig); | ||
| } | ||
|
|
||
| // Loops through forecast hours | ||
| for ( int ihrs = 0; ihrs < nhrs; ihrs++ ) { | ||
| // Get DateTime for forecast hour | ||
| const util::Duration fcstHour(3600*std::stoi(fcstHours[ihrs]) - 3*3600); | ||
| util::DateTime currentCycle = windowBegin + fcstHour; | ||
|
|
||
| // Get elements of individual additions configurations | ||
| const eckit::LocalConfiguration bkgConfig(additionsConfig[ihrs], "background"); | ||
| const eckit::LocalConfiguration incrConfig(additionsConfig[ihrs], "increment"); | ||
| const eckit::LocalConfiguration anlEnsMeanConfig(additionsConfig[ihrs], \ | ||
| "ensemble mean analysis"); | ||
| const eckit::LocalConfiguration incrCorConfig(additionsConfig[ihrs], \ | ||
| "correction increment"); | ||
|
|
||
| // Read background | ||
| fv3jedi::State xxBkg(bkgGeom, stateVars, currentCycle); | ||
| xxBkg.read(bkgConfig); | ||
|
|
||
| // Read increment | ||
| fv3jedi::Increment dx(incrGeom, incrVars, currentCycle); | ||
| dx.read(incrConfig); | ||
|
|
||
| // Read ensemble mean analysis | ||
| fv3jedi::State xxAnlEnsMean(anlEnsMeanGeom, incrVars, currentCycle); | ||
| xxAnlEnsMean.read(anlEnsMeanConfig); | ||
|
|
||
| // Compute deterministic analysis | ||
| fv3jedi::State xxAnlDet(bkgGeom, xxBkg); | ||
| xxAnlDet += dx; | ||
|
|
||
| // Interpolate full resolution analysis to ensemble resolution and then change variables | ||
| fv3jedi::State xxAnlDetEnsRes(incrCorGeom, fv3jedi::State(incrVars, xxAnlDet)); | ||
|
|
||
| // Compute correction increment | ||
| fv3jedi::Increment dxCor(incrCorGeom, incrVars, xxBkg.validTime()); | ||
| dxCor.diff(xxAnlEnsMean, xxAnlDetEnsRes); | ||
|
|
||
| // Write correction increment | ||
| dxCor.write(incrCorConfig); | ||
| } | ||
|
|
||
| return 0; | ||
| } | ||
|
|
||
| private: | ||
| std::string appname() const { | ||
| return "gdasapp::recenter"; | ||
| } | ||
| }; | ||
| } // namespace gdasapp |