429 lines
13 KiB
C++
429 lines
13 KiB
C++
/*
|
||
TetraGM4.0 四面体网络正演程序
|
||
2015/11/12
|
||
为避免之前版本中的冗余计算过程，在4.0版本中我们将对程序的数据结构做出调整（大改）。
|
||
通过模型对比实验，程序的计算效率有了质的提升。同时4.0版本更适合与并行计算的要求。调整后
|
||
的程序结构任未达到最优化状态，但进一步的优化需要获取更多的网络信息（每一条边的索引、每
|
||
一个面的索引）。准备从其他格式的网络文件（如.msh文件）中寻找突破口。
|
||
2015/11/13
|
||
今天在程序中加入了并行执行的命令，程序的运行效率有了进一步的提升。在i5，4G内存平台上
|
||
提升率约为原计算效率的两倍。相信在更高主频的平台能取得更好的成绩。
|
||
|
||
程序数据结构：调整后基本数据结构如下。
|
||
point3d（矢量结构）：矢量的坐标（x,y,z）
|
||
sphere_point（球面点/地理坐标系）：顶点的经纬值与球坐标半径
|
||
vertex（顶点结构）：顶点的索引号，点坐标
|
||
edge（边结构体）：边的编号，边的顶点索引，边的属性，边的长度，边的张量积E
|
||
face（面结构体）：面的编号，面的顶点索引，面的属性，面的张量积F
|
||
tetra（四面体结构体）：四面体的索引号、
|
||
四面体四个顶点的索引号、
|
||
重新排序后的四组三角形的索引号（记录排序后信息有利于后续的利用，避免函数的重复调用）
|
||
四个面的张量积F、
|
||
六条边的张量积E与边的长度（F、E与边长四面体是的固有属性，首先计算出来可以有效减少冗余计算）
|
||
四面体属性值
|
||
程序文件结构：Data_Func.h文件定义了数据类型与需要使用到的矢量运算函数
|
||
Commond_Line.h文件定义了程序中规定的命令行操作符
|
||
res_reg.h文件定义了计算结果的暂存与写入文件函数
|
||
Tetra_G.h文件定义了重力场正演类
|
||
Tetra_M.h文件定义了磁场正演类
|
||
main.cpp识别命令行指令，调取相应函数响应
|
||
*/
|
||
|
||
#include "Tetra_G.h"
|
||
#include "Commond_Line.h"
|
||
|
||
int caltype = NULL;//计算目标
|
||
int fietype = NULL;//观测点类型
|
||
int attritype = NULL;//元属性类型
|
||
int elementtype = NULL;//计算元类型
|
||
int space_num;//字符串中空格的个数
|
||
double attri[3];//元属性
|
||
double ob_info[7];//自定义观测面信息
|
||
string _attri;
|
||
string meshname;
|
||
string fiename;
|
||
string savename;
|
||
stringstream stemp;
|
||
|
||
void disp_info()
|
||
{
|
||
cout<<"tetragm 0.5.0 - forward modeling of gravity and magnetic data and their gradient data based on unstructured grid"<<endl<<endl
|
||
<<"usage: tetragm -gc|ggc|gs|ggs|mc|ms|mgc|mgs -a[<para1>/<para2>/<para3>/<para4>] -v<mesh-file>|s<mesh-file>"<<endl
|
||
<<" -F<fie-file>|M[<xmin>/<dx>/<xmax>/<ymin>/<dy>/<ymax>/<altitude>][<lati_min>/<d_lati>/<lati_max>/"<<endl
|
||
<<" <longi_min>/<d_longi>/<longi_max>/<radius>] -o<output-file>"<<endl
|
||
<<" -gc forward modeling gravity data in Cartesian coordinate"<<endl
|
||
<<" -ggc forward modeling gravity gradient data in Cartesian coordinate"<<endl
|
||
<<" -gs forward modeling gravity data in spherical coordinate"<<endl
|
||
<<" -ggs forward modeling gravity gradient data in spherical coordinate"<<endl
|
||
<<" -mc forward modeling magnetic data in Cartesian coordinate"<<endl
|
||
<<" -mgc forward modeling magnetic gradient data in Cartesian coordinate"<<endl
|
||
<<" -ms forward modeling magnetic data in spherical coordinate"<<endl
|
||
<<" -mgs forward modeling magnetic gradient data in spherical coordinate"<<endl
|
||
<<" -a define cells' physical attributes. If no arguments are followed behind -a, it means that the cells'"<<endl
|
||
<<" attributes are defined by the mesh-file individually. For forward modeling gravity and its gradient"<<endl
|
||
<<" data, density is the only parameter the program need. In this case, you just need to specify 'para1'."<<endl
|
||
<<" For example '1.0///'. For forward modeling magnetic and its gradient data. Parameters are as follow"<<endl
|
||
<<" para1 "<<endl
|
||
<<"-Vstring: use volume element, followed by mesh file's name"<<endl
|
||
<<"-Sstring: use surface element, followed by mesh file's name"<<endl
|
||
<<"-Fstring: read observation points from a .fie file, followed by file name"<<endl
|
||
<<"-Mdouble/double/double/double/double/double/double: user defined observation points, format:"<<endl
|
||
<<"xmin/dx/xmax/ymin/dy/ymax/altitude"<<endl
|
||
<<"lati_min/d_lati/lati_max/longi_min/d_longi/longi_max/radius"<<endl
|
||
<<"-Ostring: Output file, followed by file name"<<endl;
|
||
}
|
||
|
||
int typeget(char *str, const char* str2,string &str3)
|
||
{
|
||
char* strp = strstr(str, str2); // 从字符串str中查找str2，
|
||
if (strp == NULL)
|
||
{
|
||
return 0; // 如果没有找到，返回
|
||
}
|
||
string temp = str;
|
||
str3=temp.substr(strlen(str2));//提出除str2外的子句
|
||
return 1;
|
||
}
|
||
|
||
string& replace_all(string& str,const string& old_value,const string& new_value,int &num) //替换str中所有lod_value为new_value,返回被替换的lod_value的个数
|
||
{
|
||
int count = 0;
|
||
for(string::size_type pos(0);pos!=string::npos;pos+=new_value.length()){
|
||
if((pos=str.find(old_value,pos))!=string::npos){
|
||
str.replace(pos,old_value.length(),new_value);
|
||
count++;
|
||
}
|
||
else break;
|
||
}
|
||
num = count;
|
||
return str;
|
||
}
|
||
|
||
/*
|
||
caltype -GC:1 -GGC:2 -GS:3 -GGS:4 -MC:5 -MGC:6 -MS:7 -MGS:8
|
||
elementtype -V:1 -S:2
|
||
attritype -F:1 -M:2
|
||
*/
|
||
struct Call_func
|
||
{
|
||
int callback_func(int caltype,int elementtype,string filename,string fiename,string savename)
|
||
{
|
||
switch (caltype)
|
||
{
|
||
case 1:{
|
||
Tetra_G _Project;
|
||
if(_Project.info_taker_run(caltype,filename,fiename,savename)) return 1;
|
||
}break;
|
||
case 2:{
|
||
Tetra_G _Project;
|
||
if(_Project.info_taker_run(caltype,filename,fiename,savename)) return 1;
|
||
}break;
|
||
case 3:{
|
||
Tetra_G _Project;
|
||
if(_Project.info_taker_run(caltype,filename,fiename,savename)) return 1;
|
||
}break;
|
||
case 4:{
|
||
Tetra_G _Project;
|
||
if(_Project.info_taker_run(caltype,filename,fiename,savename)) return 1;
|
||
}break;
|
||
case 5:{
|
||
cout<<"Unfished!"<<endl;
|
||
}break;
|
||
case 6:{
|
||
cout<<"Unfished!"<<endl;
|
||
}break;
|
||
case 7:{
|
||
cout<<"Unfished!"<<endl;
|
||
}break;
|
||
case 8:{
|
||
cout<<"Unfished!"<<endl;
|
||
}break;
|
||
default:{
|
||
cout<<"An error picked at 'callback_func()', program stopped...";
|
||
exit(0);
|
||
}break;
|
||
}
|
||
return 0;
|
||
}
|
||
|
||
int callback_func(int caltype,int elementtype,string filename,double* ob_info,string savename)
|
||
{
|
||
switch (caltype)
|
||
{
|
||
case 1:{
|
||
Tetra_G _Project;
|
||
if(_Project.info_taker_run(caltype,filename,ob_info,savename)) return 1;
|
||
}break;
|
||
case 2:{
|
||
Tetra_G _Project;
|
||
if(_Project.info_taker_run(caltype,filename,ob_info,savename)) return 1;
|
||
}break;
|
||
case 3:{
|
||
Tetra_G _Project;
|
||
if(_Project.info_taker_run(caltype,filename,ob_info,savename)) return 1;
|
||
}break;
|
||
case 4:{
|
||
Tetra_G _Project;
|
||
if(_Project.info_taker_run(caltype,filename,ob_info,savename)) return 1;
|
||
}break;
|
||
case 5:{
|
||
cout<<"Unfished!"<<endl;
|
||
}break;
|
||
case 6:{
|
||
cout<<"Unfished!"<<endl;
|
||
}break;
|
||
case 7:{
|
||
cout<<"Unfished!"<<endl;
|
||
}break;
|
||
case 8:{
|
||
cout<<"Unfished!"<<endl;
|
||
}break;
|
||
default:{
|
||
cout<<"An error picked at 'callback_func()', program stopped...";
|
||
exit(0);
|
||
}break;
|
||
}
|
||
return 0;
|
||
}
|
||
|
||
int callback_func(int caltype,int elementtype,string filename,string fiename,string savename,double* attri)
|
||
{
|
||
switch (caltype)
|
||
{
|
||
case 1:{
|
||
Tetra_G _Project;
|
||
if(_Project.info_taker_run(caltype,filename,fiename,savename,attri)) return 1;
|
||
}break;
|
||
case 2:{
|
||
Tetra_G _Project;
|
||
if(_Project.info_taker_run(caltype,filename,fiename,savename,attri)) return 1;
|
||
}break;
|
||
case 3:{
|
||
Tetra_G _Project;
|
||
if(_Project.info_taker_run(caltype,filename,fiename,savename,attri)) return 1;
|
||
}break;
|
||
case 4:{
|
||
Tetra_G _Project;
|
||
if(_Project.info_taker_run(caltype,filename,fiename,savename,attri)) return 1;
|
||
}break;
|
||
case 5:{
|
||
cout<<"Unfished!"<<endl;
|
||
}break;
|
||
case 6:{
|
||
cout<<"Unfished!"<<endl;
|
||
}break;
|
||
case 7:{
|
||
cout<<"Unfished!"<<endl;
|
||
}break;
|
||
case 8:{
|
||
cout<<"Unfished!"<<endl;
|
||
}break;
|
||
default:{
|
||
cout<<"An error picked at 'callback_func()', program stopped...";
|
||
exit(0);
|
||
}break;
|
||
}
|
||
return 0;
|
||
}
|
||
|
||
int callback_func(int caltype,int elementtype,string filename,double* ob_info,string savename,double* attri)
|
||
{
|
||
switch (caltype)
|
||
{
|
||
case 1:{
|
||
Tetra_G _Project;
|
||
if(_Project.info_taker_run(caltype,filename,ob_info,savename,attri)) return 1;
|
||
}break;
|
||
case 2:{
|
||
Tetra_G _Project;
|
||
if(_Project.info_taker_run(caltype,filename,ob_info,savename,attri)) return 1;
|
||
}break;
|
||
case 3:{
|
||
Tetra_G _Project;
|
||
if(_Project.info_taker_run(caltype,filename,ob_info,savename,attri)) return 1;
|
||
}break;
|
||
case 4:{
|
||
Tetra_G _Project;
|
||
if(_Project.info_taker_run(caltype,filename,ob_info,savename,attri)) return 1;
|
||
}break;
|
||
case 5:{
|
||
cout<<"Unfished!"<<endl;
|
||
}break;
|
||
case 6:{
|
||
cout<<"Unfished!"<<endl;
|
||
}break;
|
||
case 7:{
|
||
cout<<"Unfished!"<<endl;
|
||
}break;
|
||
case 8:{
|
||
cout<<"Unfished!"<<endl;
|
||
}break;
|
||
default:{
|
||
cout<<"An error picked at 'callback_func()', program stopped...";
|
||
exit(0);
|
||
}break;
|
||
}
|
||
return 0;
|
||
}
|
||
};
|
||
|
||
int main(int argc,char* argv[])
|
||
{
|
||
if(argc==1){
|
||
disp_info();
|
||
}
|
||
else{
|
||
Call_func Call;
|
||
|
||
for(int i=1;i<=argc-1;i++){
|
||
//解析计算目标
|
||
if(!strcmp(argv[i],ForGC)) caltype=1;
|
||
else if(!strcmp(argv[i],ForGGC)) caltype=2;
|
||
else if(!strcmp(argv[i],ForGS)) caltype=3;
|
||
else if(!strcmp(argv[i],ForGGS)) caltype=4;
|
||
else if(!strcmp(argv[i],ForMC)) caltype=5;
|
||
else if(!strcmp(argv[i],ForMGC)) caltype=6;
|
||
else if(!strcmp(argv[i],ForMS)) caltype=7;
|
||
else if(!strcmp(argv[i],ForMGS)) caltype=8;
|
||
|
||
//解析元属性，需要滞后
|
||
if(typeget(argv[i],ATTRI,_attri)){
|
||
if(_attri!=""){
|
||
attritype=2;
|
||
replace_all(_attri,"/"," ",space_num);
|
||
stemp.clear();
|
||
stemp.str(_attri);
|
||
if(space_num==3){
|
||
for(int i=0;i<4;i++)
|
||
{
|
||
if(!(stemp>>attri[i])){
|
||
attri[i]=DBL_MAX;
|
||
}
|
||
}
|
||
}
|
||
else{
|
||
cout<<"Insufficient attributes information, program stoped...";//自定义观测面信息错误或不完整，程序停止
|
||
system("pause");
|
||
exit(0);
|
||
}
|
||
|
||
}
|
||
else attritype=1;//若无给定元信息
|
||
}
|
||
|
||
//解析元类型与网络文件名
|
||
if(typeget(argv[i],VolumeElement,meshname)){
|
||
elementtype = 1;
|
||
if(meshname==""){
|
||
cout<<"No mesh file name, program stoped...";//无网络文件名，程序停止
|
||
system("pause");
|
||
exit(0);
|
||
}
|
||
}
|
||
else if(typeget(argv[i],SurfaceElement,meshname)){
|
||
elementtype = 2;
|
||
if(meshname==""){
|
||
cout<<"No mesh file name, program stoped...";//无网络文件名，程序停止
|
||
system("pause");
|
||
exit(0);
|
||
}
|
||
}
|
||
|
||
//解析观测信息
|
||
if(typeget(argv[i],FILE,fiename)){
|
||
fietype=1;
|
||
if(fiename==""){
|
||
cout<<"No observation points' information, program stoped...";//无观测点文件，程序停止
|
||
system("pause");
|
||
exit(0);
|
||
}
|
||
}
|
||
else if(typeget(argv[i],MANUAL,fiename)){
|
||
fietype=2;
|
||
if(fiename==""){
|
||
cout<<"No observation points' information, program stoped...";//无自定义观测面信息，程序停止
|
||
system("pause");
|
||
exit(0);
|
||
}
|
||
else{
|
||
replace_all(fiename,"/"," ",space_num);
|
||
stemp.clear();
|
||
stemp.str(fiename);
|
||
if(space_num==6){
|
||
for(int i=0;i<7;i++)
|
||
{
|
||
if(!(stemp>>ob_info[i])){
|
||
ob_info[i]=DBL_MAX;
|
||
}
|
||
}
|
||
for(int i=0;i<7;i++)
|
||
{
|
||
if(ob_info[i]==DBL_MAX){
|
||
cout<<"Wrong observation points' information, program stoped...";//自定义观测面信息错误或不完整，程序停止
|
||
system("pause");
|
||
exit(0);
|
||
}
|
||
}
|
||
if(ob_info[1]<0||ob_info[4]<0||ob_info[0]>ob_info[2]||ob_info[3]>ob_info[5]||(ob_info[0]+ob_info[1])>ob_info[2]||(ob_info[3]+ob_info[4])>ob_info[5]){
|
||
cout<<"Wrong observation points' information, program stoped...";//自定义观测面信息错误或不完整，程序停止
|
||
system("pause");
|
||
exit(0);
|
||
}
|
||
}
|
||
else{
|
||
cout<<"Wrong observation points' information, program stoped...";//自定义观测面信息错误或不完整，程序停止
|
||
system("pause");
|
||
exit(0);
|
||
}
|
||
|
||
}
|
||
}
|
||
|
||
//解析输出文件名
|
||
if(typeget(argv[i],OUTFILE,savename)){
|
||
if(savename==""){
|
||
cout<<"No output file name, program stoped...";//无输出文件名，程序停止
|
||
system("pause");
|
||
exit(0);
|
||
}
|
||
}
|
||
}
|
||
|
||
if(caltype==NULL||fietype==NULL||attritype==NULL||elementtype==NULL||savename==""){
|
||
cout<<"Wrong commond line, program stoped...";
|
||
system("pause");
|
||
exit(0);
|
||
}
|
||
|
||
if(fietype==1&&attritype==1){
|
||
if(Call.callback_func(caltype,elementtype,meshname,fiename,savename)){
|
||
system("pause");
|
||
exit(0);
|
||
}
|
||
}
|
||
else if(fietype==1&&attritype==2){
|
||
if(Call.callback_func(caltype,elementtype,meshname,fiename,savename,attri)){
|
||
system("pause");
|
||
exit(0);
|
||
}
|
||
}
|
||
else if(fietype==2&&attritype==1){
|
||
if(Call.callback_func(caltype,elementtype,meshname,ob_info,savename)){
|
||
system("pause");
|
||
exit(0);
|
||
}
|
||
}
|
||
else if(fietype==2&&attritype==2){
|
||
if(Call.callback_func(caltype,elementtype,meshname,ob_info,savename,attri)){
|
||
system("pause");
|
||
exit(0);
|
||
}
|
||
}
|
||
else{
|
||
cout<<"A program error is detected, program stopped...";
|
||
system("pause");
|
||
exit(0);
|
||
}
|
||
}
|
||
return 0;
|
||
} |