// savepict.java
// d.g.gilbert, 1997

package dclap;
// pack edu.indiana.bio.dclap;


import java.awt.*;
import java.applet.*;
import java.net.*;
import java.io.*;
import java.lang.*;
import java.util.*;

import Acme.JPM.Encoders.*;



public interface ImageLast {

}



public class SavePICT
{
	public static Point gTranslate= new Point(0,0);
	Applet app = null;
	Component view;
	OutputStream os;
	int	format;
	
	public SavePICT() { }
	public SavePICT( Applet app) { this.app= app; }
	public SavePICT( int format, Component theView, OutputStream os) {
		this(null, format, theView, os);
		}
	public SavePICT( Applet app, int format, Component theView, OutputStream os) { 
		this.app= app; 
		this.format= format;
		this.view= theView;
		this.os= os;
		saveComponent();
		}

	public void setView( Component theView ) { this.view= theView; }
	public void setStream( OutputStream os) { this.os= os; }
	public void setFormat( int format) { this.format= format; }

	public void saveComponent() {
		if (view!=null && os!=null)
		 switch (format) {
			case DocFormat.PICT: toPICT( view, os); break;
			case DocFormat.JPEG: toJPEG( view, os); break;
			case DocFormat.GIF: toGIF( view, os); break;
			case DocFormat.POSTSCRIPT: toPostscript( view, os); break;
			}
		}
			
	protected void showStatus( String msg) {
		if (app==null) Debug.println( msg);	
		else app.showStatus( msg);
		}
			
	public boolean saveToFile(String file, String data) {  
    boolean okay= false;
    try {
      PrintStream f = new PrintStream(new FileOutputStream(file));
      f.print(data);
      f.close();
      showStatus("Printing Complete.");
      okay= true;
    	} 
    catch (Throwable t) {
      showStatus("Error while saving file.");
      System.out.println(t);
    	}
    return okay;
	} 

	public boolean saveThruServer(String host, int port, String path, String data) {
    boolean okay= false;
    try {
      //Socket datapipe = new Socket(host, port);
      URL datau= new URL("http", host, port, "");
      URLConnection datapipe= datau.openConnection();
      
      DataInputStream dis = new DataInputStream(datapipe.getInputStream());
      PrintStream dos = new PrintStream(datapipe.getOutputStream());
      
      dos.println("POST " + path + " HTTP/1.0");
      dos.print("Content-length: ");
      dos.println(data.length());
      dos.println();
      dos.println(data);
      
      while (!(dis.readLine().equals(""))); // protocol responses
      String newfile = dis.readLine();
      dis.close();
      dos.close();
      
      URL u = new URL(newfile);
      // display the data in a new frame
      if (app!=null) app.getAppletContext().showDocument(u,"_pict");
      // otherwise, do what with it?? presumably can't save to disk...
      showStatus("Printing Complete.");
      okay= true;
    	} 
    catch (Throwable t) {
      showStatus("Error during data transmission.");
      System.out.println(t);
    	}
    return okay;
		}
		

	public void toPICT(Component theView, OutputStream os) {
		Gr2PICT pict= new Gr2PICT(os, theView.getGraphics());
    paintInside(theView,pict);
		pict.finalize(); // make sure pict end is written
		}

  public void toPostscript(Component theView, OutputStream os) {
		dclap.util.PSGr pict = new dclap.util.PSGr(os, theView.getGraphics());
    paintInside(theView,pict);
		pict.finalize();  
		}
	  
	public void toGIF(Component theView, OutputStream os) {
		Image img= getImage( theView);
		if (img==null) return; // debug code
		try {
			ImageEncoder enc= new GifEncoder(img, os);
			enc.encode();
			}
    catch (Throwable t) {
      showStatus("Error during image encoding.");
      System.out.println(t);
    	}
		}
		
	public void toJPEG(Component theView, OutputStream os) {
		/**
		Image img= getImage( theView);
		if (img==null) return;  
		try {
			//!! JpegEncoder is incomplete - writes only header !?
			ImageEncoder enc= new JpegEncoder(img, os);
			enc.encode();
			}
    catch (Throwable t) {
      showStatus("Error during image encoding.");
      System.out.println(t);
    	}
    **/
		}

/***
		// debugging !!!!
	protected void grabImage(Image img) {
		Debug.println("SavePICT.getImage - grabbing pixels...");
		int w = (short)img.getWidth(null);  
		int h = (short)img.getHeight(null); 
		//if (w>50) w= 50; if (h>50) h= 50;
		Debug.println("SavePICT.grabImage - PixelConsumer...");
    dclap.util.PixelConsumer pc = new dclap.util.PixelConsumer(img);
		Hashtable colortab = new Hashtable();
  	for (int y=0; y<pc.ydim; y++)
  		for (int x=0; x<pc.xdim; x++) {
  			int n = pc.pix[x][y]; // pix[x][y] is bass-ackwards of natural storage!?
				Integer color= new Integer(n);
				colortab.put(color,color);
				}
				
		Debug.println("SavePICT.grabImage - #colors= " + colortab.size());
		}
****/
				
/***/
		// debugging !!!!
	Frame fr;
	
	protected void showImage(Image img) {
		int w = (short)img.getWidth(null);  
		int h = (short)img.getHeight(null); 
		Debug.println("SavePICT.getImage - drawn, showing... [" + w + "," + h + "]");
		fr= new Frame("Component image test");
		IconButton imgcan= new IconButton( null, img, Frame.DEFAULT_CURSOR, fr);
		fr.setLayout( new FlowLayout());
		fr.add( imgcan);
		fr.resize(w+5,h+25);
		fr.pack();
		fr.show();
		imgcan.repaint();
		}
/****/
			  
