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