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.bcel.generic;
018
019import java.io.DataOutputStream;
020import java.io.IOException;
021
022import org.apache.bcel.classfile.ConstantUtf8;
023import org.apache.bcel.classfile.ElementValue;
024import org.apache.bcel.classfile.EnumElementValue;
025
026/**
027 * @since 6.0
028 */
029public class EnumElementValueGen extends ElementValueGen {
030    // For enum types, these two indices point to the type and value
031    private final int typeIdx;
032
033    private final int valueIdx;
034
035    public EnumElementValueGen(final EnumElementValue value, final ConstantPoolGen cpool, final boolean copyPoolEntries) {
036        super(ENUM_CONSTANT, cpool);
037        if (copyPoolEntries) {
038            typeIdx = cpool.addUtf8(value.getEnumTypeString());// was
039                                                               // addClass(value.getEnumTypeString());
040            valueIdx = cpool.addUtf8(value.getEnumValueString()); // was
041                                                                  // addString(value.getEnumValueString());
042        } else {
043            typeIdx = value.getTypeIndex();
044            valueIdx = value.getValueIndex();
045        }
046    }
047
048    /**
049     * This ctor assumes the constant pool already contains the right type and value - as indicated by typeIdx and valueIdx.
050     * This ctor is used for deserialization
051     */
052    protected EnumElementValueGen(final int typeIdx, final int valueIdx, final ConstantPoolGen cpool) {
053        super(ElementValueGen.ENUM_CONSTANT, cpool);
054        if (super.getElementValueType() != ENUM_CONSTANT) {
055            throw new IllegalArgumentException("Only element values of type enum can be built with this ctor - type specified: " + super.getElementValueType());
056        }
057        this.typeIdx = typeIdx;
058        this.valueIdx = valueIdx;
059    }
060
061    public EnumElementValueGen(final ObjectType t, final String value, final ConstantPoolGen cpool) {
062        super(ElementValueGen.ENUM_CONSTANT, cpool);
063        typeIdx = cpool.addUtf8(t.getSignature());// was addClass(t);
064        valueIdx = cpool.addUtf8(value);// was addString(value);
065    }
066
067    @Override
068    public void dump(final DataOutputStream dos) throws IOException {
069        dos.writeByte(super.getElementValueType()); // u1 type of value (ENUM_CONSTANT == 'e')
070        dos.writeShort(typeIdx); // u2
071        dos.writeShort(valueIdx); // u2
072    }
073
074    /**
075     * Return immutable variant of this EnumElementValue
076     */
077    @Override
078    public ElementValue getElementValue() {
079        System.err.println("Duplicating value: " + getEnumTypeString() + ":" + getEnumValueString());
080        return new EnumElementValue(super.getElementValueType(), typeIdx, valueIdx, getConstantPool().getConstantPool());
081    }
082
083    // BCELBUG: Should we need to call utility.signatureToString() on the output
084    // here?
085    public String getEnumTypeString() {
086        // Constant cc = getConstantPool().getConstant(typeIdx);
087        // ConstantClass cu8 =
088        // (ConstantClass)getConstantPool().getConstant(typeIdx);
089        // return
090        // ((ConstantUtf8)getConstantPool().getConstant(cu8.getNameIndex())).getBytes();
091        return ((ConstantUtf8) getConstantPool().getConstant(typeIdx)).getBytes();
092        // return Utility.signatureToString(cu8.getBytes());
093    }
094
095    public String getEnumValueString() {
096        return ((ConstantUtf8) getConstantPool().getConstant(valueIdx)).getBytes();
097        // ConstantString cu8 =
098        // (ConstantString)getConstantPool().getConstant(valueIdx);
099        // return
100        // ((ConstantUtf8)getConstantPool().getConstant(cu8.getStringIndex())).getBytes();
101    }
102
103    public int getTypeIndex() {
104        return typeIdx;
105    }
106
107    public int getValueIndex() {
108        return valueIdx;
109    }
110
111    @Override
112    public String stringifyValue() {
113        final ConstantUtf8 cu8 = (ConstantUtf8) getConstantPool().getConstant(valueIdx);
114        return cu8.getBytes();
115        // ConstantString cu8 =
116        // (ConstantString)getConstantPool().getConstant(valueIdx);
117        // return
118        // ((ConstantUtf8)getConstantPool().getConstant(cu8.getStringIndex())).getBytes();
119    }
120}