/shared/development/google_code/rssoft/library/lib/MultiplicityMatrix.h

Go to the documentation of this file.

/*
 Copyright 2013 Edouard Griffiths <f4exb at free dot fr>

 This file is part of RSSoft. A Reed-Solomon Soft Decoding library

 This program is free software; you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation; either version 2 of the License, or
 (at your option) any later version.

 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with this program; if not, write to the Free Software
 Foundation, Inc., 51 Franklin Street, Boston, MA  02110-1301  USA

 Multiplicity Matrix class

 */
 #ifndef __MULTIPLICITY_MATRIX_H__
 #define __MULTIPLICITY_MATRIX_H__
 
 #include <utility>
 #include <map>
 #include <iostream>

namespace rssoft
{

class ReliabilityMatrix;

class MultiplicityMatrix_SparseOrdering
{
public:
    MultiplicityMatrix_SparseOrdering() {}
    ~MultiplicityMatrix_SparseOrdering() {}
    
    bool operator()(const std::pair<unsigned int, unsigned int>& index1, const std::pair<unsigned int, unsigned int>& index2) const
    {
        if (index1.second == index2.second) // same column
        {
            return index1.first < index2.first; // row order
        }
        else
        {
            return index1.second < index2.second; // column order
        }
    }
};

class MultiplicityMatrix : public std::map<std::pair<unsigned int, unsigned int>, unsigned int, MultiplicityMatrix_SparseOrdering>
{
public:
    class traversing_iterator : public std::map<std::pair<unsigned int, unsigned int>, unsigned int, MultiplicityMatrix_SparseOrdering>::const_iterator
    {
    public:
        traversing_iterator(std::map<std::pair<unsigned int, unsigned int>, unsigned int, MultiplicityMatrix_SparseOrdering>::const_iterator const &c) :
                std::map<std::pair<unsigned int, unsigned int>, unsigned int, MultiplicityMatrix_SparseOrdering>::const_iterator(c) {}

        unsigned int iX()
        {
                std::map<std::pair<unsigned int, unsigned int>, unsigned int, MultiplicityMatrix_SparseOrdering>::const_iterator *c_it = this;
                return (*c_it)->first.second;
        }

        unsigned int iY()
        {
                std::map<std::pair<unsigned int, unsigned int>, unsigned int, MultiplicityMatrix_SparseOrdering>::const_iterator *c_it = this;
                return (*c_it)->first.first;
        }

        unsigned int multiplicity()
        {
                std::map<std::pair<unsigned int, unsigned int>, unsigned int, MultiplicityMatrix_SparseOrdering>::const_iterator *c_it = this;
                return (*c_it)->second;
        }
    };

    MultiplicityMatrix(const ReliabilityMatrix& relmat, unsigned int multiplicity, bool soft_decision=true);

    MultiplicityMatrix(const ReliabilityMatrix& relmat, float lambda);
    
    ~MultiplicityMatrix();
    
    unsigned int cost() const
    {
        return _cost;
    }
    
        unsigned int get_nb_symbols_log2() const
        {
                return _nb_symbols_log2;
        }

        unsigned int get_nb_symbols() const
        {
                return _nb_symbols;
        }

        unsigned int get_message_length() const
        {
                return _message_length;
        }

        unsigned int operator()(unsigned int i_row, unsigned int i_col) const;

        friend std::ostream& operator <<(std::ostream& os, const MultiplicityMatrix& matrix);
    

protected:
    typedef std::map<std::pair<unsigned int, unsigned int>, unsigned int, MultiplicityMatrix_SparseOrdering>::const_iterator const_iterator;

    typedef std::map<std::pair<unsigned int, unsigned int>, unsigned int, MultiplicityMatrix_SparseOrdering>::iterator iterator;

        iterator operator()(unsigned int i_row, unsigned int i_col)
    {
        return find(std::make_pair(i_row, i_col));
    }

        unsigned int _nb_symbols_log2; 
        unsigned int _nb_symbols; 
        unsigned int _message_length; 
    unsigned int _cost; 
};
 
} // namespace rssoft

 #endif // __MULTIPLICITY_MATRIX_H__