HiddenMarkovModel.h

Go to the documentation of this file.

// Copyright (c) 2002-2023, The OpenMS Team -- EKU Tuebingen, ETH Zurich, and FU Berlin
// SPDX-License-Identifier: BSD-3-Clause
//
// --------------------------------------------------------------------------
// $Maintainer: Timo Sachsenberg $
// $Authors: Andreas Bertsch $
// --------------------------------------------------------------------------
 
 
#pragma once
 
#include <vector>
#include <set>
 
#include <OpenMS/CONCEPT/Types.h>
#include <OpenMS/DATASTRUCTURES/StringListUtils.h>
#include <OpenMS/DATASTRUCTURES/String.h>
 
#include <utility>
#include <map>
 
namespace OpenMS
{
  class OPENMS_DLLAPI HMMState
  {
public:
 
    HMMState();
 
    HMMState(const HMMState & state);
 
    HMMState(const String & name, bool hidden = true);
 
    virtual ~HMMState();
 
    HMMState & operator=(const HMMState &);
 
    void setName(const String & name);
 
    const String & getName() const;
 
    void setHidden(bool hidden);
 
    bool isHidden() const;
 
    void addPredecessorState(HMMState * state);
 
    void deletePredecessorState(HMMState * state);
 
    void addSuccessorState(HMMState * state);
 
    void deleteSuccessorState(HMMState * state);
 
    const std::set<HMMState *> & getPredecessorStates() const;
 
    const std::set<HMMState *> & getSuccessorStates() const;
 
protected:
 
    bool hidden_;
 
    String name_;
 
    std::set<HMMState *> pre_states_;
 
    std::set<HMMState *> succ_states_;
  };
 
 
  class OPENMS_DLLAPI HiddenMarkovModel
  {
public:
 
    HiddenMarkovModel();
 
    HiddenMarkovModel(const HiddenMarkovModel & hmm_new);
 
    virtual ~HiddenMarkovModel();
 
    HiddenMarkovModel & operator=(const HiddenMarkovModel &);
 
    void writeGraphMLFile(const String & filename);
 
    void write(std::ostream & out) const;
 
    double getTransitionProbability(const String & s1, const String & s2) const;
 
    void setTransitionProbability(const String & s1, const String & s2, double prob);
 
    Size getNumberOfStates() const;
 
    void addNewState(HMMState * state);
 
    void addNewState(const String & name);
 
    void addSynonymTransition(const String & name1, const String & name2, const String & synonym1, const String & synonym2);
 
    void evaluate();
 
    void train();
 
    void setInitialTransitionProbability(const String & state, double prob);
 
    void clearInitialTransitionProbabilities();
 
    void setTrainingEmissionProbability(const String & state, double prob);
 
    void clearTrainingEmissionProbabilities();
 
    void enableTransition(const String & s1, const String & s2);
 
    void disableTransition(const String & s1, const String & s2);
 
    void disableTransitions();
 
    void calculateEmissionProbabilities(std::map<HMMState *, double> & emission_probs);
 
    void dump();
 
    void forwardDump();
 
    //void buildSynonyms();
 
    void estimateUntrainedTransitions();
 
    HMMState * getState(const String & name);
 
    const HMMState * getState(const String & name) const;
 
    void clear();
 
    void setPseudoCounts(double pseudo_counts);
 
    double getPseudoCounts() const;
 
    void setVariableModifications(const StringList & modifications);
 
protected:
 
    void disableTransition_(HMMState * s1, HMMState * s2);
 
    void enableTransition_(HMMState * s1, HMMState * s2);
 
    void setTrainingEmissionProbability_(HMMState * state, double prob);
 
    void setTransitionProbability_(HMMState * s1, HMMState * s2, double prob);
 
    double getTransitionProbability_(HMMState * s1, HMMState * s2) const;
 
 
    void calculateForwardPart_();
 
    void calculateBackwardPart_();
 
    double getForwardVariable_(HMMState *);
 
    double getBackwardVariable_(HMMState *);
 
private:
 
    // transition probs
    std::map<HMMState *, std::map<HMMState *, double> > trans_;
 
    // transition prob counts
    std::map<HMMState *, std::map<HMMState *, double> > count_trans_;
 
    std::map<HMMState *, std::map<HMMState *, std::vector<double> > > count_trans_all_;
 
    // all transition probs of all training steps (for model checking)
    std::map<HMMState *, std::map<HMMState *, std::vector<double> > > train_count_trans_all_;
 
    // number of training steps of the transitions
    std::map<HMMState *, std::map<HMMState *, Size> > training_steps_count_;
 
    // forward variables
    std::map<HMMState *, double> forward_;
 
    // backward variables
    std::map<HMMState *, double> backward_;
 
    // name to state Mapping
    std::map<String, HMMState *> name_to_state_;
 
    // emission probabilities
    std::map<HMMState *, double> train_emission_prob_;
 
    // initial transition probabilities
    std::map<HMMState *, double> init_prob_;
 
    // all states of the HMM
    std::set<HMMState *> states_;
 
    // trained transitions
    std::set<std::pair<HMMState *, HMMState *> > trained_trans_;
 
    // synonym transitions Mapping
    std::map<String, std::map<String, std::pair<String, String> > > synonym_trans_names_;
 
    // synonym transitions
    std::map<HMMState *, std::map<HMMState *, std::pair<HMMState *, HMMState *> > > synonym_trans_;
 
    // transitions which are enabled
    std::map<HMMState *, std::set<HMMState *> > enabled_trans_;
 
    // pseudocounts used in this instance
    double pseudo_counts_;
 
    // copy all the stuff from one HMM to this
    void copy_(const HiddenMarkovModel & source);
 
    StringList var_modifications_;
  };
}