/** * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you 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. */ <@pp.dropOutputFile /> <@pp.changeOutputFile name="/org/apache/drill/exec/vector/complex/impl/ComplexCopier.java" /> <#include "/@includes/license.ftl" /> package org.apache.drill.exec.vector.complex.impl; <#include "/@includes/vv_imports.ftl" /> /* * This class is generated using freemarker and the ${.template_name} template. */ @SuppressWarnings("unused") public class ComplexCopier { /** * Do a deep copy of the value in input into output * @param in * @param out */ public static void copy(FieldReader input, FieldWriter output) { writeValue(input, output); } private static void writeValue(FieldReader reader, FieldWriter writer) { final DataMode m = reader.getType().getMode(); final MinorType mt = reader.getType().getMinorType(); switch(m){ case OPTIONAL: case REQUIRED: switch (mt) { case LIST: writer.startList(); while (reader.next()) { writeValue(reader.reader(), getListWriterForReader(reader.reader(), writer)); } writer.endList(); break; case MAP: writer.start(); if (reader.isSet()) { for(String name : reader){ FieldReader childReader = reader.reader(name); if(childReader.isSet()){ writeValue(childReader, getMapWriterForReader(childReader, writer, name)); } } } writer.end(); break; <#list vv.types as type><#list type.minor as minor><#assign name = minor.class?cap_first /> <#assign fields = minor.fields!type.fields /> <#assign uncappedName = name?uncap_first/> <#if !minor.class?starts_with("Decimal")> case ${name?upper_case}: if (reader.isSet()) { Nullable${name}Holder ${uncappedName}Holder = new Nullable${name}Holder(); reader.read(${uncappedName}Holder); if (${uncappedName}Holder.isSet == 1) { writer.write${name}(<#list fields as field>${uncappedName}Holder.${field.name}<#if field_has_next>, </#if></#list>); } } break; </#if> </#list></#list> } break; } } private static FieldWriter getMapWriterForReader(FieldReader reader, MapWriter writer, String name) { switch (reader.getType().getMinorType()) { <#list vv.types as type><#list type.minor as minor><#assign name = minor.class?cap_first /> <#assign fields = minor.fields!type.fields /> <#assign uncappedName = name?uncap_first/> <#if !minor.class?starts_with("Decimal")> case ${name?upper_case}: return (FieldWriter) writer.<#if name == "Int">integer<#else>${uncappedName}</#if>(name); </#if> </#list></#list> case MAP: return (FieldWriter) writer.map(name); case LIST: return (FieldWriter) writer.list(name); default: throw new UnsupportedOperationException(reader.getType().toString()); } } private static FieldWriter getListWriterForReader(FieldReader reader, ListWriter writer) { switch (reader.getType().getMinorType()) { <#list vv.types as type><#list type.minor as minor><#assign name = minor.class?cap_first /> <#assign fields = minor.fields!type.fields /> <#assign uncappedName = name?uncap_first/> <#if !minor.class?starts_with("Decimal")> case ${name?upper_case}: return (FieldWriter) writer.<#if name == "Int">integer<#else>${uncappedName}</#if>(); </#if> </#list></#list> case MAP: return (FieldWriter) writer.map(); case LIST: return (FieldWriter) writer.list(); default: throw new UnsupportedOperationException(reader.getType().toString()); } } }