kopia lustrzana https://github.com/njcrawford/EmbroideryReader
585 wiersze
20 KiB
C#
585 wiersze
20 KiB
C#
/*
|
|
Embroidery Reader - an application to view .pes embroidery designs
|
|
|
|
Copyright (C) 2013 Nathan Crawford
|
|
|
|
This program is free software; you can redistribute it and/or
|
|
modify it under the terms of the GNU General Public License
|
|
as published by the Free Software Foundation; either version 2
|
|
of the License, or (at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program; if not, write to the Free Software
|
|
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
|
|
02111-1307, USA.
|
|
|
|
A copy of the full GPL 2 license can be found in the docs directory.
|
|
You can contact me at http://www.njcrawford.com/contact/.
|
|
*/
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Text;
|
|
using System.Drawing;
|
|
|
|
namespace PesFile
|
|
{
|
|
public enum statusEnum { NotOpen, IOError, ParseError, Ready };
|
|
|
|
public struct intPair
|
|
{
|
|
public int a;
|
|
public int b;
|
|
}
|
|
|
|
public class PesFile
|
|
{
|
|
System.IO.BinaryReader fileIn;
|
|
int imageWidth;
|
|
int imageHeight;
|
|
string _filename;
|
|
public List<Int64> pesHeader = new List<Int64>();
|
|
public List<Int16> embOneHeader = new List<short>();
|
|
public List<Int16> sewSegHeader = new List<short>();
|
|
public List<Int16> embPunchHeader = new List<short>();
|
|
public List<Int16> sewFigSegHeader = new List<short>();
|
|
public List<StitchBlock> blocks = new List<StitchBlock>();
|
|
public List<intPair> colorTable = new List<intPair>();
|
|
private statusEnum readyStatus = statusEnum.NotOpen;
|
|
Int64 startStitches = 0;
|
|
string lastError = "";
|
|
Point translateStart;
|
|
UInt16 pesVersion;
|
|
|
|
|
|
// If set to true, this variable means we couldn't figure out some or
|
|
// all of the colors and white will be used instead of those colors.
|
|
private bool colorWarning = false;
|
|
|
|
private bool formatWarning = false;
|
|
|
|
private bool classWarning = false;
|
|
|
|
public PesFile(string filename)
|
|
{
|
|
OpenFile(filename);
|
|
}
|
|
|
|
// Returns an Int16 representation of the 12 bit signed int contained in
|
|
// high and low bytes
|
|
private Int16 get12Bit2sComplement(byte high, byte low)
|
|
{
|
|
Int32 retval;
|
|
|
|
// Get the bottom 4 bits of the high byte
|
|
retval = high & 0x0f;
|
|
|
|
// Shift those bits up where they belong
|
|
retval = retval << 8;
|
|
|
|
// Add in the bottom 8 bits
|
|
retval += low;
|
|
|
|
// Check for a negative number (check if 12th bit is 1)
|
|
if ((retval & 0x0800) == 0x0800)
|
|
{
|
|
// Make the number negative by subtracting 4096, which is the
|
|
// number of values a 12 bit integer can represent.
|
|
// This is a shortcut for getting the 2's complement value.
|
|
retval -= 4096;
|
|
}
|
|
|
|
return (Int16)retval;
|
|
}
|
|
|
|
// Returns a signed byte representation of the 7 bit signed int contained
|
|
// in b.
|
|
private SByte get7Bit2sComplement(byte b)
|
|
{
|
|
SByte retval;
|
|
|
|
// Ignore the 8th bit. (make sure it's 0)
|
|
b &= 0x7f;
|
|
|
|
// Check for a negative number (check if 7th bit is 1)
|
|
if ((b & 0x40) == 0x40)
|
|
{
|
|
// Make the number negative by subtracting 128, which is the
|
|
// number of values a 7 bit integer can represent.
|
|
// This is a shortcut for getting the 2's complement value.
|
|
retval = (SByte)(b - 128);
|
|
}
|
|
else
|
|
{
|
|
// Positive number - no modification needed
|
|
retval = (SByte)b;
|
|
}
|
|
|
|
return retval;
|
|
}
|
|
|
|
private void OpenFile(string filename)
|
|
{
|
|
#if !DEBUG
|
|
try
|
|
{
|
|
#endif
|
|
_filename = filename;
|
|
fileIn = new System.IO.BinaryReader(System.IO.File.Open(filename, System.IO.FileMode.Open, System.IO.FileAccess.Read));
|
|
|
|
string startFileSig = "";
|
|
for (int i = 0; i < 4; i++) // 4 bytes
|
|
{
|
|
startFileSig += fileIn.ReadChar();
|
|
}
|
|
if (startFileSig != "#PES")
|
|
{
|
|
// This is not a file that we can read
|
|
readyStatus = statusEnum.ParseError;
|
|
lastError = "Missing #PES at beginning of file";
|
|
fileIn.Close();
|
|
return;
|
|
}
|
|
|
|
// PES version
|
|
string versionString = "";
|
|
for (int i = 0; i < 4; i++) // 4 bytes
|
|
{
|
|
versionString += fileIn.ReadChar();
|
|
}
|
|
pesVersion = Convert.ToUInt16(versionString);
|
|
|
|
int pecstart = fileIn.ReadInt32();
|
|
|
|
// Read number of colors in this design
|
|
fileIn.BaseStream.Position = pecstart + 48;
|
|
int numColors = fileIn.ReadByte() +1;
|
|
List<byte> colorList = new List<byte>();
|
|
for (int x = 0; x < numColors; x++)
|
|
{
|
|
colorList.Add(fileIn.ReadByte());
|
|
}
|
|
|
|
// Read stitch data
|
|
fileIn.BaseStream.Position = pecstart + 532;
|
|
bool thisPartIsDone = false;
|
|
StitchBlock curBlock;
|
|
int prevX = 0;
|
|
int prevY = 0;
|
|
int maxX = 0;
|
|
int minX = 0;
|
|
int maxY = 0;
|
|
int minY = 0;
|
|
int colorNum = -1;
|
|
int colorIndex = 0;
|
|
List<Stitch> tempStitches = new List<Stitch>();
|
|
while (!thisPartIsDone)
|
|
{
|
|
byte val1;
|
|
byte val2;
|
|
val1 = fileIn.ReadByte();
|
|
val2 = fileIn.ReadByte();
|
|
if (val1 == 0xff && val2 == 0x00)
|
|
{
|
|
//end of stitches
|
|
thisPartIsDone = true;
|
|
|
|
//add the last block
|
|
curBlock = new StitchBlock();
|
|
curBlock.stitches = new Stitch[tempStitches.Count];
|
|
tempStitches.CopyTo(curBlock.stitches);
|
|
curBlock.stitchesTotal = tempStitches.Count;
|
|
colorNum++;
|
|
colorIndex = colorList[colorNum];
|
|
curBlock.colorIndex = colorIndex;
|
|
curBlock.color = getColorFromIndex(colorIndex);
|
|
blocks.Add(curBlock);
|
|
}
|
|
else if (val1 == 0xfe && val2 == 0xb0)
|
|
{
|
|
//color switch, start a new block
|
|
|
|
curBlock = new StitchBlock();
|
|
curBlock.stitches = new Stitch[tempStitches.Count];
|
|
tempStitches.CopyTo(curBlock.stitches);
|
|
curBlock.stitchesTotal = tempStitches.Count;
|
|
colorNum++;
|
|
colorIndex = colorList[colorNum];
|
|
curBlock.colorIndex = colorIndex;
|
|
curBlock.color = getColorFromIndex(colorIndex);
|
|
blocks.Add(curBlock);
|
|
|
|
tempStitches = new List<Stitch>();
|
|
|
|
//read useless(?) byte
|
|
byte useless = fileIn.ReadByte();
|
|
}
|
|
else
|
|
{
|
|
int deltaX = 0;
|
|
int deltaY = 0;
|
|
if ((val1 & 0x80) == 0x80)
|
|
{
|
|
//this is a jump stitch (more than 64 pixels away?)
|
|
deltaX = get12Bit2sComplement(val1, val2);
|
|
|
|
//read next byte for Y value
|
|
val2 = fileIn.ReadByte();
|
|
}
|
|
else
|
|
{
|
|
//normal stitch
|
|
deltaX = get7Bit2sComplement(val1);
|
|
}
|
|
|
|
if ((val2 & 0x80) == 0x80)
|
|
{
|
|
//this is a jump stitch (more than 64 pixels away?)
|
|
byte val3 = fileIn.ReadByte();
|
|
deltaY = get12Bit2sComplement(val2, val3);
|
|
}
|
|
else
|
|
{
|
|
//normal stitch
|
|
deltaY = get7Bit2sComplement(val2);
|
|
}
|
|
tempStitches.Add(
|
|
new Stitch(
|
|
new Point(prevX, prevY),
|
|
new Point(prevX + deltaX, prevY + deltaY)
|
|
)
|
|
);
|
|
prevX = prevX + deltaX;
|
|
prevY = prevY + deltaY;
|
|
if (prevX > maxX)
|
|
{
|
|
maxX = prevX;
|
|
}
|
|
else if (prevX < minX)
|
|
{
|
|
minX = prevX;
|
|
}
|
|
|
|
if (prevY > maxY)
|
|
{
|
|
maxY = prevY;
|
|
}
|
|
else if (prevY < minY)
|
|
{
|
|
minY = prevY;
|
|
}
|
|
}
|
|
}
|
|
imageWidth = maxX - minX;
|
|
imageHeight = maxY - minY;
|
|
translateStart.X = -minX;
|
|
translateStart.Y = -minY;
|
|
readyStatus = statusEnum.Ready;
|
|
|
|
// Close the file
|
|
fileIn.Close();
|
|
|
|
#if !DEBUG
|
|
}
|
|
catch (System.IO.IOException ioex)
|
|
{
|
|
readyStatus = statusEnum.IOError;
|
|
lastError = ioex.Message;
|
|
if (fileIn != null)
|
|
{
|
|
fileIn.Close();
|
|
}
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
readyStatus = statusEnum.ParseError;
|
|
lastError = ex.Message;
|
|
if (fileIn != null)
|
|
{
|
|
fileIn.Close();
|
|
}
|
|
}
|
|
#endif
|
|
}
|
|
|
|
/*void readCSewFigSeg(System.IO.BinaryReader file)
|
|
{
|
|
startStitches = fileIn.BaseStream.Position;
|
|
|
|
bool doneWithStitches = false;
|
|
int xValue = -100;
|
|
int yValue = -100;
|
|
stitchBlock currentBlock;
|
|
int blockType; //if this is equal to newColorMarker, it's time to change color
|
|
int colorIndex = 0;
|
|
int remainingStitches;
|
|
List<Point> stitchData;
|
|
stitchData = new List<Point>();
|
|
currentBlock = new stitchBlock();
|
|
|
|
while (!doneWithStitches)
|
|
{
|
|
//reset variables
|
|
xValue = 0;
|
|
yValue = 0;
|
|
|
|
blockType = file.ReadInt16();
|
|
if (blockType == 16716)
|
|
break;
|
|
colorIndex = file.ReadInt16();
|
|
if (colorIndex == 16716)
|
|
break;
|
|
remainingStitches = file.ReadInt16();
|
|
if (remainingStitches == 16716)
|
|
break;
|
|
while (remainingStitches >= 0)
|
|
{
|
|
xValue = file.ReadInt16();
|
|
if (xValue == -32765)
|
|
{
|
|
break;//drop out before we start eating into the next section
|
|
}
|
|
if (remainingStitches == 0)
|
|
{
|
|
int junk2 = 0;
|
|
junk2 = blocks.Count;
|
|
|
|
file.ReadBytes(24);
|
|
if (file.ReadInt16() == -1)
|
|
doneWithStitches = true;
|
|
|
|
currentBlock.stitches = new Point[stitchData.Count];
|
|
stitchData.CopyTo(currentBlock.stitches);
|
|
currentBlock.colorIndex = colorIndex;
|
|
currentBlock.color = getColorFromIndex(colorIndex);
|
|
currentBlock.stitchesTotal = stitchData.Count;
|
|
blocks.Add(currentBlock);
|
|
stitchData = new List<Point>();
|
|
currentBlock = new stitchBlock();
|
|
|
|
file.ReadBytes(48);
|
|
|
|
break;
|
|
}
|
|
else if (xValue == 16716 || xValue == 8224)
|
|
{
|
|
doneWithStitches = true;
|
|
break;
|
|
}
|
|
yValue = fileIn.ReadInt16();
|
|
if (yValue == 16716 || yValue == 8224)
|
|
{
|
|
doneWithStitches = true;
|
|
break;
|
|
}
|
|
stitchData.Add(new Point(xValue - translateStart.X, yValue + imageHeight - translateStart.Y));
|
|
remainingStitches--;
|
|
}
|
|
}
|
|
if (stitchData.Count > 1)
|
|
{
|
|
currentBlock.stitches = new Point[stitchData.Count];
|
|
stitchData.CopyTo(currentBlock.stitches);
|
|
currentBlock.colorIndex = colorIndex;
|
|
currentBlock.color = getColorFromIndex(colorIndex);
|
|
currentBlock.stitchesTotal = stitchData.Count;
|
|
blocks.Add(currentBlock);
|
|
}
|
|
}*/
|
|
|
|
public int GetWidth()
|
|
{
|
|
return imageWidth;
|
|
}
|
|
|
|
public int GetHeight()
|
|
{
|
|
return imageHeight;
|
|
}
|
|
|
|
public string GetFileName()
|
|
{
|
|
if (_filename == null)
|
|
{
|
|
return "";
|
|
}
|
|
else
|
|
{
|
|
return _filename;
|
|
}
|
|
}
|
|
|
|
public void saveDebugInfo()
|
|
{
|
|
System.IO.StreamWriter outfile = new System.IO.StreamWriter(System.IO.Path.ChangeExtension(_filename, ".txt"));
|
|
outfile.Write(getDebugInfo());
|
|
outfile.Close();
|
|
}
|
|
|
|
public string getDebugInfo()
|
|
{
|
|
System.IO.StringWriter outfile = new System.IO.StringWriter();
|
|
string name = "";
|
|
outfile.WriteLine("PES header");
|
|
outfile.WriteLine("PES version:\t" + pesVersion);
|
|
for (int i = 0; i < pesHeader.Count; i++)
|
|
{
|
|
name = (i + 1).ToString();
|
|
outfile.WriteLine(name + "\t" + pesHeader[i].ToString());
|
|
}
|
|
if (embOneHeader.Count > 0)
|
|
{
|
|
outfile.WriteLine("CEmbOne header");
|
|
for (int i = 0; i < embOneHeader.Count; i++)
|
|
{
|
|
switch (i + 1)
|
|
{
|
|
case 22:
|
|
name = "translate x";
|
|
break;
|
|
case 23:
|
|
name = "translate y";
|
|
break;
|
|
case 24:
|
|
name = "width";
|
|
break;
|
|
case 25:
|
|
name = "height";
|
|
break;
|
|
default:
|
|
name = (i + 1).ToString();
|
|
break;
|
|
}
|
|
|
|
outfile.WriteLine(name + "\t" + embOneHeader[i].ToString());
|
|
}
|
|
}
|
|
if (embPunchHeader.Count > 0)
|
|
{
|
|
outfile.WriteLine("CEmbPunch header");
|
|
for (int i = 0; i < embPunchHeader.Count; i++)
|
|
{
|
|
switch (i + 1)
|
|
{
|
|
default:
|
|
name = (i + 1).ToString();
|
|
break;
|
|
}
|
|
|
|
outfile.WriteLine(name + "\t" + embPunchHeader[i].ToString());
|
|
}
|
|
}
|
|
|
|
outfile.WriteLine("stitches start: " + startStitches.ToString());
|
|
outfile.WriteLine("block info");
|
|
outfile.WriteLine("number\tcolor\tstitches");
|
|
for (int i = 0; i < this.blocks.Count; i++)
|
|
{
|
|
outfile.WriteLine((i + 1).ToString() + "\t" + blocks[i].colorIndex.ToString() + "\t" + blocks[i].stitchesTotal.ToString());
|
|
}
|
|
outfile.WriteLine("color table");
|
|
outfile.WriteLine("number\ta\tb");
|
|
for (int i = 0; i < colorTable.Count; i++)
|
|
{
|
|
outfile.WriteLine((i + 1).ToString() + "\t" + colorTable[i].a.ToString() + ", " + colorTable[i].b.ToString());
|
|
}
|
|
if (blocks.Count > 0)
|
|
{
|
|
outfile.WriteLine("Extended stitch debug info");
|
|
for (int blocky = 0; blocky < blocks.Count; blocky++)
|
|
{
|
|
outfile.WriteLine("block " + (blocky + 1).ToString() + " start");
|
|
for (int stitchy = 0; stitchy < blocks[blocky].stitches.Length; stitchy++)
|
|
{
|
|
outfile.WriteLine(blocks[blocky].stitches[stitchy].ToString());
|
|
}
|
|
}
|
|
}
|
|
outfile.Close();
|
|
return outfile.ToString();
|
|
}
|
|
|
|
public statusEnum getStatus()
|
|
{
|
|
return readyStatus;
|
|
}
|
|
|
|
public string getLastError()
|
|
{
|
|
return lastError;
|
|
}
|
|
|
|
public bool getColorWarning()
|
|
{
|
|
return colorWarning;
|
|
}
|
|
|
|
public bool getFormatWarning()
|
|
{
|
|
return formatWarning;
|
|
}
|
|
|
|
public bool getClassWarning()
|
|
{
|
|
return classWarning;
|
|
}
|
|
|
|
private Color getColorFromIndex(int index)
|
|
{
|
|
if(index >= 1 && index <= 64)
|
|
{
|
|
return Color.FromArgb(
|
|
PesColors.colorMap[index,0],
|
|
PesColors.colorMap[index,1],
|
|
PesColors.colorMap[index,2]
|
|
);
|
|
}
|
|
else
|
|
{
|
|
colorWarning = true;
|
|
return Color.White;
|
|
}
|
|
}
|
|
|
|
public Bitmap designToBitmap(Single threadThickness, bool filterUglyStitches, double filterUglyStitchesThreshold)
|
|
{
|
|
Bitmap DrawArea;
|
|
Graphics xGraph;
|
|
|
|
DrawArea = new Bitmap(GetWidth() + (int)(threadThickness * 2), GetHeight() + (int)(threadThickness * 2));
|
|
|
|
xGraph = Graphics.FromImage(DrawArea);
|
|
xGraph.TranslateTransform(threadThickness+translateStart.X, threadThickness+translateStart.Y);
|
|
//xGraph.FillRectangle(Brushes.White, 0, 0, DrawArea.Width, DrawArea.Height);
|
|
|
|
for (int i = 0; i < blocks.Count; i++)
|
|
{
|
|
Pen tempPen = new Pen(blocks[i].color, threadThickness);
|
|
tempPen.StartCap = System.Drawing.Drawing2D.LineCap.Round;
|
|
tempPen.EndCap = System.Drawing.Drawing2D.LineCap.Round;
|
|
tempPen.LineJoin = System.Drawing.Drawing2D.LineJoin.Round;
|
|
xGraph.SmoothingMode = System.Drawing.Drawing2D.SmoothingMode.HighQuality;
|
|
foreach (Stitch thisStitch in blocks[i].stitches)
|
|
{
|
|
if (filterUglyStitches && !formatWarning) // Only filter stitches if we think we understand the format
|
|
{
|
|
if (thisStitch.calcLength() > filterUglyStitchesThreshold)
|
|
{
|
|
// This stitch is too long, so skip it
|
|
continue;
|
|
}
|
|
}
|
|
xGraph.DrawLine(tempPen, thisStitch.a, thisStitch.b);
|
|
}
|
|
}
|
|
xGraph.Dispose();
|
|
return DrawArea;
|
|
}
|
|
}
|
|
}
|