void PoissonGaussianSolution::setBcCoefs( const boost::shared_ptr<pdat::ArrayData<double> >& acoef_data, const boost::shared_ptr<pdat::ArrayData<double> >& bcoef_data, const boost::shared_ptr<pdat::ArrayData<double> >& gcoef_data, const boost::shared_ptr<hier::Variable>& variable, const hier::Patch& patch, const hier::BoundaryBox& bdry_box, const double fill_time) const { NULL_USE(variable); NULL_USE(fill_time); boost::shared_ptr<geom::CartesianPatchGeometry> patch_geom( BOOST_CAST<geom::CartesianPatchGeometry, hier::PatchGeometry>( patch.getPatchGeometry())); TBOX_ASSERT(patch_geom); /* * Set to an inhomogeneous Dirichlet boundary condition. */ hier::Box patch_box(patch.getBox()); const double* xlo = patch_geom->getXLower(); const double* xup = patch_geom->getXUpper(); const double* dx = patch_geom->getDx(); if (bdry_box.getBoundaryType() != 1) { // Must be a face boundary. TBOX_ERROR("Bad boundary type in\n" << "PoissonGaussianSolution::setBcCoefs \n"); } const hier::Box& box = bdry_box.getBox(); hier::Index lower = box.lower(); hier::Index upper = box.upper(); if (d_dim == tbox::Dimension(2)) { double* a_array = acoef_data ? acoef_data->getPointer() : 0; double* b_array = bcoef_data ? bcoef_data->getPointer() : 0; double* g_array = gcoef_data ? gcoef_data->getPointer() : 0; int i, j, ibeg, iend, jbeg, jend; double x, y; switch (bdry_box.getLocationIndex()) { case 0: // min i edge jbeg = box.lower()[1]; jend = box.upper()[1]; x = xlo[0]; for (j = jbeg; j <= jend; ++j) { y = xlo[1] + dx[1] * (j - patch_box.lower()[1] + 0.5); if (a_array) a_array[j - jbeg] = 1.0; if (b_array) b_array[j - jbeg] = 0.0; if (g_array) g_array[j - jbeg] = exactFcn(x, y); } break; case 1: // max i edge jbeg = box.lower()[1]; jend = box.upper()[1]; x = xup[0]; for (j = jbeg; j <= jend; ++j) { y = xlo[1] + dx[1] * (j - patch_box.lower()[1] + 0.5); if (a_array) a_array[j - jbeg] = 1.0; if (b_array) b_array[j - jbeg] = 0.0; if (g_array) g_array[j - jbeg] = exactFcn(x, y); } break; case 2: // min j edge ibeg = box.lower()[0]; iend = box.upper()[0]; y = xlo[1]; for (i = ibeg; i <= iend; ++i) { x = xlo[0] + dx[0] * (i - patch_box.lower()[0] + 0.5); if (a_array) a_array[i - ibeg] = 1.0; if (b_array) b_array[i - ibeg] = 0.0; if (g_array) g_array[i - ibeg] = exactFcn(x, y); } break; case 3: // max j edge ibeg = box.lower()[0]; iend = box.upper()[0]; y = xup[1]; for (i = ibeg; i <= iend; ++i) { x = xlo[0] + dx[0] * (i - patch_box.lower()[0] + 0.5); if (a_array) a_array[i - ibeg] = 1.0; if (b_array) b_array[i - ibeg] = 0.0; if (g_array) g_array[i - ibeg] = exactFcn(x, y); } break; default: TBOX_ERROR("Invalid location index in\n" << "PoissonGaussianSolution::setBcCoefs"); } } if (d_dim == tbox::Dimension(3)) { MDA_Access<double, 3, MDA_OrderColMajor<3> > a_array, b_array, g_array; if (acoef_data) a_array = pdat::ArrayDataAccess::access<3, double>( *acoef_data); if (bcoef_data) b_array = pdat::ArrayDataAccess::access<3, double>( *bcoef_data); if (gcoef_data) g_array = pdat::ArrayDataAccess::access<3, double>( *gcoef_data); int i, j, k, ibeg, iend, jbeg, jend, kbeg, kend; double x, y, z; switch (bdry_box.getLocationIndex()) { case 0: // min i side jbeg = box.lower()[1]; jend = box.upper()[1]; kbeg = box.lower()[2]; kend = box.upper()[2]; i = box.lower()[0] + 1; x = xlo[0]; for (k = kbeg; k <= kend; ++k) { z = xlo[2] + dx[2] * (k - patch_box.lower()[2] + 0.5); for (j = jbeg; j <= jend; ++j) { y = xlo[1] + dx[1] * (j - patch_box.lower()[1] + 0.5); if (a_array) a_array(i, j, k) = 1.0; if (b_array) b_array(i, j, k) = 0.0; if (g_array) g_array(i, j, k) = exactFcn(x, y, z); } } break; case 1: // max i side jbeg = box.lower()[1]; jend = box.upper()[1]; kbeg = box.lower()[2]; kend = box.upper()[2]; i = box.upper()[0]; x = xup[0]; for (k = kbeg; k <= kend; ++k) { z = xlo[2] + dx[2] * (k - patch_box.lower()[2] + 0.5); for (j = jbeg; j <= jend; ++j) { y = xlo[1] + dx[1] * (j - patch_box.lower()[1] + 0.5); if (a_array) a_array(i, j, k) = 1.0; if (b_array) b_array(i, j, k) = 0.0; if (g_array) g_array(i, j, k) = exactFcn(x, y, z); } } break; case 2: // min j side ibeg = box.lower()[0]; iend = box.upper()[0]; kbeg = box.lower()[2]; kend = box.upper()[2]; j = box.lower()[1] + 1; y = xlo[1]; for (k = kbeg; k <= kend; ++k) { z = xlo[2] + dx[2] * (k - patch_box.lower()[2] + 0.5); for (i = ibeg; i <= iend; ++i) { x = xlo[0] + dx[0] * (i - patch_box.lower()[0] + 0.5); if (a_array) a_array(i, j, k) = 1.0; if (b_array) b_array(i, j, k) = 0.0; if (g_array) g_array(i, j, k) = exactFcn(x, y, z); } } break; case 3: // max j side ibeg = box.lower()[0]; iend = box.upper()[0]; kbeg = box.lower()[2]; kend = box.upper()[2]; j = box.upper()[1]; y = xup[1]; for (k = kbeg; k <= kend; ++k) { z = xlo[2] + dx[2] * (k - patch_box.lower()[2] + 0.5); for (i = ibeg; i <= iend; ++i) { x = xlo[0] + dx[0] * (i - patch_box.lower()[0] + 0.5); if (a_array) a_array(i, j, k) = 1.0; if (b_array) b_array(i, j, k) = 0.0; if (g_array) g_array(i, j, k) = exactFcn(x, y, z); } } break; case 4: // min k side ibeg = box.lower()[0]; iend = box.upper()[0]; jbeg = box.lower()[1]; jend = box.upper()[1]; k = box.lower()[2] + 1; z = xlo[2]; for (j = jbeg; j <= jend; ++j) { y = xlo[1] + dx[1] * (j - patch_box.lower()[1] + 0.5); for (i = ibeg; i <= iend; ++i) { x = xlo[0] + dx[0] * (i - patch_box.lower()[0] + 0.5); if (a_array) a_array(i, j, k) = 1.0; if (b_array) b_array(i, j, k) = 0.0; if (g_array) g_array(i, j, k) = exactFcn(x, y, z); } } break; case 5: // max k side ibeg = box.lower()[0]; iend = box.upper()[0]; jbeg = box.lower()[1]; jend = box.upper()[1]; k = box.upper()[2]; z = xup[2]; for (j = jbeg; j <= jend; ++j) { y = xlo[1] + dx[1] * (j - patch_box.lower()[1] + 0.5); for (i = ibeg; i <= iend; ++i) { x = xlo[0] + dx[0] * (i - patch_box.lower()[0] + 0.5); if (a_array) a_array(i, j, k) = 1.0; if (b_array) b_array(i, j, k) = 0.0; if (g_array) g_array(i, j, k) = exactFcn(x, y, z); } } break; default: TBOX_ERROR("Invalid location index in\n" << "PoissonGaussianSolution::setBcCoefs"); } } }
int SkeletonBoundaryUtilities2::checkBdryData( const string& varname, const hier::Patch& patch, int data_id, int depth, const hier::IntVector& gcw_to_check, const hier::BoundaryBox& bbox, int bcase, double bstate) { TBOX_ASSERT(!varname.empty()); TBOX_ASSERT(data_id >= 0); TBOX_ASSERT(depth >= 0); int num_bad_values = 0; int btype = bbox.getBoundaryType(); int bloc = bbox.getLocationIndex(); std::shared_ptr<hier::PatchGeometry> pgeom(patch.getPatchGeometry()); std::shared_ptr<pdat::CellData<double> > vardata( SAMRAI_SHARED_PTR_CAST<pdat::CellData<double>, hier::PatchData>( patch.getPatchData(data_id))); TBOX_ASSERT(vardata); string bdry_type_str; if (btype == Bdry::EDGE2D) { bdry_type_str = "EDGE"; } else if (btype == Bdry::NODE2D) { bdry_type_str = "NODE"; } else { TBOX_ERROR( "Unknown btype " << btype << " passed to SkeletonBoundaryUtilities2::checkBdryData()! " << endl); } tbox::plog << "\n\nCHECKING 2D " << bdry_type_str << " BDRY DATA..." << endl; tbox::plog << "varname = " << varname << " : depth = " << depth << endl; tbox::plog << "bbox = " << bbox.getBox() << endl; tbox::plog << "btype, bloc, bcase = " << btype << ", = " << bloc << ", = " << bcase << endl; tbox::Dimension::dir_t idir; double valfact = 0.0, constval = 0.0, dxfact = 0.0; int offsign; get2dBdryDirectionCheckValues(idir, offsign, btype, bloc, bcase); if (btype == Bdry::EDGE2D) { if (bcase == BdryCond::FLOW) { valfact = 1.0; constval = 0.0; dxfact = 0.0; } else if (bcase == BdryCond::REFLECT) { valfact = -1.0; constval = 0.0; dxfact = 0.0; } else if (bcase == BdryCond::DIRICHLET) { valfact = 0.0; constval = bstate; dxfact = 0.0; } else { TBOX_ERROR( "Unknown bcase " << bcase << " passed to SkeletonBoundaryUtilities2::checkBdryData()" << "\n for " << bdry_type_str << " at location " << bloc << endl); } } else if (btype == Bdry::NODE2D) { if (bcase == BdryCond::XFLOW || bcase == BdryCond::YFLOW) { valfact = 1.0; constval = 0.0; dxfact = 0.0; } else if (bcase == BdryCond::XREFLECT || bcase == BdryCond::YREFLECT) { valfact = -1.0; constval = 0.0; dxfact = 0.0; } else if (bcase == BdryCond::XDIRICHLET || bcase == BdryCond::YDIRICHLET) { valfact = 0.0; constval = bstate; dxfact = 0.0; } else { TBOX_ERROR( "Unknown bcase " << bcase << " passed to SkeletonBoundaryUtilities2::checkBdryData()" << "\n for " << bdry_type_str << " at location " << bloc << endl); } } hier::Box gbox_to_check( vardata->getGhostBox() * pgeom->getBoundaryFillBox(bbox, patch.getBox(), gcw_to_check)); hier::Box cbox(gbox_to_check); hier::Box dbox(gbox_to_check); hier::Index ifirst(vardata->getBox().lower()); hier::Index ilast(vardata->getBox().upper()); if (offsign == -1) { cbox.setLower(idir, ifirst(idir) - 1); cbox.setUpper(idir, ifirst(idir) - 1); dbox.setLower(idir, ifirst(idir)); dbox.setUpper(idir, ifirst(idir)); } else { cbox.setLower(idir, ilast(idir) + 1); cbox.setUpper(idir, ilast(idir) + 1); dbox.setLower(idir, ilast(idir)); dbox.setUpper(idir, ilast(idir)); } pdat::CellIterator id(pdat::CellGeometry::begin(dbox)); pdat::CellIterator icend(pdat::CellGeometry::end(cbox)); for (pdat::CellIterator ic(pdat::CellGeometry::begin(cbox)); ic != icend; ++ic) { double checkval = valfact * (*vardata)(*id, depth) + constval; pdat::CellIndex check = *ic; for (int p = 0; p < gbox_to_check.numberCells(idir); ++p) { double offcheckval = checkval + dxfact * (p + 1); if ((*vardata)(check, depth) != offcheckval) { ++num_bad_values; TBOX_WARNING("Bad " << bdry_type_str << " boundary value for " << varname << " found in cell " << check << "\n found = " << (*vardata)(check, depth) << " : correct = " << offcheckval << endl); } check(idir) += offsign; } ++id; } return num_bad_values; }