task.stats.hpp

Go to the documentation of this file.

// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use these files except in compliance with the License.
// You may obtain a copy of the License at
//
//    http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

// Copyright 2012 - Gonzalo Iglesias, Adrià de Gispert, William Byrne

#ifndef STATSOUTPUTTASK_HPP
#define STATSOUTPUTTASK_HPP

namespace ucam {
namespace fsttools {

template <class Data>
class SpeedStatsTask: public ucam::util::TaskInterface<Data> {
 private:
  typedef ucam::util::oszfstream oszfstream;

  ucam::util::IntegerPatternAddress statsoutput_;

  Data *d_;

 public:

  SpeedStatsTask ( const ucam::util::RegistryPO& rg ) :
    d_ ( NULL ),
    statsoutput_ ( rg.get<std::string> ( HifstConstants::kStatsWrite ) ) {
  };

  bool run ( Data& d ) {
    if ( statsoutput_() == "" ) return false;
    d_ = &d;
    FORCELINFO ( "Writing stats to "  << statsoutput_ ( d.sidx ) );
    oszfstream o ( statsoutput_ ( d.sidx ) );
    o << "=================================================================" <<
      endl;
    o << "Sentence " << d.sidx << ": Time (ms):" << endl;
    writeSpeedStats ( o );
    o << "-----------------------------------------------------------------" <<
      endl;
    o << "Other:" << endl;
    o << d_->stats->message << endl;
    o << "=================================================================" <<
      endl;
    o.close();
    return false;
  };

 private:

  void writeSpeedStats ( oszfstream& o ) {
    d_->stats->write ( o );
  }
};

}
} // end namespaces
#endif