/*
 * Copyright (c) 2011-2015 EPFL DATA Laboratory
 * Copyright (c) 2014-2015 The Squall Collaboration (see NOTICE)
 *
 * All rights reserved.
 *
 * 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 ch.epfl.data.squall.examples.imperative.theta;

import java.util.Arrays;
import java.util.List;
import java.util.Map;

import org.apache.log4j.Logger;

import ch.epfl.data.squall.components.Component;
import ch.epfl.data.squall.components.DataSourceComponent;
import ch.epfl.data.squall.components.theta.ThetaJoinComponentFactory;
import ch.epfl.data.squall.expressions.ColumnReference;
import ch.epfl.data.squall.operators.AggregateOperator;
import ch.epfl.data.squall.operators.AggregateSumOperator;
import ch.epfl.data.squall.operators.ProjectOperator;
import ch.epfl.data.squall.query_plans.QueryBuilder;
import ch.epfl.data.squall.query_plans.QueryPlan;
import ch.epfl.data.squall.query_plans.ThetaQueryPlansParameters;
import ch.epfl.data.squall.types.DoubleType;
import ch.epfl.data.squall.types.NumericType;

public class ThetaOutputDominatedPlan extends QueryPlan {

    private static Logger LOG = Logger
	    .getLogger(ThetaOutputDominatedPlan.class);

    private final QueryBuilder _queryBuilder = new QueryBuilder();

    private static final NumericType<Double> _doubleConv = new DoubleType();

    /*
     * SELECT SUM(SUPPLIER.SUPPKEY) FROM SUPPLIER, NATION
     */
    public ThetaOutputDominatedPlan(String dataPath, String extension, Map conf) {
	final int Theta_JoinType = ThetaQueryPlansParameters
		.getThetaJoinType(conf);
	// -------------------------------------------------------------------------------------
	final List<Integer> hashSupplier = Arrays.asList(0);

	final ProjectOperator projectionSupplier = new ProjectOperator(
		new int[] { 0 });

	final DataSourceComponent relationSupplier = new DataSourceComponent(
		"SUPPLIER", dataPath + "supplier" + extension).add(
		projectionSupplier).setOutputPartKey(hashSupplier);
	_queryBuilder.add(relationSupplier);

	// -------------------------------------------------------------------------------------
	final List<Integer> hashNation = Arrays.asList(0);

	final ProjectOperator projectionNation = new ProjectOperator(
		new int[] { 1 });

	final DataSourceComponent relationNation = new DataSourceComponent(
		"NATION", dataPath + "nation" + extension)
		.add(projectionNation).setOutputPartKey(hashNation);
	_queryBuilder.add(relationNation);

	final AggregateOperator agg = new AggregateSumOperator(
		new ColumnReference(_doubleConv, 0), conf);
	// Empty parameters = Cartesian Product

	Component lastJoiner = ThetaJoinComponentFactory
		.createThetaJoinOperator(Theta_JoinType, relationSupplier,
			relationNation, _queryBuilder)
		.add(new ProjectOperator(new int[] { 0 })).add(agg);
	// lastJoiner.setPrintOut(false);

	// -------------------------------------------------------------------------------------

    }

    @Override
    public QueryBuilder getQueryPlan() {
	return _queryBuilder;
    }
}