001/** 002 * Copyright (c) 2011, The University of Southampton and the individual contributors. 003 * All rights reserved. 004 * 005 * Redistribution and use in source and binary forms, with or without modification, 006 * are permitted provided that the following conditions are met: 007 * 008 * * Redistributions of source code must retain the above copyright notice, 009 * this list of conditions and the following disclaimer. 010 * 011 * * Redistributions in binary form must reproduce the above copyright notice, 012 * this list of conditions and the following disclaimer in the documentation 013 * and/or other materials provided with the distribution. 014 * 015 * * Neither the name of the University of Southampton nor the names of its 016 * contributors may be used to endorse or promote products derived from this 017 * software without specific prior written permission. 018 * 019 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND 020 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 021 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 022 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR 023 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 024 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 025 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON 026 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 027 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 028 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 029 */ 030package org.openimaj.vis.ternary; 031 032import java.awt.Font; 033import java.io.File; 034import java.io.FileNotFoundException; 035import java.io.FileReader; 036import java.io.IOException; 037import java.text.AttributedCharacterIterator.Attribute; 038import java.util.ArrayList; 039import java.util.Collection; 040import java.util.HashMap; 041import java.util.HashSet; 042import java.util.List; 043import java.util.Map; 044import java.util.Map.Entry; 045import java.util.Set; 046 047import org.openimaj.image.DisplayUtilities; 048import org.openimaj.image.FImage; 049import org.openimaj.image.ImageUtilities; 050import org.openimaj.image.MBFImage; 051import org.openimaj.image.colour.ColourMap; 052import org.openimaj.image.colour.RGBColour; 053import org.openimaj.image.typography.FontStyle; 054import org.openimaj.image.typography.general.GeneralFont; 055import org.openimaj.image.typography.mathml.MathMLFont; 056import org.openimaj.image.typography.mathml.MathMLFontStyle; 057import org.openimaj.util.pair.IndependentPair; 058import org.openimaj.vis.ternary.TernaryPlot.TernaryData; 059 060import com.google.gson.Gson; 061import com.google.gson.GsonBuilder; 062import com.google.gson.JsonIOException; 063import com.google.gson.JsonSyntaxException; 064 065/** 066 * 067 * @author Sina Samangooei (ss@ecs.soton.ac.uk) 068 */ 069public class SED2013Ternary { 070 071 @SuppressWarnings("unchecked") 072 public static void main(String[] args) throws JsonSyntaxException, JsonIOException, IOException { 073 String[] fileroots = new String[]{ 074 "/Users/ss/Experiments/sed2013/long_regen_weights_taken_posted_geo.div", 075// "/Users/ss/Experiments/sed2013/long_regen_weights_taken_posted_tag.div", 076 "/Users/ss/Experiments/sed2013/long_regen_weights_desc_tags_title.div", 077 "/Users/ss/Experiments/sed2013/long_regen_weights_taken_desc_tag.div", 078// "/Users/ss/Experiments/sed2013/long_regen_weights_geo_desc_tag.div" 079 }; 080 boolean[] correctRange = new boolean[]{ 081 true, 082// false, 083 true, 084 true, 085// true 086 }; 087 for (int i = 0; i < correctRange.length; i++) { 088 String fileroot = fileroots[i]; 089 090 ternaryPlotJSONF(new File(fileroot + ".json"), new File(fileroot.replace(".", "_") + ".png"), correctRange[i]); 091 } 092 093 ternaryPlotSearchGrid(new File("/Users/ss/Experiments/sed2013/searchgrid.png")); 094 095 096 } 097 098 private static void ternaryPlotSearchGrid(File file) throws IOException { 099 List<IndependentPair<TernaryData, String>> labels = new ArrayList<IndependentPair<TernaryData,String>>(); 100 TernaryParams params = new TernaryParams(); 101 List<TernaryData> data = new ArrayList<TernaryData>(); 102 data.add(new TernaryData(0.3f, 0.3f, 0.3f, 1f)); 103 TernaryPlot plot = preparePlot(params, data ); 104 params .put(TernaryParams.DRAW_SCALE, false); 105 106 for (int i = 0; i < 3; i++) { 107 for (int j = 0; j < 3; j++) { 108 for (int k = 0; k < 3; k++) { 109 float sum = i + j + k; 110 if(sum == 0) continue; 111 float ip = i/sum; 112 float jp = j/sum; 113 float kp = k/sum; 114 TernaryData point = new TernaryData(ip, jp, kp, 10); 115 String label = String.format("(%2.1f,%2.1f,%2.1f)",ip,jp,kp); 116 IndependentPair<TernaryData, String> pair = IndependentPair.pair(point, label); 117 labels.add(pair ); 118 } 119 } 120 } 121 122// labels.add(IndependentPair.pair(new TernaryData(1f,0,0,0),"f_1")); 123// labels.add(IndependentPair.pair(new TernaryData(0,1,0,0),"f_2")); 124// labels.add(IndependentPair.pair(new TernaryData(0,0,1,0),"f_3")); 125 params.put(TernaryParams.LABELS,labels); 126 Map<Attribute, Object> fontAttrs = params.getTyped(TernaryParams.LABEL_FONT); 127 fontAttrs.put(MathMLFontStyle.TEXT_MODE, false); 128 fontAttrs.put(MathMLFontStyle.FONT_SIZE, 12); 129 MBFImage draw = plot.draw(params); 130 ImageUtilities.write(draw, file); 131 DisplayUtilities.display(draw); 132 } 133 134 private static void ternaryPlotJSONF(File json, File outfile, boolean correctRange) 135 throws FileNotFoundException, IOException { 136 Gson gson = new GsonBuilder().create(); 137 Map<String,Object> obj = gson.fromJson(new FileReader(json), Map.class); 138 TernaryParams params = new TernaryParams(); 139 List<TernaryData> data = prepareData(params,obj,correctRange); 140 TernaryPlot plot = preparePlot(params, data); 141 MBFImage draw = plot.draw(params); 142 ImageUtilities.write(draw, outfile); 143 DisplayUtilities.display(draw); 144 } 145 146 private static TernaryPlot preparePlot(TernaryParams params, List<TernaryData> data) { 147 TernaryPlot plot = new TernaryPlot(500, data); 148 149 params.put(TernaryParams.COLOUR_MAP, ColourMap.Greys9); 150 params.put(TernaryParams.PADDING, 60); 151 152 params.put(TernaryParams.TRIANGLE_BORDER, true); 153 params.put(TernaryParams.TRIANGLE_BORDER_THICKNESS, 3); 154 params.put(TernaryParams.TRIANGLE_BORDER_TICKS, true); 155 params.put(TernaryParams.TRIANGLE_BORDER_COLOUR, RGBColour.WHITE); 156 Map<Attribute, Object> fontAttrs = params.getTyped(TernaryParams.LABEL_FONT); 157 fontAttrs.put(FontStyle.COLOUR, RGBColour.WHITE); 158 fontAttrs.put(FontStyle.FONT_SIZE, 20); 159 fontAttrs.put(FontStyle.FONT, new MathMLFont()); 160 fontAttrs.put(MathMLFontStyle.TEXT_MODE, true); 161 162 fontAttrs = params.getTyped(TernaryParams.SCALE_FONT); 163 fontAttrs.put(FontStyle.COLOUR, RGBColour.WHITE); 164 fontAttrs.put(FontStyle.FONT_SIZE, 16); 165 fontAttrs.put(FontStyle.FONT, new MathMLFont()); 166 167// params.put(TernaryParams.BG_COLOUR, RGBColour.WHITE); 168 params.put(TernaryParams.BG_COLOUR,RGBColour.BLACK); 169 170 params.put(TernaryParams.DRAW_SCALE, true); 171// params.put(TernaryParams.LABEL_BACKGROUND, RGBColour.WHITE); 172// params.put(TernaryParams.LABEL_BORDER, RGBColour.BLACK); 173 params.put(TernaryParams.LABEL_BACKGROUND, RGBColour.BLACK); 174 params.put(TernaryParams.LABEL_BORDER, RGBColour.WHITE); 175 params.put(TernaryParams.LABEL_PADDING, 3); 176 return plot; 177 } 178 179 private static List<TernaryData> prepareData(TernaryParams params,Map<String, Object> obj, boolean correctRange) { 180 List<String> exps = new ArrayList<String>(obj.keySet()); 181 Set<String> weightedFeatures = new HashSet<String>(); 182 183 184 double f1min = 1;double f1max = 0; 185 for (String exp : exps) { 186 Map<String, Object> expVars = (Map<String, Object>) obj.get(exp); 187 Map<String,Double> expFeatureWeights = (Map<String, Double>) expVars.get("weights"); 188 for (Entry<String, Double> fw : expFeatureWeights.entrySet()) { 189 if(fw.getValue()!=0){ 190 weightedFeatures.add(fw.getKey()); 191 } 192 } 193 double avgf1 = ((Double) expVars.get("averageF1")); 194 f1min = Math.min(avgf1, f1min); 195 f1max = Math.max(avgf1, f1max); 196 } 197 if(correctRange) 198 f1min = f1min + (((f1max - f1min)/8) * 7); 199 List<String> weightedFeaturesL = new ArrayList<String>(weightedFeatures); 200 List<TernaryData> data = new ArrayList<TernaryData>(); 201 double f1rng = f1max - f1min; 202 List<IndependentPair<TernaryData, String>> labels = new ArrayList<IndependentPair<TernaryData,String>>(); 203 labels.add(IndependentPair.pair(new TernaryData(1f,0,0,0),correctLabel(weightedFeaturesL.get(0)))); 204 labels.add(IndependentPair.pair(new TernaryData(0,1,0,0),correctLabel(weightedFeaturesL.get(1)))); 205 labels.add(IndependentPair.pair(new TernaryData(0,0,1,0),correctLabel(weightedFeaturesL.get(2)))); 206 float maxa = 0, maxb = 0, maxc = 0; 207 double maxv = 0; 208 for (String exp : exps) { 209 Map<String, Object> expmap = (Map<String, Object>) obj.get(exp); 210 Map<String,Double> weights = (Map<String, Double>) ((Map<String, Object>) obj.get(exp)).get("weights"); 211 double a = weights.get(weightedFeaturesL .get(0)); 212 double b = weights.get(weightedFeaturesL .get(1)); 213 double c = weights.get(weightedFeaturesL .get(2)); 214 double af1 = (Double) expmap.get("averageF1"); 215 float value = (float)(( af1 - f1min)/f1rng); 216 value = Math.max(value, 0); 217 if(value >= af1){ 218 maxv = af1; 219 maxa = (float) a; 220 maxb = (float) b; 221 maxc = (float) c; 222 } 223 data.add(new TernaryData((float)a,(float)b,(float)c,value)); 224 } 225 labels.add(IndependentPair.pair(new TernaryData(maxa,maxb,maxc,5),String.format("max_{F_1}(%2.2f)",f1max))); 226 params.put(TernaryParams.LABELS,labels); 227 params.put(TernaryParams.SCALE_MIN,String.format("\\leq div_{F_1}(%2.2f)",f1min)); 228 params.put(TernaryParams.SCALE_MAX,String.format("div_{F_1}(%2.2f)",f1max)); 229 return data; 230 } 231 232 private static String correctLabel(String string) { 233 if(string.contains("Geo")) return "Geo"; 234 if(string.contains("TAKEN")) return "TimeTaken"; 235 if(string.contains("POSTED")) return "TimePosted"; 236 if(string.contains("Title")) return "Title"; 237 if(string.contains("Tag")) return "Tags"; 238 if(string.contains("Descrip")) return "Desc"; 239 return string; 240 } 241 242}