package ij.plugin;
import ij.*;
import ij.io.*;
import ij.process.*;
import ij.gui.ImageWindow;
import java.awt.*;
import java.io.*;
import java.awt.image.*;
import java.net.*;

/** Opens NIH Image look-up tables (LUTs), 768 byte binary LUTs
    (256 reds, 256 greens and 256 blues), LUTs in text format, 
    or generates the LUT specified by the string argument 
    passed to the run() method. */
public class LutLoader extends ImagePlus implements PlugIn {

    private static String defaultDirectory = null;
    private boolean suppressErrors;
    
    /** Returns the LUT 'name' as an IndexColorModel, where
     * 'name' can be any entry in the Image/Lookup Tables menu.
     * @see ij.IJ#getLuts
     * See: Help>Examples>JavaScript/Show all LUTs
    */
    public static IndexColorModel getLut(String name) {
        if (name==null) return null;
        LutLoader ll = new LutLoader();
        FileInfo fi = ll.getBuiltInLut(name.toLowerCase());
        if (fi.fileName!=null)
            return new IndexColorModel(8, 256, fi.reds, fi.greens, fi.blues);
        String path = IJ.getDir("luts")+name+".lut";
        IndexColorModel lut = LutLoader.openLut("noerror:"+path);
        if (lut==null) {
            path = IJ.getDir("luts")+name.replaceAll(" ","_")+".lut";
            lut = LutLoader.openLut("noerror:"+path);
        }
        return lut;
    }

    /** If 'arg'="", displays a file open dialog and opens the specified
        LUT. If 'arg' is a path, opens the LUT specified by the path. If
        'arg'="fire", "ice", etc., uses a method to generate the LUT. */
    public void run(String arg) {
        if (arg.equals("invert")) {
            invertLut();
            return;
        }
        
        // Built in LUT
        FileInfo fi = getBuiltInLut(arg);
        if (fi.fileName!=null) {
            showLut(fi, true);
            Menus.updateMenus();            
            return;
        }
        
        // LUT in luts folder
        OpenDialog od = new OpenDialog("Open LUT...", arg);
        fi.directory = od.getDirectory();
        fi.fileName = od.getFileName();
        if (fi.fileName==null)
                return;
        if (openLut(fi))
            showLut(fi, arg.equals(""));
        IJ.showStatus("");
    }
    
    private FileInfo getBuiltInLut(String name) {
        FileInfo fi = new FileInfo();
        fi.reds = new byte[256]; 
        fi.greens = new byte[256]; 
        fi.blues = new byte[256];
        fi.lutSize = 256;
        fi.fileName = null;
        if (name==null)
            return fi;
        if (name.equals("3-3-2 rgb")) name="3-3-2 RGB";
        if (name.equals("red/green")) name="redgreen";
        int nColors = 0;
        if (name.equals("fire"))
            nColors = fire(fi.reds, fi.greens, fi.blues);
        else if (name.equals("grays"))
            nColors = grays(fi.reds, fi.greens, fi.blues);
        else if (name.equals("ice"))
            nColors = ice(fi.reds, fi.greens, fi.blues);
        else if (name.equals("spectrum"))
            nColors = spectrum(fi.reds, fi.greens, fi.blues);
        else if (name.equals("3-3-2 RGB"))
            nColors = rgb332(fi.reds, fi.greens, fi.blues);
        else if (name.equals("red"))
            nColors = primaryColor(4, fi.reds, fi.greens, fi.blues);
        else if (name.equals("green"))
            nColors = primaryColor(2, fi.reds, fi.greens, fi.blues);
        else if (name.equals("blue"))
            nColors = primaryColor(1, fi.reds, fi.greens, fi.blues);
        else if (name.equals("cyan"))
            nColors = primaryColor(3, fi.reds, fi.greens, fi.blues);
        else if (name.equals("magenta"))
            nColors = primaryColor(5, fi.reds, fi.greens, fi.blues);
        else if (name.equals("yellow"))
            nColors = primaryColor(6, fi.reds, fi.greens, fi.blues);
        else if (name.equals("redgreen"))
            nColors = redGreen(fi.reds, fi.greens, fi.blues);
        if (nColors>0) {
            if (nColors<256)
                interpolate(fi.reds, fi.greens, fi.blues, nColors);
            fi.fileName = name;
        }
        return fi;
    }
    
