src.main/CjLinkSelector.hh

/* -*- Mode: C; indent-tabs-mode: t; c-basic-offset: 4; tab-width: 4 -*- */
/*
 * mitie
 * Copyright (C) Jason Spencer 2009 <mitie@jasonspencer.org>
 * 
 * mitie 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 3 of the License, or
 * (at your option) any later version.
 * 
 * mitie 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, see <http://www.gnu.org/licenses/>.
 */

#ifndef _CJLINKSELECTOR_H_
#define _CJLINKSELECTOR_H_ 1

#include <string>
#include "CjNLTopology.hh"
#include "CjMatrix.hh"
#include <boost/random/mersenne_twister.hpp>
#include "CjNetworkProcess.hh"

namespace jmitie {

struct CjLinkSelector_ctor_va_t {
        bool m_add_remove;
        boost::mt19937 * m_rng;
        const CjNLTopology * m_top;
        std::string m_parent_name;
        
        CjLinkSelector_ctor_va_t( CjNLTopology * top, boost::mt19937 * rng, bool isRemover ):m_add_remove(isRemover),m_rng(rng),m_top(top),m_parent_name("unknown object") {}
        CjLinkSelector_ctor_va_t( const CjNetworkProcess_ctor_va_t & temp, bool isRemover, std::string parent_name ):m_add_remove(isRemover),m_rng(temp.m_rng),m_top(temp.m_top),m_parent_name(parent_name) {}
};

struct CjLinkSelector_select_va_t {
        int m_epoch;
        boost::mt19937 * const m_rng;
        
        unsigned int m_node_a;
        unsigned int m_node_b;

        bool m_selfloop_ok;
        bool m_bridge_ok;

        CjLinkSelector_select_va_t( int epoch, boost::mt19937 * rng, bool selfloop_ok, bool bridge_ok ): \
                m_epoch(epoch),
                m_rng(rng),
                m_selfloop_ok(selfloop_ok),
                m_bridge_ok(bridge_ok)
                {}

        CjLinkSelector_select_va_t( const CjNetworkProcess_act_va_t & temp, bool selfloop_ok, bool bridge_ok ): \
                m_epoch(temp.m_epoch),
                m_rng(temp.m_rng),
                m_selfloop_ok(selfloop_ok),
                m_bridge_ok(bridge_ok)
                {}
};

struct wheelVal_t {
        union weight_t {
                double as_double;
                int as_int;
                unsigned int as_uint;
        } weight;
        unsigned int node_a;
        unsigned int node_b;
        bool valid;
        wheelVal_t( unsigned int node_a_, unsigned int node_b_ ):node_a(node_a_),node_b(node_b_),valid(true)  { }
        wheelVal_t( double weight_, unsigned int node_a_, unsigned int node_b_ ):node_a(node_a_),node_b(node_b_),valid(true)  { weight.as_double = weight_; }
        wheelVal_t( int weight_, unsigned int node_a_, unsigned int node_b_, int ):node_a(node_a_),node_b(node_b_),valid(true)  { weight.as_int = weight_; }
        wheelVal_t( unsigned int weight_, unsigned int node_a_, unsigned int node_b_, int, int ):node_a(node_a_),node_b(node_b_),valid(true)  { weight.as_uint = weight_; }
};

class CjLinkSelector
{
public:
typedef CjLinkSelector_ctor_va_t ctor_va_t;     // the variable argument type for the derived constructor
typedef CjLinkSelector_select_va_t select_arg_t;        // the variable argument type for the selectNode member function

static std::string getName_static() { throw std::logic_error("CjLinkSelector::getName_static() called."); }
static std::string getProperty_static(const std::string &) { throw std::logic_error("CjLinkSelector::getProperty_static(..) called."); }
virtual std::string getName() const = 0;
virtual std::string getProperty(const std::string &) const = 0;

CjLinkSelector(const ctor_va_t & va):m_top(va.m_top),m_add_remove(va.m_add_remove),m_parent_name(va.m_parent_name) {}

virtual bool selectLink( select_arg_t &, unsigned int & from, unsigned int & to  ) = 0;
virtual ~CjLinkSelector() {};

protected:
const CjNLTopology * m_top;
bool m_add_remove;
std::string m_parent_name;

static std::vector<wheelVal_t> m_wheel;

private:

};

} // namespace jmitie

#endif // _CJLINKSELECTOR_H_