zhangyiss/TomoATT
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

initial upload

Browse Source
This commit is contained in:
张壹
2025-12-17 10:53:43 +08:00
commit f3f1778f77
308 changed files with 129940 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

98
include/iterator_selector.h Normal file
View File

@@ -0,0 +1,98 @@
#ifndef ITERATOR_SELECTOR_H
#define ITERATOR_SELECTOR_H
#include <memory>
#include "iterator.h"
#include "iterator_legacy.h"
#include "iterator_level.h"
void select_iterator(InputParams& IP, Grid& grid, Source& src, IO_utils& io, const std::string& src_name, \
bool first_init, bool is_teleseismic, std::unique_ptr<Iterator>& It, bool is_second_run) {
// initialize iterator object
if (!is_teleseismic){
if (IP.get_sweep_type() == SWEEP_TYPE_LEGACY) {
if (IP.get_stencil_order() == 1){
if (IP.get_stencil_type() == UPWIND){
It = std::make_unique<Iterator_legacy_1st_order_upwind>(IP, grid, src, io, src_name, first_init, is_teleseismic, is_second_run);
} else {
It = std::make_unique<Iterator_legacy_1st_order>(IP, grid, src, io, src_name, first_init, is_teleseismic, is_second_run);
}
} else if (IP.get_stencil_order() == 3){
It = std::make_unique<Iterator_legacy_3rd_order>(IP, grid, src, io, src_name, first_init, is_teleseismic, is_second_run);
if (IP.get_stencil_type() == UPWIND){
std::cout << "WARNING: Upwind Stencil type not supported, using 3rd order non upwind scheme (LF)" << std::endl;
}
} else{
std::cout << "ERROR: Stencil order not supported" << std::endl;
exit(1);
}
} else if (IP.get_sweep_type() == SWEEP_TYPE_LEVEL){
if (IP.get_stencil_order() == 1){
if (IP.get_stencil_type() == UPWIND){
// std::cout << "WARNING: Upwind Stencil type not supported, using non upwind scheme (LF)" << std::endl;
It = std::make_unique<Iterator_level_1st_order_upwind>(IP, grid, src, io, src_name, first_init, is_teleseismic, is_second_run);
} else {
It = std::make_unique<Iterator_level_1st_order>(IP, grid, src, io, src_name, first_init, is_teleseismic, is_second_run);
}
} else if (IP.get_stencil_order() == 3){
It = std::make_unique<Iterator_level_3rd_order>(IP, grid, src, io, src_name, first_init, is_teleseismic, is_second_run);
if (IP.get_stencil_type() == UPWIND){
std::cout << "WARNING: Upwind Stencil type not supported, using 3rd order non upwind scheme (LF)" << std::endl;
}
} else{
std::cout << "ERROR: Stencil order not supported" << std::endl;
exit(1);
}
} else {
std::cout << "ERROR: Sweep type not supported" << std::endl;
exit(1);
}
} else { // teleseismic event
if (IP.get_sweep_type() == SWEEP_TYPE_LEGACY) {
if (IP.get_stencil_order() == 1){
if (IP.get_stencil_type() == UPWIND){
It = std::make_unique<Iterator_legacy_1st_order_upwind_tele>(IP, grid, src, io, src_name, first_init, is_teleseismic, is_second_run);
} else {
It = std::make_unique<Iterator_legacy_1st_order_tele>(IP, grid, src, io, src_name, first_init, is_teleseismic, is_second_run);
}
} else if (IP.get_stencil_order() == 3){
It = std::make_unique<Iterator_legacy_3rd_order_tele>(IP, grid, src, io, src_name, first_init, is_teleseismic, is_second_run);
if (IP.get_stencil_type() == UPWIND){
std::cout << "WARNING: Upwind Stencil type not supported, using non upwind scheme (LF)" << std::endl;
}
} else{
std::cout << "ERROR: Stencil order not supported" << std::endl;
exit(1);
}
} else if (IP.get_sweep_type() == SWEEP_TYPE_LEVEL){
if (IP.get_stencil_order() == 1){
if (IP.get_stencil_type() == UPWIND){
It = std::make_unique<Iterator_level_1st_order_upwind_tele>(IP, grid, src, io, src_name, first_init, is_teleseismic, is_second_run);
} else {
It = std::make_unique<Iterator_level_1st_order_tele>(IP, grid, src, io, src_name, first_init, is_teleseismic, is_second_run);
std::cout << "WARNING: SWEEP_TYPE_LEVEL for non-unwind solver does not work. Index wrong currently" << std::endl;
exit(1);
}
} else if (IP.get_stencil_order() == 3){
It = std::make_unique<Iterator_level_3rd_order_tele>(IP, grid, src, io, src_name, first_init, is_teleseismic, is_second_run);
std::cout << "WARNING: SWEEP_TYPE_LEVEL for non-unwind solver does not work. Index wrong currently" << std::endl;
exit(1);
if (IP.get_stencil_type() == UPWIND){
std::cout << "WARNING: Upwind Stencil type not supported, using non upwind scheme (LF)" << std::endl;
}
} else{
std::cout << "ERROR: Stencil order not supported" << std::endl;
exit(1);
}
} else {
std::cout << "ERROR: Sweep type not supported" << std::endl;
exit(1);
}
}
}
#endif