// Copyright (C) 2001 Jean-Marc Valin

#include "BufferedNode.h"
#include "Vector.h"
#include "DiagGMM.h"

using namespace std;

namespace FD {

class DiagGMMScore;

DECLARE_NODE(DiagGMMScore)
/*Node
 *
 * @name DiagGMMScore
 * @category HMM
 * @require DGMM
 * @description Scores a DiagGMM
 *
 * @input_name INPUT
 * @input_type Vector<float>
 * @input_description Input vector
 *
 * @input_name GMM
 * @input_type DiagGMM
 * @input_description Input GMM (pdf)
 *
 * @output_name OUTPUT
 * @output_type Float
 * @output_description GMM score
 *
END*/


class DiagGMMScore : public BufferedNode {
   
   int inputID;
   int gmmID;
   int outputID;

public:
   DiagGMMScore(string nodeName, ParameterSet params)
   : BufferedNode(nodeName, params)
   {
      inputID = addInput("INPUT");
      gmmID = addInput("GMM");
      outputID = addOutput("OUTPUT");
   }

   void calculate(int output_id, int count, Buffer &out)
   {
      ObjectRef inputValue = getInput(inputID, count);


      ObjectRef gmmValue = getInput(gmmID, count);

       
      DiagGMM &gmm = object_cast<DiagGMM> (gmmValue);
      const Vector<float> &in = object_cast<Vector<float> > (inputValue);
      int inputLength = in.size();
      if (inputLength != gmm.getDim())
	 throw new NodeException(this, "Dimension mismatch", __FILE__, __LINE__);

      out[count] = ObjectRef(Float::alloc(gmm.score(&in[0])));
      
   }

      
NO_ORDER_NODE_SPEEDUP(DiagGMMScore)
};

}//namespace FD