Poisson_Elem.hpp

#ifndef __Poisson_Elem_H
#define __Poisson_Elem_H
 
/*--------------------------------------------------------------------*/
/*    Copyright 2005 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.                    */
/*--------------------------------------------------------------------*/
 
class Poisson_Elem {
 
  public:
    Poisson_Elem();
    ~Poisson_Elem();
 
 
    GlobalID getElemID() const {return globalElemID_;};
    void setElemID(GlobalID gNID) {globalElemID_ = gNID;
                                   ID_IsSet_ = true;};
 
    int numElemRows() const {return numElemRows_;};
    void numElemRows(int gNERows) {numElemRows_ = gNERows;};
 
    int numElemNodes() const {return numElemNodes_;};
    void numElemNodes(int gNodes) {numElemNodes_ = gNodes;};
 
    double getElemLength() const {return elemLength_;};
    void setElemLength(double len) {elemLength_ = len;
                                    elemLengthIsSet_ = true;};
 
    double getTotalLength() const {return totalLength_;};
    void setTotalLength(double len) {totalLength_ = len;
                                    totalLengthIsSet_ = true;};
 
    int allocateInternals(int DOF);
    int allocateLoad(int DOF);
    int allocateStiffness(int DOF);
 
    GlobalID* getElemConnPtr(int& size);
 
    void calculateLoad();
    double* getElemLoad(int& size);
 
    void calculateStiffness();
    double** getElemStiff(int& size);
 
    double* getNodalX(int& size) {size = numElemNodes_; return(nodalX_);};
    double* getNodalY(int& size) {size = numElemNodes_; return(nodalY_);};
 
    void calculateCoords();
 
    void messageAbort(const char* str);
 
    void deleteMemory();
 
//$ temporary output for debugging...
 
    void dumpToScreen();
 
 
  private:
    GlobalID globalElemID_; // global ID number for this element
    bool ID_IsSet_;         // whether ID has been set or not.
    int numElemNodes_;      // number of nodes associated with this element
    int numElemRows_;       // number of rows in the element matrices
    
    GlobalID *nodeList_;      // list of nodes associated with this element
    double* nodalX_;          // list of nodal x-coordinates
    double* nodalY_;          // list of nodal y-coordinates
 
    double **elemStiff_;      // stiffness matrix for this element
    double *elemLoad_;        // load vector for this element
    bool internalsAllocated_;
 
    double elemLength_;       // length of a side of this 2D element
    double totalLength_;      // total length of a side of the square region
                              // that this element is in.
    bool elemLengthIsSet_;    // indicates whether length has been set.
    bool totalLengthIsSet_;    // indicates whether length has been set.
 
    bool loadAllocated_;
    bool stiffAllocated_;
};
 
#endif