Done: asp_program

Francisco Coelho
1 parent 5bcfee4d
Showing 4 changed files with 189 additions and 67 deletions Show diff stats
PASPProc.jl
drafts.jl
drafts_PASPProc.jl
symb-ltxify--drafts.jl
@@ -0,0 +1,106 @@
+module PASPProc
+
+include("../asplang/ASPLang.jl")
+using ASPLang
+
+export tree_string,
+    filter_tree,
+    filter_children,
+    has_child,
+    node_findreplace,
+    has_annotation_P,
+    repl_annotation_disjunction,
+    derived
+
+tree_string(node::AnonymousVariable; level=0, indent="\t") = "$(indent^level)Anonymous: $(node.args |> string)"
+tree_string(node::Functor; level=0, indent="\t") = "$(indent^level)Functor: $(node.args |> string)"
+tree_string(node::Annotation; level=0, indent="\t") = "$(indent^level)Annotation: $(node.args |> string)"
+tree_string(node::Variable; level=0, indent="\t") = "$(indent^level)Variable: $(node.args |> string)"
+tree_string(node::StringConstant; level=0, indent="\t") = "$(indent^level)String: $(node.args |> string)"
+tree_string(node::SymbolicConstant; level=0, indent="\t") = "$(indent^level)Symbol: $(node.args |> string)"
+tree_string(node::NumberConstant; level=0, indent="\t") = "$(indent^level)Number: $(node.args |> string)"
+#tree_print(node::Terminal; level=0, indent="\t") = node |> typeof |> string
+function tree_string(node::ASPLang.NonTerminal; level=0, indent="\t")
+    node_head = node |> typeof |> string
+    sub_results = [ "$(indent^level)$node_head" ] 
+    if isa(node.args, AbstractArray) && length(node.args) > 0
+        for child in node.args 
+            child_tree = if isa(child, ASPLang.Node)
+                tree_string(child, level = level + 1, indent=indent)
+            else
+                "$(indent^(level+1))$(child |> string)"
+            end
+            push!(sub_results, child_tree)
+        end
+    else
+        push!(sub_results, tree_string(node.args, level=level+1, indent=indent))
+    end
+    join(sub_results, "\n")
+end
+tree_string(node; level=0, indent="\t") = "$(indent^level)$(node |> typeof |> string):$(node |> string)"
+
+
+function filter_tree(node::NonTerminal, prop)
+    results = if prop(node)
+        [node]
+    else
+        []
+    end
+    if isa(node.args, AbstractArray)
+        for d in node.args
+            append!(results, filter_tree(d, prop))
+        end
+    else
+        if prop(node.args)
+            push!(results, node.args)
+        end
+    end
+    results
+end
+
+function filter_children(node::NonTerminal, prop)
+    if isa(node.args, AbstractArray)
+        filter(prop, node.args)
+    else
+        if prop(node.args)
+            node.args
+        else
+            nothing
+        end
+    end
+end
+
+function has_child(prop)
+    node::NonTerminal -> any(prop, node.args)
+end
+
+function node_findreplace(node::NonTerminal, prop, action)
+    if prop(node)
+        action(node)
+    elseif isa(node.args, AbstractArray)
+        new_args = []
+        for arg in node.args
+            push!(new_args, node_findreplace(arg, prop, action))
+        end
+        typeof(node)(new_args)
+    else
+        node
+    end
+end
+
+has_annotation_P = n -> has_child(isa(n, ASPLang.Annotation))
+function repl_annotation_disjunction(n::NonTerminal)
+    a = [xi for xi in n.args if !isa(xi, Annotation)][1]
+    Head(
+        Disjunction(
+            [a, Negated(a)]
+        ))
+end
+
+function derived(p::ASPLang.Program)
+    node_findreplace(p,
+        has_annotation_P,
+        repl_annotation_disjunction)
+end
+
+end # PASPproc
 \ No newline at end of file
