From 9a7fc597728f750313ea32db5fa99a44dea96e3b Mon Sep 17 00:00:00 2001
From: felmer <felmer>
Date: Mon, 6 Oct 2014 10:16:24 +0000
Subject: [PATCH] SSDM-946: Set meaningful colors for the channels in the
 installer example HCS dorp box script.

SVN: 32546
---
 .../4/dss/drop-boxes/hcs-dropbox/hcs-dropbox.py      | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/screening/source/core-plugins/screening/4/dss/drop-boxes/hcs-dropbox/hcs-dropbox.py b/screening/source/core-plugins/screening/4/dss/drop-boxes/hcs-dropbox/hcs-dropbox.py
index 5c27af74aaa..af34c12cb46 100644
--- a/screening/source/core-plugins/screening/4/dss/drop-boxes/hcs-dropbox/hcs-dropbox.py
+++ b/screening/source/core-plugins/screening/4/dss/drop-boxes/hcs-dropbox/hcs-dropbox.py
@@ -1,6 +1,7 @@
 import os
-from ch.systemsx.cisd.openbis.dss.etl.dto.api.v1 import SimpleImageDataConfig
-from ch.systemsx.cisd.openbis.dss.etl.dto.api.v1 import ImageMetadata
+from ch.systemsx.cisd.openbis.dss.etl.dto.api import ChannelColor
+from ch.systemsx.cisd.openbis.dss.etl.dto.api import SimpleImageDataConfig
+from ch.systemsx.cisd.openbis.dss.etl.dto.api import ImageMetadata
 from ch.systemsx.cisd.openbis.plugin.screening.shared.api.v1.dto import Geometry
 
 SPACE_CODE = "TEST"
@@ -68,6 +69,13 @@ class MyImageDataSetConfig(SimpleImageDataConfig):
         image_tokens.channelCode = channelCode
         return image_tokens
     
+    def getChannelColor(self, channelCode):
+        dict = { "GFP" : ChannelColor.GREEN, "DAPI" : ChannelColor.BLUE, "CY3" : ChannelColor.RED }
+        if channelCode in dict:
+            return dict[channelCode]
+        else:
+            return None    
+        
     def getTileGeometry(self, imageTokens, maxTileNumber):
         return Geometry.createFromRowColDimensions(maxTileNumber / 3, 3)    
 
-- 
GitLab