package ij.io;

import java.awt.*;
import java.awt.image.*;
import java.io.*;
import java.util.*;
import ij.*;
import ij.gui.*;
import ij.process.*;
import ij.util.*;
import ij.plugin.frame.Recorder;
import ij.plugin.FolderOpener;
import ij.plugin.FileInfoVirtualStack;
import ij.measure.Calibration;


/** This is a dialog box used to imports raw 8, 16, 24 and 32-bit images. */
public class ImportDialog {
    private String fileName;
    private String directory;
    static final String TYPE = "raw.type";
    static final String WIDTH = "raw.width";
    static final String HEIGHT = "raw.height";
    static final String OFFSET = "raw.offset";
    static final String N = "raw.n";
    static final String GAP = "raw.gap";
    static final String OPTIONS = "raw.options";
    static final int WHITE_IS_ZERO = 1;
    static final int INTEL_BYTE_ORDER = 2;
    static final int OPEN_ALL = 4;
    
    // default settings
    private static int sChoiceSelection = Prefs.getInt(TYPE,0);
    private static int sWidth = Prefs.getInt(WIDTH,512);
    private static int sHeight = Prefs.getInt(HEIGHT,512);
    private static long sOffset = Prefs.getInt(OFFSET,0);
    private static int sNImages = Prefs.getInt(N,1);
    private static int sGapBetweenImages = Prefs.getInt(GAP,0);
    private static boolean sWhiteIsZero;
    private static boolean sIntelByteOrder;
    private static boolean sVirtual;
    private int choiceSelection = sChoiceSelection;
    private int width = sWidth;
    private int height = sHeight;
    private long offset = sOffset;
    private int nImages = sNImages;
    private int gapBetweenImages = sGapBetweenImages;
    private boolean whiteIsZero = sWhiteIsZero;
    private boolean intelByteOrder = sIntelByteOrder;
    private boolean virtual = sVirtual;

    private static int options;
    private static FileInfo lastFileInfo;
    private boolean openAll;
    private static String[] types = {"8-bit", "16-bit Signed", "16-bit Unsigned",
        "32-bit Signed", "32-bit Unsigned", "32-bit Real", "64-bit Real", "24-bit RGB", 
        "24-bit RGB Planar", "24-bit BGR", "24-bit Integer", "32-bit ARGB", "32-bit ABGR", "1-bit Bitmap"};
        
    static {
        options = Prefs.getInt(OPTIONS,0);
        sWhiteIsZero = (options&WHITE_IS_ZERO)!=0;
        sIntelByteOrder = (options&INTEL_BYTE_ORDER)!=0;
    }
    
    public ImportDialog(String fileName, String directory) {
        this.fileName = fileName;
        this.directory = directory;
        IJ.showStatus("Importing: " + fileName);
    }

    public ImportDialog() {
    }

    boolean showDialog() {
        boolean macro = Macro.getOptions()!=null;
        if (macro) {
            width = height = 512;
            offset = gapBetweenImages = 0;
            nImages = 1;
            whiteIsZero = intelByteOrder = virtual = false;
        }
        if (choiceSelection>=types.length)
            choiceSelection = 0;
        getDimensionsFromName(fileName);
        GenericDialog gd = new GenericDialog("Import>Raw...");
        gd.addChoice("Image type:", types, types[choiceSelection]);
        gd.addNumericField("Width:", width, 0, 6, "pixels");
        gd.addNumericField("Height:", height, 0, 6, "pixels");
        gd.addNumericField("Offset to first image:", offset, 0, 6, "bytes");
        gd.addNumericField("Number of images:", nImages, 0, 6, null);
        gd.addNumericField("Gap between images:", gapBetweenImages, 0, 6, "bytes");
        gd.addCheckbox("White is zero", whiteIsZero);
        gd.addCheckbox("Little-endian byte order", intelByteOrder);
        gd.addCheckbox("Open all files in folder", openAll);
        gd.addCheckbox("Use virtual stack", virtual);
        gd.addHelp(IJ.URL+"/docs/menus/file.html#raw");
        gd.showDialog();
        if (gd.wasCanceled())
            return false;
        choiceSelection = gd.getNextChoiceIndex();
        width = (int)gd.getNextNumber();
        height = (int)gd.getNextNumber();
        gd.setSmartRecording(offset==0);
        offset = (long)gd.getNextNumber();
        gd.setSmartRecording(nImages==1);
        nImages = (int)gd.getNextNumber();
        gd.setSmartRecording(gapBetweenImages==0);
        gapBetweenImages = (int)gd.getNextNumber();
        gd.setSmartRecording(false);
        whiteIsZero = gd.getNextBoolean();
        intelByteOrder = gd.getNextBoolean();
        openAll = gd.getNextBoolean();
        virtual = gd.getNextBoolean();
        IJ.register(ImportDialog.class);
        if (!macro) {
            sChoiceSelection = choiceSelection;
            sWidth = width;
            sHeight = height;
            sOffset = offset;
            sNImages = nImages;
            sGapBetweenImages = gapBetweenImages;
            sWhiteIsZero = whiteIsZero;
            sIntelByteOrder = intelByteOrder;
            sVirtual = virtual;
        }
        return true;
    }
    
