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;
020import java.util.Arrays;
021
022import org.apache.commons.imaging.ImagingException;
023import org.apache.commons.imaging.common.Allocator;
024import org.apache.commons.imaging.common.ByteConversions;
025import org.apache.commons.imaging.common.RationalNumber;
026import org.apache.commons.imaging.formats.tiff.TiffField;
027
028public class FieldTypeRational extends AbstractFieldType {
029    public FieldTypeRational(final int type, final String name) {
030        super(type, name, 8);
031    }
032
033    @Override
034    public Object getValue(final TiffField entry) {
035        final byte[] bytes = entry.getByteArrayValue();
036        final boolean unsignedType = entry.getFieldType() != SRATIONAL;
037        if (entry.getCount() == 1) {
038            return ByteConversions.toRational(bytes, entry.getByteOrder(), unsignedType);
039        }
040        return ByteConversions.toRationals(bytes, entry.getByteOrder(), unsignedType);
041    }
042
043    @Override
044    public byte[] writeData(final Object o, final ByteOrder byteOrder) throws ImagingException {
045        if (o instanceof RationalNumber) {
046            return ByteConversions.toBytes((RationalNumber) o, byteOrder);
047        }
048        if (o instanceof RationalNumber[]) {
049            return ByteConversions.toBytes((RationalNumber[]) o, byteOrder);
050        }
051        if (o instanceof Number) {
052            final Number number = (Number) o;
053            return ByteConversions.toBytes(RationalNumber.valueOf(number.doubleValue()), byteOrder);
054        }
055        if (o instanceof Number[]) {
056            final Number[] numbers = (Number[]) o;
057            final RationalNumber[] rationalNumbers = Allocator.array(numbers.length, RationalNumber[]::new, RationalNumber.SHALLOW_SIZE);
058            Arrays.setAll(rationalNumbers, RationalNumber::valueOf);
059            return ByteConversions.toBytes(rationalNumbers, byteOrder);
060        }
061        if (!(o instanceof double[])) {
062            throw new ImagingException("Invalid data", o);
063        }
064        final double[] numbers = (double[]) o;
065        final RationalNumber[] rationalNumbers = Allocator.array(numbers.length, RationalNumber[]::new, RationalNumber.SHALLOW_SIZE);
066        Arrays.setAll(rationalNumbers, RationalNumber::valueOf);
067        return ByteConversions.toBytes(rationalNumbers, byteOrder);
068    }
069}