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 "Model.h"
#include <initializer_list>

namespace TBTK{
    class FileReader;

class Geometry{
public:
    Geometry(int dimensions, int numSpecifiers, Model *parentModel);

    ~Geometry();

    void setCoordinates(
        const Index &index,
        std::initializer_list<double> coordinates,
        std::initializer_list<int> specifiers = {}
    );

    void setCoordinates(
        const Index &index,
        const std::vector<double> &coordinates,
        const std::vector<int> &specifiers = {}
    );

    int getDimensions() const;

    int getNumSpecifiers() const;

    const double* getCoordinates(const Index &index) const;

    const double* getCoordinates(int index) const;

    const double* getCoordinates() const;

    void translate(std::initializer_list<double> translation);

    int getSpecifier(const Index &index, int specifier) const;

    int getSpecifier(int index, int specifier) const;

    const int* getSpecifiers(const Index &index) const;

    const int* getSpecifiers(int index) const;

    const int* getSpecifiers() const;

    double getDistance(const Index &index1, const Index &index2) const;

    double getDistance(int index1, int index2) const;

    int getBasisSize() const;
private:
    unsigned int dimensions;

    unsigned int numSpecifiers;

    double *coordinates;

    int *specifiers;

    Model *parentModel;

    friend class FileReader;
};

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

inline int Geometry::getNumSpecifiers() const{
    return numSpecifiers;
}

inline const double* Geometry::getCoordinates(const Index &index) const{
    return &(coordinates[dimensions*parentModel->getBasisIndex(index)]);
}

inline const double* Geometry::getCoordinates(int index) const{
    return &(coordinates[dimensions*index]);
}

inline const double* Geometry::getCoordinates() const{
    return coordinates;
}

inline int Geometry::getSpecifier(const Index &index, int specifier) const{
    return specifiers[numSpecifiers*parentModel->getBasisIndex(index) + specifier];
}

inline int Geometry::getSpecifier(int index, int specifier) const{
    return specifiers[numSpecifiers*index + specifier];
}

inline const int* Geometry::getSpecifiers(const Index& index) const{
    return &(specifiers[numSpecifiers*parentModel->getBasisIndex(index)]);
}

inline const int* Geometry::getSpecifiers(const int index) const{
    return &(specifiers[numSpecifiers*index]);
}

inline const int* Geometry::getSpecifiers() const{
    return specifiers;
}

inline double Geometry::getDistance(int index1, int index2) const{
    double distanceSquared = 0.;
    for(unsigned int n = 0; n < dimensions; n++){
        double difference = coordinates[index1] - coordinates[index2];
        distanceSquared += difference*difference;
    }

    return sqrt(distanceSquared);
}

inline int Geometry::getBasisSize() const{
    return parentModel->getBasisSize();
}

};  //End of namespace TBTK

#endif