#! Run from base dir with

.guix-shell -- guile -L . -s ./scripts/precompute/precompute-hits.scm

!#

(use-modules (dbi dbi)
             (gn db mysql)
             (gn data strains)
             (gn util convert)
             (rnrs base)
             (ice-9 match)
             (srfi srfi-1)
             )

;; potentially you want to test connection with mysql client:
;;
;;    mysql -uwebqtlout -pwebqtlout -A -h 127.0.0.1 -P 3306 db_webqtl -e "show tables;"
;;
;; for now update Locus_old with
;;
;;    update ProbeSetXRef set Locus_old=NULL;

(call-with-db
 (lambda (db)
   (begin
     ;(newline)
     ; (dbi-query db "SELECT StrainId,Strain.Name FROM Strain, StrainXRef WHERE StrainXRef.StrainId = Strain.Id AND StrainXRef.InbredSetId = 1 AND Used_for_mapping='Y' ORDER BY StrainId;")
     ; (let [(row (get-row db))]
     ;  (display row)
     ;  )
     ;(let [(result (get-rows-apply db (lambda (r) `(,(assoc-ref r "StrainId") . ,(assoc-ref r "Name"))) '()))]
     ;  (display (car result)))

     ;(newline)
     (define bxd-strains (bxd-strain-id-names #:map? #t))
     (display (assoc 64728 bxd-strains))
     (newline)
     ;(newline)
     ;(dbi-query db "SELECT * FROM ProbeSetXRef LIMIT 3")
     ;(let [(row (get-row db))]
     ;  (display row)
     ;  )
     ;(db-check2 db)
     ;(newline)
     ;; ---- get first available dataset for precompute:
     (dbi-query db "select Locus, DataId, ProbeSetId from ProbeSetXRef where Locus_old is NULL LIMIT 1")
     (define hit (get-row db))
     (display hit)
     (define data-id (assoc-ref hit "DataId"))
     (display data-id)
     (newline)
     ;; ---- Get strains and phenotypes for this dataset
     (dbi-query db (string-append "SELECT StrainId,value from ProbeSetData WHERE Id=" (int-to-string data-id)))
     (define traits (get-rows-apply db
                              (lambda (r) `(,(assoc-ref r "StrainId") . ,(assoc-ref r "value")))
                              '()))
     ;; ---- Now we need to make sure that all strains belong to BXD
     (define non-bxd (fold
               (lambda (strain lst)
                 (let ([id (car strain)])
                   (if (assoc id bxd-strains)
                       lst
                       (append lst `(,id)))))

               '()
               traits))
     (if (= 0 (length non-bxd))
         (begin
           (display "WE HAVE OUR BXD DATASET for precompute!")
           (display traits)
           (newline)
         ))
)))