fei_Vector_Local.hpp

/*--------------------------------------------------------------------*/
/*    Copyright 2007 Sandia Corporation.                              */
/*    Under the 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.  Export of this program may require     */
/*    a license from the United States Government.                    */
/*--------------------------------------------------------------------*/
 
#ifndef _fei_Vector_Local_hpp_
#define _fei_Vector_Local_hpp_
 
#include <fei_iosfwd.hpp>
#include <fei_SharedPtr.hpp>
#include <fei_VectorSpace.hpp>
#include <fei_Vector.hpp>
 
namespace fei {
 
class Vector_Local : public fei::Vector {
 public:
  Vector_Local(fei::SharedPtr<fei::VectorSpace> vecSpace);
 
  virtual ~Vector_Local();
 
  const char* typeName() const { return("fei::Vector_Local"); }

    int update(double a,
               const fei::Vector* x,
               double b);

    int scatterToOverlap();
 
    void setCommSizes();

    int gatherFromOverlap(bool accumulate = true);

    int putScalar(double scalar);

    int sumIn(int numValues, const int* indices, const double* values,
              int vectorIndex=0);

    int copyIn(int numValues, const int* indices, const double* values,
               int vectorIndex=0);

    fei::SharedPtr<fei::VectorSpace> getVectorSpace() const;

    void setVectorSpace(fei::SharedPtr<fei::VectorSpace> vecSpace);

    int sumInFieldData(int fieldID,
                       int idType,
                       int numIDs,
                       const int* IDs,
                       const double* data,
                       int vectorIndex=0);

    int copyInFieldData(int fieldID,
                        int idType,
                        int numIDs,
                        const int* IDs,
                        const double* data,
                        int vectorIndex=0);
 
    int copyInFieldDataLocalIDs(int fieldID,
                        int idType,
                        int numIDs,
                        const int* localIDs,
                        const double* data,
                        int vectorIndex=0);

    int copyOutFieldData(int fieldID,
                         int idType,
                         int numIDs,
                         const int* IDs,
                         double* data,
                         int vectorIndex=0);
 
    int copyOut(int numValues, const int* indices,
                double* values, int vectorIndex=0) const;
 
    int writeToFile(const char* filename,
                    bool matrixMarketFormat=true);
 
    int writeToStream(FEI_OSTREAM& ostrm,
                      bool matrixMarketFormat=true);
 
    std::vector<double>& getCoefs();
 
 private:
  int giveToVector(int numValues, const int* indices,
                           const double* values,
                           bool sumInto, int vectorIndex);
 
  int assembleFieldData(int fieldID,
                       int idType,
                       int numIDs,
                       const int* IDs,
                       const double* data,
                       bool sumInto,
                       int vectorIndex);
 
  int assembleFieldDataLocalIDs(int fieldID,
                       int idType,
                       int numIDs,
                       const int* localIDs,
                       const double* data,
                       bool sumInto,
                       int vectorIndex);
 
  fei::SharedPtr<fei::VectorSpace> vecSpace_;
  std::vector<double> coefs_;
  std::map<int,int> global_to_local_;
  std::vector<int> work_indices_;
};//class Vector_Local
 
}//namespace fei
 
#endif