Geometry.h

Go to the documentation of this file.

/* Copyright 2016 Kristofer Björnson
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef COM_DAFER45_TBTK_GEOMETRY
#define COM_DAFER45_TBTK_GEOMETRY

#include "TBTK/IndexedDataTree.h"
#include "TBTK/Serializable.h"
#include "TBTK/SerializableVector.h"

namespace TBTK{
    class FileReader;

class Geometry : public Serializable{
public:
    Geometry();

    Geometry(
        const std::string &serialization,
        Mode mode
    );

    virtual ~Geometry();

    void setCoordinate(
        const Index &index,
        const std::vector<double> &coordinate
    );

    const std::vector<double>& getCoordinate(const Index &index) const;

    int getDimensions() const;

    void translate(const std::vector<double> &translation);

    std::string serialize(Mode mode) const;
private:
    int dimensions;

    IndexedDataTree<SerializableVector<double>> coordinates;

    friend class FileReader;
};

inline void Geometry::setCoordinate(
    const Index &index,
    const std::vector<double> &coordinate
){
    if(dimensions == -1)
        dimensions = coordinate.size();

    TBTKAssert(
        coordinate.size() == (unsigned int)dimensions,
        "Geometry::setCoordinate()",
        "Incompatible dimensions. A coordinate with dimension '"
        << dimensions << "' has already been added, which is"
        << " incompatible with the current coordinate dimension of '"
        << coordinate.size() << "'.",
        ""
    );

    coordinates.add(coordinate, index);
}

inline const std::vector<double>& Geometry::getCoordinate(
    const Index &index
) const{
    return coordinates.get(index);
}

inline int Geometry::getDimensions() const{
    return dimensions;
}

};  //End of namespace TBTK

#endif