initial upload
This commit is contained in:
273
archive/linearmap/linearmap.h
Normal file
273
archive/linearmap/linearmap.h
Normal file
@@ -0,0 +1,273 @@
|
||||
#include "iostream"
|
||||
#include "fstream"
|
||||
#include "iomanip"
|
||||
#include "stdio.h"
|
||||
#include "stdlib.h"
|
||||
#include "unistd.h"
|
||||
#include "cmath"
|
||||
#include "string.h"
|
||||
#include "sstream"
|
||||
#include "vector"
|
||||
#include "string.h"
|
||||
|
||||
#define pi (atan(1.0)*4.0)
|
||||
#define BOLDRED "\033[1m\033[31m"
|
||||
#define RESET "\033[0m"
|
||||
|
||||
using namespace std;
|
||||
|
||||
struct dpoint
|
||||
{
|
||||
double x,y,val;
|
||||
dpoint()
|
||||
{
|
||||
x = y = val = 0.0;
|
||||
}
|
||||
};
|
||||
typedef vector<dpoint> _dpointArray;
|
||||
|
||||
typedef vector<double> _1dArray;
|
||||
|
||||
int open_infile(ifstream &infile,char* filename)
|
||||
{
|
||||
infile.open(filename);
|
||||
if (!infile)
|
||||
{
|
||||
cout << BOLDRED << "error ==> " << RESET << "file not found: " << filename << endl;
|
||||
return -1;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
int open_outfile(ofstream &outfile,char* filename)
|
||||
{
|
||||
outfile.open(filename);
|
||||
if (!outfile)
|
||||
{
|
||||
cout << BOLDRED << "error ==> " << RESET << "fail to create the file: " << filename << endl;
|
||||
return -1;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
class linearmap
|
||||
{
|
||||
public:
|
||||
linearmap(){}
|
||||
~linearmap(){}
|
||||
_dpointArray readxyz(char*,char*,char*,char*);
|
||||
int outRes(char*);
|
||||
int thetaMap(char*,char*,char*,char*,char*,char*);
|
||||
int THDR(char*,char*,char*,char*,char*,char*);
|
||||
int ASA(char*,char*,char*,char*,char*,char*,char*);
|
||||
int TAHG(char*,char*,char*,char*,char*,char*,char*);
|
||||
private:
|
||||
_dpointArray gradx;
|
||||
_dpointArray grady;
|
||||
_dpointArray gradz;
|
||||
_dpointArray res;
|
||||
};
|
||||
|
||||
_dpointArray linearmap::readxyz(char* filename,char* range,char* interval,char* order)
|
||||
{
|
||||
_dpointArray input_data;
|
||||
_1dArray tempRow;
|
||||
double x,y,data,temp_d;
|
||||
double xmin,xmax,ymin,ymax;
|
||||
double dx,dy;
|
||||
int xnum,ynum;
|
||||
int orders[3];
|
||||
int data_index;
|
||||
string temp_str;
|
||||
stringstream temp_ss;
|
||||
|
||||
//解析数据范围
|
||||
if (4 != sscanf(range,"%lf/%lf/%lf/%lf",&xmin,&xmax,&ymin,&ymax))
|
||||
{
|
||||
cout << BOLDRED << "error ==> " << RESET << "wrong range parameters for the file: " << filename << endl;
|
||||
return input_data;
|
||||
}
|
||||
else if (2 != sscanf(interval,"%lf/%lf",&dx,&dy))
|
||||
{
|
||||
cout << BOLDRED << "error ==> " << RESET << "wrong interval parameters for the file: " << filename << endl;
|
||||
return input_data;
|
||||
}
|
||||
else if (3 != sscanf(order,"%d,%d,%d",&orders[0],&orders[1],&orders[2]))
|
||||
{
|
||||
cout << BOLDRED << "error ==> " << RESET << "wrong order parameters for the file: " << filename << endl;
|
||||
return input_data;
|
||||
}
|
||||
|
||||
//打开文件
|
||||
ifstream datain;
|
||||
if (open_infile(datain,filename)) return input_data;
|
||||
|
||||
//设置input_data大小
|
||||
xnum = round((xmax-xmin)/dx)+1;
|
||||
ynum = round((ymax-ymin)/dy)+1;
|
||||
input_data.resize(xnum*ynum);
|
||||
|
||||
//读入数据
|
||||
while (getline(datain,temp_str))
|
||||
{
|
||||
if (*(temp_str.begin()) == '#') continue;
|
||||
else
|
||||
{
|
||||
//读入行
|
||||
temp_ss.str("");
|
||||
temp_ss.clear();
|
||||
temp_ss.str(temp_str);
|
||||
//解析数据行
|
||||
if(!tempRow.empty()) tempRow.clear();
|
||||
while (temp_ss >> temp_d)
|
||||
{
|
||||
tempRow.push_back(temp_d);
|
||||
}
|
||||
//读入指定位置的数据
|
||||
x = tempRow.at(orders[0]);
|
||||
y = tempRow.at(orders[1]);
|
||||
data = tempRow.at(orders[2]);
|
||||
//存入input_data 从左下到右上
|
||||
data_index = round((x-xmin)/dx) + round((y-ymin)/dy)*xnum;
|
||||
input_data.at(data_index).x = x;
|
||||
input_data.at(data_index).y = y;
|
||||
input_data.at(data_index).val = data;
|
||||
}
|
||||
}
|
||||
|
||||
datain.close();
|
||||
return input_data;
|
||||
}
|
||||
|
||||
int linearmap::outRes(char* filename)
|
||||
{
|
||||
ofstream outfile;
|
||||
if (open_outfile(outfile,filename)) return -1;
|
||||
|
||||
for (int i = 0; i < res.size(); i++)
|
||||
{
|
||||
outfile << res.at(i).x << " " << res.at(i).y << " " << setprecision(16) << res.at(i).val << endl;
|
||||
}
|
||||
outfile.close();
|
||||
return 0;
|
||||
}
|
||||
|
||||
int linearmap::thetaMap(char* xfile,char* zfile,char* ofile,char* ran,char* inter,char* col)
|
||||
{
|
||||
gradx = readxyz(xfile,ran,inter,col);
|
||||
if (gradx.empty()) return -1;
|
||||
|
||||
gradz = readxyz(zfile,ran,inter,col);
|
||||
if (gradz.empty()) return -1;
|
||||
|
||||
if (gradx.size() != gradz.size())
|
||||
{
|
||||
cout << BOLDRED << "error ==> " << RESET << "data amount does not match for thetaMap" << endl;
|
||||
return -1;
|
||||
}
|
||||
|
||||
res.resize(gradx.size());
|
||||
for (int i = 0; i < res.size(); i++)
|
||||
{
|
||||
res.at(i).x = gradx.at(i).x;
|
||||
res.at(i).y = gradx.at(i).y;
|
||||
if (fabs(gradx.at(i).val) < 1e-20 && gradz.at(i).val > 0)
|
||||
{
|
||||
res.at(i).val = 0.5*pi;
|
||||
}
|
||||
else if (fabs(gradx.at(i).val) < 1e-20 && gradz.at(i).val < 0)
|
||||
{
|
||||
res.at(i).val = -0.5*pi;
|
||||
}
|
||||
else res.at(i).val = atan(gradz.at(i).val/gradx.at(i).val);
|
||||
//转换为度
|
||||
res.at(i).val = res.at(i).val*180.0/pi;
|
||||
}
|
||||
|
||||
if (outRes(ofile)) return -1;
|
||||
return 0;
|
||||
}
|
||||
|
||||
int linearmap::THDR(char* xfile,char* yfile,char* ofile,char* ran,char* inter,char* col)
|
||||
{
|
||||
gradx = readxyz(xfile,ran,inter,col);
|
||||
if (gradx.empty()) return -1;
|
||||
|
||||
grady = readxyz(yfile,ran,inter,col);
|
||||
if (grady.empty()) return -1;
|
||||
|
||||
if (gradx.size() != grady.size())
|
||||
{
|
||||
cout << BOLDRED << "error ==> " << RESET << "data amount does not match for thetaMap" << endl;
|
||||
return -1;
|
||||
}
|
||||
|
||||
res.resize(gradx.size());
|
||||
for (int i = 0; i < res.size(); i++)
|
||||
{
|
||||
res.at(i).x = gradx.at(i).x;
|
||||
res.at(i).y = gradx.at(i).y;
|
||||
res.at(i).val = sqrt(grady.at(i).val*grady.at(i).val + gradx.at(i).val*gradx.at(i).val);
|
||||
}
|
||||
|
||||
if (outRes(ofile)) return -1;
|
||||
return 0;
|
||||
}
|
||||
|
||||
int linearmap::ASA(char* xfile,char* yfile,char* zfile,char* ofile,char* ran,char* inter,char* col)
|
||||
{
|
||||
gradx = readxyz(xfile,ran,inter,col);
|
||||
if (gradx.empty()) return -1;
|
||||
|
||||
grady = readxyz(yfile,ran,inter,col);
|
||||
if (grady.empty()) return -1;
|
||||
|
||||
gradz = readxyz(zfile,ran,inter,col);
|
||||
if (gradz.empty()) return -1;
|
||||
|
||||
if (gradx.size() != grady.size() || gradx.size() != gradz.size() || grady.size() != gradz.size())
|
||||
{
|
||||
cout << BOLDRED << "error ==> " << RESET << "data amount does not match for thetaMap" << endl;
|
||||
return -1;
|
||||
}
|
||||
|
||||
res.resize(gradx.size());
|
||||
for (int i = 0; i < res.size(); i++)
|
||||
{
|
||||
res.at(i).x = gradx.at(i).x;
|
||||
res.at(i).y = gradx.at(i).y;
|
||||
res.at(i).val = sqrt(grady.at(i).val*grady.at(i).val + gradx.at(i).val*gradx.at(i).val + gradz.at(i).val*gradz.at(i).val);
|
||||
}
|
||||
|
||||
if (outRes(ofile)) return -1;
|
||||
return 0;
|
||||
}
|
||||
|
||||
int linearmap::TAHG(char* xfile,char* yfile,char* zfile,char* ofile,char* ran,char* inter,char* col)
|
||||
{
|
||||
gradx = readxyz(xfile,ran,inter,col);
|
||||
if (gradx.empty()) return -1;
|
||||
|
||||
grady = readxyz(yfile,ran,inter,col);
|
||||
if (grady.empty()) return -1;
|
||||
|
||||
gradz = readxyz(zfile,ran,inter,col);
|
||||
if (gradz.empty()) return -1;
|
||||
|
||||
if (gradx.size() != grady.size() || gradx.size() != gradz.size() || grady.size() != gradz.size())
|
||||
{
|
||||
cout << BOLDRED << "error ==> " << RESET << "data amount does not match for thetaMap" << endl;
|
||||
return -1;
|
||||
}
|
||||
|
||||
res.resize(gradx.size());
|
||||
for (int i = 0; i < res.size(); i++)
|
||||
{
|
||||
res.at(i).x = gradx.at(i).x;
|
||||
res.at(i).y = gradx.at(i).y;
|
||||
res.at(i).val = sqrt(gradz.at(i).val/sqrt(grady.at(i).val*grady.at(i).val + gradx.at(i).val*gradx.at(i).val));
|
||||
}
|
||||
|
||||
if (outRes(ofile)) return -1;
|
||||
return 0;
|
||||
}
|
||||
Reference in New Issue
Block a user