dmolony-DiskBrowser/src/com/bytezone/diskbrowser/wizardry/MessageBlock.java

package com.bytezone.diskbrowser.wizardry;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import com.bytezone.common.Utility;

public class MessageBlock implements Iterable<MessageDataBlock>
{
  //  private final byte[] buffer;
  private final int indexOffset;
  private final int indexLength;

  private final List<MessageDataBlock> messageDataBlocks =
      new ArrayList<MessageDataBlock> ();

  private Huffman huffman;

  public MessageBlock (byte[] buffer)
  {
    indexOffset = Utility.getWord (buffer, 0);
    indexLength = Utility.getWord (buffer, 2);

    int ptr = indexOffset * 512;

    for (int i = 0, max = indexLength / 2; i < max; i++)
    {
      int firstMessageNo = Utility.getWord (buffer, ptr + i * 2);
      byte[] data = new byte[512];
      System.arraycopy (buffer, i * 512, data, 0, data.length);
      MessageDataBlock messageDataBlock =
          new MessageDataBlock ("Block " + firstMessageNo, data, firstMessageNo);
      messageDataBlocks.add (messageDataBlock);
    }
  }

  public void setHuffman (Huffman huffman)
  {
    this.huffman = huffman;
    for (MessageDataBlock messageDataBlock : messageDataBlocks)
      messageDataBlock.setHuffman (huffman);
  }

  public byte[] getMessage (int messageNo)
  {
    for (int i = 0; i < messageDataBlocks.size (); i++)
    {
      MessageDataBlock messageDataBlock = messageDataBlocks.get (i);
      if (messageDataBlock.firstMessageNo > messageNo)
        return messageDataBlocks.get (i - 1).getMessage (messageNo);
    }
    return null;
  }

  @Override
  public Iterator<MessageDataBlock> iterator ()
  {
    return messageDataBlocks.iterator ();
  }
}