test_tbb_dfp_mce.cxx File Reference

#include "tbb_mock.h"
#include "WireCellTbb/NodeWrapper.h"
#include "WireCellTbb/SinkCat.h"
#include "WireCellTbb/SourceCat.h"
#include "WireCellTbb/QueuedoutCat.h"
#include "WireCellUtil/Testing.h"
#include <tbb/flow_graph.h>
#include <string>
#include <deque>
#include <iostream>

Go to the source code of this file.

Functions
WireCell::INode::pointer	get_node (const std::string &node_desc)
WireCellTbb::Node	make_node (tbb::flow::graph &graph, const std::string &node_desc)
bool	connect (WireCellTbb::Node sender, WireCellTbb::Node receiver, size_t sport=0, size_t rport=0)
int	main ()

Function Documentation

bool connect	(	WireCellTbb::Node	sender,
		WireCellTbb::Node	receiver,
		size_t	sport = 0,
		size_t	rport = 0
	)

Definition at line 68 of file test_tbb_dfp_mce.cxx.

{
    using namespace WireCellTbb;

    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];
    receiver_type* r = rports[rport];
    Assert(s);
    Assert(r);

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

WireCell::INode::pointer get_node

(

const std::string &

node_desc

)

Definition at line 22 of file test_tbb_dfp_mce.cxx.

{
    using namespace WireCell;

    if (node_desc == "source") { // note actual desc should be class or class:inst
        return INode::pointer(new WireCellTbb::MockDepoSource);
    }
    if (node_desc == "drift") { // note actual desc should be class or class:inst
        return INode::pointer(new WireCellTbb::MockDrifter);
    }
    if (node_desc == "sink") { // note actual desc should be class or class:inst
        return INode::pointer(new WireCellTbb::MockDepoSink);
    }
    return nullptr;
}

int main

(

void

)

Definition at line 90 of file test_tbb_dfp_mce.cxx.

{
    using namespace WireCellTbb;

    tbb::flow::graph graph;
    Node source = make_node(graph, "source");
    Assert(source);
    Node drift = make_node(graph, "drift");
    Assert(drift);
    Node sink = make_node(graph, "sink");
    Assert(sink);

    Assert (connect(source, drift));
    Assert (connect(drift, sink));

    // fixme: in general all nodes should be initialize()'d but so far only source nodes need it.
    source->initialize();

    graph.wait_for_all();

    return 0;
}

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

Definition at line 40 of file test_tbb_dfp_mce.cxx.

{
    using namespace WireCell;
    using namespace WireCellTbb;

    INode::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 INode::sourceNode: 
        return Node(new SourceNodeWrapper(graph, wcnode));
    case INode::sinkNode:
        return Node(new SinkNodeWrapper(graph, wcnode));
    // case INode::functionNode:
    //  return Node(new FunctionWrapper(graph, wcnode));
    case INode::queuedoutNode:
        return Node(new QueuedoutWrapper(graph, wcnode));
    default:
        return nullptr;
    }
    return nullptr;
}