//save a mesh
public static void SaveMesh(SegmentMesh mesh, PatternItem info, String outDir)
{
String outPath = Path.Combine(outDir, info.Directory);
Directory.CreateDirectory(outPath);
mesh.WriteToFile(Path.Combine(outPath, Util.ConvertFileName(info.Name, "", ".txt")));
}
//Test the quantization, connected components on different color layers, and output the descriptor
private void OutputPatterns_Click(object sender, RoutedEventArgs e)
{
//Create output directories
Directory.CreateDirectory(outdir + "\\cc\\");
Directory.CreateDirectory(outdir + "\\quantized\\");
Directory.CreateDirectory(outdir + "\\mesh\\");
//read in the patterns and save out their layers
String[] files = System.IO.Directory.GetFiles(System.IO.Path.Combine(imagedir));
List <PatternItem> patterns = PatternIO.GetPatterns(imagedir);
foreach (PatternItem p in patterns)
{
Bitmap image = new Bitmap(p.FullPath);
String basename = p.Name;
//TODO: sometimes keys are not found in patterns.csv...will need to look into recovering missing info. For now, just ignore those patterns
if (!palettes.ContainsKey(basename))
{
continue;
}
PaletteData palette = palettes[basename];
ColorTemplate template = new ColorTemplate(image, palette);
//output the template descriptor
SegmentMesh mesh = new SegmentMesh(template);
PatternIO.SaveMesh(mesh, p, Path.Combine(outdir, "mesh"));
if (outputDebugImages)
{
Bitmap result = template.DebugQuantization();
PatternIO.SavePattern(result, p, Path.Combine(outdir, "quantized"), "_quantized");
PatternIO.SavePattern(result, p, Path.Combine(outdir, "quantized"), "_original");
//save the connected components
UnionFind <Color> uf = new UnionFind <Color>((a, b) => (a.GetHashCode() == b.GetHashCode()));
Color[,] resultArray = Util.BitmapToArray(result);
int[,] cc = uf.ConnectedComponentsNoiseRemoval(resultArray);
int numColors = palette.colors.Count();
for (int i = 0; i < numColors; i++)
{
Bitmap debug = uf.RenderComponents(cc, resultArray, palette.colors[i]);
PatternIO.SavePattern(debug, p, Path.Combine(outdir, "cc"), "_" + i);
debug.Dispose();
}
result.Dispose();
}
image.Dispose();
}
}
private void RenderPreviews()
{
//render the segment previews for the visualization
Directory.CreateDirectory(Path.Combine(outdir, "\\previews\\"));
//read in the patterns and save out their layers
List <PatternItem> patterns = PatternIO.GetPatterns(imagedir);
int hpadding = 30;
foreach (PatternItem p in patterns)
{
Bitmap image = new Bitmap(p.FullPath);
String basename = p.Name;
if (!palettes.ContainsKey(basename))
{
continue;
}
PaletteData palette = palettes[basename];
ColorTemplate template = new ColorTemplate(image, palette);
SegmentMesh mesh = new SegmentMesh(template);
//create a pattern directory (Not using PatternIO here, since each pattern has its own directory anyways)
String patternDir = Path.Combine(outdir, "previews", Util.ConvertFileName(basename, "", ""));
Directory.CreateDirectory(patternDir);
//for each segment, pair of adjacent segments, and group, output a preview image
List <Segment> segs = mesh.getSegments();
List <SegmentGroup> groups = mesh.getGroups();
Bitmap original = (renderFinal)? image : template.DebugQuantization();
Bitmap previewBase = new Bitmap(original.Width * 2 + hpadding, original.Height);
//draw the original image on the right
Graphics g = Graphics.FromImage(previewBase);
g.DrawImage(original, original.Width + hpadding, 0);
//draw a grayscaled image on the left
for (int i = 0; i < original.Width; i++)
{
for (int j = 0; j < original.Height; j++)
{
int gray = (int)Math.Round(255 * original.GetPixel(i, j).GetBrightness());
previewBase.SetPixel(i, j, Color.FromArgb(gray, gray, gray));
}
}
//color in orange and blue
for (int i = 0; i < segs.Count(); i++)
{
Bitmap unary = new Bitmap(previewBase);
foreach (var point in segs[i].points)
{
unary.SetPixel(point.X, point.Y, Color.Orange);
}
unary.Save(Path.Combine(patternDir, "s" + i + ".png"));
foreach (int j in segs[i].adjacencies)
{
if (j > i)
{
Bitmap binary = new Bitmap(unary);
Segment neighbor = segs[j];
foreach (var point in neighbor.points)
{
binary.SetPixel(point.X, point.Y, Color.ForestGreen);
}
binary.Save(Path.Combine(patternDir, "s" + i + "-s" + j + ".png"));
binary.Dispose();
}
}
unary.Dispose();
}
for (int i = 0; i < groups.Count(); i++)
{
Bitmap group = new Bitmap(previewBase);
foreach (int j in groups[i].members)
{
Segment member = segs[j];
//color in the points
foreach (var point in member.points)
{
group.SetPixel(point.X, point.Y, Color.Orange);
}
}
group.Save(Path.Combine(patternDir, "g" + i + ".png"));
group.Dispose();
}
original.Dispose();
previewBase.Dispose();
}
}
//save a mesh
public static void SaveMesh(SegmentMesh mesh, PatternItem info, String outDir)
{
String outPath = Path.Combine(outDir, info.Directory);
Directory.CreateDirectory(outPath);
mesh.WriteToFile(Path.Combine(outPath, Util.ConvertFileName(info.Name, "", ".txt")));
}
private void RenderPreviews()
{
//render the segment previews for the visualization
Directory.CreateDirectory(Path.Combine(outdir, "\\previews\\"));
//read in the patterns and save out their layers
List<PatternItem> patterns = PatternIO.GetPatterns(imagedir);
int hpadding = 30;
foreach (PatternItem p in patterns)
{
Bitmap image = new Bitmap(p.FullPath);
String basename = p.Name;
if (!palettes.ContainsKey(basename))
continue;
PaletteData palette = palettes[basename];
ColorTemplate template = new ColorTemplate(image, palette);
SegmentMesh mesh = new SegmentMesh(template);
//create a pattern directory (Not using PatternIO here, since each pattern has its own directory anyways)
String patternDir = Path.Combine(outdir, "previews", Util.ConvertFileName(basename, "",""));
Directory.CreateDirectory(patternDir);
//for each segment, pair of adjacent segments, and group, output a preview image
List<Segment> segs = mesh.getSegments();
List<SegmentGroup> groups = mesh.getGroups();
Bitmap original = (renderFinal)? image : template.DebugQuantization();
Bitmap previewBase = new Bitmap(original.Width*2+hpadding, original.Height);
//draw the original image on the right
Graphics g = Graphics.FromImage(previewBase);
g.DrawImage(original, original.Width+hpadding, 0);
//draw a grayscaled image on the left
for (int i = 0; i < original.Width; i++)
{
for (int j = 0; j < original.Height; j++)
{
int gray = (int)Math.Round(255*original.GetPixel(i, j).GetBrightness());
previewBase.SetPixel(i,j, Color.FromArgb(gray, gray, gray));
}
}
//color in orange and blue
for (int i = 0; i < segs.Count(); i++)
{
Bitmap unary = new Bitmap(previewBase);
foreach (var point in segs[i].points)
unary.SetPixel(point.X, point.Y, Color.Orange);
unary.Save(Path.Combine(patternDir, "s" + i + ".png"));
foreach (int j in segs[i].adjacencies)
{
if (j > i)
{
Bitmap binary = new Bitmap(unary);
Segment neighbor = segs[j];
foreach (var point in neighbor.points)
binary.SetPixel(point.X, point.Y, Color.ForestGreen);
binary.Save(Path.Combine(patternDir, "s" + i + "-s" + j + ".png"));
binary.Dispose();
}
}
unary.Dispose();
}
for (int i = 0; i < groups.Count(); i++)
{
Bitmap group = new Bitmap(previewBase);
foreach (int j in groups[i].members)
{
Segment member = segs[j];
//color in the points
foreach (var point in member.points)
group.SetPixel(point.X, point.Y, Color.Orange);
}
group.Save(Path.Combine(patternDir, "g" + i + ".png"));
group.Dispose();
}
original.Dispose();
previewBase.Dispose();
}
}
//Test the quantization, connected components on different color layers, and output the descriptor
private void OutputPatterns_Click(object sender, RoutedEventArgs e)
{
//Create output directories
Directory.CreateDirectory(outdir + "\\cc\\");
Directory.CreateDirectory(outdir + "\\quantized\\");
Directory.CreateDirectory(outdir + "\\mesh\\");
//read in the patterns and save out their layers
String[] files = System.IO.Directory.GetFiles(System.IO.Path.Combine(imagedir));
List<PatternItem> patterns = PatternIO.GetPatterns(imagedir);
foreach (PatternItem p in patterns)
{
Bitmap image= new Bitmap(p.FullPath);
String basename = p.Name;
//TODO: sometimes keys are not found in patterns.csv...will need to look into recovering missing info. For now, just ignore those patterns
if (!palettes.ContainsKey(basename))
continue;
PaletteData palette = palettes[basename];
ColorTemplate template = new ColorTemplate(image, palette);
//output the template descriptor
SegmentMesh mesh = new SegmentMesh(template);
PatternIO.SaveMesh(mesh, p, Path.Combine(outdir, "mesh"));
if (outputDebugImages)
{
Bitmap result = template.DebugQuantization();
PatternIO.SavePattern(result, p, Path.Combine(outdir, "quantized"), "_quantized");
PatternIO.SavePattern(result, p, Path.Combine(outdir, "quantized"), "_original");
//save the connected components
UnionFind<Color> uf = new UnionFind<Color>((a, b) => (a.GetHashCode() == b.GetHashCode()));
Color[,] resultArray = Util.BitmapToArray(result);
int[,] cc = uf.ConnectedComponentsNoiseRemoval(resultArray);
int numColors = palette.colors.Count();
for (int i = 0; i < numColors; i++)
{
Bitmap debug = uf.RenderComponents(cc, resultArray, palette.colors[i]);
PatternIO.SavePattern(debug, p, Path.Combine(outdir, "cc"), "_" + i);
debug.Dispose();
}
result.Dispose();
}
image.Dispose();
}
}