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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
|
/*
* BeagleJNIjava
*
*/
package beagle;
/*
* BeagleJNIjava
*
* @author Andrew Rambaut
* @author Marc Suchard
*
*/
public class BeagleJNIWrapper {
public static final String LIBRARY_NAME = getPlatformSpecificLibraryName();
/**
* private constructor to enforce singleton instance
*/
private BeagleJNIWrapper() {
}
public native String getVersion();
public native String getCitation();
public native ResourceDetails[] getResourceList();
public native int createInstance(
int tipCount,
int partialsBufferCount,
int compactBufferCount,
int stateCount,
int patternCount,
int eigenBufferCount,
int matrixBufferCount,
int categoryCount,
int scaleBufferCount,
final int[] resourceList,
int resourceCount,
long preferenceFlags,
long requirementFlags,
InstanceDetails returnInfo);
public native int finalize(int instance);
public native int setPatternWeights(int instance,
final double[] patternWeights);
public native int setTipStates(int instance, int tipIndex, final int[] inStates);
public native int getTipStates(int instance, int tipIndex, final int[] inStates);
public native int setTipPartials(int instance, int tipIndex, final double[] inPartials);
public native int setPartials(int instance, int bufferIndex, final double[] inPartials);
public native int getPartials(int instance, int bufferIndex, int scaleIndex,
final double[] outPartials);
public native int setEigenDecomposition(int instance,
int eigenIndex,
final double[] eigenVectors,
final double[] inverseEigenValues,
final double[] eigenValues);
public native int setStateFrequencies(int instance,
int stateFrequenciesIndex,
final double[] stateFrequencies);
public native int setCategoryWeights(int instance,
int categoryWeightsIndex,
final double[] categoryWeights);
public native int setCategoryRates(int instance,
final double[] inCategoryRates);
public native int setTransitionMatrix(int instance, int matrixIndex, final double[] inMatrix, double paddedValue);
public native int getTransitionMatrix(int instance, int matrixIndex, final double[] outMatrix);
public native int convolveTransitionMatrices(int instance,
final int[] firstIndices,
final int[] secondIndices,
final int[] resultIndices,
int matrixCount);
public native int updateTransitionMatrices(int instance, int eigenIndex,
final int[] probabilityIndices,
final int[] firstDerivativeIndices,
final int[] secondDervativeIndices,
final double[] edgeLengths,
int count);
public native int updatePartials(final int instance,
final int[] operations,
int operationCount,
int cumulativeScalingIndex);
public native int waitForPartials(final int instance,
final int[] destinationPartials,
int destinationPartialsCount);
public native int accumulateScaleFactors(final int instance,
final int[] scaleIndices,
final int count,
final int cumulativeScalingIndex);
public native int removeScaleFactors(final int instance,
final int[] scaleIndices,
final int count,
final int cumulativeScalingIndex);
public native int resetScaleFactors(final int instance,
final int cumulativeScalingIndex);
public native int copyScaleFactors(final int instance,
final int destScalingIndex,
final int srcScalingIndex);
public native int calculateRootLogLikelihoods(int instance,
final int[] bufferIndices,
final int[] categoryWeightsIndices,
final int[] stateFrequenciesIndices,
final int[] cumulativeScaleIndices,
int count,
final double[] outSumLogLikelihood);
public native int calculateEdgeLogLikelihoods(int instance,
final int[] parentBufferIndices,
final int[] childBufferIndices,
final int[] probabilityIndices,
final int[] firstDerivativeIndices,
final int[] secondDerivativeIndices,
final int[] categoryWeightsIndices,
final int[] stateFrequenciesIndices,
final int[] scalingFactorsIndices,
int count,
final double[] outSumLogLikelihood,
final double[] outSumFirstDerivative,
final double[] outSumSecondDerivative);
public native int getSiteLogLikelihoods(final int instance,
final double[] outLogLikelihoods);
/* Library loading routines */
private static String getPlatformSpecificLibraryName()
{
String osName = System.getProperty("os.name").toLowerCase();
String osArch = System.getProperty("os.arch").toLowerCase();
if (osName.startsWith("windows")) {
if(osArch.equals("x86")||osArch.equals("i386")) return "hmsbeagle32";
if(osArch.startsWith("amd64")||osArch.startsWith("x86_64")) return "hmsbeagle64";
}
return "hmsbeagle-jni";
}
public static void loadBeagleLibrary() throws UnsatisfiedLinkError {
String path = "";
if (System.getProperty("beagle.library.path") != null) {
path = System.getProperty("beagle.library.path");
if (path.length() > 0 && !path.endsWith("/")) {
path += "/";
}
}
System.loadLibrary(path + LIBRARY_NAME);
INSTANCE = new BeagleJNIWrapper();
}
public static BeagleJNIWrapper INSTANCE;
}
|