    /** Opens all the images in the directory. */
    void openAll(String[] list, FileInfo fi) {
        FolderOpener fo = new FolderOpener();
        list = fo.trimFileList(list);
        list = fo.sortFileList(list);
        if (list==null) return;
        ImageStack stack=null;
        ImagePlus imp=null;
        double min = Double.MAX_VALUE;
        double max = -Double.MAX_VALUE;
        int digits = 0;
        for (int i=0; i<list.length; i++) {
            if (list[i].startsWith("."))
                continue;
            fi.fileName = list[i];
            imp = new FileOpener(fi).openImage();
            if (imp==null)
                IJ.log(list[i] + ": unable to open");
            else {
                if (stack==null)
                    stack = imp.createEmptyStack();
                try {
                    ImageStack stack2 = imp.getStack();
                    int slices = stack2.getSize();
                    if (digits==0) {
                        digits = 2;
                        if (slices>99) digits=3;
                        if (slices>999) digits=4;
                        if (slices>9999) digits=5;
                    }
                    for (int n=1; n<=slices; n++) {
                        ImageProcessor ip = stack2.getProcessor(n);
                        if (ip.getMin()<min) min = ip.getMin();
                        if (ip.getMax()>max) max = ip.getMax();
                        String label = list[i];
                        if (slices>1) label += "-" + IJ.pad(n,digits);
                        stack.addSlice(label, ip);
                    }
                } catch(OutOfMemoryError e) {
                    IJ.outOfMemory("OpenAll");
                    stack.trim();
                    break;
                }
                IJ.showStatus((stack.getSize()+1) + ": " + list[i]);
            }
        }
        if (stack!=null) {
            imp = new ImagePlus("Imported Stack", stack);
            if (imp.getBitDepth()==16 || imp.getBitDepth()==32)
                imp.getProcessor().setMinAndMax(min, max);
                Calibration cal = imp.getCalibration();
                if (fi.fileType==FileInfo.GRAY16_SIGNED)
                    cal.setSigned16BitCalibration();
            imp.show();
        }
    }
    
    /** Displays the dialog and opens the specified image or images.
        Does nothing if the dialog is canceled. */
    public void openImage() {
        FileInfo fi = getFileInfo();
        if (fi==null) return;
        if (openAll) {
            if (virtual) {
                virtual = false;
                IJ.error("Import Raw", "\"Open All\" does not currently support virtual stacks");
                return;
            }
            String[] list = new File(directory).list();
            if (list==null) return;
            openAll(list, fi);
        } else if (virtual)
            new FileInfoVirtualStack(fi);
        else {
            FileOpener fo = new FileOpener(fi);
            ImagePlus imp = fo.openImage();
            if (imp!=null) {
                imp.show();
                int n = imp.getStackSize();
                if (n>1) {
                    imp.setSlice(n/2);
                    ImageProcessor ip = imp.getProcessor();
                    ip.resetMinAndMax();
                    imp.setDisplayRange(ip.getMin(),ip.getMax());
                }
            }
        }
    }

