ManyParticleContext.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_MANY_PARTICLE_CONTEXT
#define COM_DAFER45_TBTK_MANY_PARTICLE_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 ManyParticleContext{
public:
    ManyParticleContext();
 
//  ManyParticleContext(FockSpace<BitRegister> *fockSpace);
    ManyParticleContext(
        const SingleParticleContext *singleParticleContext
    );
 
//  ManyParticleContext(FockSpace<ExtensiveBitRegister> *fockSpace);
 
    ~ManyParticleContext();
 
    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 ManyParticleContext::wrapsBitRegister(){
    if(brFockSpace.get() != NULL)
        return true;
    else
        return false;
}
 
inline bool ManyParticleContext::wrapsExtensiveBitRegister(){
    if(ebrFockSpace.get() != NULL)
        return true;
    else
        return false;
}
 
inline FockSpace<BitRegister>* ManyParticleContext::getFockSpaceBitRegister(){
    TBTKAssert(
        wrapsBitRegister(),
        "ManyParticleContext::getFockSpaceBitRegister()",
        "Use ManyParticleContext::getFockSpaceExtensiveBitRegister()"
        << " instead.",
        ""
    );
    return brFockSpace.get();
}
 
inline FockSpace<ExtensiveBitRegister>* ManyParticleContext::getFockSpaceExtensiveBitRegister(){
    TBTKAssert(
        wrapsExtensiveBitRegister(),
        "ManyParticleContext::getFockSpaceExtensiveBitRegister()",
        "Use ManyParticleContext::getFockSpaceBitRegister()"
        << " instead.",
        ""
    );
    return ebrFockSpace.get();
}
 
inline void ManyParticleContext::addFockStateRule(const FockStateRule::WrapperRule rule){
    fockStateRuleSet.addFockStateRule(rule);
}
 
inline void ManyParticleContext::addIA(InteractionAmplitude ia){
    interactionAmplitudeSet.get()->addIA(ia);
}
 
inline const InteractionAmplitudeSet* ManyParticleContext::getInteractionAmplitudeSet() const{
    return interactionAmplitudeSet.get();
}
 
inline const FockStateRuleSet& ManyParticleContext::getFockStateRuleSet() const{
    return fockStateRuleSet;
}
 
};  //End of namespace TBTK
 
#endif