/** * Copyright 2012-2017 Gunnar Morling (http://www.gunnarmorling.de/) * and/or other contributors as indicated by the @authors tag. See the * copyright.txt file in the distribution for a full listing of all * contributors. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package org.mapstruct.ap.internal.model.assignment; import java.util.HashSet; import java.util.Set; import org.mapstruct.ap.internal.model.common.Type; /** * * @author Sjaak Derksen */ public class EnumConstantWrapper extends AssignmentWrapper { private final Type enumType; public EnumConstantWrapper(Assignment decoratedAssignment, Type enumType ) { super( decoratedAssignment, false ); this.enumType = enumType; } @Override public Set<Type> getImportTypes() { Set<Type> imported = new HashSet<Type>( getAssignment().getImportTypes() ); imported.add( enumType ); return imported; } @Override public String toString() { return enumType.getName() + "." + getAssignment(); } }