【OpenFOAM/C++】OpenFOAM生成大量probes脚本

在OpenFOAM的probes脚本中设置大量的固定距离增加的监测点。
用C++编写脚本自动写probes。

/*
* 用于生成大量的Probes文件用
*/


#include <iostream>
#include <vector>
#include <fstream>
#include <iomanip>
#include <string>

using namespace std;

void GetY(double yPlus_Pre[], double nu, double u_tau, int size, double Lx, double Ly, double Lz);
void writeFile(double yPlus_Pre[], int size, vector<vector<double>> result);

int main() {

	/*
	* Physical parameter
	*/
	const double Lx = 6.4;
	const double Ly = 2;
	const double Lz = 3.2;
    const double nu    = 2e-5;
    const double u_tau = 0.0036;


	double yPlus_Pre[] = {1,5,11,14,30,41,60,90,100,120,150,180};
	int size = sizeof(yPlus_Pre) / sizeof(yPlus_Pre[0]);

	GetY(yPlus_Pre, 2e-5, 0.0036, size, Lx, Ly, Lz);
	
    ttttDebug();


}


void GetY(double yPlus_Pre[], double nu, double u_tau, int size , double Lx, double Ly, double Lz) {
	//double yValue[size];
	double* yValue = new double[size];

	for( int i = 0; i < size; ++i ){
		yValue[i] = yPlus_Pre[i]* nu / u_tau;
		cout << "yPlus==  "<< yPlus_Pre[i]<< " to physical length  " << yValue[i] << endl;
	}

	//文件输出模块
	int divide_num_z = 16;
	int divide_num_x = 16;

	double node_in_z = Lz/divide_num_z ;
	double node_in_x = Lx/divide_num_x ;
   vector<vector<double>> result; // 存储生成的向量
  // vector<vector<vector<double>>> result;

    /*
    * 在一定的y值处，在xz平面取16*16个点
    */

/*
* 核心就是在这里用让x z的值用二重循环按照固定值逐渐增加
*/
    double arr[] = { 0, yValue[0], 0 }; // 初始数组
     for(int j=0; j < divide_num_z; ++j){
        arr[0] += node_in_x;
         for (int i = 0; i < divide_num_z; i++) {
             arr[2] += node_in_z; // 在第二个元素上加上node_in_z
             vector<double> vec(arr, arr + 3); // 将三个元素添加到一个向量中
             result.push_back(vec); // 将向量添加到结果中
         }
          arr[2] = 0;//防止超过几何域
      }


    // 输出结果

    for (auto v : result) {
        for (auto d : v) {
            cout << d << " ";
        }
        cout << endl;
    }
    
    for (int k = 0; k < size; ++k) {
        writeFile(yPlus_Pre, size, result);
    }

}

void writeFile(double yPlus_Pre[], int size, vector<vector<double>> result)
{
    // 定义输出文件流
    for (int i = 0; i < size; ++i) {
        string out_file_name = "probes_" + std::to_string((int)yPlus_Pre[i]);
        ofstream outFile(out_file_name);

        //使用R可以进行免去使用\进行跨行
        outFile << R"(
/*--------------------------------*- C++ -*----------------------------------*\
  =========                 |
  \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   \\    /   O peration     | Website:  https://openfoam.org
    \\  /    A nd           | Version:  6
     \\/     M anipulation  |
-------------------------------------------------------------------------------
Description
    Writes out values of fields from cells nearest to specified locations.

\*---------------------------------------------------------------------------*/

type            probes;

libs            ("libsampling.so");

//writeControl    writeTime;

writeControl    timeStep;
writeInterval   1;

fields
(
    U 
    p 
); 
 )";


        outFile << "\n" << "probeLocations" << endl << "(" << endl;

        for (auto coord : result) {
            outFile << "    ( ";
            for (auto val : coord) {
                outFile << fixed << setprecision(4) << val << " ";
            }
            outFile << ")" << endl;
        }


        outFile << ");" << endl;

        // 关闭文件流
        outFile.close();
    }
}