package edu.brown.benchmark.tpce.generators;
import edu.brown.benchmark.tpce.generators.TradeGenerator.TradeType;
import edu.brown.benchmark.tpce.util.EGenDate;
public class CE {
private void initialize( TDriverCETxnSettings txnParamSettings ){
logger.sendToLogger(driverGlobalSettings);
if( txnParamSettings != null){
setTxnTunables( txnParamSettings );
}
else{
setTxnTunables( driverCETxnSettings );
}
}
private void autoSetRNGSeeds( int uniqueID ){
int baseYear, baseMonth, baseDay, millisec;
baseYear = EGenDate.getYear();
baseMonth = EGenDate.getMonth();
baseDay = EGenDate.getDay();
millisec = (EGenDate.getHour() * EGenDate.MinutesPerHour + EGenDate.getMinute()) * EGenDate.SecondsPerMinute + EGenDate.getSecond();
baseYear -= ( baseYear % 5 );
long Seed;
Seed = millisec / 100;
Seed <<= 11;
Seed += EGenDate.getDayNo(baseYear, baseMonth, baseDay) - EGenDate.getDayNo(baseYear, 1, 1);
Seed <<= 33;
Seed += uniqueID;
txnMixGenerator.setRNGSeed( Seed );
driverCESettings.cur_TxnMixRNGSeed = Seed;
Seed |= 0x0000000100000000L;
txnInputGenerator.setRNGSeed( Seed );
driverCESettings.cur_TxnInputRNGSeed = Seed;
}
/*
* Constructor - no partitioning by C_ID, automatic RNG seed generation (requires unique input)
*/
public CE( CESUTInterface sut, BaseLogger logger, final TPCEGenerator inputFiles, long configuredCustomerCount, long activeCustomerCount,
int scaleFactor, int daysOfInitialTrades, int uniqueID, final TDriverCETxnSettings driverCETxnSettings ){
this.driverCETxnSettings = new TDriverCETxnSettings();
driverGlobalSettings = new DriverGlobalSettings( configuredCustomerCount, activeCustomerCount, scaleFactor, daysOfInitialTrades );
driverCESettings = new DriverCESettings( uniqueID, 0, 0 );
this.sut = sut;
this.logger = logger;
txnMixGenerator = new CETxnMixGenerator(driverCETxnSettings, this.logger );
txnInputGenerator = new CETxnInputGenerator( inputFiles, configuredCustomerCount, activeCustomerCount, scaleFactor,
daysOfInitialTrades * EGenDate.HoursPerWorkDay, this.logger, driverCETxnSettings );
bClearBufferBeforeGeneration = false;
this.logger.sendToLogger("CE object constructed using constructor 1 (valid for publication: YES).");
initialize( driverCETxnSettings );
autoSetRNGSeeds( uniqueID );
this.logger.sendToLogger(driverCESettings);
}
/*
* Constructor - no partitioning by C_ID, RNG seeds provided
*/
public CE( CESUTInterface sut, BaseLogger logger, final TPCEGenerator inputFiles, long configuredCustomerCount, long activeCustomerCount,
int scaleFactor, int daysOfInitialTrades, int uniqueID,long TxnMixRNGSeed, long TxnInputRNGSeed, final TDriverCETxnSettings driverCETxnSettings ){
this.driverCETxnSettings = new TDriverCETxnSettings();
driverGlobalSettings = new DriverGlobalSettings( configuredCustomerCount, activeCustomerCount, scaleFactor, daysOfInitialTrades );
driverCESettings = new DriverCESettings( uniqueID, TxnMixRNGSeed, TxnInputRNGSeed );
this.sut = sut;
this.logger = logger;
txnMixGenerator = new CETxnMixGenerator(driverCETxnSettings, TxnMixRNGSeed, this.logger );
txnInputGenerator = new CETxnInputGenerator( inputFiles, configuredCustomerCount, activeCustomerCount, scaleFactor, daysOfInitialTrades * EGenDate.HoursPerWorkDay, TxnInputRNGSeed, this.logger, driverCETxnSettings );
bClearBufferBeforeGeneration = false;
this.logger.sendToLogger("CE object constructed using constructor 2 (valid for publication: YES).");
initialize( driverCETxnSettings );
autoSetRNGSeeds( uniqueID );
this.logger.sendToLogger(driverCESettings);
}
/*
* Constructor - partitioning by C_ID, automatic RNG seed generation (requires unique input)
*/
public CE( CESUTInterface sut, BaseLogger logger, final TPCEGenerator inputFiles, long configuredCustomerCount, long activeCustomerCount,
long iMyStartingCustomerId, long iMyCustomerCount, int iPartitionPercent, int scaleFactor, int daysOfInitialTrades,
int uniqueID, final TDriverCETxnSettings driverCETxnSettings ){
this.driverCETxnSettings = new TDriverCETxnSettings();
driverGlobalSettings = new DriverGlobalSettings( configuredCustomerCount, activeCustomerCount, scaleFactor, daysOfInitialTrades );
driverCESettings = new DriverCESettings( uniqueID, 0, 0 );
driverCEPartitionSettings = new DriverCEPartitionSettings( iMyStartingCustomerId, iMyCustomerCount, iPartitionPercent );
this.sut = sut;
this.logger = logger;
txnMixGenerator = new CETxnMixGenerator(driverCETxnSettings, this.logger );
txnInputGenerator = new CETxnInputGenerator( inputFiles, configuredCustomerCount, activeCustomerCount, scaleFactor,
daysOfInitialTrades * EGenDate.HoursPerWorkDay, iMyStartingCustomerId, iMyCustomerCount, iPartitionPercent, this.logger, driverCETxnSettings );
bClearBufferBeforeGeneration = false;
this.logger.sendToLogger("CE object constructed using constructor 3 (valid for publication: YES).");
initialize( driverCETxnSettings );
autoSetRNGSeeds( uniqueID );
this.logger.sendToLogger(driverCEPartitionSettings);
this.logger.sendToLogger(driverCESettings);
}
/*
* Constructor - partitioning by C_ID, RNG seeds provided
*/
public CE( CESUTInterface sut, BaseLogger logger, final TPCEGenerator inputFiles, long configuredCustomerCount, long activeCustomerCount,
long iMyStartingCustomerId, long iMyCustomerCount, int iPartitionPercent, int scaleFactor, int daysOfInitialTrades,
int uniqueID, long TxnMixRNGSeed, long TxnInputRNGSeed, final TDriverCETxnSettings driverCETxnSettings ){
this.driverCETxnSettings = new TDriverCETxnSettings();
driverGlobalSettings = new DriverGlobalSettings( configuredCustomerCount, activeCustomerCount, scaleFactor, daysOfInitialTrades );
driverCESettings = new DriverCESettings( uniqueID, TxnMixRNGSeed, TxnInputRNGSeed );
driverCEPartitionSettings = new DriverCEPartitionSettings( iMyStartingCustomerId, iMyCustomerCount, iPartitionPercent );
this.sut = sut;
this.logger = logger;
txnMixGenerator = new CETxnMixGenerator(driverCETxnSettings, TxnMixRNGSeed, this.logger );
txnInputGenerator = new CETxnInputGenerator( inputFiles, configuredCustomerCount, activeCustomerCount, scaleFactor,
daysOfInitialTrades * EGenDate.HoursPerWorkDay, iMyStartingCustomerId, iMyCustomerCount, iPartitionPercent,
TxnInputRNGSeed, this.logger, driverCETxnSettings );
bClearBufferBeforeGeneration = false;
this.logger.sendToLogger("CE object constructed using constructor 4 (valid for publication: YES).");
initialize( driverCETxnSettings );
autoSetRNGSeeds( uniqueID );
this.logger.sendToLogger(driverCEPartitionSettings);
this.logger.sendToLogger(driverCESettings);
}
public long getTxnInputGeneratorRNGSeed( ){
return( txnInputGenerator.getRNGSeed() );
}
public long getTxnMixGeneratorRNGSeed( ){
return( txnMixGenerator.getRNGSeed() );
}
public boolean setTxnTunables( final TDriverCETxnSettings txnParamSettings ){
if( txnParamSettings.isValid() == true ){
if (txnParamSettings != driverCETxnSettings){
driverCETxnSettings = txnParamSettings;
}
txnMixGenerator.updateTunables();
txnInputGenerator.updateTunables();
return true;
}
else{
return false;
}
}
public void doTxn(){
int iTxnType = txnMixGenerator.generateNextTxnType( );
if (bClearBufferBeforeGeneration){
zeroInputBuffer(iTxnType);
}
switch( iTxnType ){
case CETxnMixGenerator.BROKER_VOLUME:
txnInputGenerator.generateBrokerVolumeInput( brokerVolumeTxnInput );
sut.BrokerVolume( brokerVolumeTxnInput );
break;
case CETxnMixGenerator.CUSTOMER_POSITION:
txnInputGenerator.generateCustomerPositionInput( customerPositionTxnInput );
sut.CustomerPosition( customerPositionTxnInput );
break;
case CETxnMixGenerator.MARKET_WATCH:
txnInputGenerator.generateMarketWatchInput( marketWatchTxnInput );
sut.MarketWatch( marketWatchTxnInput );
break;
case CETxnMixGenerator.SECURITY_DETAIL:
txnInputGenerator.generateSecurityDetailInput( securityDetailTxnInput );
sut.SecurityDetail( securityDetailTxnInput );
break;
case CETxnMixGenerator.TRADE_LOOKUP:
txnInputGenerator.generateTradeLookupInput( tradeLookupTxnInput );
sut.TradeLookup( tradeLookupTxnInput );
break;
case CETxnMixGenerator.TRADE_ORDER:
/*
* These two variables will be modified in the GenerateTradeOrderInput
*/
boolean bExecutorIsAccountOwner = true;
int iTradeType = TradeType.eLimitBuy.ordinal();
txnInputGenerator.generateTradeOrderInput( tradeOrderTxnInput, iTradeType, bExecutorIsAccountOwner );
sut.TradeOrder( tradeOrderTxnInput, iTradeType, bExecutorIsAccountOwner );
break;
case CETxnMixGenerator.TRADE_STATUS:
txnInputGenerator.generateTradeStatusInput( tradeStatusTxnInput );
sut.TradeStatus( tradeStatusTxnInput );
break;
case CETxnMixGenerator.TRADE_UPDATE:
txnInputGenerator.generateTradeUpdateInput( tradeUpdateTxnInput );
sut.TradeUpdate( tradeUpdateTxnInput );
break;
default:
System.err.println("CE: Generated illegal transaction");
System.exit(1);
}
}
public void zeroInputBuffer(int iTxnType){
switch( iTxnType ){
case CETxnMixGenerator.BROKER_VOLUME:
brokerVolumeTxnInput = new TBrokerVolumeTxnInput();
break;
case CETxnMixGenerator.CUSTOMER_POSITION:
customerPositionTxnInput = new TCustomerPositionTxnInput();
break;
case CETxnMixGenerator.MARKET_WATCH:
marketWatchTxnInput = new TMarketWatchTxnInput();
break;
case CETxnMixGenerator.SECURITY_DETAIL:
securityDetailTxnInput = new TSecurityDetailTxnInput();
break;
case CETxnMixGenerator.TRADE_LOOKUP:
tradeLookupTxnInput = new TTradeLookupTxnInput();
break;
case CETxnMixGenerator.TRADE_ORDER:
tradeOrderTxnInput = new TTradeOrderTxnInput();
break;
case CETxnMixGenerator.TRADE_STATUS:
tradeStatusTxnInput = new TTradeStatusTxnInput();
break;
case CETxnMixGenerator.TRADE_UPDATE:
tradeUpdateTxnInput = new TTradeUpdateTxnInput();
break;
}
}
public void setClearBufferOption(boolean bClearBufferBeforeGeneration){
this.bClearBufferBeforeGeneration = bClearBufferBeforeGeneration;
}
public CETxnMixGenerator getCETxnMixGenerator(){
return txnMixGenerator;
}
public CETxnInputGenerator getCETxnInputGenerator(){
return txnInputGenerator;
}
private DriverGlobalSettings driverGlobalSettings;
private DriverCESettings driverCESettings;
private DriverCEPartitionSettings driverCEPartitionSettings;
private TDriverCETxnSettings driverCETxnSettings;
private CESUTInterface sut;
private BaseLogger logger;
private CETxnMixGenerator txnMixGenerator;
private CETxnInputGenerator txnInputGenerator;
private TBrokerVolumeTxnInput brokerVolumeTxnInput;
private TCustomerPositionTxnInput customerPositionTxnInput;
private TMarketWatchTxnInput marketWatchTxnInput;
private TSecurityDetailTxnInput securityDetailTxnInput;
private TTradeLookupTxnInput tradeLookupTxnInput;
private TTradeOrderTxnInput tradeOrderTxnInput;
private TTradeStatusTxnInput tradeStatusTxnInput;
private TTradeUpdateTxnInput tradeUpdateTxnInput;
private boolean bClearBufferBeforeGeneration;
}