source: trunk/src/org/expeditee/io/HTMLBWriter.java@ 309

package org.expeditee.io;

import java.awt.Image;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.util.List;

import org.expeditee.actions.Misc;
import org.expeditee.gui.FrameIO;
import org.expeditee.items.Item;
import org.expeditee.items.Picture;
import org.expeditee.items.Text;

import sun.awt.image.ToolkitImage;

public class HTMLBWriter extends AbstractHTMLWriter {

        @Override
        protected void writeTitle(Text toWrite, List<Item> items)
                        throws IOException {
                int indent = getIndent();
                if(indent == 0)
                        return;
                
                String heading = toWrite.getText();
                String tag = "h" + indent;
                // indenting for tag
                indent();
                _writer.write("<" + tag + ">" + heading + "</" + tag + ">"
                                + ItemWriter.NEW_LINE);
        }

        @Override
        protected void writeText(Text text) throws IOException {
                indent();
                _writer.write("<p>");
                _writer.write(text.getText().replace('\n', ' '));
                _writer.write("</p>" + ItemWriter.NEW_LINE);
        }

        @Override
        protected void writePicture(Picture pic) throws IOException {
                //TODO: Deal with cropped images!
                
                Image image = pic.getImage();
                String filesFolder = getFilesFolder();
                String fileName;
                // If its a bufferedImage then just write it out to the files directory
                //This means it is probably a FrameImage
                if (image instanceof BufferedImage || image instanceof ToolkitImage) {
                        String link = pic.getAbsoluteLink();
                        // Account for the possiblitly of an unlinked buffered image
                        fileName = link == null ? ("Image" + pic.getID()) : link;
                        fileName = Misc.SaveImage((BufferedImage) image, "PNG", FrameIO.EXPORTS_DIR
                                        + filesFolder, fileName);
                } else {//It is a normal Image stored somewhere
                        
                        Text source = pic.getSource();

                        fileName = source.getFirstLine();
                        fileName = fileName.substring(fileName.indexOf(":") + 1).trim();

                        String oldImageName = FrameIO.IMAGES_PATH + fileName;
                        String newImageName = FrameIO.EXPORTS_DIR + filesFolder + fileName;
                        try {
                                FrameIO.copyFile(oldImageName, newImageName);
                        } catch (Exception e) {
                                filesFolder = "";
                        }
                        if (filesFolder.equals("")) {
                                filesFolder = "../" + FrameIO.IMAGES_FOLDER;
                        }
                }
                indent();
                StringBuffer imageTag = new StringBuffer("<img src=\"");
                imageTag.append(filesFolder).append(fileName);
                imageTag.append("\" height=").append(pic.getHeight());
                imageTag.append(" width=").append(pic.getWidth());
                imageTag.append(" border=1>");
                _writer.write(imageTag.toString());
                _writer.write(ItemWriter.NEW_LINE);
        }
}