Panzer  Version of the Day
Panzer_STK_Utilities.hpp
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 
43 #ifndef __Panzer_STK_Utilities_hpp__
44 #define __Panzer_STK_Utilities_hpp__
45 
46 #include "Panzer_STK_Interface.hpp"
47 
48 #include "Epetra_Vector.h"
49 #include "Epetra_MultiVector.h"
50 
51 namespace panzer {
52  template <typename LO,typename GO> class UniqueGlobalIndexer;
53 #ifdef PANZER_HAVE_FEI
54  template <typename LO,typename GO> class DOFManagerFEI;
55 #endif
56 }
57 
58 namespace panzer_stk {
59 
68 void write_cell_data(panzer_stk::STK_Interface & mesh,const std::vector<double> & data,const std::string & fieldName);
69 
70 template <typename GlobalOrdinal>
71 void write_solution_data(const panzer::UniqueGlobalIndexer<int,GlobalOrdinal> & dofMngr,panzer_stk::STK_Interface & mesh,const Epetra_MultiVector & x,const std::string & prefx="",const std::string & postfix="");
72 template <typename GlobalOrdinal>
73 void write_solution_data(const panzer::UniqueGlobalIndexer<int,GlobalOrdinal> & dofMngr,panzer_stk::STK_Interface & mesh,const Epetra_Vector & x,const std::string & prefix="",const std::string & postfix="");
74 
75 #ifdef PANZER_HAVE_FEI
76 void read_solution_data(const panzer::DOFManagerFEI<int,int> & dofMngr,const panzer_stk::STK_Interface & mesh,Epetra_MultiVector & x);
77 void read_solution_data(const panzer::DOFManagerFEI<int,int> & dofMngr,const panzer_stk::STK_Interface & mesh,Epetra_Vector & x);
78 #endif
79 
86 template <typename RAContainer,class Compare>
87 void sorted_permutation(const RAContainer & cont,std::vector<std::size_t> & permutation,const Compare & comp);
88 
95 template <typename RAContainer>
96 void sorted_permutation(const RAContainer & cont,std::vector<std::size_t> & permutation);
97 
98 }
99 
100 namespace panzer_stk {
101 // utility class used by the sorted permutation objects
102 template <typename RAContainer,typename Compare>
103 struct PermFunctor {
104  PermFunctor(const RAContainer & cont,const Compare & comp)
105  : compare(comp), values(cont) {}
107  : compare(p.compare), values(p.values) {}
108 
109  bool operator()(std::size_t a,std::size_t b) const
110  { return compare(values[a],values[b]); }
111 
112 private:
113  const Compare & compare;
114  const RAContainer & values;
115 
116  PermFunctor();
117 };
118 
119 template <typename RAContainer>
120 void sorted_permutation(const RAContainer & cont,std::vector<std::size_t> & permutation)
121 {
122  std::less<typename RAContainer::value_type> comp;
123  sorted_permutation(cont,permutation,comp);
124 }
125 
126 template <typename RAContainer,class Compare>
127 void sorted_permutation(const RAContainer & cont,std::vector<std::size_t> & permutation,const Compare & comp)
128 {
129  PermFunctor<RAContainer,Compare> pf(cont,comp);
130 
131  permutation.resize(cont.size());
132  for(std::size_t i=0;i<cont.size();i++)
133  permutation[i] = i;
134 
135  std::sort(permutation.begin(),permutation.end(),pf);
136 }
137 
138 }
139 
140 #endif
bool operator()(std::size_t a, std::size_t b) const
void write_solution_data(const panzer::UniqueGlobalIndexer< int, GlobalOrdinal > &dofMngr, panzer_stk::STK_Interface &mesh, const Epetra_MultiVector &x, const std::string &prefix, const std::string &postfix)
PermFunctor(const RAContainer &cont, const Compare &comp)
void sorted_permutation(const RAContainer &cont, std::vector< std::size_t > &permutation, const Compare &comp)
PermFunctor(const PermFunctor &p)
void write_cell_data(panzer_stk::STK_Interface &mesh, const std::vector< double > &data, const std::string &fieldName)