FinleyElements.h

Go to the documentation of this file.

/*******************************************************
*
* Copyright (c) 2003-2010 by University of Queensland
* Earth Systems Science Computational Center (ESSCC)
* http://www.uq.edu.au/esscc
*
* Primary Business: Queensland, Australia
* Licensed under the Open Software License version 3.0
* http://www.opensource.org/licenses/osl-3.0.php
*
*******************************************************/

#ifndef __WEIPA_FINLEYELEMENTS_H__
#define __WEIPA_FINLEYELEMENTS_H__

#include <weipa/ElementData.h>
#include <weipa/FinleyNodes.h>

extern "C" {
#include <dudley/ElementType.h> // for Dudley_ElementTypeId
#include <finley/ReferenceElements.h> // for Finley_ElementTypeId
}

class DBfile;
class NcFile;

struct Dudley_ElementFile;
struct Finley_ElementFile;

namespace weipa {
 
struct FinleyElementInfo
{
    ZoneType elementType, reducedElementType;
    int elementFactor;
    int elementSize, reducedElementSize;
    const size_t* multiCellIndices;
    bool useQuadNodes;
    int quadDim;
};

class FinleyElements;
typedef boost::shared_ptr<FinleyElements> FinleyElements_ptr;

class FinleyElements : public ElementData
{
public:

    FinleyElements(const std::string& elementName, FinleyNodes_ptr nodes);

    FinleyElements(const FinleyElements& e);

    virtual ~FinleyElements() {}

    bool initFromDudley(const Dudley_ElementFile* dudleyFile);

    bool initFromFinley(const Finley_ElementFile* finleyFile);

    bool readFromNc(NcFile* ncfile);

    void reorderGhostZones(int ownIndex);

    void removeGhostZones(int ownIndex);

    virtual void writeConnectivityVTK(std::ostream& os);

    bool writeToSilo(DBfile* dbfile, const std::string& siloPath,
                     const StringVec& labels, const StringVec& units,
                     bool writeMeshData);

    virtual StringVec getMeshNames() const;

    virtual StringVec getVarNames() const;

    virtual int getNumElements() const { return numElements; }

    virtual int getNodesPerElement() const { return nodesPerElement; }

    virtual int getGhostCount() const { return numGhostElements; }

    virtual ZoneType getType() const { return type; }

    Finley_ElementTypeId getFinleyTypeId() const { return finleyTypeId; }

    virtual const IntVec& getNodeList() const { return nodes; }

    virtual const IntVec& getIDs() const { return ID; }

    virtual const IntVec& getVarDataByName(const std::string varName) const;

    virtual NodeData_ptr getNodes() const { return nodeMesh; }

    virtual ElementData_ptr getReducedElements() const { return reducedElements; }
 
    virtual const QuadMaskInfo& getQuadMask(int functionSpace) const;
 
    virtual int getElementFactor() const { return elementFactor; }

private:
    FinleyElements() {}
    FinleyElementInfo getDudleyTypeInfo(Dudley_ElementTypeId typeId);
    FinleyElementInfo getFinleyTypeInfo(Finley_ElementTypeId typeId);
    void buildMeshes();
    void buildReducedElements(const FinleyElementInfo& f);
    IntVec prepareGhostIndices(int ownIndex);
    void reorderArray(IntVec& v, const IntVec& idx, int elementsPerIndex);
    QuadMaskInfo buildQuadMask(const CoordArray& quadNodes, int numQNodes);

    FinleyElements_ptr reducedElements;
    FinleyNodes_ptr nodeMesh;
    FinleyNodes_ptr originalMesh;
    std::string name;
    int numElements;
    int numGhostElements;
    int nodesPerElement;
    ZoneType type;
    Finley_ElementTypeId finleyTypeId;
    IntVec nodes;
    IntVec color, ID, tag;
    IntVec owner;
    QuadMaskInfo quadMask, reducedQuadMask;
    int elementFactor;
};

} // namespace weipa

#endif // __WEIPA_FINLEYELEMENTS_H__