/* This file is part of VoltDB.
 * Copyright (C) 2008-2010 VoltDB L.L.C.
 *
 * This file contains original code and/or modifications of original code.
 * Any modifications made by VoltDB L.L.C. are licensed under the following
 * terms and conditions:
 *
 * 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.
 */
/* Copyright (C) 2008
 * Evan Jones
 * Massachusetts Institute of Technology
 *
 * 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 org.voltdb.benchmark.tpcc;

/** Stores the scaling parameters for loading and running. */
public class ScaleParameters {
    
    public ScaleParameters(int items, int warehouses, int firstWarehouse, int districtsPerWarehouse,
                           int customersPerDistrict, int newOrdersPerDistrict) {
        
        // ITEMS
        assert(items > 0) : "Invalid # of items: " + items; 
        this.num_items = items;
        
        // WAREHOUSES
        assert warehouses > 0;
        this.warehouses = warehouses;
        this.starting_warehouse = firstWarehouse;
        this.last_warehouse = (this.warehouses + this.starting_warehouse - 1);
        
        // DISTRICTS
        assert(districtsPerWarehouse > 0) : "Invalid # of districtsPerWarehouse: " + districtsPerWarehouse;
        this.districtsPerWarehouse = districtsPerWarehouse;
        
        // CUSTOMERS
        assert(customersPerDistrict > 0) : "Invalid # of customersPerDistrict: " + customersPerDistrict;
        this.customersPerDistrict = customersPerDistrict;
        
        // NEWORDERS
        assert(newOrdersPerDistrict > 0) : "Invalid # of newOrdersPerDistrict: " + newOrdersPerDistrict;
        this.newOrdersPerDistrict = newOrdersPerDistrict;
    }

    public static ScaleParameters makeDefault(int warehouses) {
        return new ScaleParameters(TPCCConstants.NUM_ITEMS,
                                   warehouses,
                                   TPCCConstants.STARTING_WAREHOUSE,
                                   TPCCConstants.DISTRICTS_PER_WAREHOUSE,
                                   TPCCConstants.CUSTOMERS_PER_DISTRICT,
                                   TPCCConstants.INITIAL_NEW_ORDERS_PER_DISTRICT);
    }

    public static ScaleParameters makeWithScaleFactor(TPCCConfig config, double scaleFactor) {
        int items = TPCCConstants.NUM_ITEMS;
        if (config.scale_items) {
            items = (int)Math.max(1, TPCCConstants.NUM_ITEMS * scaleFactor);
        }
        int districts = TPCCConstants.DISTRICTS_PER_WAREHOUSE;
        int customers = (int)Math.max(1, TPCCConstants.CUSTOMERS_PER_DISTRICT * scaleFactor);
        int newOrders = (int)Math.max(1, TPCCConstants.INITIAL_NEW_ORDERS_PER_DISTRICT * scaleFactor);

        return new ScaleParameters(items,
                                   config.num_warehouses,
                                   config.first_warehouse,
                                   districts,
                                   customers,
                                   newOrders);
    }

    public String toString() {
        String out = num_items + " items\n";
        out += warehouses + " warehouses\n";
        out += districtsPerWarehouse + " districts/warehouse\n";
        out += customersPerDistrict + " customers/district\n";
        out += newOrdersPerDistrict + " initial new orders/district";
        return out;
    }

    public final int num_items;
    public final int warehouses;
    public final int starting_warehouse;
    public final int last_warehouse;
    public final int districtsPerWarehouse;
    public final int customersPerDistrict;
    public final int newOrdersPerDistrict;
}