task.hifst.expandednumstates.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 TASK_HIFST_EXPANDEDNUMSTATES_HPP
#define TASK_HIFST_EXPANDEDNUMSTATES_HPP

namespace ucam {
namespace hifst {

template<class Arc>
class ExpandedNumStatesRTN {
 private:
  unordered_map< uint , uint > rtnnumstates_;
 public:
  ExpandedNumStatesRTN() {};

  inline uint operator() ( uint cc, uint x, uint y ) {
    uint hieroindex = APBASETAG + cc * APCCTAG + x * APXTAG + y * APYTAG;
    return rtnnumstates_[hieroindex];
  };

  void update ( uint cc, uint x, uint y, fst::VectorFst<Arc> *myfst ) {
    uint hieroindex = APBASETAG + cc * APCCTAG + x * APXTAG + y * APYTAG;
    uint& ns = rtnnumstates_[hieroindex];
    ns = myfst->NumStates();
    LDEBUG ( "AT: " << cc << "," << x << "," << y << ":" << ",NS=" <<
             rtnnumstates_[hieroindex] );
    typedef typename Arc::StateId StateId;
    for ( fst::StateIterator< fst::VectorFst<Arc> > si ( *myfst ); !si.Done();
          si.Next() ) {
      StateId state_id = si.Value();
      for ( fst::ArcIterator< fst::VectorFst<Arc> > ai ( *myfst , si.Value() );
            !ai.Done(); ai.Next() ) {
        const Arc& arc = ai.Value();
        if ( arc.ilabel < APBASETAG ) continue;
        ns += rtnnumstates_[arc.ilabel] - 2;
      }
    }
    LINFO ( "AT: " << cc << "," << x << "," << y << ":" << "NS (expanded) =" <<
            ns );
  };

 private:
  DISALLOW_COPY_AND_ASSIGN ( ExpandedNumStatesRTN );
};

}
} // end namespaces

#endif //ifndef TASK_HIFST_EXPANDEDNUMSTATES_HPP