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