artdaq_demo  v3_02_00
NthEvent_transfer.cc
1 #include "artdaq/TransferPlugins/TransferInterface.hh"
2 #include "artdaq/TransferPlugins/MakeTransferPlugin.hh"
3 #include "artdaq-core/Data/Fragment.hh"
4 #include "artdaq-core/Utilities/ExceptionHandler.hh"
5 #include "cetlib/BasicPluginFactory.h"
6 
7 #include "messagefacility/MessageLogger/MessageLogger.h"
8 #include "fhiclcpp/ParameterSet.h"
9 
10 #include <boost/tokenizer.hpp>
11 
12 #include <sys/shm.h>
13 #include <memory>
14 #include <iostream>
15 #include <string>
16 #include <limits>
17 #include <sstream>
18 
22 namespace artdaq
23 {
28  class NthEventTransfer : public TransferInterface
29  {
30  public:
38  NthEventTransfer(fhicl::ParameterSet const& ps, artdaq::TransferInterface::Role role);
39 
46  TransferInterface::CopyStatus
47  copyFragment(artdaq::Fragment& fragment, size_t send_timeout_usec) override;
48 
55  TransferInterface::CopyStatus
56  moveFragment(artdaq::Fragment&& fragment) override;
57 
64  int receiveFragment(artdaq::Fragment& fragment,
65  size_t receiveTimeout) override
66  {
67  // nth-event discarding is done at the send side. Pass receive calls through to underlying transfer
68  return physical_transfer_->receiveFragment(fragment, receiveTimeout);
69  }
70 
77  int receiveFragmentHeader(detail::RawFragmentHeader& header, size_t receiveTimeout) override
78  {
79  return physical_transfer_->receiveFragmentHeader(header, receiveTimeout);
80  }
81 
88  int receiveFragmentData(RawDataType* destination, size_t wordCount) override
89  {
90  return physical_transfer_->receiveFragmentData(destination, wordCount);
91  }
92 
97  int source_rank() const override { return physical_transfer_->source_rank(); }
98 
103  int destination_rank() const override { return physical_transfer_->destination_rank(); }
104 
105 
110  bool isRunning() override { return physical_transfer_->isRunning(); }
111 
112  private:
113 
114  bool pass(const artdaq::Fragment&) const;
115 
116  std::unique_ptr<TransferInterface> physical_transfer_;
117  size_t nth_;
118  size_t offset_;
119  };
120 
121  NthEventTransfer::NthEventTransfer(fhicl::ParameterSet const& pset, artdaq::TransferInterface::Role role) :
122  TransferInterface(pset, role)
123  , nth_(pset.get<size_t>("nth")),
124  offset_(pset.get<size_t>("offset", 0))
125  {
126  if (pset.has_key("source_rank") || pset.has_key("destination_rank")) {
127  throw cet::exception("NthEvent") << "The parameters \"source_rank\" and \"destination_rank\" must be explicitly defined in the body of the physical_transfer_plugin table, and not outside of it";
128  }
129 
130 
131  if (offset_ >= nth_) {
132  throw cet::exception("NthEvent") << "Offset value of " << offset_ <<
133  " must not be larger than the modulus value of " << nth_;
134  }
135 
136  if (nth_ == 0)
137  {
138  mf::LogWarning("NthEventTransfer") << "0 was passed as the nth parameter to NthEventTransfer. Will change to 1 (0 is undefined behavior)";
139  nth_ = 1;
140  }
141  // Instantiate the TransferInterface plugin used to effect transfers
142  physical_transfer_ = MakeTransferPlugin(pset, "physical_transfer_plugin", role);
143  }
144 
145 
146  TransferInterface::CopyStatus
147  NthEventTransfer::copyFragment(artdaq::Fragment& fragment,
148  size_t send_timeout_usec)
149  {
150 
151  if (!pass(fragment))
152  {
153  // Do not transfer but return success. Fragment is discarded
154  return TransferInterface::CopyStatus::kSuccess;
155  }
156 
157  // This is the nth Fragment, transfer
158  return physical_transfer_->copyFragment(fragment, send_timeout_usec);
159  }
160 
161  TransferInterface::CopyStatus
162  NthEventTransfer::moveFragment(artdaq::Fragment&& fragment)
163  {
164  if (!pass(fragment))
165  {
166  // Do not transfer but return success. Fragment is discarded
167  return TransferInterface::CopyStatus::kSuccess;
168  }
169 
170  // This is the nth Fragment, transfer
171  return physical_transfer_->moveFragment(std::move(fragment));
172  }
173 
174  bool
175  NthEventTransfer::pass(const artdaq::Fragment& fragment) const
176  {
177  bool passed = false;
178 
179  if (fragment.type() == artdaq::Fragment::DataFragmentType) {
180  passed = (fragment.sequenceID() + nth_ - offset_) % nth_ == 0 ? true : false;
181  }
182  else {
183  passed = true;
184  }
185 
186  return passed;
187  }
188 }
189 
190 DEFINE_ARTDAQ_TRANSFER(artdaq::NthEventTransfer)
191 
192 // Local Variables:
193 // mode: c++
194 // End:
Demonstration TransferInterface plugin showing how to discard events Intended for use in the transfer...
int receiveFragment(artdaq::Fragment &fragment, size_t receiveTimeout) override
Receive a fragment from the transfer plugin.
int receiveFragmentHeader(detail::RawFragmentHeader &header, size_t receiveTimeout) override
Receive a Fragment Header from the transport mechanism.
TransferInterface::CopyStatus copyFragment(artdaq::Fragment &fragment, size_t send_timeout_usec) override
Copy a fragment, using the non-reliable channel.
bool isRunning() override
Determine whether the TransferInterface plugin is able to send/receive data.
NthEventTransfer(fhicl::ParameterSet const &ps, artdaq::TransferInterface::Role role)
NthEventTransfer Constructor.
int destination_rank() const override
Get the destination rank from the physical transfer.
TransferInterface::CopyStatus moveFragment(artdaq::Fragment &&fragment) override
Copy a fragment, using the reliable channel. moveFragment assumes ownership of the fragment...
int receiveFragmentData(RawDataType *destination, size_t wordCount) override
Receive the body of a Fragment to the given destination pointer.
int source_rank() const override
Get the source rank from the physical transfer.