1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
/**
* 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.
*/
package org.apache.bigtop.bigpetstore.datagenerator.generators.transaction;
import java.util.List;
import java.util.Map;
import org.apache.bigtop.bigpetstore.datagenerator.Constants;
import org.apache.bigtop.bigpetstore.datagenerator.datamodels.Product;
import org.apache.bigtop.bigpetstore.datagenerator.framework.SeedFactory;
import org.apache.bigtop.bigpetstore.datagenerator.framework.samplers.ConditionalSampler;
import org.apache.bigtop.bigpetstore.datagenerator.framework.samplers.RouletteWheelSampler;
import org.apache.bigtop.bigpetstore.datagenerator.framework.samplers.Sampler;
import org.apache.bigtop.bigpetstore.datagenerator.framework.wfs.ConditionalWeightFunction;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
public class TransactionPurchasesHiddenMarkovModel implements ConditionalSampler<List<Product>, Double>
{
protected final static String STOP_STATE = "STOP";
final ConditionalSampler<Product, String> purchasingProcesses;
final ConditionalWeightFunction<Double, Double> categoryWF;
final CustomerInventory inventory;
final SeedFactory seedFactory;
public TransactionPurchasesHiddenMarkovModel(ConditionalSampler<Product, String> purchasingProcesses,
ConditionalWeightFunction<Double, Double> categoryWF, CustomerInventory inventory,
SeedFactory seedFactory)
{
this.purchasingProcesses = purchasingProcesses;
this.inventory = inventory;
this.categoryWF = categoryWF;
this.seedFactory = seedFactory;
}
protected String chooseCategory(double transactionTime, int numPurchases) throws Exception
{
ImmutableMap<String, Double> exhaustionTimes = this.inventory.getExhaustionTimes();
Map<String, Double> weights = Maps.newHashMap();
for(Map.Entry<String, Double> entry : exhaustionTimes.entrySet())
{
String category = entry.getKey();
double weight = this.categoryWF.weight(entry.getValue(), transactionTime);
weights.put(category, weight);
}
if(numPurchases > 0)
{
weights.put(STOP_STATE, Constants.STOP_CATEGORY_WEIGHT);
}
Sampler<String> sampler = RouletteWheelSampler.create(weights, seedFactory);
return sampler.sample();
}
protected Product chooseProduct(String category) throws Exception
{
return this.purchasingProcesses.sample(category);
}
public List<Product> sample(Double transactionTime) throws Exception
{
int numPurchases = 0;
List<Product> purchasedProducts = Lists.newArrayList();
String category;
while(true)
{
category = this.chooseCategory(transactionTime, numPurchases);
if(category.equals(STOP_STATE))
{
break;
}
Product product = this.chooseProduct(category);
purchasedProducts.add(product);
this.inventory.simulatePurchase(transactionTime, product);
numPurchases += 1;
}
return purchasedProducts;
}
}
|