    private void showLut(FileInfo fi, boolean showImage) {
        ImagePlus imp = WindowManager.getCurrentImage();
        if (imp!=null) {
            if (imp.getProcessor().getNChannels()!=1) {
                IJ.error("LUTs cannot be assiged to RGB Images.");
            } else if (imp.isComposite() && ((CompositeImage)imp).getMode()==IJ.GRAYSCALE) {
                CompositeImage cimp = (CompositeImage)imp;
                cimp.setMode(IJ.COLOR);
                int saveC = cimp.getChannel();
                IndexColorModel cm = new IndexColorModel(8, 256, fi.reds, fi.greens, fi.blues);
                for (int c=1; c<=cimp.getNChannels(); c++) {
                    cimp.setC(c);
                    cimp.setChannelColorModel(cm);
                }
                imp.setC(saveC);
                imp.updateAndRepaintWindow();
            } else {
                ImageProcessor ip = imp.getChannelProcessor();
                IndexColorModel cm = new IndexColorModel(8, 256, fi.reds, fi.greens, fi.blues);
                if (imp.isComposite())
                    ((CompositeImage)imp).setChannelColorModel(cm);
                else {
                    ip.setColorModel(cm);
                    //ip.setMinAndMax(ip.getMin(), ip.getMax());
                    if (imp.getWindow()==null)
                        imp.setProcessor(ip);
                }
                if (imp.getStackSize()>1)
                    imp.getStack().setColorModel(cm);
                imp.updateAndRepaintWindow();
                if (IJ.isMacro() && imp.getWindow()!=null)
                    IJ.wait(25);
            }
            saveLUTName(imp, fi);
        } else
            createImage(fi, showImage);
    }
    
    private void saveLUTName(ImagePlus imp, FileInfo fi) {
        if (imp!=null && fi!=null && fi.fileName!=null) {
            String name = fi.fileName;
            if (name.endsWith(".lut"))
                name = name.substring(0,name.length()-4);
            if (name.equals("grays"))
                imp.setProp(LUT.nameKey, null);
            else
                imp.setProp(LUT.nameKey, name);
        }
    }
    
    void invertLut() {
        ImagePlus imp = WindowManager.getCurrentImage();
        if (imp==null)
            {IJ.noImage(); return;}
        if (imp.getProcessor().getNChannels()==3) {
            IJ.error("RGB images do not use LUTs");
            return;
        }
        if (imp.isComposite()) {
            CompositeImage ci = (CompositeImage)imp;
            LUT lut = ci.getChannelLut();
            if (lut!=null)
                ci.setChannelLut(lut.createInvertedLut());
        } else {
            ImageProcessor ip = imp.getProcessor();
            ip.invertLut();
            if (imp.getStackSize()>1)
                imp.getStack().setColorModel(ip.getColorModel());
        }
        imp.updateAndRepaintWindow();
    }

    int fire(byte[] reds, byte[] greens, byte[] blues) {
        int[] r = {0,0,1,25,49,73,98,122,146,162,173,184,195,207,217,229,240,252,255,255,255,255,255,255,255,255,255,255,255,255,255,255};
        int[] g = {0,0,0,0,0,0,0,0,0,0,0,0,0,14,35,57,79,101,117,133,147,161,175,190,205,219,234,248,255,255,255,255};
        int[] b = {0,61,96,130,165,192,220,227,210,181,151,122,93,64,35,5,0,0,0,0,0,0,0,0,0,0,0,35,98,160,223,255};
        for (int i=0; i<r.length; i++) {
            reds[i] = (byte)r[i];
            greens[i] = (byte)g[i];
            blues[i] = (byte)b[i];
        }
        return r.length;
    }

    int grays(byte[] reds, byte[] greens, byte[] blues) {
        for (int i=0; i<256; i++) {
            reds[i] = (byte)i;
            greens[i] = (byte)i;
            blues[i] = (byte)i;
        }
        return 256;
    }
    
    int primaryColor(int color, byte[] reds, byte[] greens, byte[] blues) {
        for (int i=0; i<256; i++) {
            if ((color&4)!=0)
                reds[i] = (byte)i;
            if ((color&2)!=0)
                greens[i] = (byte)i;
            if ((color&1)!=0)
                blues[i] = (byte)i;
        }
        return 256;
    }
    
