ManyBodyContext.h

Go to the documentation of this file.

/* 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_MANY_BODY_CONTEXT
#define COM_DAFER45_TBTK_MANY_BODY_CONTEXT

#include "TBTK/BitRegister.h"
#include "TBTK/ExtensiveBitRegister.h"
#include "TBTK/FockSpace.h"
#include "TBTK/FockStateRuleSet.h"
#include "TBTK/InteractionAmplitudeSet.h"
#include "TBTK/SingleParticleContext.h"

#include <memory>

namespace TBTK{

class ManyBodyContext{
public:
//  ManyBodyContext(FockSpace<BitRegister> *fockSpace);
    ManyBodyContext(const SingleParticleContext *singleParticleContext);

//  ManyBodyContext(FockSpace<ExtensiveBitRegister> *fockSpace);

    ~ManyBodyContext();

    bool wrapsBitRegister();

    bool wrapsExtensiveBitRegister();

    FockSpace<BitRegister>* getFockSpaceBitRegister();

    FockSpace<ExtensiveBitRegister>* getFockSpaceExtensiveBitRegister();

    void addFockStateRule(const FockStateRule::WrapperRule rule);

    void addIA(InteractionAmplitude ia);

    const InteractionAmplitudeSet* getInteractionAmplitudeSet() const;

    const FockStateRuleSet& getFockStateRuleSet() const;
private:
    std::shared_ptr<FockSpace<BitRegister>> brFockSpace;

    std::shared_ptr<FockSpace<ExtensiveBitRegister>> ebrFockSpace;

//  std::vector<FockStateRule::WrapperRule> fockStateRules;
    FockStateRuleSet fockStateRuleSet;

    std::shared_ptr<InteractionAmplitudeSet> interactionAmplitudeSet;
};

inline bool ManyBodyContext::wrapsBitRegister(){
    if(brFockSpace.get() != NULL)
        return true;
    else
        return false;
}

inline bool ManyBodyContext::wrapsExtensiveBitRegister(){
    if(ebrFockSpace.get() != NULL)
        return true;
    else
        return false;
}

inline FockSpace<BitRegister>* ManyBodyContext::getFockSpaceBitRegister(){
    return brFockSpace.get();
}

inline FockSpace<ExtensiveBitRegister>* ManyBodyContext::getFockSpaceExtensiveBitRegister(){
    return ebrFockSpace.get();
}

inline void ManyBodyContext::addFockStateRule(const FockStateRule::WrapperRule rule){
    fockStateRuleSet.addFockStateRule(rule);
}

inline void ManyBodyContext::addIA(InteractionAmplitude ia){
    interactionAmplitudeSet.get()->addIA(ia);
}

inline const InteractionAmplitudeSet* ManyBodyContext::getInteractionAmplitudeSet() const{
    return interactionAmplitudeSet.get();
}

inline const FockStateRuleSet& ManyBodyContext::getFockStateRuleSet() const{
    return fockStateRuleSet;
}

};  //End of namespace TBTK

#endif