diff --git a/pom.xml b/pom.xml
index 3acf453..e4accae 100644
--- a/pom.xml
+++ b/pom.xml
@@ -8,7 +8,7 @@
com.google.cloud.genomics
google-genomics-dataflow
jar
- v1beta2-0.22-SNAPSHOT
+ v1-0.1-SNAPSHOT
Google
@@ -114,7 +114,7 @@
com.google.apis
google-api-services-genomics
- v1beta2-rev87-1.20.0
+ v1-rev73-1.21.0
@@ -127,7 +127,7 @@
com.google.cloud.genomics
google-genomics-utils
- v1beta2-0.44
+ v1-0.1
diff --git a/src/main/java/com/google/cloud/genomics/dataflow/functions/transmission/ExtractAlleleTransmissionStatus.java b/src/main/java/com/google/cloud/genomics/dataflow/functions/transmission/ExtractAlleleTransmissionStatus.java
index 7d63d54..ea8ad84 100644
--- a/src/main/java/com/google/cloud/genomics/dataflow/functions/transmission/ExtractAlleleTransmissionStatus.java
+++ b/src/main/java/com/google/cloud/genomics/dataflow/functions/transmission/ExtractAlleleTransmissionStatus.java
@@ -15,8 +15,8 @@
*/
package com.google.cloud.genomics.dataflow.functions.transmission;
-import com.google.api.services.genomics.model.Call;
import com.google.api.services.genomics.model.Variant;
+import com.google.api.services.genomics.model.VariantCall;
import com.google.cloud.dataflow.sdk.transforms.DoFn;
import com.google.cloud.dataflow.sdk.values.KV;
import com.google.cloud.genomics.dataflow.model.Allele;
@@ -93,8 +93,8 @@ public void processElement(ProcessContext c) {
}
@VisibleForTesting
- Call getSample(Variant variant, String sampleName) {
- for (Call call : variant.getCalls()) {
+ VariantCall getSample(Variant variant, String sampleName) {
+ for (VariantCall call : variant.getCalls()) {
if (call.getCallSetName() == sampleName)
return call;
}
diff --git a/src/main/java/com/google/cloud/genomics/dataflow/pipelines/AnnotateVariants.java b/src/main/java/com/google/cloud/genomics/dataflow/pipelines/AnnotateVariants.java
index 6061e7f..d934cbf 100644
--- a/src/main/java/com/google/cloud/genomics/dataflow/pipelines/AnnotateVariants.java
+++ b/src/main/java/com/google/cloud/genomics/dataflow/pipelines/AnnotateVariants.java
@@ -18,8 +18,6 @@
import com.google.api.services.genomics.model.Annotation;
import com.google.api.services.genomics.model.AnnotationSet;
import com.google.api.services.genomics.model.ListBasesResponse;
-import com.google.api.services.genomics.model.QueryRange;
-import com.google.api.services.genomics.model.RangePosition;
import com.google.api.services.genomics.model.SearchAnnotationsRequest;
import com.google.api.services.genomics.model.SearchVariantsRequest;
import com.google.api.services.genomics.model.Variant;
@@ -219,17 +217,15 @@ private ListMultimap, Annotation> retrieveVariantAnnotations(
Paginator.Annotations.create(genomics, ShardBoundary.Requirement.OVERLAPS).search(
new SearchAnnotationsRequest()
.setAnnotationSetIds(variantAnnotationSetIds)
- .setRange(new QueryRange()
- .setReferenceName(canonicalizeRefName(request.getReferenceName()))
- .setStart(request.getStart())
- .setEnd(request.getEnd())));
+ .setReferenceName(canonicalizeRefName(request.getReferenceName()))
+ .setStart(request.getStart())
+ .setEnd(request.getEnd()));
for (Annotation annotation : annotationIter) {
- RangePosition pos = annotation.getPosition();
long start = 0;
- if (pos.getStart() != null) {
- start = pos.getStart();
+ if (annotation.getStart() != null) {
+ start = annotation.getStart();
}
- annotationMap.put(Range.closedOpen(start, pos.getEnd()), annotation);
+ annotationMap.put(Range.closedOpen(start, annotation.getEnd()), annotation);
}
LOG.info(String.format("read %d variant annotations in %s (%.2f / s)", annotationMap.size(),
stopwatch, (double)annotationMap.size() / stopwatch.elapsed(TimeUnit.SECONDS)));
@@ -243,13 +239,11 @@ private IntervalTree retrieveTranscripts(Genomics genomics, SearchVa
Paginator.Annotations.create(genomics, ShardBoundary.Requirement.OVERLAPS).search(
new SearchAnnotationsRequest()
.setAnnotationSetIds(transcriptSetIds)
- .setRange(new QueryRange()
- .setReferenceName(canonicalizeRefName(request.getReferenceName()))
- .setStart(request.getStart())
- .setEnd(request.getEnd())));
+ .setReferenceName(canonicalizeRefName(request.getReferenceName()))
+ .setStart(request.getStart())
+ .setEnd(request.getEnd()));
for (Annotation annotation : transcriptIter) {
- RangePosition pos = annotation.getPosition();
- transcripts.put(pos.getStart().intValue(), pos.getEnd().intValue(), annotation);
+ transcripts.put(annotation.getStart().intValue(), annotation.getEnd().intValue(), annotation);
}
LOG.info(String.format("read %d transcripts in %s (%.2f / s)", transcripts.size(),
stopwatch, (double)transcripts.size() / stopwatch.elapsed(TimeUnit.SECONDS)));
@@ -258,23 +252,22 @@ private IntervalTree retrieveTranscripts(Genomics genomics, SearchVa
private String getCachedTranscriptBases(Genomics genomics, Annotation transcript)
throws IOException {
- RangePosition pos = transcript.getPosition();
- Range rng = Range.closedOpen(pos.getStart(), pos.getEnd());
+ Range rng = Range.closedOpen(transcript.getStart(), transcript.getEnd());
if (!refBaseCache.containsKey(rng)) {
- refBaseCache.put(rng, retrieveReferenceBases(genomics, pos));
+ refBaseCache.put(rng, retrieveReferenceBases(genomics, transcript));
}
return refBaseCache.get(rng);
}
- private String retrieveReferenceBases(Genomics genomics, RangePosition pos) throws IOException {
+ private String retrieveReferenceBases(Genomics genomics, Annotation annotation) throws IOException {
StringBuilder b = new StringBuilder();
String pageToken = "";
while (true) {
// TODO: Support full request parameterization for Paginator.References.Bases.
ListBasesResponse response = genomics.references().bases()
- .list(pos.getReferenceId())
- .setStart(pos.getStart())
- .setEnd(pos.getEnd())
+ .list(annotation.getReferenceId())
+ .setStart(annotation.getStart())
+ .setEnd(annotation.getEnd())
.setPageToken(pageToken)
.execute();
b.append(response.getSequence());
@@ -337,7 +330,7 @@ private static void validateRefsetForAnnotationSets(
Genomics genomics, List annosetIds) throws IOException {
String refsetId = null;
for (String annosetId : annosetIds) {
- String gotId = genomics.annotationSets().get(annosetId).execute().getReferenceSetId();
+ String gotId = genomics.annotationsets().get(annosetId).execute().getReferenceSetId();
if (refsetId == null) {
refsetId = gotId;
} else if (!refsetId.equals(gotId)) {
@@ -351,7 +344,7 @@ private static List validateAnnotationSetsFlag(
Genomics genomics, String flagValue, String wantType) throws IOException {
List annosetIds = ImmutableList.copyOf(flagValue.split(","));
for (String annosetId : annosetIds) {
- AnnotationSet annoset = genomics.annotationSets().get(annosetId).execute();
+ AnnotationSet annoset = genomics.annotationsets().get(annosetId).execute();
if (!wantType.equals(annoset.getType())) {
throw new IllegalArgumentException("annotation set " + annosetId + " has type " +
annoset.getType() + ", wanted type " + wantType);
diff --git a/src/main/java/com/google/cloud/genomics/dataflow/pipelines/CalculateCoverage.java b/src/main/java/com/google/cloud/genomics/dataflow/pipelines/CalculateCoverage.java
index d658f02..752cc39 100644
--- a/src/main/java/com/google/cloud/genomics/dataflow/pipelines/CalculateCoverage.java
+++ b/src/main/java/com/google/cloud/genomics/dataflow/pipelines/CalculateCoverage.java
@@ -15,11 +15,11 @@
import com.google.api.client.util.Strings;
import com.google.api.services.genomics.Genomics;
+import com.google.api.services.genomics.Genomics.Annotationsets;
import com.google.api.services.genomics.model.Annotation;
import com.google.api.services.genomics.model.AnnotationSet;
import com.google.api.services.genomics.model.BatchCreateAnnotationsRequest;
import com.google.api.services.genomics.model.Position;
-import com.google.api.services.genomics.model.RangePosition;
import com.google.cloud.dataflow.sdk.Pipeline;
import com.google.cloud.dataflow.sdk.options.Default;
import com.google.cloud.dataflow.sdk.options.Description;
@@ -421,17 +421,16 @@ public void processElement(ProcessContext c) throws GeneralSecurityException, IO
Position bucket = c.element().getKey();
Annotation a = new Annotation()
.setAnnotationSetId(asId)
- .setPosition(new RangePosition()
- .setStart(bucket.getPosition())
- .setEnd(bucket.getPosition() + pOptions.getBucketWidth())
- .setReferenceName(bucket.getReferenceName()))
+ .setStart(bucket.getPosition())
+ .setEnd(bucket.getPosition() + pOptions.getBucketWidth())
+ .setReferenceName(bucket.getReferenceName())
.setType("GENERIC")
- .setInfo(new HashMap>());
+ .setInfo(new HashMap>());
for (KV> mappingQualityKV : c.element().getValue()) {
- List output = Lists.newArrayList();
+ List