HiddenMarkovModel.h

Go to the documentation of this file.

// --------------------------------------------------------------------------
//                   OpenMS -- Open-Source Mass Spectrometry
// --------------------------------------------------------------------------
// Copyright The OpenMS Team -- Eberhard Karls University Tuebingen,
// ETH Zurich, and Freie Universitaet Berlin 2002-2012.
//
// This software is released under a three-clause BSD license:
//  * Redistributions of source code must retain the above copyright
//    notice, this list of conditions and the following disclaimer.
//  * Redistributions in binary form must reproduce the above copyright
//    notice, this list of conditions and the following disclaimer in the
//    documentation and/or other materials provided with the distribution.
//  * Neither the name of any author or any participating institution
//    may be used to endorse or promote products derived from this software
//    without specific prior written permission.
// For a full list of authors, refer to the file AUTHORS.
// --------------------------------------------------------------------------
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
// ARE DISCLAIMED. IN NO EVENT SHALL ANY OF THE AUTHORS OR THE CONTRIBUTING
// INSTITUTIONS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
// OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
// WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
// OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
// ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
//
// --------------------------------------------------------------------------
// $Maintainer: Andreas Bertsch $
// $Authors: Andreas Bertsch $
// --------------------------------------------------------------------------


#ifndef OPENMS_ANALYSIS_ID_HIDDENMARKOVMODEL_H
#define OPENMS_ANALYSIS_ID_HIDDENMARKOVMODEL_H

#include <vector>
#include <set>

#include <OpenMS/DATASTRUCTURES/Map.h>
#include <OpenMS/CONCEPT/Types.h>
#include <OpenMS/DATASTRUCTURES/StringList.h>
#include <OpenMS/DATASTRUCTURES/String.h>

#include <utility>

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;

    DoubleReal getTransitionProbability(const String & s1, const String & s2) const;

    void setTransitionProbability(const String & s1, const String & s2, DoubleReal 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, DoubleReal prob);

    void clearInitialTransitionProbabilities();

    void setTrainingEmissionProbability(const String & state, DoubleReal prob);

    void clearTrainingEmissionProbabilities();

    void enableTransition(const String & s1, const String & s2);

    void disableTransition(const String & s1, const String & s2);

    void disableTransitions();

    void calculateEmissionProbabilities(Map<HMMState *, DoubleReal> & 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(DoubleReal pseudo_counts);

    DoubleReal getPseudoCounts() const;

    void setVariableModifications(const StringList & modifications);

protected:

    void disableTransition_(HMMState * s1, HMMState * s2);

    void enableTransition_(HMMState * s1, HMMState * s2);

    void setTrainingEmissionProbability_(HMMState * state, DoubleReal prob);

    void setTransitionProbability_(HMMState * s1, HMMState * s2, DoubleReal prob);

    DoubleReal getTransitionProbability_(HMMState * s1, HMMState * s2) const;


    void calculateForwardPart_();

    void calculateBackwardPart_();

    DoubleReal getForwardVariable_(HMMState *);

    DoubleReal getBackwardVariable_(HMMState *);

private:

    // transition probs
    Map<HMMState *, Map<HMMState *, DoubleReal> > trans_;

    // transition prob counts
    Map<HMMState *, Map<HMMState *, DoubleReal> > count_trans_;

    Map<HMMState *, Map<HMMState *, std::vector<DoubleReal> > > count_trans_all_;

    // all transition probs of all training steps (for model checking)
    Map<HMMState *, Map<HMMState *, std::vector<DoubleReal> > > train_count_trans_all_;

    // number of training steps of the transitions
    Map<HMMState *, Map<HMMState *, Size> > training_steps_count_;

    // forward variables
    Map<HMMState *, DoubleReal> forward_;

    // backward variables
    Map<HMMState *, DoubleReal> backward_;

    // name to state Mapping
    Map<String, HMMState *> name_to_state_;

    // emission probabilities
    Map<HMMState *, DoubleReal> train_emission_prob_;

    // initial transition probabilities
    Map<HMMState *, DoubleReal> init_prob_;

    // all states of the HMM
    std::set<HMMState *> states_;

    // trained transitions
    std::set<std::pair<HMMState *, HMMState *> > trained_trans_;

    // synonym transitions Mapping
    Map<String, Map<String, std::pair<String, String> > > synonym_trans_names_;

    // synonym transitions
    Map<HMMState *, Map<HMMState *, std::pair<HMMState *, HMMState *> > > synonym_trans_;

    // transitions which are enabled
    Map<HMMState *, std::set<HMMState *> > enabled_trans_;

    // pseudocounts used in this instance
    DoubleReal pseudo_counts_;

    // copy all the stuff from one HMM to this
    void copy_(const HiddenMarkovModel & source);

    StringList var_modifications_;
  };
}
#endif