annotate src/main/java/jp/ac/u_ryukyu/ie/cr/tatsuki/jungle/store/index/AddNewChildrenIndexEditor.java @ 139:ec166c8ff079

add findAll and findInSubTreeAll
author one
date Tue, 28 Oct 2014 06:35:34 +0900
parents 6e9a8d26e0cf
children afbe19c98f53 72f454eb04ec
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
135
6e9a8d26e0cf refactor
one
parents:
diff changeset
1 package jp.ac.u_ryukyu.ie.cr.tatsuki.jungle.store.index;
6e9a8d26e0cf refactor
one
parents:
diff changeset
2
6e9a8d26e0cf refactor
one
parents:
diff changeset
3 import java.util.Iterator;
6e9a8d26e0cf refactor
one
parents:
diff changeset
4
6e9a8d26e0cf refactor
one
parents:
diff changeset
5 import fj.F;
6e9a8d26e0cf refactor
one
parents:
diff changeset
6 import fj.Ord;
6e9a8d26e0cf refactor
one
parents:
diff changeset
7 import fj.P2;
6e9a8d26e0cf refactor
one
parents:
diff changeset
8 import fj.data.List;
6e9a8d26e0cf refactor
one
parents:
diff changeset
9 import fj.data.Option;
6e9a8d26e0cf refactor
one
parents:
diff changeset
10 import fj.data.TreeMap;
6e9a8d26e0cf refactor
one
parents:
diff changeset
11 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.JungleTreeEditor;
6e9a8d26e0cf refactor
one
parents:
diff changeset
12 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.NodePath;
6e9a8d26e0cf refactor
one
parents:
diff changeset
13 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.TreeEditor;
6e9a8d26e0cf refactor
one
parents:
diff changeset
14 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.impl.DefaultNodePath;
6e9a8d26e0cf refactor
one
parents:
diff changeset
15 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.impl.TreeNode;
6e9a8d26e0cf refactor
one
parents:
diff changeset
16 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.impl.logger.TreeOperationLog;
6e9a8d26e0cf refactor
one
parents:
diff changeset
17 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.transaction.IndexJungleTreeEditor;
6e9a8d26e0cf refactor
one
parents:
diff changeset
18 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.transaction.TransactionManager;
6e9a8d26e0cf refactor
one
parents:
diff changeset
19 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.util.DefaultEither;
6e9a8d26e0cf refactor
one
parents:
diff changeset
20 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.util.Either;
6e9a8d26e0cf refactor
one
parents:
diff changeset
21 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.util.Error;
6e9a8d26e0cf refactor
one
parents:
diff changeset
22 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.util.Pair;
6e9a8d26e0cf refactor
one
parents:
diff changeset
23
6e9a8d26e0cf refactor
one
parents:
diff changeset
24 public class AddNewChildrenIndexEditor implements IndexEditor {
6e9a8d26e0cf refactor
one
parents:
diff changeset
25
6e9a8d26e0cf refactor
one
parents:
diff changeset
26 NodePath editNodePath;
6e9a8d26e0cf refactor
one
parents:
diff changeset
27
6e9a8d26e0cf refactor
one
parents:
diff changeset
28 public AddNewChildrenIndexEditor(int pos, NodePath path) {
6e9a8d26e0cf refactor
one
parents:
diff changeset
29 this.editNodePath = path.add(pos);
6e9a8d26e0cf refactor
one
parents:
diff changeset
30 }
6e9a8d26e0cf refactor
one
parents:
diff changeset
31
6e9a8d26e0cf refactor
one
parents:
diff changeset
32 @Override
6e9a8d26e0cf refactor
one
parents:
diff changeset
33 public Either<Error, IndexJungleTreeEditor> edit(
6e9a8d26e0cf refactor
one
parents:
diff changeset
34 TreeNode root,
6e9a8d26e0cf refactor
one
parents:
diff changeset
35 TransactionManager txManager,
6e9a8d26e0cf refactor
one
parents:
diff changeset
36 TreeEditor editor,
6e9a8d26e0cf refactor
one
parents:
diff changeset
37 TreeOperationLog log,
6e9a8d26e0cf refactor
one
parents:
diff changeset
38 TreeMap<String, TreeMap<String, List<Pair<TreeNode, NodePath>>>> index) {
6e9a8d26e0cf refactor
one
parents:
diff changeset
39
6e9a8d26e0cf refactor
one
parents:
diff changeset
40 TreeMap<String, TreeMap<String, List<Pair<TreeNode, NodePath>>>> newIndex = editIndex(index);
6e9a8d26e0cf refactor
one
parents:
diff changeset
41 IndexJungleTreeEditor newEditor = new IndexJungleTreeEditor(root, txManager, editor, newIndex);
6e9a8d26e0cf refactor
one
parents:
diff changeset
42 return DefaultEither.newB(newEditor);
6e9a8d26e0cf refactor
one
parents:
diff changeset
43 }
6e9a8d26e0cf refactor
one
parents:
diff changeset
44
6e9a8d26e0cf refactor
one
parents:
diff changeset
45 public TreeMap<String, TreeMap<String, List<Pair<TreeNode, NodePath>>>> editIndex(
6e9a8d26e0cf refactor
one
parents:
diff changeset
46 TreeMap<String, TreeMap<String, List<Pair<TreeNode, NodePath>>>> index) {
6e9a8d26e0cf refactor
one
parents:
diff changeset
47
6e9a8d26e0cf refactor
one
parents:
diff changeset
48 if (!index.isEmpty()) {
6e9a8d26e0cf refactor
one
parents:
diff changeset
49 List<String> keyList = index.keys();
6e9a8d26e0cf refactor
one
parents:
diff changeset
50 TreeMap<String, TreeMap<String, List<Pair<TreeNode, NodePath>>>> newIndex = TreeMap.empty(Ord.stringOrd);
6e9a8d26e0cf refactor
one
parents:
diff changeset
51 TreeMap<String, List<Pair<TreeNode, NodePath>>> newInnerIndex = TreeMap.empty(Ord.stringOrd);
6e9a8d26e0cf refactor
one
parents:
diff changeset
52
6e9a8d26e0cf refactor
one
parents:
diff changeset
53 for (String indexKey : keyList) {
6e9a8d26e0cf refactor
one
parents:
diff changeset
54 TreeMap<String, List<Pair<TreeNode, NodePath>>> innerIndex = index.get(indexKey).some();
6e9a8d26e0cf refactor
one
parents:
diff changeset
55 List<String> innerIndexKeyList = innerIndex.keys();
6e9a8d26e0cf refactor
one
parents:
diff changeset
56
6e9a8d26e0cf refactor
one
parents:
diff changeset
57 for (String innerIndexKey : innerIndexKeyList) {
6e9a8d26e0cf refactor
one
parents:
diff changeset
58 List<Pair<TreeNode, NodePath>> pairList = innerIndex.get(innerIndexKey).some();
6e9a8d26e0cf refactor
one
parents:
diff changeset
59 List<Pair<TreeNode, NodePath>> list = checkPath(pairList);
6e9a8d26e0cf refactor
one
parents:
diff changeset
60 if (!list.isEmpty()){
139
ec166c8ff079 add findAll and findInSubTreeAll
one
parents: 135
diff changeset
61 //System.out.println(new String(list.head().left().getAttributes().get("KEY").array()));
135
6e9a8d26e0cf refactor
one
parents:
diff changeset
62 newInnerIndex = newInnerIndex.set(innerIndexKey, list);
6e9a8d26e0cf refactor
one
parents:
diff changeset
63 }
6e9a8d26e0cf refactor
one
parents:
diff changeset
64 }
6e9a8d26e0cf refactor
one
parents:
diff changeset
65 newIndex = newIndex.set(indexKey, newInnerIndex);
6e9a8d26e0cf refactor
one
parents:
diff changeset
66 }
6e9a8d26e0cf refactor
one
parents:
diff changeset
67 return newIndex;
6e9a8d26e0cf refactor
one
parents:
diff changeset
68 } else {
6e9a8d26e0cf refactor
one
parents:
diff changeset
69 return index;
6e9a8d26e0cf refactor
one
parents:
diff changeset
70 }
6e9a8d26e0cf refactor
one
parents:
diff changeset
71 }
6e9a8d26e0cf refactor
one
parents:
diff changeset
72
6e9a8d26e0cf refactor
one
parents:
diff changeset
73 public List<Pair<TreeNode, NodePath>> checkPath(List<Pair<TreeNode, NodePath>> pairList){
6e9a8d26e0cf refactor
one
parents:
diff changeset
74
6e9a8d26e0cf refactor
one
parents:
diff changeset
75 List<Pair<TreeNode, NodePath>> list = List.nil();
6e9a8d26e0cf refactor
one
parents:
diff changeset
76 for (Pair<TreeNode, NodePath> pair : pairList) {
6e9a8d26e0cf refactor
one
parents:
diff changeset
77
6e9a8d26e0cf refactor
one
parents:
diff changeset
78 NodePath path = pair.right();
139
ec166c8ff079 add findAll and findInSubTreeAll
one
parents: 135
diff changeset
79 //System.out.println("oldPath = " + path.toString());
135
6e9a8d26e0cf refactor
one
parents:
diff changeset
80 NodePath newPath = new DefaultNodePath();
6e9a8d26e0cf refactor
one
parents:
diff changeset
81
6e9a8d26e0cf refactor
one
parents:
diff changeset
82
6e9a8d26e0cf refactor
one
parents:
diff changeset
83 if (editNodePath.size() > path.size()) {
6e9a8d26e0cf refactor
one
parents:
diff changeset
84 list = list.cons(pair);
6e9a8d26e0cf refactor
one
parents:
diff changeset
85 continue;
6e9a8d26e0cf refactor
one
parents:
diff changeset
86 }
6e9a8d26e0cf refactor
one
parents:
diff changeset
87
6e9a8d26e0cf refactor
one
parents:
diff changeset
88 Pair<Integer, NodePath> editNodePathCopy = editNodePath.pop();
6e9a8d26e0cf refactor
one
parents:
diff changeset
89 int loopCount = 0;
6e9a8d26e0cf refactor
one
parents:
diff changeset
90
6e9a8d26e0cf refactor
one
parents:
diff changeset
91 for (Integer pathInt : path) {
6e9a8d26e0cf refactor
one
parents:
diff changeset
92 loopCount++;
6e9a8d26e0cf refactor
one
parents:
diff changeset
93
6e9a8d26e0cf refactor
one
parents:
diff changeset
94 if (pathInt == -1)
6e9a8d26e0cf refactor
one
parents:
diff changeset
95 continue;
6e9a8d26e0cf refactor
one
parents:
diff changeset
96
6e9a8d26e0cf refactor
one
parents:
diff changeset
97 if (editNodePathCopy.right().size() > 0) {
6e9a8d26e0cf refactor
one
parents:
diff changeset
98 editNodePathCopy = editNodePathCopy.right().pop();
6e9a8d26e0cf refactor
one
parents:
diff changeset
99
6e9a8d26e0cf refactor
one
parents:
diff changeset
100 if (loopCount == editNodePath.size() && editNodePathCopy.left() <= pathInt) {
6e9a8d26e0cf refactor
one
parents:
diff changeset
101 newPath = newPath.add(pathInt + 1);
6e9a8d26e0cf refactor
one
parents:
diff changeset
102 continue;
6e9a8d26e0cf refactor
one
parents:
diff changeset
103 }
6e9a8d26e0cf refactor
one
parents:
diff changeset
104
6e9a8d26e0cf refactor
one
parents:
diff changeset
105 if (!(editNodePathCopy.left() == pathInt)) {
6e9a8d26e0cf refactor
one
parents:
diff changeset
106 newPath = path;
6e9a8d26e0cf refactor
one
parents:
diff changeset
107 break;
6e9a8d26e0cf refactor
one
parents:
diff changeset
108 }
6e9a8d26e0cf refactor
one
parents:
diff changeset
109 }
6e9a8d26e0cf refactor
one
parents:
diff changeset
110
6e9a8d26e0cf refactor
one
parents:
diff changeset
111 newPath = newPath.add(pathInt);
6e9a8d26e0cf refactor
one
parents:
diff changeset
112
6e9a8d26e0cf refactor
one
parents:
diff changeset
113 }
6e9a8d26e0cf refactor
one
parents:
diff changeset
114
139
ec166c8ff079 add findAll and findInSubTreeAll
one
parents: 135
diff changeset
115 //System.out.println("newPath = " + newPath.toString());
135
6e9a8d26e0cf refactor
one
parents:
diff changeset
116 Pair<TreeNode, NodePath> newPair = new Pair<TreeNode, NodePath>(pair.left(), newPath);
6e9a8d26e0cf refactor
one
parents:
diff changeset
117 list = list.cons(newPair);
6e9a8d26e0cf refactor
one
parents:
diff changeset
118 }
6e9a8d26e0cf refactor
one
parents:
diff changeset
119
6e9a8d26e0cf refactor
one
parents:
diff changeset
120 return list;
6e9a8d26e0cf refactor
one
parents:
diff changeset
121 }
6e9a8d26e0cf refactor
one
parents:
diff changeset
122
6e9a8d26e0cf refactor
one
parents:
diff changeset
123 }