PredictorMfeEns.h

Go to the documentation of this file.

 
#ifndef INTARNA_PREDICTORMFEENS_H_
#define INTARNA_PREDICTORMFEENS_H_
 
 
#include "IntaRNA/PredictorMfe.h"
 
#include "IntaRNA/IndexRangeList.h"
 
#include <list>
#include <utility>
 
namespace IntaRNA {
 
class PredictorMfeEns : public PredictorMfe {
 
 
public:
 
    PredictorMfeEns( const InteractionEnergy & energy
                , OutputHandler & output
                , PredictionTracker * predTracker );
 
    virtual ~PredictorMfeEns() override;
 
protected:
 
    struct ZPartition {
        size_t i1;
        size_t j1;
        size_t i2;
        size_t j2;
        Z_type partZ;
    };
 
    using PredictorMfe::energy;
 
    using PredictorMfe::output;
 
    using PredictorMfe::predTracker;
 
    std::unordered_map<Interaction::Boundary, Z_type, Interaction::Boundary::Hash> Z_partition;
 
    void updateExactCompleteZ(size_t i1, size_t j1, size_t i2, size_t j2,
            Z_type hybrid, Z_type coefficient);
 
    const std::pair<Z_type,Z_type> * exactContribution = nullptr;
 
    bool updateZisComplete;
 
 
    virtual
    void
    initZ();
 
    virtual
    void
    updateZ( const size_t i1, const size_t j1
                , const size_t i2, const size_t j2
                , const Z_type partFunct
                , const bool isHybridZ );
 
    void
    updateCompleteZ( const size_t i1, const size_t j1
                , const size_t i2, const size_t j2
                , const Z_type partFunct
                , const bool isHybridZ );
 
    virtual
    void
    reportOptima() override;
 
    virtual
    void
    traceBack( Interaction & interaction ) override;
 
private:
 
    bool
    addPartitionContribution( const size_t i1, const size_t j1
                , const size_t i2, const size_t j2
                , const Z_type partFunct
                , const bool isHybridZ
                , Z_type & partZ_noED );
 
    virtual
    void
    updateOptimaUsingZ();
 
};
 
} // namespace
 
#endif /* INTARNA_PREDICTORMFEENS_H_ */