(GSoC) Generic Assay Categorical/Binary data implementation Backend

model/src/main/java/org/cbioportal/model/GenericAssayBinaryEnrichment.java

@@ -0,0 +1,19 @@
+package org.cbioportal.model;
+
+import javax.validation.constraints.NotNull;
+import java.math.BigDecimal;
+import java.util.List;
+
+public class GenericAssayBinaryEnrichment extends GenericAssayEnrichment {
+    @NotNull
+    private List<GenericAssayCountSummary> counts;
+
+    public List<GenericAssayCountSummary> getCounts() {
+        return counts;
+    }
+
+    public void setCounts(List<GenericAssayCountSummary> counts) {
+        this.counts = counts;
+    }
+
+}

model/src/main/java/org/cbioportal/model/GenericAssayCategoricalEnrichment.java

@@ -0,0 +1,18 @@
+package org.cbioportal.model;
+
+import javax.validation.constraints.NotNull;
+import java.math.BigDecimal;
+
+public class GenericAssayCategoricalEnrichment extends GenericAssayEnrichment {
+    @NotNull
+    private BigDecimal qValue;
+
+    public BigDecimal getqValue() {
+        return qValue;
+    }
+
+    public void setqValue(BigDecimal qValue) {
+        this.qValue = qValue;
+    }
+
+}

model/src/main/java/org/cbioportal/model/GenericAssayCountSummary.java

@@ -0,0 +1,38 @@
+package org.cbioportal.model;
+
+import javax.validation.constraints.NotNull;
+import java.io.Serializable;
+
+public class GenericAssayCountSummary implements Serializable {
+
+    @NotNull
+    private String name;
+    @NotNull
+    private Integer count;
+    @NotNull
+    private Integer totalCount;
+
+    public Integer getCount() {
+        return count;
+    }
+
+    public void setCount(Integer count) {
+        this.count = count;
+    }
+
+    public String getName() {
+        return name;
+    }
+
+    public void setName(String name) {
+        this.name = name;
+    }
+
+    public Integer getTotalCount() {
+        return totalCount;
+    }
+
+    public void setTotalCount(Integer totalCount) {
+        this.totalCount = totalCount;
+    }
+}

model/src/main/java/org/cbioportal/model/GenericAssayEnrichment.java

@@ -1,6 +1,7 @@
 package org.cbioportal.model;
 
 import java.io.Serializable;
+import java.math.BigDecimal;
 import java.util.HashMap;
 
 import javax.validation.constraints.NotNull;
@@ -11,6 +12,8 @@ public class GenericAssayEnrichment extends ExpressionEnrichment implements Seri
 	private String stableId;
 	@NotNull
 	private String name;
+    @NotNull
+    private BigDecimal qValue;
 	@NotNull
 	private HashMap<String, String> genericEntityMetaProperties;
 
@@ -38,4 +41,14 @@ public void setGenericEntityMetaProperties(HashMap<String, String> genericEntity
 		this.genericEntityMetaProperties = genericEntityMetaProperties;
 	}
 
+    public static int compare(GenericAssayEnrichment c1, GenericAssayEnrichment c2) {
+        return c1.getpValue().compareTo(c2.getpValue());
+    }
+    public BigDecimal getqValue() {
+        return qValue;
+    }
+
+    public void setqValue(BigDecimal qValue) {
+        this.qValue = qValue;
+    }
 }

service/src/main/java/org/cbioportal/service/ExpressionEnrichmentService.java

@@ -7,6 +7,8 @@
 import org.cbioportal.model.GenericAssayEnrichment;
 import org.cbioportal.model.GenomicEnrichment;
 import org.cbioportal.model.MolecularProfileCaseIdentifier;
+import org.cbioportal.model.GenericAssayBinaryEnrichment;
+import org.cbioportal.model.GenericAssayCategoricalEnrichment;
 import org.cbioportal.service.exception.MolecularProfileNotFoundException;
 
 public interface ExpressionEnrichmentService {
@@ -15,8 +17,18 @@ List<GenomicEnrichment> getGenomicEnrichments(String molecularProfileId,
             Map<String, List<MolecularProfileCaseIdentifier>> molecularProfileCaseSets, EnrichmentType enrichmentType)
             throws MolecularProfileNotFoundException;
 
-    List<GenericAssayEnrichment> getGenericAssayEnrichments(String molecularProfileId,
+    List<GenericAssayEnrichment> getGenericAssayNumericalEnrichments(String molecularProfileId,
             Map<String, List<MolecularProfileCaseIdentifier>> molecularProfileCaseSets, EnrichmentType enrichmentType)
             throws MolecularProfileNotFoundException;
+    List<GenericAssayBinaryEnrichment> getGenericAssayBinaryEnrichments(
+        String molecularProfileId,
+        Map<String, List<MolecularProfileCaseIdentifier>> molecularProfileCaseSets,
+        EnrichmentType enrichmentType)
+        throws MolecularProfileNotFoundException;
+
+    List<GenericAssayCategoricalEnrichment> getGenericAssayCategoricalEnrichments(String molecularProfileId,
+                                                                                  Map<String, List<MolecularProfileCaseIdentifier>> molecularProfileCaseSets,
+                                                                                  EnrichmentType enrichmentType)
+        throws MolecularProfileNotFoundException;
 
 }