Panzer  Version of the Day
Panzer_SubcellConnectivity.cpp
Go to the documentation of this file.
1 // @HEADER
2 // ***********************************************************************
3 //
4 // Panzer: A partial differential equation assembly
5 // engine for strongly coupled complex multiphysics systems
6 // Copyright (2011) Sandia Corporation
7 //
8 // Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
9 // the U.S. Government retains certain rights in this software.
10 //
11 // Redistribution and use in source and binary forms, with or without
12 // modification, are permitted provided that the following conditions are
13 // met:
14 //
15 // 1. Redistributions of source code must retain the above copyright
16 // notice, this list of conditions and the following disclaimer.
17 //
18 // 2. Redistributions in binary form must reproduce the above copyright
19 // notice, this list of conditions and the following disclaimer in the
20 // documentation and/or other materials provided with the distribution.
21 //
22 // 3. Neither the name of the Corporation nor the names of the
23 // contributors may be used to endorse or promote products derived from
24 // this software without specific prior written permission.
25 //
26 // THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
27 // EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
28 // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
29 // PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
30 // CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
31 // EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
32 // PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
33 // PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
34 // LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
35 // NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
36 // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
37 //
38 // Questions? Contact Roger P. Pawlowski (rppawlo@sandia.gov) and
39 // Eric C. Cyr (eccyr@sandia.gov)
40 // ***********************************************************************
41 // @HEADER
42 
44 
45 #include "Teuchos_Assert.hpp"
46 
47 #include "Panzer_LocalMeshInfo.hpp"
48 
49 namespace panzer
50 {
51 
52 int
54 numSubcellsOnCell(const int cell) const
55 {
56 #ifdef PANZER_DEBUG
57  TEUCHOS_ASSERT(cell >= 0 and cell < _num_cells);
58 #endif
59  return _cell_to_subcells_adj(cell+1)-_cell_to_subcells_adj(cell);
60 }
61 
62 int
64 numCellsOnSubcell(const int subcell) const
65 {
66 #ifdef PANZER_DEBUG
67  TEUCHOS_ASSERT(subcell >= 0 and subcell < _num_subcells);
68 #endif
69  return _subcell_to_cells_adj(subcell+1)-_subcell_to_cells_adj(subcell);
70 }
71 
72 int
74 subcellForCell(const int cell, const int local_subcell_index) const
75 {
76 #ifdef PANZER_DEBUG
77  TEUCHOS_ASSERT(cell >= 0 and cell < _num_cells);
78  TEUCHOS_ASSERT(local_subcell_index < numSubcellsOnCell(cell));
79 #endif
80  const int index = _cell_to_subcells_adj(cell)+local_subcell_index;
81  return _cell_to_subcells(index);
82 }
83 
84 int
86 cellForSubcell(const int subcell, const int local_cell_index) const
87 {
88 #ifdef PANZER_DEBUG
89  TEUCHOS_ASSERT(subcell >= 0 and subcell < _num_subcells);
90  TEUCHOS_ASSERT(local_cell_index < numCellsOnSubcell(subcell));
91 #endif
92  const int index = _subcell_to_cells_adj(subcell)+local_cell_index;
93  return _subcell_to_cells(index);
94 }
95 
96 int
98 localSubcellForSubcell(const int subcell, const int local_cell_index) const
99 {
100 #ifdef PANZER_DEBUG
101  TEUCHOS_ASSERT(subcell >= 0 and subcell < _num_subcells);
102  TEUCHOS_ASSERT(local_cell_index < numCellsOnSubcell(subcell));
103 #endif
104  const int index = _subcell_to_cells_adj(subcell)+local_cell_index;
105  return _subcell_to_local_subcells(index);
106 }
107 
108 // ========================================================================
109 
110 void
113 {
114  const int num_cells = partition.cell_to_faces.extent(0);
115  const int num_faces = partition.face_to_cells.extent(0);
116  const int num_faces_per_cell = partition.cell_to_faces.extent(1);
117  const int num_cells_per_face = 2;
118 
119  _num_subcells = num_faces;
120  _num_cells = num_cells;
121 
122  _subcell_to_cells_adj = Kokkos::View<int*>("subcell_to_cells_adj", num_faces+1);
123  _subcell_to_cells = Kokkos::View<int*>("subcell_to_cells", num_faces*num_cells_per_face);
124  _subcell_to_local_subcells = Kokkos::View<int*>("subcell_to_local_subcells", num_faces*num_cells_per_face);
125  _cell_to_subcells_adj = Kokkos::View<int*>("cell_to_subcells_adj", num_cells+1);
126  _cell_to_subcells = Kokkos::View<int*>("cell_to_subcells", num_cells*num_faces_per_cell);
127 
129  for(int face=0;face<num_faces;++face){
130  _subcell_to_cells_adj(face+1) =_subcell_to_cells_adj(face)+num_cells_per_face;
131  _subcell_to_cells(num_cells_per_face*face + 0) = partition.face_to_cells(face,0);
132  _subcell_to_cells(num_cells_per_face*face + 1) = partition.face_to_cells(face,1);
133  _subcell_to_local_subcells(num_cells_per_face*face + 0) = partition.face_to_lidx(face,0);
134  _subcell_to_local_subcells(num_cells_per_face*face + 1) = partition.face_to_lidx(face,1);
135  }
136 
138  for(int cell=0;cell<num_cells;++cell){
139  _cell_to_subcells_adj(cell+1) =_cell_to_subcells_adj(cell)+num_faces_per_cell;
140  for(int local_face=0;local_face<num_faces_per_cell;++local_face){
141  _cell_to_subcells(num_faces_per_cell*cell+local_face) = partition.cell_to_faces(cell,local_face);
142  }
143  }
144 
145 }
146 
147 }
Kokkos::View< int *, PHX::Device > _subcell_to_cells
Mapping from subcells to cells.
int cellForSubcell(const int subcell, const int local_cell_index) const
Get the cell for a given subcell and a local_cell_index.
int numSubcellsOnCell(const int cell) const
gives number of subcells (e.g. faces) found on a given cell
int _num_subcells
Number of subcells for a given number of cells.
Kokkos::View< int *, PHX::Device > _cell_to_subcells_adj
Adjacency array for indexing into cell_to_subcells array.
Kokkos::View< int *, PHX::Device > _cell_to_subcells
Mapping from cells to subcells.
int numCellsOnSubcell(const int subcell) const
Returns the number of cells attached to a given subcell.
void setup(const panzer::LocalMeshPartition< int, panzer::Ordinal64 > &partition)
Setup the face connectivity from a partition of the local mesh.
int subcellForCell(const int cell, const int local_subcell_index) const
Get the subcell index for a given cell and local subcell index.
Kokkos::View< LO *[2]> face_to_cells
Kokkos::View< int *, PHX::Device > _subcell_to_local_subcells
Mapping from subcell indexes to local subcell indexes.
Kokkos::View< LO ** > cell_to_faces
Kokkos::View< LO *[2]> face_to_lidx
Kokkos::View< int *, PHX::Device > _subcell_to_cells_adj
Adjacency array for indexing into subcell_to_cells array.
#define TEUCHOS_ASSERT(assertion_test)
int localSubcellForSubcell(const int subcell, const int local_cell_index) const
Get the local subcell index given a subcell and a local cell index.