summaryrefslogtreecommitdiff
path: root/src/test_msgpack.cc
blob: 70cf1c8b290e54b5325128b186ffef3f5d6f1c7f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
#include <iostream>
#include <fstream>
#include <string>

/*
 * http://msgpack.org/
 *
 */
#include <msgpack.hpp>
#include <msgpack/fbuffer.hpp>

using namespace std;


struct Node {
  int id;
  string cat;
  vector<int> span;

  MSGPACK_DEFINE(id, cat, span);
};

struct Vector {
  double CountEF;
  double EgivenFCoherent;
  double Glue;
  double IsSingletonF;
  double IsSingletonFE;
  double LanguageModel;
  double LanguageModel_OOV;
  double MaxLexFgivenE;
  double MaxLexEgivenF;
  double PassThrough;
  double PassThrough_1;
  double PassThrough_2;
  double PassThrough_3;
  double PassThrough_4;
  double PassThrough_5;
  double PassThrough_6;
  double SampleCountF;
  double WordPenalty;

	MSGPACK_DEFINE(CountEF, EgivenFCoherent, Glue, IsSingletonF, IsSingletonFE, LanguageModel, LanguageModel_OOV, MaxLexEgivenF, MaxLexFgivenE, PassThrough, PassThrough_1, PassThrough_2, PassThrough_3, PassThrough_4, PassThrough_5, PassThrough_6, SampleCountF, WordPenalty);
};

struct Edge {
  int head;
  string rule;
  vector<int> tails;
  Vector f;
  double weight;

	MSGPACK_DEFINE(head, rule, tails, f, weight);
};

struct Hg {
  Vector weights;
  vector<Node> nodes;
  vector<Edge> edges;

  MSGPACK_DEFINE(weights, nodes, edges);
};

int
main(int argc, char** argv)
{
  ifstream ifs(argv[1]);
  string str((istreambuf_iterator<char>(ifs)),
                   (istreambuf_iterator<char>()));

	msgpack::zone zone;
	msgpack::object obj;
	msgpack::unpack(str.data(), str.size(), NULL, &zone, &obj);
  Hg hg;
	obj.convert(&hg);
  Edge last_edge = hg.edges.back();
	cerr << last_edge.rule.substr(1, 4) << endl;

  return 0;
}