EpetraExt Package Browser (Single Doxygen Collection) Development
Loading...
Searching...
No Matches
EpetraExt_TCrsGraph_MapColoringIndex.h
Go to the documentation of this file.
1//@HEADER
2// ***********************************************************************
3//
4// EpetraExt: Epetra Extended - Linear Algebra Services Package
5// Copyright (2011) Sandia Corporation
6//
7// Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
8// the U.S. Government retains certain rights in this software.
9//
10// Redistribution and use in source and binary forms, with or without
11// modification, are permitted provided that the following conditions are
12// met:
13//
14// 1. Redistributions of source code must retain the above copyright
15// notice, this list of conditions and the following disclaimer.
16//
17// 2. Redistributions in binary form must reproduce the above copyright
18// notice, this list of conditions and the following disclaimer in the
19// documentation and/or other materials provided with the distribution.
20//
21// 3. Neither the name of the Corporation nor the names of the
22// contributors may be used to endorse or promote products derived from
23// this software without specific prior written permission.
24//
25// THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
26// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
27// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
28// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
29// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
30// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
31// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
32// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
33// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
34// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
35// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
36//
37// Questions? Contact Michael A. Heroux (maherou@sandia.gov)
38//
39// ***********************************************************************
40//@HEADER
41
42#ifndef EpetraExt_TCRSGRAPH_MAPCOLORINGINDEX_H
43#define EpetraExt_TCRSGRAPH_MAPCOLORINGINDEX_H
44
45#include <EpetraExt_Transform.h>
47#include <Epetra_MapColoring.h>
48
49#include <vector>
50#include <map>
51
52class Epetra_CrsGraph;
54
55namespace EpetraExt {
56
58
62template<typename int_type>
64: public StructuralTransform< Epetra_CrsGraph,std::vector<typename Epetra_GIDTypeVector<int_type>::impl> > {
65
67
68 protected:
69
71
74
76
80 : ColorMap_( ColorMap )
81 {}
82public:
85
87 typedef typename Base::NewTypeRef NewTypeRef;
90};
91
93
94template<typename int_type>
98{
99 if(!orig.RowMap(). template GlobalIndicesIsType<int_type>())
100 throw "EpetraExt::TCrsGraph_MapColoringIndex::operator(): Global indices mismatch.";
101
102 Base::origObj_ = &orig;
103
104 const Epetra_BlockMap & RowMap = orig.RowMap();
105 int nRows = RowMap.NumMyElements();
106
107 int NumColors = ColorMap_.NumColors();
108 int * ListOfColors = ColorMap_.ListOfColors();
109
110 std::map<int,int> MapOfColors;
111 for( int i = 0; i < NumColors; ++i ) MapOfColors[ ListOfColors[i] ] = i;
112
113 //initial setup of stl vector of IntVectors for indexing
114 std::vector<int_type> dummy( nRows, -1 );
115 typename Base::NewTypePtr IndexVec = new typename Base::NewType( NumColors, typename Epetra_GIDTypeVector<int_type>::impl( Copy, RowMap, &dummy[0] ) );
116
117 int MaxNumIndices = orig.MaxNumIndices();
118 int NumIndices;
119 std::vector<int_type> Indices( MaxNumIndices );
120
121 for( int i = 0; i < nRows; ++i )
122 {
123 orig.ExtractGlobalRowCopy( (int_type) orig.GRID64(i), MaxNumIndices, NumIndices, &Indices[0] );
124
125 for( int j = 0; j < NumIndices; ++j )
126 (*IndexVec)[ MapOfColors[ColorMap_(Indices[j])] ][i] = Indices[j];
127 }
128
129 Base::newObj_ = IndexVec;
130
131 return *IndexVec;
132}
133
134} //namespace EpetraExt
135
136#endif // EpetraExt_TCRSGRAPH_MAPCOLORINGINDEX_H
Copy
Generates a std::vector of Epetra_IntVector's to be used to map perturbation contributions to a CrsGr...
TCrsGraph_MapColoringIndex(const Epetra_MapColoring &ColorMap)
Constructor input param ColorMap defines the perturbation coloring.
StructuralTransform< Epetra_CrsGraph, std::vector< typename Epetra_GIDTypeVector< int_type >::impl > > Base
NewTypeRef operator()(OriginalTypeRef orig)
Analysis of transform operation on original object and construction of new object.
Base::NewTypeRef NewTypeRef
Generates a std::vector<Epetra_IntVector> from the input Epetra_CrsGraph.
int NumMyElements() const
EpetraExt::BlockCrsMatrix: A class for constructing a distributed block matrix.