---

ARFF2PNG.java

gehe zur Dokumentation dieser Datei

/*
 * $Source: /shared/cvsroot/diplom/app/src/java/de/picana/converter/ARFF2PNG.java,v $
 * $Author: mstolpe $
 * $Date: 2003/04/22 09:51:28 $
 * $Revision: 1.4 $
 * $Release$
 *
 * Created on 15. November 2002, 19:28
 *
 * Copyright 2002 by Marco Stolpe
 */

package de.picana.converter;

import de.picana.control.*;
import de.picana.logging.*;

import weka.core.*;

import java.io.*;
import java.util.*;
import java.awt.image.*;
import javax.imageio.*;
import javax.imageio.stream.*;


public class ARFF2PNG extends Task {
    
    private String infile;
    private String outfile;
    private int width;
    private int height;
    
    private Instances instances;
    
    
    public ARFF2PNG() {
    }

    public void init(ParameterSet params, Logger logger) {
        
        super.init(params, logger);
        
        width = 0;
        height = 0;
        
        infile = (String)params.getParameter("in");
        outfile = (String)params.getParameter("out");

        try {
            width = Integer.parseInt((String)params.getParameter("width"));
        } catch (NumberFormatException nfe) {}
        
        try {
            height = Integer.parseInt((String)params.getParameter("height"));
        } catch (NumberFormatException nfe) {}
    }
    
    public void start() throws TaskException {
     
        logger.info(LOGSRC, "Started.");
        
        try {
            File input = new File(infile);
            FileInputStream fis = new FileInputStream(input);
            InputStreamReader reader = new InputStreamReader(fis);
            instances = new Instances(reader);

            BufferedImage image = new BufferedImage(width, height,
                BufferedImage.TYPE_INT_RGB);
        
            for (int y=0; y < height; y++) {
             
                //System.out.println(y);
                //System.out.flush();
                
                for (int x=0; x < width; x++) {
                
                    int index = y * width + x;
                    Instance ex = instances.instance(index);
                    
                    int red = (int)(ex.value(0) * 256);
                    int green = (int)(ex.value(1) * 256);
                    int blue = (int)(ex.value(2) * 256);
                    int rgb = (red << 16) + (green << 8) + blue;
                    
                    image.setRGB(x, y, rgb);
                }   
            }
            
            Iterator writers = ImageIO.getImageWritersByFormatName("png");
            ImageWriter wr = (ImageWriter)writers.next();
        
            ImageOutputStream ios = ImageIO.createImageOutputStream(
                new FileOutputStream(outfile));
            wr.setOutput(ios);
            wr.write(image);

            logger.info(LOGSRC, "Stopped.");
            
        } catch (Exception e) {
            throw new TaskException(e.toString());    
        }
    }
    
    public void stop() {
    }
    
    public void pause() {
    }
    
    public void resume() {
    }
}

---