sparel.h

/*Copyright 2014 Francisco Alvaro

 This file is part of SESHAT.

    SESHAT is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    SESHAT is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with SESHAT.  If not, see <http://www.gnu.org/licenses/>.
 */
#pragma once

class CellCYK;

#include <cstdio>
#include "hypothesis.h"
#include "cellcyk.h"
#include "gmm.h"
#include "sample.h"

class SpaRel {
public:
    static const int NRELS = 6;
    static const int NFEAT = 9;

private:
    GMM *model;
    Sample *mue;
    float probs[NRELS];

    double compute_prob(Hypothesis *h1, Hypothesis *h2, int k);
    void smooth(float *post);

public:
    SpaRel(GMM *gmm, Sample *m);
    ~SpaRel();

    void getFeas(Hypothesis *a, Hypothesis *b, float *sample, int ry);

    double getHorProb(Hypothesis *ha, Hypothesis *hb);
    double getSubProb(Hypothesis *ha, Hypothesis *hb);
    double getSupProb(Hypothesis *ha, Hypothesis *hb);
    double getVerProb(Hypothesis *ha, Hypothesis *hb, bool strict = false);
    double getInsProb(Hypothesis *ha, Hypothesis *hb);
    double getMrtProb(Hypothesis *ha, Hypothesis *hb);
};