Subversion Repositories aoocs

/*
 * Copyright 2010, Aleksander Osman, alfik@poczta.fm. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without modification, are
 * permitted provided that the following conditions are met:
 *
 *  1. Redistributions of source code must retain the above copyright notice, this list of
 *     conditions and the following disclaimer.
 *
 *  2. Redistributions in binary form must reproduce the above copyright notice, this list
 *     of conditions and the following disclaimer in the documentation and/or other materials
 *     provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
 * FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR
 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
 
package aoOCS_tool;
 
import java.io.File;
import java.io.OutputStream;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.nio.ByteBuffer;
import java.nio.channels.FileChannel;
import java.util.LinkedList;
import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.util.Vector;
 
public class Main {
    /**
     * Print program call arguments description and exit.
     */
    static void print_call_arguments() {
        System.out.println("Can not parse program arguments.");
        System.out.println("");
        System.out.println("aoOCS_tool accepts the following arguments:");
        System.out.println("<operation> [operation arguments]");
        System.out.println("");
        System.out.println("The following   <operations> are available:");
        System.out.println("\t sd_disk      <intro image> <directory with ROM> <directory with AFS> <output disk image>");
        System.out.println("\t control_osd  <output control_osd.mif>");
        System.out.println("\t spec_extract <input> <output>");
        System.out.println("\t vga_to_png   <input file> <output directory>");
        System.out.println("");
        System.out.println("For more information please read the aoOCS IP core documentation.");
 
        System.exit(1);
    }
 
    static class LoadedFile {
        String name;
        long position_in_sectors;
    }
 
    static void p(String s) throws Exception {
        if(p_out == null) {
            System.out.println(s);
        }
        else {
            s += "\n";
            p_out.write(s.getBytes());
        }
    }
    static int print_32bits(int offset, int a, int b, int c, int d) throws Exception {
        String ha = Integer.toHexString(a);
        String hb = Integer.toHexString(b);
        String hc = Integer.toHexString(c);
        String hd = Integer.toHexString(d);
 
        if(ha.length() < 2) ha = "0" + ha;
        if(hb.length() < 2) hb = "0" + hb;
        if(hc.length() < 2) hc = "0" + hc;
        if(hd.length() < 2) hd = "0" + hd;
 
        p("" + (offset/4) + ": " + ha + hb + hc + hd + ";");
        return offset+4;
    }
    static int print_string(String str, int val_a, int val_b, int offset) throws Exception {
        for(int j=0; j<24; j+=4) {
            int a = str.charAt(j +0);
            int b = str.charAt(j +1);
            int c = str.charAt(j +2);
            int d = str.charAt(j +3);
 
            offset = print_32bits(offset, a,b,c,d);
        }
        offset = print_32bits(offset, (val_a>>24)&0xFF, (val_a>>16)&0xFF, (val_a>>8)&0xFF, (val_a)&0xFF);
        offset = print_32bits(offset, (val_b>>24)&0xFF, (val_b>>16)&0xFF, (val_b>>8)&0xFF, (val_b>>0)&0xFF);
        return offset;
    }
    static void load_files(String directory, LinkedList<LoadedFile> list, FileChannel channel) throws Exception {
        File rom_dir = new File(directory);
        if(rom_dir.exists() == false || rom_dir.isDirectory() == false) throw new Exception("Can not find directory: " + rom_dir.getCanonicalPath());
 
        for(File f : rom_dir.listFiles()) {
            if(f.isFile()) {
                byte bytes[] = new byte[(int)f.length()];
                FileInputStream in = new FileInputStream(f);
                if(in.read(bytes) != bytes.length) throw new Exception("Can not read file: " + f.getCanonicalPath());
                in.close();
 
                LoadedFile loaded = new LoadedFile();
                loaded.name = "  " + f.getName() + "                        ";
                if(loaded.name.length() > 24) loaded.name = loaded.name.substring(0, 24);
                loaded.position_in_sectors = channel.position()/512;
                list.addLast(loaded);
                System.out.println("" + f.getName() + ": " + channel.position());
                channel.write(ByteBuffer.wrap(bytes));
 
                if((channel.position() % 512) != 0) channel.position(channel.position() + 512 - (channel.position() % 512));
 
            }
        }
    }
    static void write_header(LinkedList<LoadedFile> list, FileChannel channel, int first, int middle, int last) throws Exception {
        for(int i=0; i<list.size(); i++) {
            LoadedFile f = list.get(i);
 
            byte line[] = new byte[32];
            System.arraycopy(f.name.getBytes(), 0, line, 0, 24);
            int a = 0;
            if(i == 0)                          a |= first;
            if(i == list.size()-1)              a |= last;
            if(i != 0 && i != list.size()-1)    a |= middle;
            int b = (int)f.position_in_sectors;
 
            line[24] = (byte)(a >> 24);
            line[25] = (byte)(a >> 16);
            line[26] = (byte)(a >> 8);
            line[27] = (byte)(a);
            line[28] = (byte)(b >> 24);
            line[29] = (byte)(b >> 16);
            line[30] = (byte)(b >> 8);
            line[31] = (byte)(b);
 
            channel.write(ByteBuffer.wrap(line));
        }
    }
    public static void main(String[] args) throws Exception {
        // check program call arguments
        if(args.length == 0)                                                print_call_arguments();
        else if(args[0].equals("sd_disk") == true && args.length != 5)      print_call_arguments();
        else if(args[0].equals("control_osd") == true && args.length != 2)  print_call_arguments();
        else if(args[0].equals("spec_extract") == true && args.length != 3) print_call_arguments();
        else if(args[0].equals("vga_to_png") == true && args.length != 3)   print_call_arguments();
 
        int SELECTABLE      = 1;
        int TOP             = 2;
        int BOTTOM          = 4;
        int RESET           = 8;
        int JOYSTICK        = 16;
        int FLOPPY          = 32;
        int WRITE_ENABLED   = 64;
 
        if(args[0].equals("control_osd")) {
            File f = new File(args[1]);
            p_out = new FileOutputStream(f);
 
            p("DEPTH = 1024;");
            p("WIDTH = 32;");
            p("ADDRESS_RADIX = DEC;");
            p("DATA_RADIX = HEX;");
            p("CONTENT");
            p("BEGIN");
            int offset = 0;
 
            offset = print_string("aoOCS version 1.0       ", 0,0, offset);
            offset = print_string("Initializing SD card....", 0,0, offset);
            offset = print_string("SD fatal error.         ", 0,0, offset);
            offset = print_string("SD ready. Select ROM:   ", 0,0, offset);
            offset += 32; // blank line
            offset = print_string("Inserted floppy:        ", 0,                            0, offset);
            offset = print_string("None                    ", 0,                            0, offset);
            offset = print_string("Joystick(kbd arrows) OFF", TOP | JOYSTICK | SELECTABLE,  0, offset);
            offset = print_string("Joystick(kbd arrows) ON ", TOP | JOYSTICK | SELECTABLE,  0, offset);
            offset = print_string("Floppy write enabled OFF", WRITE_ENABLED | SELECTABLE,   0, offset);
            offset = print_string("Floppy write enabled ON ", WRITE_ENABLED | SELECTABLE,   0, offset);
            offset = print_string("Reset                   ", RESET | SELECTABLE,           0, offset);
            offset = print_string("Eject floppy            ", BOTTOM | FLOPPY | SELECTABLE, 0, offset);
 
            p("END;");
            p_out.close();
        }
        else if(args[0].equals("sd_disk")) {
            // read image file
            File intro_image_file = new File(args[1]);
            if(intro_image_file.exists() == false) throw new Exception("Can not read intro image file: " + intro_image_file.getCanonicalPath());
 
            BufferedImage img = ImageIO.read(intro_image_file);
 
            // prepare output file
            FileOutputStream out = new FileOutputStream(args[4]);
            FileChannel channel = out.getChannel();
 
            channel.position(0);
            for(int y=0; y<256; y++) {
                for(int x=0; x<640; ) {
                    long three_pixels = 0;
 
                    for(int z=0; z<3; z++) {
                        int r,g,b;
 
                        if(x >= img.getWidth() || y >= img.getHeight()) {
                            r = g = b = 0;
                        }
                        else {
                            r = (img.getRGB(x, y)>>16) & 0xFF;
                            g = (img.getRGB(x, y)>>8) & 0xFF;
                            b = (img.getRGB(x, y)) & 0xFF;
                        }
 
                        three_pixels <<= 3;
                        three_pixels |= (r>>5) & 0x7;
                        three_pixels <<= 3;
                        three_pixels |= (g>>5) & 0x7;
                        three_pixels <<= 3;
                        three_pixels |= (b>>5) & 0x7;
 
                        x++;
                    }
                    three_pixels <<= 5;
 
                    byte word[] = new byte[4];
                    word[0] = (byte)((three_pixels >> 24) & 0xFF);
                    word[1] = (byte)((three_pixels >> 16) & 0xFF);
                    word[2] = (byte)((three_pixels >> 8) & 0xFF);
                    word[3] = (byte)((three_pixels) & 0xFF);
 
                    channel.write(ByteBuffer.wrap(word));
                }
                // padding
                byte padding[] = new byte[4*2];
                channel.write(ByteBuffer.wrap(padding));
            }
            System.out.println("Start channel pos: " + channel.position());
 
            channel.position(216*4*256 + 4096);
 
            LinkedList<LoadedFile> roms = new LinkedList<LoadedFile>();
            load_files(args[2], roms, channel);
 
            LinkedList<LoadedFile> floppies = new LinkedList<LoadedFile>();
            load_files(args[3], floppies, channel);
 
            channel.position(216*4*256 + 0);
            write_header(roms, channel, TOP | SELECTABLE, SELECTABLE, BOTTOM | SELECTABLE);
 
            channel.position(216*4*256 + 512);
            write_header(floppies, channel, FLOPPY | SELECTABLE, FLOPPY | SELECTABLE, BOTTOM | FLOPPY | SELECTABLE);
 
            channel.close();
            out.close();
 
            /* Sample zero and first SD sector.
            * The 0 sector is loaded at byte offset 512 in control_osd|display_ram_inst.
            * The 1-7 sectors are loaded starting at byte offset 1024 in control_osd|display_ram_inst.
            */
            /*
            offset = 512;
            offset = print_string("  Some ROM              ", TOP | SELECTABLE,             0, offset);
            offset = print_string("  R1                    ", SELECTABLE,                   0, offset);
            offset = print_string("  R2                    ", SELECTABLE,                   0, offset);
            offset = print_string("  R3                    ", SELECTABLE,                   0, offset);
            offset = print_string("  R4                    ", SELECTABLE,                   0, offset);
            offset = print_string("  R5                    ", SELECTABLE,                   0, offset);
            offset = print_string("  Some ROM2             ", BOTTOM | SELECTABLE,          0, offset);
 
            offset = 1024;
            offset = print_string("  Floppy1               ", FLOPPY | SELECTABLE,          1, offset);
            offset = print_string("  F2                    ", FLOPPY | SELECTABLE,          1, offset);
            offset = print_string("  F3                    ", FLOPPY | SELECTABLE,          1, offset);
            offset = print_string("  F4                    ", FLOPPY | SELECTABLE,          1, offset);
            offset = print_string("  F5                    ", FLOPPY | SELECTABLE,          1, offset);
            offset = print_string("  F6                    ", FLOPPY | SELECTABLE,          1, offset);
            offset = print_string("  F7                    ", FLOPPY | SELECTABLE,          1, offset);
            offset = print_string("  F8                    ", FLOPPY | SELECTABLE,          1, offset);
            offset = print_string("  F9                    ", FLOPPY | SELECTABLE,          1, offset);
            offset = print_string("  F10                   ", FLOPPY | SELECTABLE,          1, offset);
            offset = print_string("  F11                   ", FLOPPY | SELECTABLE,          1, offset);
            offset = print_string("  F12                   ", FLOPPY | SELECTABLE,          1, offset);
            offset = print_string("  F13                   ", FLOPPY | SELECTABLE,          1, offset);
            offset = print_string("  F14                   ", FLOPPY | SELECTABLE,          1, offset);
            offset = print_string("  F15                   ", FLOPPY | SELECTABLE,          1, offset);
            offset = print_string("  F16                   ", FLOPPY | SELECTABLE,          1, offset);
            offset = print_string("  F17                   ", FLOPPY | SELECTABLE,          1, offset);
            offset = print_string("  F18                   ", FLOPPY | SELECTABLE,          1, offset);
            offset = print_string("  F19                   ", FLOPPY | SELECTABLE,          1, offset);
            offset = print_string("  F20                   ", BOTTOM | FLOPPY | SELECTABLE, 1, offset);
            */
        }
        else if(args[0].equals("spec_extract")) {
            DocumentationTool.extract(args[1], args[2]);
        }
        else if(args[0].equals("vga_to_png")) {
            int screen[][] = new int[480][640];
            int line_inserted[] = new int[480];
 
            File in_file = new File(args[1]);
            if(in_file.exists() == false || in_file.isDirectory() == true) throw new Exception("Can not open input file: " + in_file.getCanonicalPath());
 
            FileInputStream in = new FileInputStream(in_file);
            int b;
            int frame_number = 0;
            while(true) {
                // read line number
                int line_number;
 
                b = in.read(); if(b == -1) break;
                line_number = b;
                b = in.read(); if(b == -1) break;
                line_number |= (b<<8);
 
                // read full line
                byte line[] = new byte[960];
                if(in.read(line) != line.length) break;
 
                // convert bytes to bits
                int bits[] = new int[960*8];
                for(int i=0; i<line.length; i++) {
                    for(int j=0; j<8; j++) {
                        bits[i*8+j] = (line[i] >> j) & 0x01;
                    }
                }
 
                // fill in screen array
                if(line_number < 0 || line_number >= 480) throw new Exception("Illegal line number: " + line_number);
                for(int i=0; i<640; i++) {
                    int red = 0, green = 0, blue = 0;
                    for(int j=0; j<4; j++) blue  |= ((bits[i*12 +j +0]) << j);
                    for(int j=0; j<4; j++) green |= ((bits[i*12 +j +4]) << j);
                    for(int j=0; j<4; j++) red   |= ((bits[i*12 +j +8]) << j);
                    screen[line_number][i] = (0xFF << 24) | (red << 20) | (green << 12) | (blue << 4);
                }
 
                // check is screen frame ready
                line_inserted[line_number] = 1;
 
                boolean ready = true;
                for(int v : line_inserted) if(v != 1) ready = false;
 
                // create PNG image file when frame ready
                if(ready) {
                    BufferedImage img = new BufferedImage(640,480, BufferedImage.TYPE_INT_ARGB);
                    for(int i=0; i<480; i++) {
                        for(int j=0; j<640; j++) {
                            img.setRGB(j, i, screen[i][j]);
                        }
                    }
                    File output_dir = new File(args[2]);
                    if(output_dir.exists() == false || output_dir.isDirectory() == false) throw new Exception("Output directory illegal: " + output_dir.getCanonicalPath());
 
                    File output_file = new File(output_dir.getCanonicalFile() + File.separator + "frame_" + frame_number);
                    boolean written = ImageIO.write(img, "PNG", output_file);
                    if(written == false) throw new Exception("Could not write image file: " + output_file.getCanonicalPath());
 
                    System.out.println("Written frame number: " + frame_number);
                    frame_number++;
                    for(int i=0; i<line_inserted.length; i++) line_inserted[i] = 0;
                }
            }
        }
    }
    static OutputStream p_out;
}