IntVectorAnalyzer_module.cc

Go to the documentation of this file.

// vim: set sw=2 expandtab :

#include "art/Framework/Core/ModuleMacros.h"
#include "art/Framework/Core/SharedAnalyzer.h"
#include "art/Framework/Principal/Event.h"
#include "art/Framework/Principal/View.h"
#include "cetlib_except/exception.h"
#include "fhiclcpp/types/Atom.h"

#include <cassert>
#include <cstddef>
#include <iostream>
#include <string>
#include <vector>

using namespace std;
using namespace art;
using namespace fhicl;

namespace arttest {

  class IntVectorAnalyzer : public SharedAnalyzer {
  public:
    struct Config {
      Atom<string> input_label{Name("input_label")};
      Atom<size_t> nvalues{Name("nvalues")};
    };
    using Parameters = Table<Config>;
    explicit IntVectorAnalyzer(Parameters const& p, ProcessingFrame const&);

  private:
    void analyze(Event const&, ProcessingFrame const&) override;

    string const moduleLabel_;
    size_t const nvalues_;
    ViewToken<int> const viewToken_;
  };

  IntVectorAnalyzer::IntVectorAnalyzer(Parameters const& p,
                                       ProcessingFrame const&)
    : SharedAnalyzer{p}
    , moduleLabel_{p().input_label()}
    , nvalues_{p().nvalues()}
    , viewToken_{consumesView<int>(moduleLabel_)}
  {
    async<InEvent>();
  }

  void
  IntVectorAnalyzer::analyze(Event const& e, ProcessingFrame const&)
  {
    {
      vector<int const*> ptrs;
      size_t sz = e.getView(moduleLabel_, ptrs);
      if (sz != nvalues_) {
        cerr << "SizeMismatch expected a view of size " << nvalues_
             << " but the obtained size is " << sz << '\n';
        throw cet::exception("SizeMismatch")
          << "Expected a view of size " << nvalues_
          << " but the obtained size is " << sz << '\n';
      }
      for (size_t k = 0; k != sz; ++k) {
        if (*ptrs[k] != (int)(e.id().event() + k)) {
          cerr << "ValueMismatch at position " << k << " expected value "
               << e.id().event() + k << " but obtained " << *ptrs[k] << '\n';
          throw cet::exception("ValueMismatch")
            << "At position " << k << " expected value " << e.id().event() + k
            << " but obtained " << *ptrs[k] << '\n';
        }
      }
    }
    {
      View<int> vw;
      e.getView(viewToken_, vw);
      assert(vw.isValid());
      // Test that the range-for loop compiles.
      for (auto it : vw) {
        assert(*it >= 0);
      }
    }
  }

} // namespace arttest

DEFINE_ART_MODULE(arttest::IntVectorAnalyzer)