Skip to content

Commit

Permalink
Expose additional RTE and RRTMGP functions (#11)
Browse files Browse the repository at this point in the history
* Expose additional RTE and RRTMGP functions

* Update submodule
  • Loading branch information
tcmetzger authored Mar 22, 2024
1 parent bc198c5 commit ad9bbab
Show file tree
Hide file tree
Showing 4 changed files with 1,474 additions and 65 deletions.
6 changes: 5 additions & 1 deletion CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,10 @@ add_custom_command(
)

# Compile C bindings
set(HEADERS ${CMAKE_SOURCE_DIR}/rte-rrtmgp/rte-kernels/api/rte_kernels.h)
set(HEADERS
${CMAKE_SOURCE_DIR}/rte-rrtmgp/rte-kernels/api/rte_kernels.h
${CMAKE_SOURCE_DIR}/rte-rrtmgp/rrtmgp-kernels/api/rrtmgp_kernels.h
)
set(SOURCES ${CMAKE_SOURCE_DIR}/pybind_interface.cpp)
set(TARGET_NAME pyrte)

Expand All @@ -38,6 +41,7 @@ add_dependencies(pyrte compile-rte-rrtmgp)

target_include_directories(${TARGET_NAME} PUBLIC
${CMAKE_SOURCE_DIR}/rte-rrtmgp/rte-kernels/api/
${CMAKE_SOURCE_DIR}/rte-rrtmgp/rrtmgp-kernels/api/
)

target_link_directories(${TARGET_NAME} PUBLIC
Expand Down
54 changes: 29 additions & 25 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -36,34 +36,34 @@ Currently, the following functions are available in the `pyrte` package:
| **OPTICAL PROPS - INCREMENT** | |
| `rte_increment_1scalar_by_1scalar` ||
| `rte_increment_1scalar_by_2stream` ||
| `rte_increment_1scalar_by_nstream` | 🔲 |
| `rte_increment_2stream_by_1scalar` | 🔲 |
| `rte_increment_2stream_by_2stream` | 🔲 |
| `rte_increment_2stream_by_nstream` | 🔲 |
| `rte_increment_nstream_by_1scalar` | 🔲 |
| `rte_increment_nstream_by_2stream` | 🔲 |
| `rte_increment_nstream_by_nstream` | 🔲 |
| `rte_increment_1scalar_by_nstream` | |
| `rte_increment_2stream_by_1scalar` | |
| `rte_increment_2stream_by_2stream` | |
| `rte_increment_2stream_by_nstream` | |
| `rte_increment_nstream_by_1scalar` | |
| `rte_increment_nstream_by_2stream` | |
| `rte_increment_nstream_by_nstream` | |
| **OPTICAL PROPS - INCREMENT BYBND** | |
| `rte_inc_1scalar_by_1scalar_bybnd` | 🔲 |
| `rte_inc_1scalar_by_2stream_bybnd` | 🔲 |
| `rte_inc_1scalar_by_1scalar_bybnd` | |
| `rte_inc_1scalar_by_2stream_bybnd` | |
| `rte_inc_1scalar_by_nstream_bybnd` | 🔲 |
| `rte_inc_2stream_by_1scalar_bybnd` | 🔲 |
| `rte_inc_2stream_by_2stream_bybnd` | 🔲 |
| `rte_inc_2stream_by_nstream_bybnd` | 🔲 |
| `rte_inc_nstream_by_1scalar_bybnd` | 🔲 |
| `rte_inc_nstream_by_2stream_bybnd` | 🔲 |
| `rte_inc_nstream_by_nstream_bybnd` | 🔲 |
| `rte_inc_2stream_by_1scalar_bybnd` | |
| `rte_inc_2stream_by_2stream_bybnd` | |
| `rte_inc_2stream_by_nstream_bybnd` | |
| `rte_inc_nstream_by_1scalar_bybnd` | |
| `rte_inc_nstream_by_2stream_bybnd` | |
| `rte_inc_nstream_by_nstream_bybnd` | |
| **OPTICAL PROPS - DELTA SCALING** | |
| `rte_delta_scale_2str_k` | 🔲 |
| `rte_delta_scale_2str_f_k` | 🔲 |
| `rte_delta_scale_2str_k` | |
| `rte_delta_scale_2str_f_k` | |
| **OPTICAL PROPS - SUBSET** | |
| `rte_extract_subset_dim1_3d` | 🔲 |
| `rte_extract_subset_dim2_4d` | 🔲 |
| `rte_extract_subset_absorption_tau` | 🔲 |
| `rte_extract_subset_dim1_3d` | |
| `rte_extract_subset_dim2_4d` | |
| `rte_extract_subset_absorption_tau` | |
| **Fluxes - Reduction** | |
| `rte_sum_broadband` | 🔲 |
| `rte_net_broadband_full` | 🔲 |
| `rte_net_broadband_precalc` | 🔲 |
| `rte_sum_broadband` | |
| `rte_net_broadband_full` | |
| `rte_net_broadband_precalc` | |
| `rte_sum_byband` | 🔲 |
| `rte_net_byband_full` | 🔲 |
| **Array Utilities** | |
Expand All @@ -74,8 +74,12 @@ Currently, the following functions are available in the `pyrte` package:

### RRTMGP Functions

RRTMGP functions are not yet available in the `pyrte` package.
Covering those functions is a future goal of this project.
| Function name | Covered |
|-----------------------------------------|:-------:|
| `rrtmgp_interpolation` ||
| `rrtmgp_compute_tau_absorption` ||
| `rrtmgp_compute_tau_rayleigh` ||
| `rrtmgp_compute_Planck_source` ||

## Setup Instructions

Expand Down
Loading

0 comments on commit ad9bbab

Please sign in to comment.