semparser.ml
7.66 KB
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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
(*
* ENIAMexec implements ENIAM processing stream
* Copyright (C) 2016-2017 Wojciech Jaworski <wjaworski atSPAMfree mimuw dot edu dot pl>
* Copyright (C) 2016-2017 Institute of Computer Science Polish Academy of Sciences
*
* This library is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This library 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 Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*)
open Xstd
open ENIAMsubsyntaxTypes
open ENIAMtokenizerTypes
let rules = ENIAM_LCGlexicon.make_rules false ENIAM_LCGlexiconTypes.user_lexicon_filename
let load_cats_map filename =
File.fold_tab filename StringMap.empty (fun map -> function
[lemma;cat] -> StringMap.add_inc map lemma [cat] (fun l -> cat :: l)
| l -> failwith ("load_senses_map: " ^ String.concat "\t" l))
let cats_map = load_cats_map ENIAM_LCGlexiconTypes.user_cats_filename
let examples = [
(* "liceum","W 1984-89 uczęszczał do VII Liceum Ogólnokształcącego im. K.K. Baczyńskiego w Szczecinie."; *)
"studia","Następnie studiował architekturę na Politechnice Szczecińskiej, dyplom uzyskał w 1994.";
]
(*
type output = Text | Xml | Html | Marsh | Graphviz
let output = ref Text
let comm_stdio = ref true
let sentence_split = ref true
let port = ref 0
*)
let subsyntax_built_in = ref true
let subsyntax_host = ref "localhost"
let subsyntax_port = ref 5739
let verbosity = ref 1
let img = ref 1
let timeout = ref 30.
let select_sentence_modes_flag = ref false
let select_sentences_flag = ref true
let output_dir = ref "results/"
let spec_list = [
(* "-s", Arg.Unit (fun () -> sentence_split:=true), "Split input into sentences (default)";
"-n", Arg.Unit (fun () -> sentence_split:=false), "Do not split input into sentences";
"-i", Arg.Unit (fun () -> comm_stdio:=true), "Communication using stdio (default)";
"-p", Arg.Int (fun p -> comm_stdio:=false; port:=p), "<port> Communication using sockets on given port number";
"-t", Arg.Unit (fun () -> output:=Text), "Output as plain text (default)";
"-x", Arg.Unit (fun () -> output:=Xml), "Output as XML";
"-m", Arg.Unit (fun () -> output:=Marsh), "Output as marshalled Ocaml data structure";
"-h", Arg.Unit (fun () -> output:=Html), "Output as HTML";
"-g", Arg.Unit (fun () -> output:=Graphviz; sentence_split:=false), "Output as graphviz dot file; turns sentence split off";*)
(* "-r", Arg.String (fun p ->
ENIAMtokenizerTypes.set_resource_path p;
ENIAMmorphologyTypes.set_resource_path p;
ENIAMsubsyntaxTypes.set_resource_path p), "<path> Set resource path"; *)
"-b", Arg.Unit (fun () -> subsyntax_built_in:=true), "Use built in version of ENIAMsubsyntax (default)";
"--port", Arg.Int (fun p -> subsyntax_built_in:=false; subsyntax_port:=p), "<port> Connect to ENIAMsubsyntax on a given port";
"--host", Arg.String (fun s -> subsyntax_built_in:=false; subsyntax_host:=s), "<hostname> Connect to ENIAMsubsyntax on a given host (by default localhost)";
]
let usage_msg =
"Usage: semparser <options>\nInput is a sequence of lines. Empty line ends the sequence and invoke parsing. Double empty line shutdown parser.\nOptions are:"
let message = "ENIAM_LCGsemparser, a parser for Logical Categorial Grammar formalism\n\
Copyright (C) 2017 Wojciech Jaworski <wjaworski atSPAMfree mimuw dot edu dot pl>\n\
Copyright (C) 2017 Institute of Computer Science Polish Academy of Sciences"
let anon_fun s = raise (Arg.Bad ("invalid argument: " ^ s))
let input_text channel =
let s = ref (try input_line channel with End_of_file -> "") in
let lines = ref [] in
while !s <> "" do
lines := !s :: !lines;
s := try input_line channel with End_of_file -> ""
done;
String.concat "\n" (List.rev !lines)
let get_cats cats_map = function
Interp orth -> (try StringMap.find cats_map orth with Not_found -> ["X"])
| Lemma(lemma,_,_) -> (try StringMap.find cats_map lemma with Not_found -> ["X"])
| Proper(_,_,_,cats) -> cats
| _ -> ["X"]
let assign_lex_sems cats_map tokens =
let lex_sems = ExtArray.make (ExtArray.size tokens) ENIAMlexSemanticsTypes.empty_lex_sem in
let _ = ExtArray.add lex_sems ENIAMlexSemanticsTypes.empty_lex_sem in
Int.iter 1 (ExtArray.size tokens - 1) (fun i ->
let cats = get_cats cats_map (ExtArray.get tokens i).token in
let lex_sem = {ENIAMlexSemanticsTypes.empty_lex_sem with ENIAMlexSemanticsTypes.cats=cats} in
let _ = ExtArray.add lex_sems lex_sem in
());
lex_sems
let rec main_loop sub_in sub_out in_chan out_chan =
let text = input_text in_chan in
if text = "" then () else (
let text,tokens,msg =
if !subsyntax_built_in then ENIAMsubsyntax.catch_parse_text text else (
Printf.fprintf sub_out "%s\n\n%!" text;
(Marshal.from_channel sub_in : ENIAMsubsyntaxTypes.text * token_env ExtArray.t * string)) in
if msg <> "" then print_endline msg else (
let lex_sems = assign_lex_sems cats_map tokens in
let text = ENIAMexec.translate_text text in
let text = ENIAMexec.parse !timeout !verbosity rules tokens lex_sems text in
let text = if !select_sentence_modes_flag then ENIAMselectSent.select_sentence_modes_text text else text in
let text = if !select_sentences_flag then ENIAMselectSent.select_sentences_text ENIAMexecTypes.Struct text else text in
ENIAMvisualization.print_html_text !output_dir "parsed_text" text !img !verbosity tokens)
(* print_endline "input text begin";
print_endline text;
print_endline "input text end"; *)
(*if !sentence_split then
let text,tokens = ENIAMsubsyntax.parse_text text in
(match !output with
Text -> output_string out_chan (ENIAMsubsyntaxStringOf.text "" tokens text ^ "\n" ^ ENIAMsubsyntaxStringOf.token_extarray tokens ^ "\n\n")
| Xml -> output_string out_chan (Xml.to_string (ENIAMsubsyntaxXMLof.text_and_tokens text tokens) ^ "\n\n")
| Html -> output_string out_chan (ENIAMsubsyntaxHTMLof.text_and_tokens text tokens ^ "\n\n")
| Marsh -> Marshal.to_channel out_chan (text,tokens) []
| Graphviz -> failwith "main_loop: ni")
else
let tokens = ENIAMsubsyntax.parse text in
(match !output with
Text -> output_string out_chan (ENIAMsubsyntaxStringOf.token_list tokens ^ "\n\n")
| Xml -> output_string out_chan (Xml.to_string (ENIAMsubsyntaxXMLof.token_list tokens) ^ "\n\n")
| Html -> output_string out_chan (ENIAMsubsyntaxHTMLof.token_list tokens ^ "\n\n")
| Marsh -> Marshal.to_channel out_chan tokens []
| Graphviz -> output_string out_chan (ENIAMsubsyntaxGraphOf.token_list tokens ^ "\n\n"))*);
flush out_chan;
main_loop sub_in sub_out in_chan out_chan)
let get_sock_addr host_name port =
let he = Unix.gethostbyname host_name in
let addr = he.Unix.h_addr_list in
Unix.ADDR_INET(addr.(0),port)
let _ =
prerr_endline message;
ENIAMcategoriesPL.initialize ();
Arg.parse spec_list anon_fun usage_msg;
if !subsyntax_built_in then ENIAMsubsyntax.initialize ();
Gc.compact ();
let sub_in,sub_out =
if !subsyntax_built_in then stdin,stdout
else Unix.open_connection (get_sock_addr !subsyntax_host !subsyntax_port) in
prerr_endline "Ready!";
(*if !comm_stdio then*) main_loop sub_in sub_out stdin stdout
(*else
let sockaddr = Unix.ADDR_INET(Unix.inet_addr_any,!port) in
Unix.establish_server main_loop sockaddr*)