dxt1compressor.java

来自「world wind java sdk 源码」· Java 代码 · 共 128 行

/* Copyright (C) 2001, 2008 United States Government as represented by
the Administrator of the National Aeronautics and Space Administration.
All Rights Reserved.
*/
package gov.nasa.worldwind.formats.dds;

import gov.nasa.worldwind.util.Logging;

/**
 * @author dcollins
 * @version $Id: DXT1Compressor.java 8991 2009-02-25 01:20:20Z dcollins $
 */
public class DXT1Compressor implements DXTCompressor
{
    public DXT1Compressor()
    {
    }

    public int getDXTFormat()
    {
        return DDSConstants.D3DFMT_DXT1;
    }

    public int getCompressedSize(java.awt.image.BufferedImage image, DXTCompressionAttributes attributes)
    {
        if (image == null)
        {
            String message = Logging.getMessage("nullValue.ImageIsNull");
            Logging.logger().severe(message);
            throw new IllegalArgumentException(message);
        }
        if (attributes == null)
        {
            String message = Logging.getMessage("nullValue.AttributesIsNull");
            Logging.logger().severe(message);
            throw new IllegalArgumentException(message);
        }

        // TODO: comment, provide documentation reference

        int width = Math.max(image.getWidth(), 4);
        int height = Math.max(image.getHeight(), 4);

        return (width * height) / 2;
    }

    public void compressImage(java.awt.image.BufferedImage image, DXTCompressionAttributes attributes,
        java.nio.ByteBuffer buffer)
    {
        if (image == null)
        {
            String message = Logging.getMessage("nullValue.ImageIsNull");
            Logging.logger().severe(message);
            throw new IllegalArgumentException(message);
        }
        if (attributes == null)
        {
            String message = Logging.getMessage("nullValue.AttributesIsNull");
            Logging.logger().severe(message);
            throw new IllegalArgumentException(message);
        }
        if (buffer == null)
        {
            String message = Logging.getMessage("nullValue.BufferNull");
            Logging.logger().severe(message);
            throw new IllegalArgumentException(message);
        }

        // If it is determined that the image and block have no alpha component, then we compress with DXT1 using a
        // four color palette. Otherwise, we use the three color palette (with the fourth color as transparent black).

        ColorBlock4x4 colorBlock = new ColorBlock4x4();
        ColorBlockExtractor colorBlockExtractor = this.getColorBlockExtractor(image);

        BlockDXT1 dxt1Block = new BlockDXT1();
        BlockDXT1Compressor dxt1Compressor = new BlockDXT1Compressor();

        int width = image.getWidth();
        int height = image.getHeight();

        boolean imageHasAlpha = image.getColorModel().hasAlpha();
        boolean enableAlpha = attributes.isEnableDXT1Alpha();
        int alphaThreshold = attributes.getDXT1AlphaThreshold();

        for (int j = 0; j < height; j += 4)
        {
            for (int i = 0; i < width; i += 4)
            {
                colorBlockExtractor.extractColorBlock4x4(attributes, i, j, colorBlock);

                if (enableAlpha && imageHasAlpha && blockHasDXT1Alpha(colorBlock, alphaThreshold))
                {
                    dxt1Compressor.compressBlockDXT1a(colorBlock, attributes, dxt1Block);
                }
                else
                {
                    dxt1Compressor.compressBlockDXT1(colorBlock, attributes, dxt1Block);
                }

                buffer.putShort((short) dxt1Block.color0);
                buffer.putShort((short) dxt1Block.color1);
                buffer.putInt((int) dxt1Block.colorIndexMask);
            }
        }
    }

    protected boolean blockHasDXT1Alpha(ColorBlock4x4 colorBlock, int alphaThreshold)
    {
        // DXT1 provides support for binary alpha. Therefore we determine treat a color block as needing alpha support
        // if any of the alpha values are less than a certain threshold.

        for (int i = 0; i < 16; i++)
        {
            if (colorBlock.color[i].a < alphaThreshold)
            {
                return true;
            }
        }

        return false;
    }

    protected ColorBlockExtractor getColorBlockExtractor(java.awt.image.BufferedImage image)
    {
        return new BasicColorBlockExtractor(image);
    }
}