data.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 STATS_HPP
#define STATS_HPP

namespace ucam {
namespace fsttools {

struct SpeedStatsData {
  unordered_map<std::string, std::vector<timeb> > time1;
  unordered_map<std::string, std::vector<timeb> > time2;

  inline void setTimeStart ( const std::string& key ) {
    timeb t;
    time1[key].push_back ( t );
    ftime ( &time1[key][time1[key].size() - 1] );
  };
  inline void setTimeEnd ( const std::string& key ) {
    timeb t;
    ftime ( &t );
    time2[key].push_back ( t );
  };
  void write ( std::ostream &o ) {
    for ( unordered_map<std::string, std::vector<timeb> >::iterator itx =
            time2.begin(); itx != time2.end(); itx++ ) {
      USER_CHECK ( time1[itx->first].size() == itx->second.size(),
                   "Mismatch with SpeedStats (each setTimeStart needs a setTimeEnd" );
      o << std::setw ( 30 ) << setiosflags ( std::ios::right ) << itx->first << ":";
      int64 fst_time = 0;
      for ( unsigned k = 0; k < itx->second.size() - 1 ; ++k ) {
        fst_time += ( itx->second[k].time - time1[itx->first][k].time ) * 1000;
        fst_time += itx->second[k].millitm - time1[itx->first][k].millitm;
      }
      int64 last_time = ( itx->second[itx->second.size() - 1].time -
                          time1[itx->first][itx->second.size() - 1].time ) * 1000
                        + itx->second[itx->second.size() - 1].millitm -
                        time1[itx->first][itx->second.size() - 1].millitm;
      fst_time += last_time;
      o << std::setw ( 10 ) << last_time;
      o << std::setw ( 10 ) << fst_time ;
      o << " (" << itx->second.size() << " times )" << endl;
    }
  }

};

struct StatsData {

  StatsData() :
    lpcount ( 0 ),
    numcats ( 0 ) {
  }

  unsigned lpcount;
  unsigned numcats;

  unordered_map<std::string, std::vector<timeb> > time1;
  unordered_map<std::string, std::vector<timeb> > time2;

  unordered_map<unsigned, unsigned> rulecounts;

  unordered_map<unsigned, unsigned> numstates;
  unordered_map<unsigned, unsigned> numprunedstates;

  std::string message;

  inline void setTimeStart ( const std::string& key ) {
    timeb t;
    time1[key].push_back ( t );
    ftime ( &time1[key][time1[key].size() - 1] );
  };
  inline void setTimeEnd ( const std::string& key ) {
    timeb t;
    ftime ( &t );
    time2[key].push_back ( t );
  };

  void write ( ucam::util::oszfstream& o ) {
    for ( unordered_map<std::string, std::vector<timeb> >::iterator itx =
            time2.begin(); itx != time2.end(); itx++ ) {
      USER_CHECK ( time1[itx->first].size() == itx->second.size(),
                   "Mismatch with SpeedStats (each setTimeStart needs a setTimeEnd" );
      o << std::setw ( 30 ) << setiosflags ( std::ios::right ) << itx->first << ":";
      int64 fst_time = 0;
      for ( unsigned k = 0; k < itx->second.size() - 1 ; ++k ) {
        fst_time += ( itx->second[k].time - time1[itx->first][k].time ) * 1000;
        fst_time += itx->second[k].millitm - time1[itx->first][k].millitm;
      }
      int64 last_time = ( itx->second[itx->second.size() - 1].time -
                          time1[itx->first][itx->second.size() - 1].time ) * 1000
                        + itx->second[itx->second.size() - 1].millitm -
                        time1[itx->first][itx->second.size() - 1].millitm;
      fst_time += last_time;
      o << std::setw ( 10 ) << last_time;
      o << std::setw ( 10 ) << fst_time ;
      o << " ms  (" << itx->second.size() << " times )" << endl;
    }
  }
};

}
}  // end namespaces

#endif