source: trunk/grails-app/controllers/dbnp/studycapturing/AssayController.groovy @ 2052

Last change on this file since 2052 was 2052, checked in by s.h.sikkema@…, 11 years ago

fixed breaking commit of yesterday

  • Property svn:keywords set to Rev Author Date
File size: 18.5 KB
Line 
1package dbnp.studycapturing
2
3import grails.plugins.springsecurity.Secured
4
5@Secured(['IS_AUTHENTICATED_REMEMBERED'])
6class AssayController {
7
8        def assayService
9        def authenticationService
10        def fileService
11
12        static allowedMethods = [save: "POST", update: "POST", delete: "POST"]
13
14        def index = {
15                redirect(action: "list", params: params)
16        }
17
18        def list = {
19                params.max = Math.min(params.max ? params.int('max') : 10, 100)
20                [assayInstanceList: Assay.list(params), assayInstanceTotal: Assay.count()]
21        }
22
23        def create = {
24                def assayInstance = new Assay()
25                assayInstance.properties = params
26                return [assayInstance: assayInstance]
27        }
28
29        def save = {
30                def assayInstance = new Assay(params)
31
32                // The following lines deviate from the generate-all generated code.
33                // See http://jira.codehaus.org/browse/GRAILS-3783 for why we have this shameful workaround...
34                def study = assayInstance.parent
35                study.addToAssays(assayInstance)
36
37                if (assayInstance.save(flush: true)) {
38                        flash.message = "${message(code: 'default.created.message', args: [message(code: 'assay.label', default: 'Assay'), assayInstance.id])}"
39                        redirect(action: "show", id: assayInstance.id)
40                }
41                else {
42                        render(view: "create", model: [assayInstance: assayInstance])
43                }
44        }
45
46        def show = {
47                def assayInstance = Assay.get(params.id)
48                if (!assayInstance) {
49                        flash.message = "${message(code: 'default.not.found.message', args: [message(code: 'assay.label', default: 'Assay'), params.id])}"
50                        redirect(action: "list")
51                }
52                else {
53                        [assayInstance: assayInstance]
54                }
55        }
56
57        def showByToken = {
58                def assayInstance = Assay.findByAssayUUID(params.id)
59                if (!assayInstance) {
60                        flash.message = "${message(code: 'default.not.found.message', args: [message(code: 'assay.label', default: 'Assay'), params.id])}"
61                        redirect(action: "list")
62                }
63                else {
64                        redirect(action: "show", id: assayInstance.id)
65                }
66        }
67
68        def edit = {
69                def assayInstance = Assay.get(params.id)
70                if (!assayInstance) {
71                        flash.message = "${message(code: 'default.not.found.message', args: [message(code: 'assay.label', default: 'Assay'), params.id])}"
72                        redirect(action: "list")
73                }
74                else {
75                        return [assayInstance: assayInstance]
76                }
77        }
78
79        def update = {
80                def assayInstance = Assay.get(params.id)
81                if (assayInstance) {
82                        if (params.version) {
83                                def version = params.version.toLong()
84                                if (assayInstance.version > version) {
85
86                                        assayInstance.errors.rejectValue("version", "default.optimistic.locking.failure", [message(code: 'assay.label', default: 'Assay')] as Object[], "Another user has updated this Assay while you were editing")
87                                        render(view: "edit", model: [assayInstance: assayInstance])
88                                        return
89                                }
90                        }
91                        assayInstance.properties = params
92                        if (!assayInstance.hasErrors() && assayInstance.save(flush: true)) {
93                                flash.message = "${message(code: 'default.updated.message', args: [message(code: 'assay.label', default: 'Assay'), assayInstance.id])}"
94                                redirect(action: "show", id: assayInstance.id)
95                        }
96                        else {
97                                render(view: "edit", model: [assayInstance: assayInstance])
98                        }
99                }
100                else {
101                        flash.message = "${message(code: 'default.not.found.message', args: [message(code: 'assay.label', default: 'Assay'), params.id])}"
102                        redirect(action: "list")
103                }
104        }
105
106        def delete = {
107                def assayInstance = Assay.get(params.id)
108                if (assayInstance) {
109                        try {
110                                assayInstance.delete(flush: true)
111                                flash.message = "${message(code: 'default.deleted.message', args: [message(code: 'assay.label', default: 'Assay'), params.id])}"
112                                redirect(action: "list")
113                        }
114                        catch (org.springframework.dao.DataIntegrityViolationException e) {
115                                flash.message = "${message(code: 'default.not.deleted.message', args: [message(code: 'assay.label', default: 'Assay'), params.id])}"
116                                redirect(action: "show", id: params.id)
117                        }
118                }
119                else {
120                        flash.message = "${message(code: 'default.not.found.message', args: [message(code: 'assay.label', default: 'Assay'), params.id])}"
121                        redirect(action: "list")
122                }
123        }
124
125        def assayExportFlow = {
126                entry {
127                        action{
128                                def user            = authenticationService.getLoggedInUser()
129                                flow.userStudies    = Study.giveReadableStudies(user)
130                        }
131                        on("success").to "selectAssay"
132                }
133
134                selectAssay {
135                        on ("submit"){
136                                flow.assay = Assay.get(params.assayId)
137
138                                // check if assay exists
139                                if (!flow.assay) throw new Exception("No assay found with id: ${params.assayId}")
140
141                                // obtain fields for each category
142                                flow.fieldMap = assayService.collectAssayTemplateFields(flow.assay, null)
143
144                                flash.errorMessage = flow.fieldMap.remove('Module Error')
145                                flow.measurementTokens = flow.fieldMap.remove('Module Measurement Data')
146                        }.to "selectFields"
147
148                        on(Exception).to "handleError"
149                }
150
151                selectFields {
152                        on ("submit"){
153
154                                def fieldMapSelection = [:]
155
156                                flow.fieldMap.eachWithIndex { cat, cat_i ->
157
158                                        if (params."cat_$cat_i" == 'on') {
159                                                fieldMapSelection[cat.key] = []
160
161                                                cat.value.eachWithIndex { field, field_i ->
162
163                                                        if (params."cat_${cat_i}_${field_i}" == 'on')
164                                                                fieldMapSelection[cat.key] += field
165                                                }
166
167                                                if (fieldMapSelection[cat.key] == [])
168                                                        fieldMapSelection.remove(cat.key)
169                                        }
170                                }
171
172                                def measurementTokens = []
173
174                                if (params."cat_4" == 'on') {
175                                        measurementTokens = params.list( "measurementToken" )
176                                }
177
178                                // collect the assay data according to user selecting
179                                def assayData           = assayService.collectAssayData(flow.assay, fieldMapSelection, measurementTokens)
180
181                                flash.errorMessage      = assayData.remove('Module Error')
182
183                                flow.rowData            = assayService.convertColumnToRowStructure(assayData)
184
185                                // prepare the assay data preview
186                                def previewRows         = Math.min(flow.rowData.size()    as int, 5) - 1
187                                def previewCols         = Math.min(flow.rowData[0].size() as int, 5) - 1
188
189                                flow.assayDataPreview   = flow.rowData[0..previewRows].collect{ it[0..previewCols] as ArrayList }
190
191                                // store the selected file type in the flow
192                                flow.exportFileType = params.exportFileType
193
194                        }.to "compileExportData"
195
196                        on(Exception).to "handleError"
197                }
198
199                compileExportData {
200                        on ("ok"){
201                                session.rowData = flow.rowData
202                                session.exportFileType = flow.exportFileType
203                        }.to "export"
204                        on ("cancel").to "selectAssay"
205                }
206
207                export {
208                        redirect(action: 'doExport')
209                }
210
211                handleError() {
212                        render(view: 'errorPage')
213                }
214        }
215       
216        /**
217         * This method is called when a user wants to fetch data from GSCF
218         */
219        def galaxyExport = {
220                def galaxy_url = params.get( "GALAXY_URL" );
221                def tool_id = params.get( "tool_id" );
222               
223                // Without galaxy url, we can't do anything
224                if( !galaxy_url || !tool_id ) {
225                        redirect( action: "assayExport" );
226                }
227               
228                def user            = authenticationService.getLoggedInUser()
229                def userStudies    = Study.giveReadableStudies(user)
230               
231                [ galaxy_url: galaxy_url, tool_id: tool_id, userStudies: userStudies ]
232        }
233       
234        def sendToGalaxy = {
235                // Generate the url to redirect the user to
236                def galaxy_url = params.get( "GALAXY_URL" );
237                def tool_id = params.get( "tool_id" );
238               
239                // Without galaxy url, we can't do anything
240                if( !galaxy_url || !tool_id  ) {
241                        redirect( action: "assayExport" );
242                }
243               
244                def assayId = params.long( "assayId" );
245                def assay = assayId ? Assay.get( assayId ) : null;
246               
247                // If no assay is given, return to galaxyExport screen
248                if( !assay ) {
249                        redirect( action: "galaxyExport", params: [ "GALAXY_URL": galaxy_url ] );
250                }
251               
252                // create a random session token that will be used to allow to module to
253                // sync with gscf prior to presenting the measurement data
254                def sessionToken = UUID.randomUUID().toString()
255                def consumer = "galaxy";
256
257                // put the session token to work
258                authenticationService.logInRemotely( consumer, sessionToken, authenticationService.getLoggedInUser() )
259               
260                // Create a url where the data can be fetched
261                def fetchUrl = g.createLink( absolute: true, controller: "assay", action: "fetchGalaxyData", params: [ assayToken: assay.assayUUID, sessionToken: sessionToken ] );
262               
263                // Generate url to redirect the user to
264                def url = galaxy_url + "?tool_id=" + URLEncoder.encode( tool_id.toString(), "UTF-8" ) + "&URL=" + URLEncoder.encode( fetchUrl.toString(), "UTF-8" ) 
265               
266                log.trace "Redirecting galaxy user back to " + url;
267               
268                redirect( url: url );
269        }
270       
271        // This method is accessible for each user. However, he should return with a valid
272        // session token
273        @Secured(['true'])
274        def fetchGalaxyData = {
275                // Check accessibility
276                def consumer = "galaxy";
277                def remoteUser = authenticationService.getRemotelyLoggedInUser( consumer, params.sessionToken );
278                if( !remoteUser ) {
279                        response.status = 401;
280                        render "You must be logged in";
281                        return
282                }
283
284                // retrieve assay
285                def assay = Assay.findByAssayUUID( params.assayToken );
286               
287                if( !assay ) {
288                        response.status = 404;
289                        render "No assay found";
290                        return
291                }
292               
293                // Return assay data
294                def fieldMap = assayService.collectAssayTemplateFields( assay, null, remoteUser )
295
296                def measurementTokens = fieldMap.remove('Module Measurement Data');
297                def assayData = assayService.collectAssayData(assay, fieldMap, measurementTokens, null, remoteUser)
298
299                def rowData   = assayService.convertColumnToRowStructure(assayData)
300
301                // Invalidate session token
302                authenticationService.logOffRemotely( consumer, params.sessionToken );
303                               
304                def outputDelimiter = '\t'
305                def outputFileExtension = 'txt'
306
307                def filename = "export.$outputFileExtension"
308                response.setHeader("Content-disposition", "attachment;filename=\"${filename}\"")
309                response.setContentType("application/octet-stream")
310                try {
311                        // Skip first row for now to support the current PLS tool in Galaxy, will change in the future
312                        assayService.exportRowWiseDataToCSVFile(rowData[1..-1], response.outputStream, outputDelimiter, java.util.Locale.US)
313                } catch (Exception e) {
314                        flash.errorMessage = e.message
315                        redirect action: 'errorPage'
316                }
317        }
318
319        /**
320         * Export the row data in session.rowData to the outputStream of the http
321         * response.
322         */
323        def doExport = {
324
325                // make sure we're coming from the export flow, otherwise redirect there
326                if (!(session.rowData && session.exportFileType))
327                        redirect(action: 'assayExportFlow')
328
329                // process requested output file type
330                def outputDelimiter, outputFileExtension, locale = java.util.Locale.US
331
332                switch(session.exportFileType) {
333                        case '2': // Comma delimited csv
334                                outputDelimiter = ','
335                                outputFileExtension = 'csv'
336                                break
337                        case '3': // Semicolon delimited csv
338                                outputDelimiter = ';'
339                                outputFileExtension = 'csv'
340                                locale = java.util.Locale.GERMAN // force use of comma as decimal separator
341                                break
342                        default: // Tab delimited with .txt extension
343                                outputDelimiter = '\t'
344                                outputFileExtension = 'txt'
345                }
346
347                def filename = "export.$outputFileExtension"
348                response.setHeader("Content-disposition", "attachment;filename=\"${filename}\"")
349                response.setContentType("application/octet-stream")
350                try {
351
352                        assayService.exportRowWiseDataToCSVFile(session.rowData, response.outputStream, outputDelimiter, locale)
353
354                        // clear the data from the session
355                        session.removeAttribute('rowData')
356                        session.removeAttribute('exportFileType')
357
358                } catch (Exception e) {
359
360                        flash.errorMessage = e.message
361                        redirect action: 'errorPage'
362
363                }
364        }
365
366        /**
367         * Method to export one or more assays to excel in separate sheets.
368         *
369         * @param       params.ids              One or more assay IDs to export
370         * @param       params.format   "list" in order to export all assays in one big excel sheet
371         *                                                      "sheets" in order to export every assay on its own sheet (default)
372         */
373        def exportToExcel = {
374                def format = params.get( 'format', 'sheets' );
375                if( format == 'list' ) {
376                        exportToExcelAsList( params );
377                } else {
378                        exportToExcelAsSheets( params );
379        }
380
381        /**
382         * Method to export one or more assays to excel in separate sheets.
383         *
384         * @param       params.ids              One or more assay IDs to export
385         */
386        def exportToExcelAsSheets = {
387                def assays = getAssaysFromParams( params );
388
389                if( !assays )
390                        return;
391
392                // Send headers to the browser so the user can download the file
393                def filename = 'export.xlsx'
394                response.setHeader("Content-disposition", "attachment;filename=\"${filename}\"")
395                response.setContentType("application/octet-stream")
396
397                try {
398                        // Loop through all assays to collect the data
399                        def rowWiseAssayData = [];
400
401                        assays.each { assay ->
402                                // Determine which fields should be exported for this assay
403                                def fieldMap = assayService.collectAssayTemplateFields(assay, null)
404                                def measurementTokens = fieldMap.remove('Module Measurement Data')
405
406                                // Retrieve row based data for this assay
407                                def assayData = assayService.collectAssayData( assay, fieldMap, measurementTokens );
408                                def rowData   = assayService.convertColumnToRowStructure(assayData)
409
410                                // Put each assay on another sheet
411                                rowWiseAssayData << rowData;
412                        }
413
414                        assayService.exportRowWiseDataForMultipleAssaysToExcelFile( rowWiseAssayData, response.getOutputStream() )
415
416                        response.outputStream.flush()
417
418                } catch (Exception e) {
419                        throw e;
420                }
421        }
422
423        /**
424         * Method to export one or more assays to excel.
425         *
426         * @param       params.ids              One or more assay IDs to export
427         */
428        def exportToExcelAsList = {
429                def assays = getAssaysFromParams( params );
430
431                if( !assays )
432                        return;
433
434                // Send headers to the browser so the user can download the file
435                def filename = 'export.csv'
436                response.setHeader("Content-disposition", "attachment;filename=\"${filename}\"")
437                response.setContentType("application/octet-stream")
438
439                try {
440                        // Loop through all assays to collect the data
441                        def columnWiseAssayData = [];
442
443                        assays.each { assay ->
444                                // Determine which fields should be exported for this assay
445                                def fieldMap = assayService.collectAssayTemplateFields(assay, null)
446                                def measurementTokens = fieldMap.remove('Module Measurement Data')
447
448                                // Retrieve row based data for this assay
449                                def assayData = assayService.collectAssayData( assay, fieldMap, measurementTokens );
450
451                                // Prepend study and assay data to the list
452                                assayData = assayService.prependAssayData( assayData, assay, assay.samples?.size() )
453                                assayData = assayService.prependStudyData( assayData, assay, assay.samples?.size() )
454
455                                // Put each assay on another sheet
456                                columnWiseAssayData << assayData;
457                        }
458
459                        // Merge data from all assays
460                        def mergedColumnWiseData = assayService.mergeColumnWiseDataOfMultipleStudies( columnWiseAssayData );
461
462                        def rowData   = assayService.convertColumnToRowStructure(mergedColumnWiseData)
463                        assayService.exportRowWiseDataToCSVFile( rowData, response.getOutputStream() )
464
465                        response.outputStream.flush()
466
467                } catch (Exception e) {
468                        throw e;
469                }
470        }
471
472        /**
473         * Method to export one or more samples to csv in separate sheets.
474         *
475         * @param       params.ids              One or more sample ids to export
476         */
477        def exportSamplesToCsv = {
478                def samples = getSamplesFromParams( params );
479
480                if( !samples ) {
481                        return;
482                }
483
484                // Determine a list of assays these samples have been involved in. That way, we can
485                // retrieve the data for that assay once, and save precious time doing HTTP calls
486                def assays = [:];
487               
488                samples.each { sample ->
489                        def thisAssays = sample.getAssays();
490                       
491                        // Loop through all assays. If it already exists, add the sample it to the list
492                        thisAssays.each { assay ->
493                                if( !assays[ assay.id ] ) {
494                                        assays[ assay.id ] = [ 'assay': assay, 'samples': [] ]
495                                }
496                               
497                                assays[ assay.id ].samples << sample
498                        }
499                }
500               
501                // Now collect data for all assays
502                try {
503                        // Loop through all assays to collect the data
504                        def columnWiseAssayData = [];
505
506                        assays.each { assayInfo ->
507                                def assay = assayInfo.value.assay;
508                                def assaySamples = assayInfo.value.samples;
509                               
510                                // Determine which fields should be exported for this assay
511                                def fieldMap = assayService.collectAssayTemplateFields(assay, null)
512                                def measurementTokens = fieldMap.remove('Module Measurement Data')
513                               
514                                // Retrieve row based data for this assay
515                                def assayData = assayService.collectAssayData( assay, fieldMap, measurementTokens, assaySamples );
516                               
517                                // Prepend study and assay data to the list
518                                assayData = assayService.prependAssayData( assayData, assay, assaySamples.size() )
519                                assayData = assayService.prependStudyData( assayData, assay, assaySamples.size() )
520                               
521                                // Make sure the assay data can be distinguished later
522                                assayData.put( "Assay data - " + assay.name, assayData.remove( "Assay Data") )
523                                assayData.put( "Module measurement data - " + assay.name, assayData.remove( "Module Measurement Data") )
524                               
525                                // Add the sample IDs to the list, in order to be able to combine
526                                // data for a sample that has been processed in multiple assays
527                                assayData[ "Sample Data" ][ "id" ] = assaySamples*.id;
528                               
529                                println "Assay data"
530                                assayData.each { println it }
531
532                                columnWiseAssayData << assayData;
533                        }
534                       
535                        def mergedColumnWiseData = assayService.mergeColumnWiseDataOfMultipleStudiesForASetOfSamples( columnWiseAssayData );
536
537                        def rowData   = assayService.convertColumnToRowStructure(mergedColumnWiseData)
538                       
539                        // Send headers to the browser so the user can download the file
540                        def filename = 'export.csv'
541                        response.setHeader("Content-disposition", "attachment;filename=\"${filename}\"")
542                        response.setContentType("application/octet-stream")
543       
544                        assayService.exportRowWiseDataToCSVFile( rowData, response.getOutputStream() )
545
546                        response.outputStream.flush()
547
548                } catch (Exception e) {
549                        throw e;
550                }
551        }
552
553
554        def getAssaysFromParams( params ) {
555                def ids = params.list( 'ids' ).findAll { it.isLong() }.collect { Long.valueOf( it ) };
556                def tokens = params.list( 'tokens' );
557
558                if( !ids && !tokens ) {
559                        flash.errorMessage = "No assay ids given";
560                        redirect( action: "errorPage" );
561                        return [];
562                }
563
564                // Find all assays for the given ids
565                def assays = [];
566                ids.each { id ->
567                        def assay = Assay.get( id );
568                        if( assay )
569                                assays << assay;
570                }
571
572                // Also accept tokens for defining studies
573                tokens.each { token ->
574                        def assay = Assay.findByAssayUUID( token );
575                        if( assay )
576                                assays << assay;
577                }
578
579                if( !assays ) {
580                        flash.errorMessage = "No assays found";
581                        redirect( action: "errorPage" );
582                        return [];
583                }
584
585                return assays.unique();
586        }
587
588        def getSamplesFromParams( params ) {
589                def ids = params.list( 'ids' ).findAll { it.isLong() }.collect { Long.valueOf( it ) };
590                def tokens = params.list( 'tokens' );
591
592                if( !ids && !tokens ) {
593                        flash.errorMessage = "No sample ids given";
594                        redirect( action: "errorPage" );
595                        return [];
596                }
597
598                // Find all assays for the given ids
599                def samples = [];
600                ids.each { id ->
601                        def sample = Sample.get( id );
602                        if( sample )
603                                samples << sample;
604                }
605
606                // Also accept tokens for defining studies
607                tokens.each { token ->
608                        def sample = Sample.findBySampleUUID( token );
609                        if( sample )
610                                samples << sample;
611                }
612
613                if( !samples ) {
614                        flash.errorMessage = "No assays found";
615                        redirect( action: "errorPage" );
616                        return [];
617                }
618
619                return samples.unique();
620        }
621       
622        def errorPage = {
623                render(view: 'assayExport/errorPage')
624        }
625}
Note: See TracBrowser for help on using the repository browser.