@Override
public void endElement(String namespaceURI, String localName, String qName) {
XmlState endState = findState(qName);
if (state == XmlState.GRAPHIC && endState == XmlState.GRAPHIC) {
pic.setFileName(pathName + txt.toString().trim());
}
}
@Override
public void startElement(String namespaceURI, String localName, String qName, Attributes atts) {
state = findState(qName);
String at;
if (state != XmlState.BDN && !valid) {
logger.error("BDN tag missing");
}
txt = null;
switch (state) {
case UNKNOWN:
logger.error("Unknown tag " + qName + "\n");
break;
case BDN:
if (valid) {
logger.error("BDN must be used only once");
} else {
valid = true;
}
break;
case NAME:
at = atts.getValue("Title");
if (at != null) {
title = at;
logger.trace("Title: " + title + "\n");
}
break;
case LANGUAGE:
at = atts.getValue("Code");
if (at != null) {
language = at;
logger.trace("Language: " + language + "\n");
}
break;
case FORMAT:
at = atts.getValue("FrameRate");
if (at != null) {
fps = SubtitleUtils.getFps(at);
fpsXml = XmlFps(fps);
logger.trace("fps: " + ToolBox.formatDouble(fps) + "\n");
}
at = atts.getValue("VideoFormat");
if (at != null) {
String res = at;
for (Resolution r : Resolution.values()) {
if (res.length() == 4
&& res.charAt(0) != '7') { // hack to rename 480p/576p to 480i/576i
res = res.replace('p', 'i');
}
if (r.getResolutionNameForXml().equalsIgnoreCase(res)) {
resolution = r;
logger.trace("Language: " + r.getResolutionNameForXml() + "\n");
break;
}
}
}
break;
case EVENTS:
at = atts.getValue("NumberofEvents");
if (at != null) {
int n = ToolBox.getInt(at);
if (n > 0) {
/* number of subtitles read from the xml */
Core.setProgressMax(n);
}
}
break;
case EVENT:
pic = new SubPictureXml();
subPictures.add(pic);
int num = subPictures.size();
logger.info("#" + num + "\n");
Core.setProgress(num);
at = atts.getValue("InTC");
if (at != null) {
pic.setStartTime(timeStrXmlToPTS(at, fpsXml));
if (pic.getStartTime() == -1) {
pic.setStartTime(0);
logger.warn("Invalid start time " + at + "\n");
}
}
at = atts.getValue("OutTC");
if (at != null) {
pic.setEndTime(timeStrXmlToPTS(at, fpsXml));
if (pic.getEndTime() == -1) {
pic.setEndTime(0);
logger.warn("Invalid end time " + at + "\n");
}
}
if (fps != fpsXml) {
pic.setStartTime((pic.getStartTime() * 1001 + 500) / 1000);
pic.setEndTime((pic.getEndTime() * 1001 + 500) / 1000);
}
at = atts.getValue("Forced");
pic.setForced(at != null && at.equalsIgnoreCase("true"));
if (pic.isForced()) {
numForcedFrames++;
}
int dim[] = resolution.getDimensions();
pic.setWidth(dim[0]);
pic.setHeight(dim[1]);
break;
case GRAPHIC:
pic.setImageWidth(ToolBox.getInt(atts.getValue("Width")));
pic.setImageHeight(ToolBox.getInt(atts.getValue("Height")));
pic.setOfsX(ToolBox.getInt(atts.getValue("X")));
pic.setOfsY(ToolBox.getInt(atts.getValue("Y")));
pic.storeOriginalOffsets();
txt = new StringBuffer();
break;
}
}
/* (non-Javadoc)
* @see deadbeef.SupTools.SubtitleStream#decode(int)
*/
@Override
public void decode(int index) throws CoreException {
try {
File f = new File(subPictures.get(index).getFileName());
if (!f.exists()) {
throw new CoreException("file " + subPictures.get(index).getFileName() + " not found.");
}
BufferedImage img = ImageIO.read(f);
int w = img.getWidth();
int h = img.getHeight();
this.palette = null;
// first try to read image and palette directly from imported image
if (img.getType() == BufferedImage.TYPE_BYTE_INDEXED) {
IndexColorModel icm = (IndexColorModel) img.getColorModel();
if (icm.getMapSize() < 255 || (icm.hasAlpha() && icm.getAlpha(255) == 0)) {
// create palette
palette = new Palette(256);
for (int i = 0; i < icm.getMapSize(); i++) {
int alpha = (icm.getRGB(i) >> 24) & 0xff;
if (alpha >= configuration.getAlphaCrop()) {
palette.setARGB(i, icm.getRGB(i));
} else {
palette.setARGB(i, 0);
}
}
// copy pixels
WritableRaster raster = img.getRaster();
bitmap =
new Bitmap(
img.getWidth(),
img.getHeight(),
(byte[]) raster.getDataElements(0, 0, img.getWidth(), img.getHeight(), null));
}
}
// if this failed, assume RGB image and quantize palette
if (palette == null) {
// grab int array (ARGB)
int[] pixels = new int[w * h];
img.getRGB(0, 0, w, h, pixels, 0, w);
// quantize image
QuantizeFilter qf = new QuantizeFilter();
bitmap = new Bitmap(img.getWidth(), img.getHeight());
int ct[] = qf.quantize(pixels, bitmap.getInternalBuffer(), w, h, 255, false, false);
int size = ct.length;
if (size > 255) {
logger.warn("Quantizer failed.\n");
size = 255;
}
// create palette
palette = new Palette(256);
for (int i = 0; i < size; i++) {
int alpha = (ct[i] >> 24) & 0xff;
if (alpha >= configuration.getAlphaCrop()) {
palette.setARGB(i, ct[i]);
} else {
palette.setARGB(i, 0);
}
}
}
primaryColorIndex =
bitmap.getPrimaryColorIndex(
palette.getAlpha(), configuration.getAlphaThreshold(), palette.getY());
// crop
BitmapBounds bounds =
bitmap.getCroppingBounds(palette.getAlpha(), configuration.getAlphaCrop());
if (bounds.yMin > 0
|| bounds.xMin > 0
|| bounds.xMax < bitmap.getWidth() - 1
|| bounds.yMax < bitmap.getHeight() - 1) {
w = bounds.xMax - bounds.xMin + 1;
h = bounds.yMax - bounds.yMin + 1;
if (w < 2) {
w = 2;
}
if (h < 2) {
h = 2;
}
bitmap = bitmap.crop(bounds.xMin, bounds.yMin, w, h);
// update picture
SubPictureXml pic = subPictures.get(index);
pic.setImageWidth(w);
pic.setImageHeight(h);
pic.setOfsX(pic.getOriginalXOffset() + bounds.xMin);
pic.setOfsY(pic.getOriginalYOffset() + bounds.yMin);
}
} catch (IOException e) {
throw new CoreException(e.getMessage());
} catch (OutOfMemoryError e) {
JOptionPane.showMessageDialog(
null,
"Out of heap! Use -Xmx256m to increase heap!",
"Error!",
JOptionPane.WARNING_MESSAGE);
throw new CoreException("Out of heap! Use -Xmx256m to increase heap!");
}
}