ParallelepipedArrayState.h

/* Copyright 2017 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_PARALLELEPIPED_ARRAY_STATE
#define COM_DAFER45_TBTK_PARALLELEPIPED_ARRAY_STATE
 
#include "TBTK/ArrayState.h"
#include "TBTK/Field.h"
#include "TBTK/ParallelepipedCell.h"
 
namespace TBTK{
 
class ParallelepipedArrayState : public ArrayState, public Field<std::complex<double>, double>{
public:
    ParallelepipedArrayState(
        const std::vector<std::vector<double>> &basisVectors,
        std::initializer_list<unsigned int> numMeshPoints
    );
 
    virtual ~ParallelepipedArrayState();
 
    virtual ParallelepipedArrayState* clone() const;
 
    virtual std::complex<double> getOverlap(const AbstractState &bra) const;
 
    virtual std::complex<double> getMatrixElement(
        const AbstractState &bra,
        const AbstractOperator &o = DefaultOperator()
    ) const;
 
    void setAmplitude(
        std::complex<double> amplitude,
        std::initializer_list<double> coordinate
    );
 
    const std::complex<double>& getAmplitude(
        std::initializer_list<double> coordinate
    ) const;
 
    const std::complex<double>& getAmplitude(
        const std::vector<double> &coordinate
    ) const;
 
    virtual std::complex<double> operator()(std::initializer_list<double> argument) const;
 
    virtual const std::vector<double>& getCoordinates() const;
 
    virtual double getExtent() const;
private:
    ParallelepipedCell parallelepiped;
};
 
};  //End of namespace TBTK
 
#endif