DataVar.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_DATAVAR_H__
#define __WEIPA_DATAVAR_H__

#include <weipa/DomainChunk.h>
#include <ostream>

class DBfile;
class NcFile;

namespace escript {
    class Data;
}

namespace weipa {

class DataVar
{
public:
    DataVar(const std::string& name);

    DataVar(const DataVar& d);

    ~DataVar();

    bool initFromEscript(escript::Data& escriptData, const_DomainChunk_ptr dom);

    bool initFromMeshData(const_DomainChunk_ptr dom, const IntVec& data,
            int fsCode, Centering c, NodeData_ptr nodes, const IntVec& id);

    bool initFromFile(const std::string& filename, const_DomainChunk_ptr dom);

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

    void writeToVTK(std::ostream& os, int ownIndex);

    int getRank() const { return rank; }

    bool isNodeCentered() const;

    std::string getMeshName() const { return meshName; }

    const IntVec& getShape() const { return shape; }

    std::string getName() const { return varName; }

    std::string getTensorDef() const;

    int getNumberOfSamples() const { return numSamples; }

    const CoordArray& getData() const { return dataArray; }

    float* getDataFlat() const;

    int getNumberOfComponents() const;

private:
    void cleanup();

    float* averageData(const float* src, size_t stride);

    IndexMap buildIndexMap();

    bool reorderSamples();

    void sampleToStream(std::ostream& os, int index);

    bool initialized;
    const_DomainChunk_ptr domain;
    std::string varName;
    int numSamples, rank, ptsPerSample, funcSpace;
    Centering centering;
    IntVec shape;
    IntVec sampleID;
    CoordArray dataArray;
    std::string meshName, siloMeshName;
};

inline IndexMap DataVar::buildIndexMap()
{
    IndexMap sampleID2idx;
    int idx = sampleID.size()-1;
    // see this thread for why this is done the way it's done:
    // http://www.tech-archive.net/Archive/VC/microsoft.public.vc.stl/2005-01/0075.html
    IntVec::const_reverse_iterator idIt = sampleID.rbegin();
    IntVec::const_reverse_iterator endIt = sampleID.rend();
    for (; idIt != endIt; idIt++, idx--)
        sampleID2idx[*idIt] = idx;

    return sampleID2idx;
}

} // namespace weipa

#endif // __WEIPA_DATAVAR_H__