@@ -1,67 +0,0 @@
-using Symbolics
-using Latexify
-
-# # γ = 0.5
-# no_s = [
-#     0   ,
-#     23  ,
-#     614 ,
-#     165 ,
-#     169 ,
-#     0   ,
-#     0   ,
-#     4   ,
-#     25  ,
-# ]
-
-# γ = 0.8
-no_s = [
-    0   ,
-    28  ,
-    632 ,
-    246 ,
-    59 ,
-    0   ,
-    0   ,
-    5   ,
-    27  ,
-]
-pr_s = (x -> x // 1000).(no_s)
-
-@variables θ
-num_e = [
-    0           , 
-    0           , 
-    7           , 
-    3 * θ       , 
-    3 * (1 - θ) , 
-    0           , 
-    0           , 
-    3           , 
-    10          ,
-]
-pr_e = (x -> x // 23).(num_e)
-
-target = expand(sum( (x -> x^2).(pr_s - pr_e) ))
-println(latexify(target))
-
-# using Plots
-# g(t) = (20869963/66125000) + (477/52900)*t + (18/529)*(t^2)
-# t = 0:0.1:1
-# plot(t, g.(t))
-
-
-
-function solve2(a, b, c)
-    delta = sqrt(Complex(b^2 - 4 * a * c))
-    return ( (-b - delta)/(2*a), (-b + delta)/(2*a) )
-end
-    
-# # g = target ~ 0
-
-a = float(18//529)
-b = -float(21903//264500)
-c = float(188207311//529000000)
-
-println("a: $a b: $b c: $c")
-println("̂θ = $(-b/a)")
@@ -0,0 +1,16 @@
+include("../asplang/ASPLang.jl")
+include("PASPProc.jl")
+
+using .ASPLang
+using .PASPProc
+
+
+root_path = "/home/fc/sci/projetos/zugzwang/code/asplang"
+asp_file = "sbf.lp"
+src = read("$(root_path)/$(asp_file)", String)
+
+println(methods(tree_string))
+# e = parse_asp(src, asp_grammar)
+# println(isa(e, NonTerminal))
+# t = tree_string(e)
+# println(t)
 \ No newline at end of file
@@ -0,0 +1,67 @@
+using Symbolics
+using Latexify
+
+# # γ = 0.5
+# no_s = [
+#     0   ,
+#     23  ,
+#     614 ,
+#     165 ,
+#     169 ,
+#     0   ,
+#     0   ,
+#     4   ,
+#     25  ,
+# ]
+
+# γ = 0.8
+no_s = [
+    0   ,
+    28  ,
+    632 ,
+    246 ,
+    59 ,
+    0   ,
+    0   ,
+    5   ,
+    27  ,
+]
+pr_s = (x -> x // 1000).(no_s)
+
+@variables θ
+num_e = [
+    0           , 
+    0           , 
+    7           , 
+    3 * θ       , 
+    3 * (1 - θ) , 
+    0           , 
+    0           , 
+    3           , 
+    10          ,
+]
+pr_e = (x -> x // 23).(num_e)
+
+target = expand(sum( (x -> x^2).(pr_s - pr_e) ))
+println(latexify(target))
+
+# using Plots
+# g(t) = (20869963/66125000) + (477/52900)*t + (18/529)*(t^2)
+# t = 0:0.1:1
+# plot(t, g.(t))
+
+
+
+function solve2(a, b, c)
+    delta = sqrt(Complex(b^2 - 4 * a * c))
+    return ( (-b - delta)/(2*a), (-b + delta)/(2*a) )
+end
+    
+# # g = target ~ 0
+
+a = float(18//529)
+b = -float(21903//264500)
+c = float(188207311//529000000)
+
+println("a: $a b: $b c: $c")
+println("̂θ = $(-b/a)")
...	...	@@ -0,0 +1,106 @@
	1	+module PASPProc
	2	+
	3	+include("../asplang/ASPLang.jl")
	4	+using ASPLang
	5	+
	6	+export tree_string,
	7	+ filter_tree,
	8	+ filter_children,
	9	+ has_child,
	10	+ node_findreplace,
	11	+ has_annotation_P,
	12	+ repl_annotation_disjunction,
	13	+ derived
	14	+
	15	+tree_string(node::AnonymousVariable; level=0, indent="\t") = "$(indent^level)Anonymous: $(node.args \|> string)"
	16	+tree_string(node::Functor; level=0, indent="\t") = "$(indent^level)Functor: $(node.args \|> string)"
	17	+tree_string(node::Annotation; level=0, indent="\t") = "$(indent^level)Annotation: $(node.args \|> string)"
	18	+tree_string(node::Variable; level=0, indent="\t") = "$(indent^level)Variable: $(node.args \|> string)"
	19	+tree_string(node::StringConstant; level=0, indent="\t") = "$(indent^level)String: $(node.args \|> string)"
	20	+tree_string(node::SymbolicConstant; level=0, indent="\t") = "$(indent^level)Symbol: $(node.args \|> string)"
	21	+tree_string(node::NumberConstant; level=0, indent="\t") = "$(indent^level)Number: $(node.args \|> string)"
	22	+#tree_print(node::Terminal; level=0, indent="\t") = node \|> typeof \|> string
	23	+function tree_string(node::ASPLang.NonTerminal; level=0, indent="\t")
	24	+ node_head = node \|> typeof \|> string
	25	+ sub_results = [ "$(indent^level)$node_head" ]
	26	+ if isa(node.args, AbstractArray) && length(node.args) > 0
	27	+ for child in node.args
	28	+ child_tree = if isa(child, ASPLang.Node)
	29	+ tree_string(child, level = level + 1, indent=indent)
	30	+ else
	31	+ "$(indent^(level+1))$(child \|> string)"
	32	+ end
	33	+ push!(sub_results, child_tree)
	34	+ end
	35	+ else
	36	+ push!(sub_results, tree_string(node.args, level=level+1, indent=indent))
	37	+ end
	38	+ join(sub_results, "\n")
	39	+end
	40	+tree_string(node; level=0, indent="\t") = "$(indent^level)$(node \|> typeof \|> string):$(node \|> string)"
	41	+
	42	+
	43	+function filter_tree(node::NonTerminal, prop)
	44	+ results = if prop(node)
	45	+ [node]
	46	+ else
	47	+ []
	48	+ end
	49	+ if isa(node.args, AbstractArray)
	50	+ for d in node.args
	51	+ append!(results, filter_tree(d, prop))
	52	+ end
	53	+ else
	54	+ if prop(node.args)
	55	+ push!(results, node.args)
	56	+ end
	57	+ end
	58	+ results
	59	+end
	60	+
	61	+function filter_children(node::NonTerminal, prop)
	62	+ if isa(node.args, AbstractArray)
	63	+ filter(prop, node.args)
	64	+ else
	65	+ if prop(node.args)
	66	+ node.args
	67	+ else
	68	+ nothing
	69	+ end
	70	+ end
	71	+end
	72	+
	73	+function has_child(prop)
	74	+ node::NonTerminal -> any(prop, node.args)
	75	+end
	76	+
	77	+function node_findreplace(node::NonTerminal, prop, action)
	78	+ if prop(node)
	79	+ action(node)
	80	+ elseif isa(node.args, AbstractArray)
	81	+ new_args = []
	82	+ for arg in node.args
	83	+ push!(new_args, node_findreplace(arg, prop, action))
	84	+ end
	85	+ typeof(node)(new_args)
	86	+ else
	87	+ node
	88	+ end
	89	+end
	90	+
	91	+has_annotation_P = n -> has_child(isa(n, ASPLang.Annotation))
	92	+function repl_annotation_disjunction(n::NonTerminal)
	93	+ a = [xi for xi in n.args if !isa(xi, Annotation)][1]
	94	+ Head(
	95	+ Disjunction(
	96	+ [a, Negated(a)]
	97	+ ))
	98	+end
	99	+
	100	+function derived(p::ASPLang.Program)
	101	+ node_findreplace(p,
	102	+ has_annotation_P,
	103	+ repl_annotation_disjunction)
	104	+end
	105	+
	106	+end # PASPproc
0	107	\ No newline at end of file
...	...
...	...	@@ -1,67 +0,0 @@
1		-using Symbolics
2		-using Latexify
3		-
4		-# # γ = 0.5
5		-# no_s = [
6		-# 0 ,
7		-# 23 ,
8		-# 614 ,
9		-# 165 ,
10		-# 169 ,
11		-# 0 ,
12		-# 0 ,
13		-# 4 ,
14		-# 25 ,
15		-# ]
16		-
17		-# γ = 0.8
18		-no_s = [
19		- 0 ,
20		- 28 ,
21		- 632 ,
22		- 246 ,
23		- 59 ,
24		- 0 ,
25		- 0 ,
26		- 5 ,
27		- 27 ,
28		-]
29		-pr_s = (x -> x // 1000).(no_s)
30		-
31		-@variables θ
32		-num_e = [
33		- 0 ,
34		- 0 ,
35		- 7 ,
36		- 3 * θ ,
37		- 3 * (1 - θ) ,
38		- 0 ,
39		- 0 ,
40		- 3 ,
41		- 10 ,
42		-]
43		-pr_e = (x -> x // 23).(num_e)
44		-
45		-target = expand(sum( (x -> x^2).(pr_s - pr_e) ))
46		-println(latexify(target))
47		-
48		-# using Plots
49		-# g(t) = (20869963/66125000) + (477/52900)t + (18/529)(t^2)
50		-# t = 0:0.1:1
51		-# plot(t, g.(t))
52		-
53		-
54		-
55		-function solve2(a, b, c)
56		- delta = sqrt(Complex(b^2 - 4 * a * c))
57		- return ( (-b - delta)/(2a), (-b + delta)/(2a) )
58		-end
59		-
60		-# # g = target ~ 0
61		-
62		-a = float(18//529)
63		-b = -float(21903//264500)
64		-c = float(188207311//529000000)
65		-
66		-println("a: $a b: $b c: $c")
67		-println("̂θ = $(-b/a)")
...	...	@@ -0,0 +1,16 @@
	1	+include("../asplang/ASPLang.jl")
	2	+include("PASPProc.jl")
	3	+
	4	+using .ASPLang
	5	+using .PASPProc
	6	+
	7	+
	8	+root_path = "/home/fc/sci/projetos/zugzwang/code/asplang"
	9	+asp_file = "sbf.lp"
	10	+src = read("$(root_path)/$(asp_file)", String)
	11	+
	12	+println(methods(tree_string))
	13	+# e = parse_asp(src, asp_grammar)
	14	+# println(isa(e, NonTerminal))
	15	+# t = tree_string(e)
	16	+# println(t)
0	17	\ No newline at end of file
...	...
...	...	@@ -0,0 +1,67 @@
	1	+using Symbolics
	2	+using Latexify
	3	+
	4	+# # γ = 0.5
	5	+# no_s = [
	6	+# 0 ,
	7	+# 23 ,
	8	+# 614 ,
	9	+# 165 ,
	10	+# 169 ,
	11	+# 0 ,
	12	+# 0 ,
	13	+# 4 ,
	14	+# 25 ,
	15	+# ]
	16	+
	17	+# γ = 0.8
	18	+no_s = [
	19	+ 0 ,
	20	+ 28 ,
	21	+ 632 ,
	22	+ 246 ,
	23	+ 59 ,
	24	+ 0 ,
	25	+ 0 ,
	26	+ 5 ,
	27	+ 27 ,
	28	+]
	29	+pr_s = (x -> x // 1000).(no_s)
	30	+
	31	+@variables θ
	32	+num_e = [
	33	+ 0 ,
	34	+ 0 ,
	35	+ 7 ,
	36	+ 3 * θ ,
	37	+ 3 * (1 - θ) ,
	38	+ 0 ,
	39	+ 0 ,
	40	+ 3 ,
	41	+ 10 ,
	42	+]
	43	+pr_e = (x -> x // 23).(num_e)
	44	+
	45	+target = expand(sum( (x -> x^2).(pr_s - pr_e) ))
	46	+println(latexify(target))
	47	+
	48	+# using Plots
	49	+# g(t) = (20869963/66125000) + (477/52900)t + (18/529)(t^2)
	50	+# t = 0:0.1:1
	51	+# plot(t, g.(t))
	52	+
	53	+
	54	+
	55	+function solve2(a, b, c)
	56	+ delta = sqrt(Complex(b^2 - 4 * a * c))
	57	+ return ( (-b - delta)/(2a), (-b + delta)/(2a) )
	58	+end
	59	+
	60	+# # g = target ~ 0
	61	+
	62	+a = float(18//529)
	63	+b = -float(21903//264500)
	64	+c = float(188207311//529000000)
	65	+
	66	+println("a: $a b: $b c: $c")
	67	+println("̂θ = $(-b/a)")
...	...