depends/univplan/src/univplan/univplanbuilder/univplanbuilder-ext-gs-proj.cc

/*
 * 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.
 */

#include "univplan/univplanbuilder/univplanbuilder-ext-gs-proj.h"

namespace univplan {

UnivPlanBuilderExtGSProject::UnivPlanBuilderExtGSProject()
    : UnivPlanBuilderNode() {
  planNode.reset(new UnivPlanExtGSProj());
  ref = planNode.get();

  ref->set_allocated_super(UnivPlanBuilderNode::basePlanNode.release());
}

UnivPlanBuilderExtGSProject::~UnivPlanBuilderExtGSProject() {}

void UnivPlanBuilderExtGSProject::setColumnIndexes(
    const std::vector<int32_t> &colIndexes) {
  for (int i = 0; i < colIndexes.size(); ++i) {
    ref->add_columnindexes(colIndexes[i]);
  }
}

std::unique_ptr<UnivPlanBuilderPlanNodePoly>
UnivPlanBuilderExtGSProject::ownPlanNode() {
  UnivPlanBuilderPlanNodePoly::uptr pn(
      new UnivPlanBuilderPlanNodePoly(UNIVPLAN_EXT_GS_PROJ, planNode->super()));
  pn->getPlanNodePoly()->set_allocated_extgsproject(planNode.release());
  return std::move(pn);
}

void UnivPlanBuilderExtGSProject::from(const UnivPlanPlanNodePoly &node) {
  assert(node.type() == UNIVPLAN_EXT_GS_PROJ);
  ref->CopyFrom(node.extgsproject());
  baseRef = ref->mutable_super();
  baseRef->clear_leftplan();
  baseRef->clear_rightplan();
}

}  // namespace univplan