task.hifst.rtn.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_RTN_HPP
#define TASK_HIFST_RTN_HPP

namespace ucam {
namespace hifst {

template<class Arc>
class RTN {

  typedef typename Arc::Label Label;
  typedef std::unordered_map< Label , boost::shared_ptr< fst::Fst<Arc> > >
  hifst_rtn_t;
 private:
  hifst_rtn_t rtn_ , rtn2_;

 public:
  RTN() {};

  inline fst::Fst<Arc> *operator() ( const unsigned cc, const unsigned x, const unsigned y ) {
    Label hieroindex = APBASETAG + cc * APCCTAG + x * APXTAG + y * APYTAG;
    typename hifst_rtn_t::iterator itx = rtn_.find ( hieroindex );
    if ( itx != rtn_.end() )
      return itx->second.get();
    return NULL;
  };

  inline void Add ( const unsigned cc,
                    const unsigned x,
                    const unsigned y,
                    fst::Fst<Arc> *fst ,
                    fst::Fst<Arc> *fst2 = NULL) {
    Label hieroindex = APBASETAG + cc * APCCTAG + x * APXTAG + y * APYTAG;
    rtn_[hieroindex] = boost::shared_ptr<fst::Fst<Arc> > ( fst );
    if (fst2 != NULL)
      rtn2_[hieroindex] = boost::shared_ptr<fst::Fst<Arc> > ( fst2 );
  };
  inline void Add ( const unsigned cc,
                    const unsigned x,
                    const unsigned y,
                    boost::shared_ptr< fst::Fst<Arc> > fst ,
                    boost::shared_ptr< fst::Fst<Arc> > fst2 ) {
    Label hieroindex = APBASETAG + cc * APCCTAG + x * APXTAG + y * APYTAG;
    rtn_[hieroindex] = fst;
    rtn2_[hieroindex] = fst2;
  };

  inline void clear() {
    rtn_.clear();
  };

  inline std::size_t size() {
    return rtn_.size();
  };

 private:
  ZDISALLOW_COPY_AND_ASSIGN ( RTN );
};

}
} // end namespaces

#endif //TASK_HIFST_RTN_HPP