	public final Image getImage(Component c) { return getImage(c,c.size()); }

	public Image getImage(Component c, Dimension d) {	
	  int w= d.width;
	  int h= d.height;
	  Image img = c.createImage(w, h);
	  Graphics igr = img.getGraphics();
  	igr.setColor(c.getBackground());
		igr.fillRect( 0, 0, w, h);		
		igr.setColor(c.getForeground());
		// hack for image map writing -- need to know translate value
		gTranslate= new Point(0,0);
    paintInside(c, igr);
// debugging....    
    //if (Debug.isOn) { showImage( img); return null; }
		//if (Debug.isOn) grabImage( img);
   
   	return img; 	
    }

	public static Point translation() { return new Point(gTranslate.x, gTranslate.y); }
	
  protected void paintInside(Component comp, Graphics g) 
  {
    boolean doshift= !(comp instanceof java.awt.Window);
			// also, don't shift if window has menubar and current item is shifted due to that @@!!
		if ((Native.SystemFlavor() != Native.kSysMac) 
			&& (comp.getParent() instanceof java.awt.Window))
				doshift= false;
				
    Point p = comp.location();
    	// set origin for this component
    if (doshift) { g.translate( p.x, p.y); gTranslate.translate( p.x, p.y); }
    updateComponent(comp,g);				// draw this component
    boolean haslast= false;
    if (comp instanceof Container) {
      Component[] comps= ((Container) comp).getComponents();
      	// ImageLast - for unruly components that bozo up following c's imaging
      for (int i=0; i<comps.length; i++) {
        Component c= comps[i];
        boolean islast= (c instanceof ImageLast);
        haslast = haslast || islast;
        if (!islast) paintInside(c, g);
        }
      if (haslast) for (int i=0; i<comps.length; i++) {
        Component c= comps[i];
        if (c instanceof ImageLast) paintInside(c, g);
        }
    	}
    	// restore the coordinate system
    if (doshift) { g.translate( -p.x, -p.y); gTranslate.translate( -p.x, -p.y); }
  }

  
  protected void updateComponent(Component c, Graphics g) {
    // draw a few special types of Component
		if (g instanceof Gr2PICT) ((Gr2PICT)g).beginPicGroup();
    Rectangle b = c.bounds();
    int halfheight = b.height/2;
		Color backc= c.getBackground(); // Color.white
		Color forec= c.getForeground(); // Color.black;
		if (c.getFont() != null) g.setFont(c.getFont());
		
    if (c instanceof Button) {
      g.setColor(backc);  
      g.fillRoundRect(0,0,b.width, b.height, 12,12);
      g.setColor(forec);  
      g.drawRoundRect(0,0,b.width, b.height, 12,12);
      String labs= ((Button) c).getLabel();
      int x= (b.width - g.getFontMetrics().stringWidth(labs)) / 2;
      g.drawString(labs, x, halfheight+3);
    	} 
    	
    else if (c instanceof Checkbox) {
      Checkbox cbx= (Checkbox) c;
    	int cwide= 12;
    	int cwide2= cwide/2;
    	int cwide3= (cwide - cwide2) / 2;
      int x= 3, y = 3;
      
      if (cbx.getCheckboxGroup() != null) {
      	// radio button
      	g.setColor(backc);  
     		g.fillOval(x,y,cwide,cwide);
      	g.setColor(forec);  
     		g.drawOval(x,y,cwide,cwide);
       	if (cbx.getState()) g.fillOval(x+cwide3,y+cwide3,cwide2,cwide2);
      	}
      	
      else {
      	// checkbox
     	 	g.setColor(backc);  
     		g.fillRect(x,y,cwide,cwide);
      	g.setColor(forec);  
     		g.drawRect(x,y,cwide,cwide);
       	if (cbx.getState()) {
       		g.drawLine(x,y,x+cwide-1,y+cwide-1);
       		g.drawLine(x+cwide-1,y,x,y+cwide-1);
       		}
      	}
   		x += cwide+2;
    	g.drawString(cbx.getLabel(), x+2, halfheight+3);
    	} 
    	
    else if (c instanceof Choice) {
      g.setColor(Color.black);  
      g.fillRect(3,3,b.width-4, b.height-4);
      g.setColor(Color.white);  
      g.fillRect(0,0,b.width-2, b.height-2);
      g.setColor(forec);  
      g.drawRect(0,0,b.width-2, b.height-2);

				// drop arrow
			int[] xp= { b.width - 16, b.width - 6,  b.width - 11, b.width - 16};
			int[] yp= { halfheight - 3, halfheight - 3,  halfheight + 2, halfheight - 3};
    	g.fillPolygon( xp, yp, 4);  
      
      g.drawString(((Choice) c).getSelectedItem(), 4, halfheight+3);
    	} 
    	
    else if (c instanceof Label) {
      g.setColor(forec);  
      g.drawString(((Label) c).getText(), 2, halfheight+3);
    	} 
    	
    else if (c instanceof TextComponent) {
      g.setColor(backc);  
      g.fillRect(0,0,b.width, b.height);
      g.setColor(forec);  
      g.drawRect(0,0,b.width, b.height);
      g.drawString(((TextComponent) c).getText().trim(), 2, halfheight+3);
    	} 
    	
    else {
      // let it draw itself
			g.setColor(backc);
      g.fillRect(0,0,b.width, b.height); // ! b may be smaller than print size!
			g.setColor(forec);
      c.print(g); // was c.update(g), which does backcolor fill; 
      		// use print instead, to subclass & set component bounds for saving
    	}
		if (g instanceof Gr2PICT) ((Gr2PICT)g).endPicGroup();
  }

}