    int ice(byte[] reds, byte[] greens, byte[] blues) {
        int[] r = {0,0,0,0,0,0,19,29,50,48,79,112,134,158,186,201,217,229,242,250,250,250,250,251,250,250,250,250,251,251,243,230};
        int[] g = {156,165,176,184,190,196,193,184,171,162,146,125,107,93,81,87,92,97,95,93,93,90,85,69,64,54,47,35,19,0,4,0};
        int[] b = {140,147,158,166,170,176,209,220,234,225,236,246,250,251,250,250,245,230,230,222,202,180,163,142,123,114,106,94,84,64,26,27};
        for (int i=0; i<r.length; i++) {
            reds[i] = (byte)r[i];
            greens[i] = (byte)g[i];
            blues[i] = (byte)b[i];
        }
        return r.length;
    }

    int spectrum(byte[] reds, byte[] greens, byte[] blues) {
        Color c;
        for (int i=0; i<256; i++) {
            c = Color.getHSBColor(i/255f, 1f, 1f);
            reds[i] = (byte)c.getRed();
            greens[i] = (byte)c.getGreen();
            blues[i] = (byte)c.getBlue();
        }
        return 256;
    }
    
    int rgb332(byte[] reds, byte[] greens, byte[] blues) {
        Color c;
        for (int i=0; i<256; i++) {
            reds[i] = (byte)(i&0xe0);
            greens[i] = (byte)((i<<3)&0xe0);
            blues[i] = (byte)((i<<6)&0xc0);
        }
        return 256;
    }

    int redGreen(byte[] reds, byte[] greens, byte[] blues) {
        for (int i=0; i<128; i++) {
            reds[i] = (byte)(i*2);
            greens[i] = (byte)0;
            blues[i] = (byte)0;
        }
        for (int i=128; i<256; i++) {
            reds[i] = (byte)0;
            greens[i] = (byte)(i*2);
            blues[i] = (byte)0;
        }
        return 256;
    }

    void interpolate(byte[] reds, byte[] greens, byte[] blues, int nColors) {
        byte[] r = new byte[nColors]; 
        byte[] g = new byte[nColors]; 
        byte[] b = new byte[nColors];
        System.arraycopy(reds, 0, r, 0, nColors);
        System.arraycopy(greens, 0, g, 0, nColors);
        System.arraycopy(blues, 0, b, 0, nColors);
        double scale = nColors/256.0;
        int i1, i2;
        double fraction;
        for (int i=0; i<256; i++) {
            i1 = (int)(i*scale);
            i2 = i1+1;
            if (i2==nColors) i2 = nColors-1;
            fraction = i*scale - i1;
            reds[i] = (byte)((1.0-fraction)*(r[i1]&255) + fraction*(r[i2]&255));
            greens[i] = (byte)((1.0-fraction)*(g[i1]&255) + fraction*(g[i2]&255));
            blues[i] = (byte)((1.0-fraction)*(b[i1]&255) + fraction*(b[i2]&255));
        }
    }
    
    /** Opens a LUT and returns it as a LUT object. */
    public static LUT openLut(String pathOrURL) {
        boolean noError = pathOrURL.startsWith("noerror:");
        if (noError)
            pathOrURL = pathOrURL.substring(8);
        FileInfo fi = new FileInfo();
        fi.reds = new byte[256]; 
        fi.greens = new byte[256]; 
        fi.blues = new byte[256];
        fi.lutSize = 256;
        int nColors = 0;
        if (pathOrURL.contains("://")) {
            fi.url = pathOrURL;
            fi.fileName = "";
        } else {
            OpenDialog od = new OpenDialog("Open LUT...", pathOrURL);
            fi.directory = od.getDirectory();
            fi.fileName = od.getFileName();
            if (fi.fileName==null)
                return null;
        }
        LutLoader loader = new LutLoader();
        loader.suppressErrors = noError;
        boolean ok = loader.openLut(fi);
        if (ok)
            return new LUT(fi.reds, fi.greens, fi.blues);
        else
            return null;
    }
    
