public MixtureModelBranchRates(
TreeModel tree,
Parameter rateCategoryQuantilesParameter,
ParametricDistributionModel[] models,
Parameter distributionIndexParameter,
boolean useQuantilesForRates,
boolean normalize,
double normalizeBranchRateTo) {
super(MixtureModelBranchRatesParser.MIXTURE_MODEL_BRANCH_RATES);
this.useQuantilesForRates = useQuantilesForRates;
this.rateCategoryQuantiles =
new TreeParameterModel(tree, rateCategoryQuantilesParameter, false);
rates = new double[tree.getNodeCount()];
this.normalize = normalize;
this.treeModel = tree;
this.distributionModels = models;
this.normalizeBranchRateTo = normalizeBranchRateTo;
this.tree = new SimpleTree(tree);
this.distributionIndexParameter = distributionIndexParameter;
addVariable(this.distributionIndexParameter);
// Force the boundaries of rateCategoryParameter to match the category count
// d Parameter.DefaultBounds bound = new Parameter.DefaultBounds(categoryCount - 1, 0,
// rateCategoryParameter.getDimension());
// d rateCategoryParameter.addBounds(bound);
// rateCategoryQuantilesParameter.;
Parameter.DefaultBounds bound =
new Parameter.DefaultBounds(1.0, 0.0, rateCategoryQuantilesParameter.getDimension());
rateCategoryQuantilesParameter.addBounds(bound);
Parameter.DefaultBounds bound2 = new Parameter.DefaultBounds(models.length, 0.0, 1);
distributionIndexParameter.addBounds(bound2);
distributionIndexParameter.setParameterValue(0, 0);
// Parameter distributionIndexParameter;
for (ParametricDistributionModel distributionModel : distributionModels) {
addModel(distributionModel);
}
// AR - commented out: changes to the tree are handled by model changed events fired by
// rateCategories
// addModel(tree);
// d addModel(rateCategories);
addModel(rateCategoryQuantiles);
// addModel(treeModel); // Maybe
// AR - commented out: changes to rateCategoryParameter are handled by model changed events
// fired by rateCategories
// addVariable(rateCategoryParameter);
if (normalize) {
tree.addModelListener(
new ModelListener() {
public void modelChangedEvent(Model model, Object object, int index) {
computeFactor();
}
public void modelRestored(Model model) {
computeFactor();
}
});
}
setupRates();
}
