Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Generalize fextract allowing to specify coords perpendicular to slicedir #2035

Merged
merged 2 commits into from
Jun 1, 2021
Merged

Generalize fextract allowing to specify coords perpendicular to slicedir #2035

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
f5eb230
Generalize fextract to allow specifying coordinates perpendicular to …
drummerdoc May 18, 2021
edacd8b
Change to a large negative number rather than a small negative number
drummerdoc Jun 1, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
62 changes: 47 additions & 15 deletions Tools/Plotfile/fextract.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,9 @@ void main_main()
bool center = true;
int coarse_level = 0;
int fine_level = -1; // This will be fixed later
Real xcoord = std::numeric_limits<Real>::lowest();
Real ycoord = std::numeric_limits<Real>::lowest();
Real zcoord = std::numeric_limits<Real>::lowest();
bool scientific = false;
bool csv = false;
int precision = 17;
Expand All @@ -36,8 +38,12 @@ void main_main()
idir = std::stoi(amrex::get_command_argument(++farg));
} else if (name == "-v" || name == "--variable") {
varnames_arg = amrex::get_command_argument(++farg);
} else if (name == "-x") {
xcoord = std::stod(amrex::get_command_argument(++farg));
} else if (name == "-y") {
ycoord = std::stod(amrex::get_command_argument(++farg));
} else if (name == "-z") {
zcoord = std::stod(amrex::get_command_argument(++farg));
} else if (name == "-l" || name == "--lower_left") {
center = false;
} else if (name == "-c" || name == "--coarse_level") {
Expand Down Expand Up @@ -80,7 +86,7 @@ void main_main()
<< " multiple variables)\n"
<< " [-l|--lower_left] : slice through lower left corner\n"
<< " instead of center\n"
<< " [-y] : y-coordinate to pass through\n"
<< " [-x][-y][-z] : (x,y,z)-coordinate to pass through\n"
<< " (overrides center/lower-left)\n"
<< " [-c|--coarse_level] coarse level : coarsest level to extract from\n"
<< " [-f|--fine_level] fine level : finest level to extract from\n"
Expand All @@ -105,19 +111,6 @@ void main_main()
}

PlotFileData pf(pltfile);
const int dim = pf.spaceDim();

if (idir < 0 || idir >= dim) {
amrex::Print() << " invalid direction\n";
return;
} else if (idir == 0) {
amrex::Print() << " slicing along x-direction and output to " << slicefile << "\n";
} else if (idir == 1) {
amrex::Print() << " slicing along y-direction and output to " << slicefile << "\n";
} else if (idir == 2) {
amrex::Print() << " slicing along z-direction and output to " << slicefile << "\n";
}

const Vector<std::string>& var_names_pf = pf.varNames();

Vector<std::string> var_names;
Expand Down Expand Up @@ -155,8 +148,21 @@ void main_main()
kloc = (hi0.z-lo0.z+1)/2 + lo0.z;
}

if (ycoord > -1.e-36 && AMREX_SPACEDIM >= 2) {
if (xcoord > -1.e36 && AMREX_SPACEDIM >= 1) {
// we specified the x value to pass through
iloc = hi0.x;
for (int i = lo0.x; i <= hi0.x; ++i) {
amrex::Real xc = problo[0] + (i+0.5)*dx0[0];
if (xc > xcoord) {
iloc = i;
break;
}
}
}

if (ycoord > -1.e36 && AMREX_SPACEDIM >= 2) {
// we specified the y value to pass through
jloc = hi0.y;
for (int j = lo0.y; j <= hi0.y; ++j) {
amrex::Real yc = problo[1] + (j+0.5)*dx0[1];
if (yc > ycoord) {
Expand All @@ -166,6 +172,32 @@ void main_main()
}
}

if (zcoord > -1.e36 && AMREX_SPACEDIM == 3) {
// we specified the z value to pass through
kloc = hi0.z;
for (int k = lo0.z; k <= hi0.z; ++k) {
amrex::Real zc = problo[2] + (k+0.5)*dx0[2];
if (zc > zcoord) {
kloc = k;
break;
}
}
}

const int dim = pf.spaceDim();

if (idir < 0 || idir >= dim) {
amrex::Print() << " invalid direction\n";
return;
} else if (idir == 0) {
amrex::Print() << " slicing along x-direction at coarse grid (j,k)=(" << jloc << "," << kloc << ") and output to " << slicefile << "\n";
} else if (idir == 1) {
amrex::Print() << " slicing along y-direction at coarse grid (i,k)=(" << iloc << "," << kloc << ") and output to " << slicefile << "\n";
} else if (idir == 2) {
amrex::Print() << " slicing along z-direction at coarse grid (i,j)=(" << iloc << "," << jloc << ") and output to " << slicefile << "\n";
}


const IntVect ivloc{AMREX_D_DECL(iloc,jloc,kloc)};

if (fine_level < 0) fine_level = pf.finestLevel();
Expand Down