    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;
    021import java.util.ArrayList;
    022import java.util.List;
    023import java.util.stream.Collectors;
    024
    025import org.apache.bcel.classfile.ArrayElementValue;
    026import org.apache.bcel.classfile.ElementValue;
    027import org.apache.commons.lang3.stream.Streams;
    028
    029/**
    030 * @since 6.0
    031 */
    032public class ArrayElementValueGen extends ElementValueGen {
    033    // J5TODO: Should we make this an array or a list? A list would be easier to
    034    // modify ...
    035    private final List<ElementValueGen> evalues;
    036
    037    /**
    038     * @param value
    039     * @param cpool
    040     */
    041    public ArrayElementValueGen(final ArrayElementValue value, final ConstantPoolGen cpool, final boolean copyPoolEntries) {
    042        super(ARRAY, cpool);
    043        evalues = new ArrayList<>();
    044        final ElementValue[] in = value.getElementValuesArray();
    045        for (final ElementValue element : in) {
    046            evalues.add(copy(element, cpool, copyPoolEntries));
    047        }
    048    }
    049
    050    public ArrayElementValueGen(final ConstantPoolGen cp) {
    051        super(ARRAY, cp);
    052        evalues = new ArrayList<>();
    053    }
    054
    055    public ArrayElementValueGen(final int type, final ElementValue[] elementValues, final ConstantPoolGen cpool) {
    056        super(type, cpool);
    057        if (type != ARRAY) {
    058            throw new IllegalArgumentException("Only element values of type array can be built with this ctor - type specified: " + type);
    059        }
    060        this.evalues = Streams.of(elementValues).map(e -> copy(e, cpool, true)).collect(Collectors.toList());
    061    }
    062
    063    public void addElement(final ElementValueGen gen) {
    064        evalues.add(gen);
    065    }
    066
    067    @Override
    068    public void dump(final DataOutputStream dos) throws IOException {
    069        dos.writeByte(super.getElementValueType()); // u1 type of value (ARRAY == '[')
    070        dos.writeShort(evalues.size());
    071        for (final ElementValueGen element : evalues) {
    072            element.dump(dos);
    073        }
    074    }
    075
    076    /**
    077     * Return immutable variant of this ArrayElementValueGen
    078     */
    079    @Override
    080    public ElementValue getElementValue() {
    081        final ElementValue[] immutableData = new ElementValue[evalues.size()];
    082        int i = 0;
    083        for (final ElementValueGen element : evalues) {
    084            immutableData[i++] = element.getElementValue();
    085        }
    086        return new ArrayElementValue(super.getElementValueType(), immutableData, getConstantPool().getConstantPool());
    087    }
    088
    089    public List<ElementValueGen> getElementValues() {
    090        return evalues;
    091    }
    092
    093    public int getElementValuesSize() {
    094        return evalues.size();
    095    }
    096
    097    @Override
    098    public String stringifyValue() {
    099        final StringBuilder sb = new StringBuilder();
    100        sb.append("[");
    101        String comma = "";
    102        for (final ElementValueGen element : evalues) {
    103            sb.append(comma);
    104            comma = ",";
    105            sb.append(element.stringifyValue());
    106        }
    107        sb.append("]");
    108        return sb.toString();
    109    }
    110}