chn/ufo
1
0
Fork 0
mirror of https://github.com/CHN-beta/ufo.git synced 2024-10-22 19:58:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

姑且是能画出来差不多的图了

Browse Source
This commit is contained in:
陈浩南 2023-09-28 16:10:27 +08:00
parent 523a4b7e4c
commit 0256f6796b
1 changed files with 38 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
src/plot.cpp
View File

@ -83,6 +83,10 @@ int main(int argc, char** argv)
total_distance += (Qpoints[i + 1] - Qpoints[i]).norm();
}
// 打印结果看一下
for (auto& point : Points)
std::cout << point.Distance << " " << point.QPoint.transpose() << std::endl;
// 对结果插值
std::vector<Point> interpolated_points;
for (unsigned i = 0; i < 1024; i++)
@ -113,7 +117,7 @@ int main(int argc, char** argv)
}
// 将结果对应到像素上的值
std::vector<std::vector<double>> weight(400, std::vector<double>(1024, 0));
std::vector<std::vector<double>> weight(1024, std::vector<double>(400, 0));
for (auto& point : interpolated_points)
{
int x = point.Distance / total_distance * 1024;
@ -128,10 +132,40 @@ int main(int argc, char** argv)
y = 0;
else if (y >= 400)
y = 399;
weight[y][x] += point.Weight(i);
weight[x][y] += point.Weight(i);
}
}
matplot::image(weight);
matplot::show();
std::ofstream fout("weight.txt");
for (unsigned i = 0; i < 400; i++)
fout << fmt::format(" {:.6f}", i * 0.1 - 5);
fout << std::endl;
for (unsigned i = 0; i < 1024; i++)
{
fout << fmt::format("{:.6f} ", total_distance / 1024 * i);
for (unsigned j = 0; j < 400; j++)
fout << fmt::format("{:.6f} ", weight[i][j]);
fout << std::endl;
}
std::vector<std::vector<double>>
r(400, std::vector<double>(1024, 0)),
g(400, std::vector<double>(1024, 0)),
b(400, std::vector<double>(1024, 0));
for (unsigned i = 0; i < 400; i++)
for (unsigned j = 0; j < 1024; j++)
{
r[i][j] = 255;
g[i][j] = 255 - weight[j][i] * 2 * 255;
if (g[i][j] < 0)
g[i][j] = 0;
b[i][j] = 255 - weight[j][i] * 2 * 255;
if (b[i][j] < 0)
b[i][j] = 0;
}
auto f = matplot::figure(true);
auto ax = f->current_axes();
ax->image(std::tie(r, g, b));
ax->y_axis().reverse(false);
f->show();
}