    /** Displays the dialog and returns a FileInfo object that can be used to
        open the image. Returns null if the dialog is canceled. The fileName 
        and directory fields are null if the no argument constructor was used. */
    public FileInfo getFileInfo() {
        if (!showDialog())
            return null;
        String imageType = types[choiceSelection];
        FileInfo fi = new FileInfo();
        fi.fileFormat = fi.RAW;
        fi.fileName = fileName;
        fi.directory = directory;
        fi.width = width;
        fi.height = height;
        if (offset>2147483647)
            fi.longOffset = offset;
        else
            fi.offset = (int)offset;
        fi.nImages = nImages;
        fi.gapBetweenImages = gapBetweenImages;
        fi.intelByteOrder = intelByteOrder;
        fi.whiteIsZero = whiteIsZero;
        if (imageType.equals("8-bit"))
            fi.fileType = FileInfo.GRAY8;
        else if (imageType.equals("16-bit Signed"))
            fi.fileType = FileInfo.GRAY16_SIGNED;
        else if (imageType.equals("16-bit Unsigned"))
            fi.fileType = FileInfo.GRAY16_UNSIGNED;
        else if (imageType.equals("32-bit Signed"))
            fi.fileType = FileInfo.GRAY32_INT;
        else if (imageType.equals("32-bit Unsigned"))
            fi.fileType = FileInfo.GRAY32_UNSIGNED;
        else if (imageType.equals("32-bit Real"))
            fi.fileType = FileInfo.GRAY32_FLOAT;
        else if (imageType.equals("64-bit Real"))
            fi.fileType = FileInfo.GRAY64_FLOAT;
        else if (imageType.equals("24-bit RGB"))
            fi.fileType = FileInfo.RGB;
        else if (imageType.equals("24-bit RGB Planar"))
            fi.fileType = FileInfo.RGB_PLANAR;
        else if (imageType.equals("24-bit BGR"))
            fi.fileType = FileInfo.BGR;
        else if (imageType.equals("24-bit Integer"))
            fi.fileType = FileInfo.GRAY24_UNSIGNED;
        else if (imageType.equals("32-bit ARGB"))
            fi.fileType = FileInfo.ARGB;
        else if (imageType.equals("32-bit ABGR"))
            fi.fileType = FileInfo.ABGR;
        else if (imageType.equals("1-bit Bitmap"))
            fi.fileType = FileInfo.BITMAP;
        else
            fi.fileType = FileInfo.GRAY8;
        if (IJ.debugMode) IJ.log("ImportDialog: "+fi);
        lastFileInfo = (FileInfo)fi.clone();
        return fi;
    }

    /** Called once when ImageJ quits. */
    public static void savePreferences(Properties prefs) {
        prefs.put(TYPE, Integer.toString(sChoiceSelection));
        prefs.put(WIDTH, Integer.toString(sWidth));
        prefs.put(HEIGHT, Integer.toString(sHeight));
        prefs.put(OFFSET, Integer.toString(sOffset>2147483647?0:(int)sOffset));
        prefs.put(N, Integer.toString(sNImages));
        prefs.put(GAP, Integer.toString(sGapBetweenImages));
        int options = 0;
        if (sWhiteIsZero)
            options |= WHITE_IS_ZERO;
        if (sIntelByteOrder)
            options |= INTEL_BYTE_ORDER;
        prefs.put(OPTIONS, Integer.toString(options));
    }
    
    /** Returns the FileInfo object used to import the last raw image,
        or null if a raw image has not been imported. */
    public static FileInfo getLastFileInfo() {
        return lastFileInfo;
    }
    
    private void getDimensionsFromName(String name) {
        if (name==null) return;
        int lastUnderscore = name.lastIndexOf("_");
        String name2 = name;
        if (lastUnderscore>=0)
            name2 = name.substring(lastUnderscore);
        char[] chars = new char[name2.length()];
        for (int i=0; i<name2.length(); i++)  // change non-digits to spaces
            chars[i] = Character.isDigit(name2.charAt(i))?name2.charAt(i):' ';
        name2 = new String(chars);
        String[] numbers = Tools.split(name2);
        int n = numbers.length;
        if (n<2 || n>3) return;
        int w = (int)Tools.parseDouble(numbers[0],0);
        if (w<10) return;
        int h = (int)Tools.parseDouble(numbers[1],0);
        if (h<10) return;
        width = w;
        height = h;
        nImages = 1;
        if (n==3) {
            int d = (int)Tools.parseDouble(numbers[2],0);
            if (d>0)
                nImages = d;
        }
        guessFormat(directory, name);
    }

    private void guessFormat(String dir, String name) {
        if (dir==null) return;
        File file = new File(dir+name);
        long imageSize = (long)width*height*nImages;
        long fileSize = file.length();
        if (fileSize==4*imageSize) {
            choiceSelection = 5; // 32-bit real
            intelByteOrder = true;
        } else if (fileSize==2*imageSize) {
            choiceSelection = 2;    // 16-bit unsigned
            intelByteOrder = true;
        } else if (fileSize==3*imageSize) {
            choiceSelection = 7;    // 24-bit RGB
        } else if (fileSize==imageSize)
            choiceSelection = 0;    // 8-bit
        if (name.endsWith("be.raw"))
            intelByteOrder = false;
    }
    
}