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