Home | History | Annotate | Download | only in script
      1 
      2 // Licensed under the Apache License, Version 2.0 (the "License");
      3 // you may not use this file except in compliance with the License.
      4 // You may obtain a copy of the License at
      5 //
      6 //     http://www.apache.org/licenses/LICENSE-2.0
      7 //
      8 // Unless required by applicable law or agreed to in writing, software
      9 // distributed under the License is distributed on an "AS IS" BASIS,
     10 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     11 // See the License for the specific language governing permissions and
     12 // limitations under the License.
     13 //
     14 // Copyright 2005-2010 Google, Inc.
     15 // Author: jpr (at) google.com (Jake Ratkiewicz)
     16 
     17 #include <string>
     18 
     19 #include <fst/arc.h>
     20 #include <fst/script/weight-class.h>
     21 
     22 namespace fst {
     23 namespace script {
     24 
     25 REGISTER_FST_WEIGHT(StdArc::Weight);
     26 REGISTER_FST_WEIGHT(LogArc::Weight);
     27 REGISTER_FST_WEIGHT(Log64Arc::Weight);
     28 
     29 WeightClass::WeightClass(const string &weight_type,
     30                          const string &weight_str)
     31   : element_type_(OTHER) {
     32   WeightClassRegister *reg = WeightClassRegister::GetRegister();
     33 
     34   StrToWeightImplBaseT stw = reg->GetEntry(weight_type);
     35 
     36   impl_ = stw(weight_str, "WeightClass", 0);
     37 };
     38 
     39 ostream& operator << (ostream &o, const WeightClass &c) {
     40   c.impl_->Print(&o);
     41   return o;
     42 }
     43 
     44 }  // namespace script
     45 }  // namespace fst
     46