/* * OrnsteinUhlenbeckPriorLikelihoodParser.java * * Copyright (c) 2002-2015 Alexei Drummond, Andrew Rambaut and Marc Suchard * * This file is part of BEAST. * See the NOTICE file distributed with this work for additional * information regarding copyright ownership and licensing. * * BEAST is free software; you can redistribute it and/or modify * it under the terms of the GNU Lesser General Public License as * published by the Free Software Foundation; either version 2 * of the License, or (at your option) any later version. * * BEAST is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with BEAST; if not, write to the * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor, * Boston, MA 02110-1301 USA */ package dr.evomodelxml.coalescent; import dr.evomodel.coalescent.OrnsteinUhlenbeckPriorLikelihood; import dr.evomodel.coalescent.VariableDemographicModel; import dr.inference.distribution.ParametricDistributionModel; import dr.inference.model.Parameter; import dr.inference.model.Statistic; import dr.xml.*; /** */ public class OrnsteinUhlenbeckPriorLikelihoodParser extends AbstractXMLObjectParser {// (todo) Parser is still in a bad state public static final String OU = "Ornstein-Uhlenbeck"; public static final String LOG_SPACE = "logUnits"; public static final String NORMALIZE = "normalize"; public static final String DATA = "data"; public static final String TIMES = "times"; public static final String MEAN = "mean"; public static final String SIGMA = "sigma"; public static final String LAMBDA = "lambda"; public String getParserDescription() { return ""; } public Class getReturnType() { return OrnsteinUhlenbeckPriorLikelihood.class; } public String getParserName() { return OU; } private Parameter getParam(XMLObject xo, String name) throws XMLParseException { final XMLObject object = xo.getChild(name); // optional if (object == null) { return null; } final Object child = object.getChild(0); if (child instanceof Parameter) { return (Parameter) child; } double x = object.getDoubleChild(0); return new Parameter.Default(x); } public Object parseXMLObject(XMLObject xo) throws XMLParseException { Parameter mean = getParam(xo, MEAN); Parameter sigma = getParam(xo, SIGMA); Parameter lambda = getParam(xo, LAMBDA); final boolean logSpace = xo.getAttribute(LOG_SPACE, false); final boolean normalize = xo.getAttribute(NORMALIZE, false); VariableDemographicModel m = (VariableDemographicModel) xo.getChild(VariableDemographicModel.class); if (m != null) { ParametricDistributionModel popMeanPrior = (ParametricDistributionModel) xo.getChild(ParametricDistributionModel.class); return new OrnsteinUhlenbeckPriorLikelihood(mean, sigma, lambda, m, logSpace, normalize, popMeanPrior); } final XMLObject cxo1 = xo.getChild(DATA); Parameter dataParameter = (Parameter) cxo1.getChild(Parameter.class); final XMLObject cxo2 = xo.getChild(TIMES); final Parameter timesParameter = (Parameter) cxo2.getChild(Parameter.class); return new OrnsteinUhlenbeckPriorLikelihood(mean, sigma, lambda, dataParameter, timesParameter, logSpace, normalize); } public XMLSyntaxRule[] getSyntaxRules() { return new XMLSyntaxRule[]{ AttributeRule.newBooleanRule(LOG_SPACE, true), AttributeRule.newBooleanRule(NORMALIZE, true), //new ElementRule(DATA, new XMLSyntaxRule[]{new ElementRule(Statistic.class)}), //new ElementRule(TIMES, new XMLSyntaxRule[]{new ElementRule(Statistic.class)}), new XORRule( new ElementRule(MEAN, Double.class), new ElementRule(MEAN, Parameter.class), true ), new XORRule( new ElementRule(SIGMA, Double.class), new ElementRule(SIGMA, Parameter.class) ), new XORRule( new ElementRule(LAMBDA, Double.class), new ElementRule(LAMBDA, Parameter.class), true ), // you can't have a XOR (b AND c), yikes // make all optional and check in parser new ElementRule(VariableDemographicModel.class, true), new ElementRule(DATA, new XMLSyntaxRule[]{new ElementRule(Statistic.class)}, true), new ElementRule(TIMES, new XMLSyntaxRule[]{new ElementRule(Statistic.class)}, true) }; } }