EpetraExt_CrsSingletonFilter_LinearProblem.h

Go to the documentation of this file.

//@HEADER
// ***********************************************************************
// 
//     EpetraExt: Epetra Extended - Linear Algebra Services Package
//                 Copyright (2001) Sandia Corporation
// 
// Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive
// license for use of this work by or on behalf of the U.S. Government.
// 
// This library is free software; you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as
// published by the Free Software Foundation; either version 2.1 of the
// License, or (at your option) any later version.
//  
// This library is distributed in the hope that it will be useful, but
// WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// Lesser General Public License for more details.
//  
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
// USA
// Questions? Contact Michael A. Heroux (maherou@sandia.gov) 
// 
// ***********************************************************************
//@HEADER
                                                                                                    
#ifndef _EpetraExt_LINEARPROBLEM_CRSSINGLETONFILTER_H_
#define _EpetraExt_LINEARPROBLEM_CRSSINGLETONFILTER_H_

#include "Epetra_Object.h"
#include "Epetra_CrsMatrix.h"
#include "Epetra_MapColoring.h"
#include "Epetra_SerialDenseVector.h"

#include "EpetraExt_Transform.h"

class Epetra_LinearProblem;
class Epetra_Map;
class Epetra_MultiVector;
class Epetra_Import;
class Epetra_Export;
class Epetra_IntVector;

namespace EpetraExt {


class LinearProblem_CrsSingletonFilter : public SameTypeTransform<Epetra_LinearProblem> {
      
 public:


  LinearProblem_CrsSingletonFilter( bool verbose = false );

  virtual ~LinearProblem_CrsSingletonFilter();

  NewTypeRef operator()( OriginalTypeRef orig );

  bool analyze( OriginalTypeRef orig );

  NewTypeRef construct();

  bool fwd();

  bool rvs();



  int Analyze(Epetra_RowMatrix * FullMatrix);

  bool SingletonsDetected() const {if (!AnalysisDone_) return(false); else return(RowMapColors_->MaxNumColors()>1);};



  int ConstructReducedProblem(Epetra_LinearProblem * Problem);


  int UpdateReducedProblem(Epetra_LinearProblem * Problem);




  int ComputeFullSolution();


  int NumRowSingletons() const {return(NumGlobalRowSingletons_);};

  int NumColSingletons() const {return(NumGlobalColSingletons_);};


  int NumSingletons() const {return(NumColSingletons()+NumRowSingletons());};

  double RatioOfDimensions() const {return(RatioOfDimensions_);};

  double RatioOfNonzeros() const {return(RatioOfNonzeros_);};



  Epetra_LinearProblem * FullProblem() const {return(FullProblem_);};

  Epetra_LinearProblem * ReducedProblem() const {return(ReducedProblem_);};

  Epetra_RowMatrix * FullMatrix() const {return(FullMatrix_);};

  Epetra_CrsMatrix * ReducedMatrix() const {return(ReducedMatrix_);};

  Epetra_MapColoring * RowMapColors() const {return(RowMapColors_);};

  Epetra_MapColoring * ColMapColors() const {return(ColMapColors_);};

  Epetra_Map * ReducedMatrixRowMap() const {return(ReducedMatrixRowMap_);};

  Epetra_Map * ReducedMatrixColMap() const {return(ReducedMatrixColMap_);};

  Epetra_Map * ReducedMatrixDomainMap() const {return(ReducedMatrixDomainMap_);};

  Epetra_Map * ReducedMatrixRangeMap() const {return(ReducedMatrixRangeMap_);};

 protected:

 

  //  This pointer will be zero if full matrix is not a CrsMatrix.
  Epetra_CrsMatrix * FullCrsMatrix() const {return(FullCrsMatrix_);};

  const Epetra_Map & FullMatrixRowMap() const {return(FullMatrix()->RowMatrixRowMap());};
  const Epetra_Map & FullMatrixColMap() const {return(FullMatrix()->RowMatrixColMap());};
  const Epetra_Map & FullMatrixDomainMap() const {return((FullMatrix()->OperatorDomainMap()));};
  const Epetra_Map & FullMatrixRangeMap() const {return((FullMatrix()->OperatorRangeMap()));};
  void InitializeDefaults();
  int ComputeEliminateMaps();
  int Setup(Epetra_LinearProblem * Problem);
  int InitFullMatrixAccess();
  int GetRow(int Row, int & NumIndices, int * & Indices);
  int GetRowGCIDs(int Row, int & NumIndices, double * & Values, int * & GlobalIndices);
  int GetRow(int Row, int & NumIndices, double * & Values, int * & Indices);
  int CreatePostSolveArrays(const Epetra_IntVector & RowIDs,
          const Epetra_MapColoring & RowMapColors,
          const Epetra_IntVector & ColProfiles,
          const Epetra_IntVector & NewColProfiles,
          const Epetra_IntVector & ColHasRowWithSingleton);
  
  int ConstructRedistributeExporter(Epetra_Map * SourceMap, Epetra_Map * TargetMap,
            Epetra_Export * & RedistributeExporter,
            Epetra_Map * & RedistributeMap);
  
  Epetra_LinearProblem * FullProblem_;
  Epetra_LinearProblem * ReducedProblem_;
  Epetra_RowMatrix * FullMatrix_;
  Epetra_CrsMatrix * FullCrsMatrix_;
  Epetra_CrsMatrix * ReducedMatrix_;
  Epetra_MultiVector * ReducedRHS_;
  Epetra_MultiVector * ReducedLHS_;
  
  Epetra_Map * ReducedMatrixRowMap_;
  Epetra_Map * ReducedMatrixColMap_;
  Epetra_Map * ReducedMatrixDomainMap_;
  Epetra_Map * ReducedMatrixRangeMap_;
  Epetra_Map * OrigReducedMatrixDomainMap_;
  Epetra_Import * Full2ReducedRHSImporter_;
  Epetra_Import * Full2ReducedLHSImporter_;
  Epetra_Export * RedistributeDomainExporter_;
  
  int * ColSingletonRowLIDs_;
  int * ColSingletonColLIDs_;
  int * ColSingletonPivotLIDs_;
  double * ColSingletonPivots_;
  
  
  int AbsoluteThreshold_;
  double RelativeThreshold_;

  int NumMyRowSingletons_;
  int NumMyColSingletons_;
  int NumGlobalRowSingletons_;
  int NumGlobalColSingletons_;
  double RatioOfDimensions_;
  double RatioOfNonzeros_;
  
  bool HaveReducedProblem_;
  bool UserDefinedEliminateMaps_;
  bool AnalysisDone_;
  bool SymmetricElimination_;
  
  Epetra_MultiVector * tempExportX_;
  Epetra_MultiVector * tempX_;
  Epetra_MultiVector * tempB_;
  Epetra_MultiVector * RedistributeReducedLHS_;
  int * Indices_;
  Epetra_SerialDenseVector Values_;
  
  Epetra_MapColoring * RowMapColors_;
  Epetra_MapColoring * ColMapColors_;
  bool FullMatrixIsCrsMatrix_;
  int MaxNumMyEntries_;

  bool verbose_;
  
  
 private:
  LinearProblem_CrsSingletonFilter(const LinearProblem_CrsSingletonFilter & Problem){};
};

} //namespace EpetraExt

#endif /* _EpetraExt_LINEARPROBLEM_CRSSINGLETONFILTER_H_ */