package com.bytezone.diskbrowser.applefile;

import java.awt.image.BufferedImage;
import java.awt.image.DataBuffer;

import com.bytezone.diskbrowser.prodos.ProdosConstants;
import com.bytezone.diskbrowser.utilities.HexFormatter;

public class SHRPictureFile2 extends HiResImage
{
  ColorTable[] colorTables;
  byte[] controlBytes;

  // see Graphics & Animation.2mg

  public SHRPictureFile2 (String name, byte[] buffer, int fileType, int auxType, int eof)
  {
    super (name, buffer, fileType, auxType, eof);

    switch (fileType)
    {
      case ProdosConstants.FILE_TYPE_PNT:
        switch (auxType)
        {
          case 0:
            System.out.printf (
                "%s: PNT aux 0 (Paintworks Packed SHR Image) not written yet%n", name);
            int background = HexFormatter.intValue (buffer[0x20], buffer[0x21]);

            byte[] palette = new byte[32];
            byte[] patterns = new byte[512];
            byte[] data = new byte[buffer.length - 0x222];

            System.arraycopy (buffer, 0x00, palette, 0, palette.length);
            System.arraycopy (buffer, 0x22, patterns, 0, patterns.length);
            System.arraycopy (buffer, 0x0222, data, 0, data.length);

            this.buffer = unpackBytes (data);
            break;

          case 1:                             // packed version of PIC/$00
            this.buffer = unpackBytes (buffer);
            controlBytes = new byte[200];
            System.arraycopy (this.buffer, 32000, controlBytes, 0, controlBytes.length);

            colorTables = new ColorTable[16];
            for (int i = 0; i < colorTables.length; i++)
              colorTables[i] = new ColorTable (i, this.buffer, 32256 + i * 32);
            break;

          case 2:                             // handled in SHRPictureFile1
            break;

          case 3:                             // packed version of PIC/$01
            System.out.printf ("%s: PNT aux 3 (Packed IIGS SHR Image) not written yet%n",
                name);
            break;

          case 4:                             // packed version of PIC/$02
            System.out.printf ("%s: PNT aux 4 (Packed SHR Brooks Image) not tested yet%n",
                name);
            this.buffer = unpackBytes (buffer);
            colorTables = new ColorTable[200];
            for (int i = 0; i < colorTables.length; i++)
            {
              colorTables[i] = new ColorTable (i, this.buffer, 32000 + i * 32);
              colorTables[i].reverse ();
            }
            break;

          default:
            System.out.printf ("%s: PNT unknown aux: %04X%n", name, auxType);
        }
        break;

      case ProdosConstants.FILE_TYPE_PIC:
        if (auxType > 2)
        {
          System.out.printf ("%s: PIC changing aux from %04X to 0%n", name, auxType);
          auxType = 0;
        }

        switch (auxType)
        {
          case 0:                             // unpacked version of PNT/$01
            controlBytes = new byte[200];
            System.arraycopy (buffer, 32000, controlBytes, 0, controlBytes.length);

            colorTables = new ColorTable[16];
            for (int i = 0; i < colorTables.length; i++)
              colorTables[i] = new ColorTable (i, buffer, 32256 + i * 32);
            break;

          case 1:                             // unpacked version of PNT/$03
            System.out.printf ("%s: PIC aux 1 not written yet%n", name);
            break;

          case 2:                             // unpacked version of PNT/$04
            colorTables = new ColorTable[200];
            for (int i = 0; i < colorTables.length; i++)
            {
              colorTables[i] = new ColorTable (i, buffer, 32000 + i * 32);
              colorTables[i].reverse ();
            }
            break;

          default:
            System.out.println ("PIC unknown aux " + auxType);
        }
        break;
      default:
        System.out.println ("unknown filetype " + fileType);
    }

    if (colorTables != null)
      createImage ();
  }

  @Override
  void createMonochromeImage ()
  {
  }

  @Override
  void createColourImage ()
  {
    image = new BufferedImage (320, 200, BufferedImage.TYPE_INT_RGB);
    DataBuffer dataBuffer = image.getRaster ().getDataBuffer ();

    int element = 0;
    int ptr = 0;
    for (int row = 0; row < 200; row++)
    {
      ColorTable colorTable =
          controlBytes != null ? colorTables[controlBytes[row] & 0x0F] : colorTables[row];

      for (int col = 0; col < 160; col++)
      {
        int left = (buffer[ptr] & 0xF0) >> 4;
        int right = buffer[ptr] & 0x0F;

        dataBuffer.setElem (element++, colorTable.entries[left].color.getRGB ());
        dataBuffer.setElem (element++, colorTable.entries[right].color.getRGB ());

        ptr++;
      }
    }
  }

  @Override
  public String getText ()
  {
    StringBuilder text = new StringBuilder (super.getText ());
    text.append ("\n\n");

    if (controlBytes != null)
    {
      text.append ("SCB\n---\n");
      for (int i = 0; i < controlBytes.length; i += 8)
      {
        for (int j = 0; j < 8; j++)
          text.append (String.format ("  %3d:  %02X  ", i + j, controlBytes[i + j]));
        text.append ("\n");
      }
      text.append ("\n");
    }

    if (colorTables != null)
    {
      text.append ("Color Table\n\n #");
      for (int i = 0; i < 16; i++)
        text.append (String.format ("   %02X ", i));
      text.append ("\n--");
      for (int i = 0; i < 16; i++)
        text.append ("  ----");
      text.append ("\n");
      for (ColorTable colorTable : colorTables)
      {
        text.append (colorTable.toLine ());
        text.append ("\n");
      }
    }

    text.append ("\nScreen lines\n\n");
    for (int i = 0; i < 200; i++)
    {
      text.append (String.format ("Line: %02X  %<3d%n", i));
      text.append (HexFormatter.format (buffer, i * 160, 160));
      text.append ("\n\n");
    }

    text.deleteCharAt (text.length () - 1);
    text.deleteCharAt (text.length () - 1);

    return text.toString ();
  }
}