/***************************************************************************
 *  Copyright (C) 2012 by H-Store Project                                  *
 *  Brown University                                                       *
 *  Massachusetts Institute of Technology                                  *
 *  Yale University                                                        *
 *                                                                         *
 *  Alex Kalinin (akalinin@cs.brown.edu)                                   *
 *  http://www.cs.brown.edu/~akalinin/                                     *
 *                                                                         *
 *  Permission is hereby granted, free of charge, to any person obtaining  *
 *  a copy of this software and associated documentation files (the        *
 *  "Software"), to deal in the Software without restriction, including    *
 *  without limitation the rights to use, copy, modify, merge, publish,    *
 *  distribute, sublicense, and/or sell copies of the Software, and to     *
 *  permit persons to whom the Software is furnished to do so, subject to  *
 *  the following conditions:                                              *
 *                                                                         *
 *  The above copyright notice and this permission notice shall be         *
 *  included in all copies or substantial portions of the Software.        *
 *                                                                         *
 *  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,        *
 *  EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF     *
 *  MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. *
 *  IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR      *
 *  OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,  *
 *  ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR  *
 *  OTHER DEALINGS IN THE SOFTWARE.                                        *
 ***************************************************************************/

package edu.brown.benchmark.tpce.generators;

import java.util.List;

import org.apache.log4j.Logger;
import org.voltdb.catalog.Table;

import edu.brown.benchmark.tpce.generators.TPCEGenerator.InputFile;

public class TaxRateGenerator extends TableGenerator {
    private final InputFileHandler tr_country_file;
    private final InputFileHandler tr_div_file;
    
    private final int countrySize;
    private final int divSize;
    private int countryCnt = 0;
    private int divCnt = 0;
    private int tuplesCnt = 0;
    
    public TaxRateGenerator(Table catalog_tbl, TPCEGenerator generator) {
        super(catalog_tbl, generator);
        
        tr_country_file = generator.getInputFile(InputFile.TAXCOUNTRY);
        tr_div_file = generator.getInputFile(InputFile.TAXDIV);
        
        countrySize = tr_country_file.getRecordsNum();
        divSize = tr_div_file.getRecordsNum();
    }
    
    @Override
    public boolean hasNext() {
        return (countryCnt < countrySize || divCnt < divSize);
    }
    
    @Override
    public Object[] next() {
        Object tuple[] = new Object[columnsNum];
        List<String[]> tuples;
        
        if (countryCnt < countrySize) {
            tuples = tr_country_file.getTuplesByIndex(countryCnt);
        }
        else {
            tuples = tr_div_file.getTuplesByIndex(divCnt);
        }
        
        String[] tr_record = tuples.get(tuplesCnt++);
        int col = 0;

        tuple[col++] = tr_record[0]; // tr_id
        tuple[col++] = tr_record[1]; // tr_name
        tuple[col++] = Double.valueOf(tr_record[2]); // tr_rate
        
        if (tuplesCnt == tuples.size()) {
            tuplesCnt = 0;
            if (countryCnt < countrySize) {
                countryCnt++;
            }
            else {
                divCnt++;
            }
        }
        
        return tuple;
    }
}