src.main/CjNodeSelector.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 _CJNODESELECTOR_H_
#define _CJNODESELECTOR_H_ 1

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

namespace jmitie {

class CjNodeSelector;


struct CjNodeSelector_ctor_va_t {
        bool m_src_dest;
        bool m_add_remove;
        boost::mt19937 * m_rng;
        const CjNLTopology * m_top;
        std::string m_parent_name;
        
        CjNodeSelector_ctor_va_t( CjNLTopology * top, boost::mt19937 * rng, bool isDest, bool isRemove, std::string parent_name ):m_src_dest(isDest),m_add_remove(isRemove),m_rng(rng),m_top(top),m_parent_name(parent_name) {}
        CjNodeSelector_ctor_va_t( const CjNetworkProcess_ctor_va_t & temp, bool isDest, bool isRemove, std::string parent_name ):m_src_dest(isDest),m_add_remove(isRemove),m_rng(temp.m_rng),m_top(temp.m_top),m_parent_name(parent_name) {}
};


struct CjNodeSelector_select_va_t {
        int m_epoch;
        boost::mt19937 * m_rng;
        

        std::vector<int> * m_invalid_choice;
        
        int m_start_node;
        bool m_skip_start;

        bool m_skip_fullconn;
        bool m_selfloop_ok;

        CjNodeSelector_select_va_t( int epoch, boost::mt19937 * rng, int start_node = -1, bool skip_start = true, bool skip_fullconn = true, bool selfloop_ok = false ): \
                m_epoch(epoch),
                m_rng(rng),
                m_invalid_choice(0),
                m_start_node(start_node),
                m_skip_start(skip_start),
                m_skip_fullconn(skip_fullconn),
                m_selfloop_ok(selfloop_ok)
                {}
        CjNodeSelector_select_va_t( const CjNetworkProcess_act_va_t & temp, int start_node = -1, bool skip_start = true, bool skip_fullconn = true, bool selfloop_ok = false ): \
                m_epoch(temp.m_epoch),
                m_rng(temp.m_rng),
                m_invalid_choice(0),
                m_start_node(start_node),
                m_skip_start(skip_start),
                m_skip_fullconn(skip_fullconn),
                m_selfloop_ok(selfloop_ok)
                {}

        CjNodeSelector_select_va_t( int epoch, std::vector<int> * invalid_choice, boost::mt19937 * rng, int start_node = -1, bool skip_start = true, bool skip_fullconn = true, bool selfloop_ok = false ): \
                m_epoch(epoch),
                m_rng(rng),
                m_invalid_choice(invalid_choice),
                m_start_node(start_node),
                m_skip_start(skip_start),
                m_skip_fullconn(skip_fullconn),
                m_selfloop_ok(selfloop_ok)
                {}
        CjNodeSelector_select_va_t( std::vector<int> * invalid_choice, const CjNetworkProcess_act_va_t & temp, int start_node = -1, bool skip_start = true, bool skip_fullconn = true, bool selfloop_ok = false ): \
                m_epoch(temp.m_epoch),
                m_rng(temp.m_rng),
                m_invalid_choice(invalid_choice),
                m_start_node(start_node),
                m_skip_start(skip_start),
                m_skip_fullconn(skip_fullconn),
                m_selfloop_ok(selfloop_ok)
                {}
};

class CjNodeSelector
{
public:
typedef CjNodeSelector_ctor_va_t ctor_va_t;
typedef CjNodeSelector_select_va_t select_arg_t;

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

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


virtual bool selectNode( select_arg_t &, unsigned int & node ) = 0;
virtual ~CjNodeSelector() {};

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


private:

};

} // namespace jmitie

#endif // _CJNODESELECTOR_H_