TFFifoHandler.cxx

Go to the documentation of this file.

///////////////////////////////////////////////////////////////////////
// TFFifoHander
//
// A module to evaluate a tensorflow model using FIFO connections to python
//
// \author $Author: Alex Himmel
////////////////////////////////////////////////////////////////////////

#include  <iostream>
#include  <string>
#include "TensorFlowHandler/TFFifoHandler.h"

namespace tensorflow
{
  TFFifoHandler::TFFifoHandler(fhicl::ParameterSet const &pset) :
    fToTF(pset.get<std::string>("PipeToTF")),
    fFromTF(pset.get<std::string>("PipeFromTF"))
  {
  }

  TFFifoHandler::~TFFifoHandler()
  {
  }

  std::vector<std::vector<double> > TFFifoHandler::Predict(const std::vector<cvn::PixelMap const*> pixellist) const
  {

    //
    // loop over pixel maps to send all to TF together
    //

    // Open the output pipe
    std::ofstream totf;
    totf.open(fToTF, std::ios::out);

    int nimages = pixellist.size();
    int nviews = 2;

    // Print summary info first to pipe
    totf << nimages << " " << nviews << " "<< pixellist[0]->NPlanePerView() << " " << pixellist[0]->NCell() << std::endl;

    // Loop over pixel maps and print them to the pipe
    for(size_t iP = 0; iP < pixellist.size(); ++iP) {
      // Print this pixel map to pipe
      pixellist[iP]->ToFile(totf, iP, true);
    }
    
    // Close the pipe
    totf.close();

    
    //
    // Read results back from all pixel maps
    //
    
    std::vector<std::vector<double>> tfoutput;

    // Open the pipe
    std::ifstream fromtf;
    fromtf.open(fFromTF);

    // Read results from the file
    // Each line corresponds to a pixel map
    std::string line; 
    while (std::getline(fromtf, line))
    {
      //std::cout << "Debug:" << line << std::endl;
      std::vector<std::string> parts;
      std::vector<double> results;
      boost::split(parts, line, boost::is_any_of(" "));
      for (std::string p : parts) {
        if (!p.empty())
          results.push_back( std::stof(p) );
      }
      tfoutput.emplace_back(results);
    }

    return tfoutput;
  }
}