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 package org.openimaj.vis.ternary;
31
32 import java.io.File;
33 import java.io.FileNotFoundException;
34 import java.io.FileReader;
35 import java.io.IOException;
36 import java.text.AttributedCharacterIterator.Attribute;
37 import java.util.ArrayList;
38 import java.util.HashSet;
39 import java.util.List;
40 import java.util.Map;
41 import java.util.Map.Entry;
42 import java.util.Set;
43
44 import org.openimaj.image.DisplayUtilities;
45 import org.openimaj.image.ImageUtilities;
46 import org.openimaj.image.MBFImage;
47 import org.openimaj.image.colour.ColourMap;
48 import org.openimaj.image.colour.RGBColour;
49 import org.openimaj.image.typography.FontStyle;
50 import org.openimaj.image.typography.mathml.MathMLFont;
51 import org.openimaj.image.typography.mathml.MathMLFontStyle;
52 import org.openimaj.util.pair.IndependentPair;
53 import org.openimaj.vis.ternary.TernaryPlot.TernaryData;
54
55 import com.google.gson.Gson;
56 import com.google.gson.GsonBuilder;
57 import com.google.gson.JsonIOException;
58 import com.google.gson.JsonSyntaxException;
59
60
61
62
63
64 public class SED2013Ternary {
65
66 public static void main(String[] args) throws JsonSyntaxException, JsonIOException, IOException {
67 final String[] fileroots = new String[] {
68 "/Users/ss/Experiments/sed2013/long_regen_weights_taken_posted_geo.div",
69
70 "/Users/ss/Experiments/sed2013/long_regen_weights_desc_tags_title.div",
71 "/Users/ss/Experiments/sed2013/long_regen_weights_taken_desc_tag.div",
72
73 };
74 final boolean[] correctRange = new boolean[] {
75 true,
76
77 true,
78 true,
79
80 };
81 for (int i = 0; i < correctRange.length; i++) {
82 final String fileroot = fileroots[i];
83
84 ternaryPlotJSONF(new File(fileroot + ".json"), new File(fileroot.replace(".", "_") + ".png"), correctRange[i]);
85 }
86
87 ternaryPlotSearchGrid(new File("/Users/ss/Experiments/sed2013/searchgrid.png"));
88
89 }
90
91 private static void ternaryPlotSearchGrid(File file) throws IOException {
92 final List<IndependentPair<TernaryData, String>> labels = new ArrayList<IndependentPair<TernaryData, String>>();
93 final TernaryParams params = new TernaryParams();
94 final List<TernaryData> data = new ArrayList<TernaryData>();
95 data.add(new TernaryData(0.3f, 0.3f, 0.3f, 1f));
96 final TernaryPlot plot = preparePlot(params, data);
97 params.put(TernaryParams.DRAW_SCALE, false);
98
99 for (int i = 0; i < 3; i++) {
100 for (int j = 0; j < 3; j++) {
101 for (int k = 0; k < 3; k++) {
102 final float sum = i + j + k;
103 if (sum == 0)
104 continue;
105 final float ip = i / sum;
106 final float jp = j / sum;
107 final float kp = k / sum;
108 final TernaryData point = new TernaryData(ip, jp, kp, 10);
109 final String label = String.format("(%2.1f,%2.1f,%2.1f)", ip, jp, kp);
110 final IndependentPair<TernaryData, String> pair = IndependentPair.pair(point, label);
111 labels.add(pair);
112 }
113 }
114 }
115
116
117
118
119 params.put(TernaryParams.LABELS, labels);
120 final Map<Attribute, Object> fontAttrs = params.getTyped(TernaryParams.LABEL_FONT);
121 fontAttrs.put(MathMLFontStyle.TEXT_MODE, false);
122 fontAttrs.put(MathMLFontStyle.FONT_SIZE, 12);
123 final MBFImage draw = plot.draw(params);
124 ImageUtilities.write(draw, file);
125 DisplayUtilities.display(draw);
126 }
127
128 private static void ternaryPlotJSONF(File json, File outfile, boolean correctRange)
129 throws FileNotFoundException, IOException
130 {
131 final Gson gson = new GsonBuilder().create();
132 @SuppressWarnings("unchecked")
133 final Map<String, Object> obj = gson.fromJson(new FileReader(json), Map.class);
134 final TernaryParams params = new TernaryParams();
135 final List<TernaryData> data = prepareData(params, obj, correctRange);
136 final TernaryPlot plot = preparePlot(params, data);
137 final MBFImage draw = plot.draw(params);
138 ImageUtilities.write(draw, outfile);
139 DisplayUtilities.display(draw);
140 }
141
142 private static TernaryPlot preparePlot(TernaryParams params, List<TernaryData> data) {
143 final TernaryPlot plot = new TernaryPlot(500, data);
144
145 params.put(TernaryParams.COLOUR_MAP, ColourMap.Greys9);
146 params.put(TernaryParams.PADDING, 60);
147
148 params.put(TernaryParams.TRIANGLE_BORDER, true);
149 params.put(TernaryParams.TRIANGLE_BORDER_THICKNESS, 3);
150 params.put(TernaryParams.TRIANGLE_BORDER_TICKS, true);
151 params.put(TernaryParams.TRIANGLE_BORDER_COLOUR, RGBColour.WHITE);
152 Map<Attribute, Object> fontAttrs = params.getTyped(TernaryParams.LABEL_FONT);
153 fontAttrs.put(FontStyle.COLOUR, RGBColour.WHITE);
154 fontAttrs.put(FontStyle.FONT_SIZE, 20);
155 fontAttrs.put(FontStyle.FONT, new MathMLFont());
156 fontAttrs.put(MathMLFontStyle.TEXT_MODE, true);
157
158 fontAttrs = params.getTyped(TernaryParams.SCALE_FONT);
159 fontAttrs.put(FontStyle.COLOUR, RGBColour.WHITE);
160 fontAttrs.put(FontStyle.FONT_SIZE, 16);
161 fontAttrs.put(FontStyle.FONT, new MathMLFont());
162
163
164 params.put(TernaryParams.BG_COLOUR, RGBColour.BLACK);
165
166 params.put(TernaryParams.DRAW_SCALE, true);
167
168
169 params.put(TernaryParams.LABEL_BACKGROUND, RGBColour.BLACK);
170 params.put(TernaryParams.LABEL_BORDER, RGBColour.WHITE);
171 params.put(TernaryParams.LABEL_PADDING, 3);
172 return plot;
173 }
174
175 @SuppressWarnings("unchecked")
176 private static List<TernaryData> prepareData(TernaryParams params, Map<String, Object> obj, boolean correctRange) {
177 final List<String> exps = new ArrayList<String>(obj.keySet());
178 final Set<String> weightedFeatures = new HashSet<String>();
179
180 double f1min = 1;
181 double f1max = 0;
182 for (final String exp : exps) {
183 final Map<String, Object> expVars = (Map<String, Object>) obj.get(exp);
184 final Map<String, Double> expFeatureWeights = (Map<String, Double>) expVars.get("weights");
185 for (final Entry<String, Double> fw : expFeatureWeights.entrySet()) {
186 if (fw.getValue() != 0) {
187 weightedFeatures.add(fw.getKey());
188 }
189 }
190 final double avgf1 = ((Double) expVars.get("averageF1"));
191 f1min = Math.min(avgf1, f1min);
192 f1max = Math.max(avgf1, f1max);
193 }
194 if (correctRange)
195 f1min = f1min + (((f1max - f1min) / 8) * 7);
196 final List<String> weightedFeaturesL = new ArrayList<String>(weightedFeatures);
197 final List<TernaryData> data = new ArrayList<TernaryData>();
198 final double f1rng = f1max - f1min;
199 final List<IndependentPair<TernaryData, String>> labels = new ArrayList<IndependentPair<TernaryData, String>>();
200 labels.add(IndependentPair.pair(new TernaryData(1f, 0, 0, 0), correctLabel(weightedFeaturesL.get(0))));
201 labels.add(IndependentPair.pair(new TernaryData(0, 1, 0, 0), correctLabel(weightedFeaturesL.get(1))));
202 labels.add(IndependentPair.pair(new TernaryData(0, 0, 1, 0), correctLabel(weightedFeaturesL.get(2))));
203 float maxa = 0, maxb = 0, maxc = 0;
204
205 for (final String exp : exps) {
206 final Map<String, Object> expmap = (Map<String, Object>) obj.get(exp);
207 final Map<String, Double> weights = (Map<String, Double>) ((Map<String, Object>) obj.get(exp)).get("weights");
208 final double a = weights.get(weightedFeaturesL.get(0));
209 final double b = weights.get(weightedFeaturesL.get(1));
210 final double c = weights.get(weightedFeaturesL.get(2));
211 final double af1 = (Double) expmap.get("averageF1");
212 float value = (float) ((af1 - f1min) / f1rng);
213 value = Math.max(value, 0);
214 if (value >= af1) {
215
216 maxa = (float) a;
217 maxb = (float) b;
218 maxc = (float) c;
219 }
220 data.add(new TernaryData((float) a, (float) b, (float) c, value));
221 }
222 labels.add(IndependentPair.pair(new TernaryData(maxa, maxb, maxc, 5), String.format("max_{F_1}(%2.2f)", f1max)));
223 params.put(TernaryParams.LABELS, labels);
224 params.put(TernaryParams.SCALE_MIN, String.format("\\leq div_{F_1}(%2.2f)", f1min));
225 params.put(TernaryParams.SCALE_MAX, String.format("div_{F_1}(%2.2f)", f1max));
226 return data;
227 }
228
229 private static String correctLabel(String string) {
230 if (string.contains("Geo"))
231 return "Geo";
232 if (string.contains("TAKEN"))
233 return "TimeTaken";
234 if (string.contains("POSTED"))
235 return "TimePosted";
236 if (string.contains("Title"))
237 return "Title";
238 if (string.contains("Tag"))
239 return "Tags";
240 if (string.contains("Descrip"))
241 return "Desc";
242 return string;
243 }
244
245 }