    001/*
    002 * Licensed to the Apache Software Foundation (ASF) under one or more
    003 * contributor license agreements.  See the NOTICE file distributed with
    004 * this work for additional information regarding copyright ownership.
    005 * The ASF licenses this file to You under the Apache License, Version 2.0
    006 * (the "License"); you may not use this file except in compliance with
    007 * the License.  You may obtain a copy of the License at
    008 *
    009 *      http://www.apache.org/licenses/LICENSE-2.0
    010 *
    011 * Unless required by applicable law or agreed to in writing, software
    012 * distributed under the License is distributed on an "AS IS" BASIS,
    013 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    014 * See the License for the specific language governing permissions and
    015 * limitations under the License.
    016 */
    017package org.apache.commons.imaging.formats.tiff.fieldtypes;
    018
    019import java.nio.ByteOrder;
    020
    021import org.apache.commons.imaging.ImageWriteException;
    022import org.apache.commons.imaging.common.ByteConversions;
    023import org.apache.commons.imaging.formats.tiff.TiffField;
    024
    025public class FieldTypeShort extends FieldType {
    026    public FieldTypeShort(final int type, final String name) {
    027        super(type, name, 2);
    028    }
    029
    030    @Override
    031    public Object getValue(final TiffField entry) {
    032        final byte[] bytes = entry.getByteArrayValue();
    033        if (entry.getCount() == 1) {
    034            return ByteConversions.toShort(bytes, entry.getByteOrder());
    035        }
    036        return ByteConversions.toShorts(bytes, entry.getByteOrder());
    037    }
    038
    039    @Override
    040    public byte[] writeData(final Object o, final ByteOrder byteOrder) throws ImageWriteException {
    041        if (o instanceof Short) {
    042            return ByteConversions.toBytes(((Short) o).shortValue(), byteOrder);
    043        } else if (o instanceof short[]) {
    044            final short[] numbers = (short[]) o;
    045            return ByteConversions.toBytes(numbers, byteOrder);
    046        } else if (o instanceof Short[]) {
    047            final Short[] numbers = (Short[]) o;
    048            final short[] values = new short[numbers.length];
    049            for (int i = 0; i < values.length; i++) {
    050                values[i] = numbers[i].shortValue();
    051            }
    052            return ByteConversions.toBytes(values, byteOrder);
    053        } else {
    054            throw new ImageWriteException("Invalid data", o);
    055        }
    056    }
    057
    058}