tropical-sparse-tuple-weight.makeweight.h

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 TROPICALSPARSETUPLEWEIGHT_MAKEWEIGHT_H_
#define TROPICALSPARSETUPLEWEIGHT_MAKEWEIGHT_H_

namespace fst {

template<typename Arc = StdArc>
struct MakeSparseVectorWeight {
  explicit MakeSparseVectorWeight ( int32_t k ) :
    k ( k ) {
  }

  inline TropicalSparseTupleWeight<float> operator() ( const typename Arc::Weight&
      w ) const {
    TropicalSparseTupleWeight<float> result;
    result.Push ( k + 1, w.Value() );
    return result;
  }

 private:
  int32_t k;
};

template<>
struct MakeSparseVectorWeight<LexStdArc> {
  explicit MakeSparseVectorWeight ( int32_t k ) :
    k ( k ) {
  }

  inline TropicalSparseTupleWeight<float> operator() ( const  LexStdArc::Weight&
      w ) const {
    TropicalSparseTupleWeight<float> result;
    result.Push ( k + 1, w.Value1().Value() );
    return result;
  }
 private:
  int32_t k;
};

template<>
struct MakeWeight<TupleArc32> {
  explicit MakeWeight ( int32_t k = 0 ) :
    k_ ( k ) {
  };

  inline TupleArc32::Weight operator () ( const float weight )  const {
    TropicalSparseTupleWeight<float> result;
    result.Push ( k_ + 1, weight );
    return result;
  };
  inline TupleArc32::Weight operator ()  ( const TupleArc32::Weight& weight )
  const  {
    return weight ;
  };

  inline void update() {
    ++k_;
  };
  inline int32_t get_k() {
    return k_;
  };
 private:
  int32_t k_;
};

} //namespace fst

#endif /* TROPICALSPARSETUPLEWEIGHT_MAKEWEIGHT_H_ */