/*
  *  MoMEMta: a modular implementation of the Matrix Element Method
  *  Copyright (C) 2016  Universite catholique de Louvain (UCL), Belgium
  *
  *  This program 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.
  *
  *  This program 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/>.
  */
 
 #include <momemta/Module.h>
 
 #include <sstream>
 
 #include <momemta/ParameterSet.h>
 #include <momemta/Solution.h>
 #include <momemta/Types.h>
 
 template<typename T>
 class Printer: public Module {
     public:
 
         Printer(PoolPtr pool, const ParameterSet& parameters): Module(pool, parameters.getModuleName()) {
             auto tag = parameters.get<InputTag>("input");
             input = pool->get<T>(tag);
             name = parameters.get<std::string>("name", tag.toString());
         };
 
         virtual Status work() override {
             LOG(info) << name << ": " << *input;
 
             return Status::OK;
         }
 
     private:
         std::string name;
 
         // Inputs
         Value<T> input;
 };
 
 template<typename T>
 class Printer<std::vector<T>>: public Module {
     public:
 
         Printer(PoolPtr pool, const ParameterSet& parameters): Module(pool, parameters.getModuleName()) {
             auto tag = parameters.get<InputTag>("input");
             input = pool->get<std::vector<T>>(tag);
             name = parameters.get<std::string>("name", tag.toString());
         };
 
         virtual Status work() override {
             std::stringstream str;
             str << name << ": {";
 
             size_t count = 1;
             for (const auto& i: *input) {
                 str << i;
                 if (count < input->size())
                     str << ", ";
                 count++;
             }
             LOG(info) << str.str() << "}";
 
             return Status::OK;
         }
 
     private:
         std::string name;
 
         // Inputs
         Value<std::vector<T>> input;
 };
 
 REGISTER_MODULE_NAME("IntPrinter", Printer<int64_t>)
         .Input("input")
         .OptionalAttr("name:string")
         .Sticky();
 REGISTER_MODULE_NAME("DoublePrinter", Printer<double>)
         .Input("input")
         .OptionalAttr("name:string")
         .Sticky();
 REGISTER_MODULE_NAME("P4Printer", Printer<LorentzVector>)
         .Input("input")
         .OptionalAttr("name:string")
         .Sticky();
 REGISTER_MODULE_NAME("SolutionPrinter", Printer<Solution>)
         .Input("input")
         .OptionalAttr("name:string")
         .Sticky();
 
 REGISTER_MODULE_NAME("IntVectorPrinter", Printer<std::vector<int64_t>>)
         .Input("input")
         .OptionalAttr("name:string")
         .Sticky();
 REGISTER_MODULE_NAME("DoubleVectorPrinter", Printer<std::vector<double>>)
         .Input("input")
         .OptionalAttr("name:string")
         .Sticky();
 REGISTER_MODULE_NAME("P4VectorPrinter", Printer<std::vector<LorentzVector>>)
         .Input("input")
         .OptionalAttr("name:string")
         .Sticky();
 REGISTER_MODULE_NAME("SolutionVectorPrinter", Printer<SolutionCollection>)
         .Input("input")
         .OptionalAttr("name:string")
         .Sticky();