    /** Opens an NIH Image LUT, 768 byte binary LUT or text LUT from a file or URL. */
    boolean openLut(FileInfo fi) {
        boolean isURL = fi.url!=null && !fi.url.equals("");
        int length = 0;
        String path = isURL?fi.url:fi.getFilePath();
        if (!isURL) {
            File f = new File(path);
            length = (int)f.length();
            if (length>10000) {
                error(path);
                return false;
            }
        }
        int size = 0;
        try {
            if (length>768)
                size = openBinaryLut(fi, isURL, false); // attempt to read NIH Image LUT
            if (size==0 && (length==0||length==768||length==970))
                size = openBinaryLut(fi, isURL, true); // otherwise read raw LUT
            if (size==0 && length>768)
                size = openTextLut(fi);
            if (size==0)
                error(path);
        } catch (IOException e) {
            if (!suppressErrors)
                IJ.error("LUT Loader", ""+e);
        }
        return size==256;
    }
    
    private void error(String path) {
        IJ.error("LUT Reader", "This is not an ImageJ or NIH Image LUT, a 768 byte \nraw LUT, or a LUT in text format.\n \n"+path);
    }

    /** Opens an NIH Image LUT or a 768 byte binary LUT. */
    int openBinaryLut(FileInfo fi, boolean isURL, boolean raw) throws IOException {
        InputStream is;
        if (isURL)
            is = new URL(fi.url+fi.fileName).openStream();
        else
            is = new FileInputStream(fi.getFilePath());
        DataInputStream f = new DataInputStream(is);
        int nColors = 256;
        if (!raw) {
            // attempt to read 32 byte NIH Image LUT header
            int id = f.readInt();
            if (id!=1229147980) { // 'ICOL'
                f.close();
                return 0;
            }
            int version = f.readShort();
            nColors = f.readShort();
            int start = f.readShort();
            int end = f.readShort();
            long fill1 = f.readLong();
            long fill2 = f.readLong();
            int filler = f.readInt();
        }
        f.read(fi.reds, 0, nColors);
        f.read(fi.greens, 0, nColors);
        f.read(fi.blues, 0, nColors);
        if (nColors<256)
            interpolate(fi.reds, fi.greens, fi.blues, nColors);
        f.close();
        return 256;
    }
    
    int openTextLut(FileInfo fi) throws IOException {
        TextReader tr = new TextReader();
        tr.hideErrorMessages();
        ImageProcessor ip = tr.open(fi.getFilePath());
        if (ip==null)
            return 0;
        int width = ip.getWidth();
        int height = ip.getHeight();
        if (width<3||width>4||height<256||height>258) 
            return 0; 
        int x = width==4?1:0; 
        int y = height>256?1:0;
        ip.setRoi(x, y, 3, 256);
        ip = ip.crop();
        for (int i=0; i<256; i++) {
            fi.reds[i] = (byte)ip.getPixelValue(0,i);
            fi.greens[i] = (byte)ip.getPixelValue(1,i);
            fi.blues[i] = (byte)ip.getPixelValue(2,i);
        }
        return 256;
    }

    private void createImage(FileInfo fi, boolean show) {
        IndexColorModel cm = new IndexColorModel(8, 256, fi.reds, fi.greens, fi.blues);
        ByteProcessor bp = createImage(cm);
        setProcessor(fi.fileName, bp);
        saveLUTName(this, fi);
        if (show) show();
    }
    
    /** Opens the specified ImageJ LUT and returns
        it as an IndexColorModel. Since 1.43t. */
    public static IndexColorModel open(String path) throws IOException {
        return open(new FileInputStream(path));
    }

    /** Opens an ImageJ LUT using an InputStream
        and returns it as an IndexColorModel. Since 1.43t. */
    public static IndexColorModel open(InputStream stream) throws IOException {
        DataInputStream f = new DataInputStream(stream);
        byte[] reds = new byte[256]; 
        byte[] greens = new byte[256]; 
        byte[] blues = new byte[256];
        f.read(reds, 0, 256);
        f.read(greens, 0, 256);
        f.read(blues, 0, 256);
        f.close();
        return new IndexColorModel(8, 256, reds, greens, blues);
    }
    
    /** Creates a 256x32 image from an IndexColorModel. Since 1.43t. */
    public static ByteProcessor createImage(IndexColorModel cm) {
        int width = 256;
        int height = 32;
        byte[] pixels = new byte[width*height];
        ByteProcessor bp = new ByteProcessor(width, height, pixels, cm);
        int[] ramp = new int[width];
        for (int i=0; i<width; i++)
            ramp[i] = i; 
        for (int y=0; y<height; y++)
            bp.putRow(0, y, ramp, width);
        return bp;
    }
    
}