test_tbb_dfp_any.cxx File Reference

#include "WireCellUtil/Testing.h"
#include <tbb/flow_graph.h>
#include <boost/any.hpp>
#include <string>
#include <deque>
#include <iostream>

Go to the source code of this file.

Classes
class	MockNode
class	MockSource
class	MockFunction
class	MockSink
class	TbbNodeWrapper
class	TbbSourceBody
class	TbbSourceNodeWrapper
class	TbbSinkBody
class	TbbSinkNodeWrapper

Typedefs
typedef std::shared_ptr< MockNode >	mock_node_pointer
typedef tbb::flow::sender< boost::any >	sender_type
typedef tbb::flow::receiver< boost::any >	receiver_type
typedef std::shared_ptr< sender_type >	sender_port_pointer
typedef std::shared_ptr< receiver_type >	receiver_port_pointer
typedef std::vector< sender_port_pointer >	sender_port_vector
typedef std::vector< receiver_port_pointer >	receiver_port_vector
typedef tbb::flow::source_node< boost::any >	source_node
typedef tbb::flow::function_node< boost::any >	sink_node
typedef std::shared_ptr< TbbNodeWrapper >	TbbNode

Enumerations
enum	NodeCategory { unknownCat, sourceCat, sinkCat, functionCat }

Functions
mock_node_pointer	get_node (const std::string &node_desc)
TbbNode	make_node (tbb::flow::graph &graph, const std::string &node_desc)
bool	connect (TbbNode sender, TbbNode receiver, size_t sport=0, size_t rport=0)
int	main ()

Typedef Documentation

typedef std::shared_ptr<MockNode> mock_node_pointer

Definition at line 26 of file test_tbb_dfp_any.cxx.

typedef std::shared_ptr<receiver_type> receiver_port_pointer

Definition at line 103 of file test_tbb_dfp_any.cxx.

typedef std::vector<receiver_port_pointer> receiver_port_vector

Definition at line 105 of file test_tbb_dfp_any.cxx.

typedef tbb::flow::receiver<boost::any> receiver_type

Definition at line 101 of file test_tbb_dfp_any.cxx.

typedef std::shared_ptr<sender_type> sender_port_pointer

Definition at line 102 of file test_tbb_dfp_any.cxx.

typedef std::vector<sender_port_pointer> sender_port_vector

Definition at line 104 of file test_tbb_dfp_any.cxx.

typedef tbb::flow::sender<boost::any> sender_type

Definition at line 100 of file test_tbb_dfp_any.cxx.

typedef tbb::flow::function_node<boost::any> sink_node

Definition at line 108 of file test_tbb_dfp_any.cxx.

typedef tbb::flow::source_node<boost::any> source_node

Definition at line 107 of file test_tbb_dfp_any.cxx.

typedef std::shared_ptr<TbbNodeWrapper> TbbNode

Definition at line 124 of file test_tbb_dfp_any.cxx.

Enumeration Type Documentation

enum NodeCategory

Enumerator
unknownCat
sourceCat
sinkCat
functionCat

Definition at line 14 of file test_tbb_dfp_any.cxx.

                  {
    unknownCat, sourceCat, sinkCat, functionCat,
};

Function Documentation

bool connect	(	TbbNode	sender,
		TbbNode	receiver,
		size_t	sport = 0,
		size_t	rport = 0
	)

Definition at line 259 of file test_tbb_dfp_any.cxx.

{
    Assert(sender);
    Assert(receiver);
    auto sports = sender->sender_ports();
    auto rports = receiver->receiver_ports();

    Assert(sports.size() > sport);
    Assert(rports.size() > rport);
    
    sender_type* s = sports[sport].get();
    receiver_type* r = rports[rport].get();
    Assert(s);
    Assert(r);

    cerr << "Connecting " << s << " and " << r << endl;
    make_edge(*s, *r);
    return true;
}

mock_node_pointer get_node

(

const std::string &

node_desc

)

Definition at line 82 of file test_tbb_dfp_any.cxx.

{

    if (node_desc == "source") { // note actual desc should be class or class:inst
        return mock_node_pointer(new MockSource);
    }
    if (node_desc == "drift") { // note actual desc should be class or class:inst
        return mock_node_pointer(new MockFunction);
    }
    if (node_desc == "sink") { // note actual desc should be class or class:inst
        return mock_node_pointer(new MockSink);
    }
    return nullptr;
}

int main

(

void

)

Definition at line 279 of file test_tbb_dfp_any.cxx.

{
    tbb::flow::graph graph;
    TbbNode source = make_node(graph, "source");
    Assert(source);
    TbbNode sink = make_node(graph, "sink");
    Assert(sink);

    Assert (connect(source, sink));

    sink->initialize();
    source->initialize();


    graph.wait_for_all();

    return 0;
}

TbbNode make_node	(	tbb::flow::graph &	graph,
		const std::string &	node_desc
	)

Definition at line 236 of file test_tbb_dfp_any.cxx.

{
    mock_node_pointer wcnode = get_node(node_desc);
    if (! wcnode) {
        cerr << "Failed to get node for " << node_desc << endl; 
        return nullptr;
    }

    cerr << "Getting node from category: " << wcnode->category() << endl;
    switch (wcnode->category()) {
    case sourceCat:
        return TbbNode(new TbbSourceNodeWrapper(graph, wcnode));
    case sinkCat:
        return TbbNode(new TbbSinkNodeWrapper(graph, wcnode));
    // case functionCat:
    //  return TbbNode(new TbbFunctionNodeWrapper(garph, wcnode));
    default:
        return nullptr;
    }
    return nullptr;
}