;;       
;; 
;; An extension for generating Python code describing the parameters of NEMO models.
;;
;; Copyright 2008-2012 Ivan Raikov and the Okinawa Institute of Science and Technology
;;
;; This program is free software: you can redistribute it and/or
;; modify it under the terms of the GNU General Public License as
;; published by the Free Software Foundation, either version 3 of the
;; License, or (at your option) any later version.
;;
;; This program is distributed in the hope that it will be useful, but
;; WITHOUT ANY WARRANTY; without even the implied warranty of
;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
;; General Public License for more details.
;;
;; A full copy of the GPL license can be found at
;; <http://www.gnu.org/licenses/>.
;;

(module nemo-pyparams

	(nemo:pyparams-translator)

	(import scheme chicken utils data-structures lolevel ports srfi-1 srfi-13 srfi-69)
	
	(require-extension lolevel matchable strictly-pretty 
			   varsubst datatype nemo-core nemo-utils
			   nemo-gate-complex)

(define python-builtin-consts
  `())

(define python-ops
  `(+ - * / > < <= >= = ^))

(define builtin-fns
  `(+ - * / pow neg abs atan asin acos sin cos exp ln
      sqrt tan cosh sinh tanh hypot gamma lgamma log10 log2 log1p ldexp cube
      > < <= >= = and or round ceiling floor max min
      ))

(define (python-name s)
  (let ((cs (string->list (->string s))))
    (let loop ((lst (list)) (cs cs))
      (if (null? cs) (string->symbol (list->string (reverse lst)))
	  (let* ((c (car cs))
		 (c1 (cond ((or (char-alphabetic? c) (char-numeric? c) (char=? c #\_)) c)
			   (else #\_))))
	    (loop (cons c1 lst) (cdr cs)))))))
	    

(define (nmodl-name s)
  (let ((cs (string->list (->string s))))
    (let loop ((lst (list)) (cs cs))
      (if (null? cs) (string->symbol (list->string (reverse lst)))
	  (let* ((c (car cs))
		 (c1 (cond ((or (char-alphabetic? c) (char-numeric? c) (char=? c #\_)) c)
			   (else #\_))))
	    (loop (cons c1 lst) (cdr cs)))))))


(define (rhsexpr/python expr)
  (match expr 
	 (('if . es)  `(if . ,(map (lambda (x) (rhsexpr/python x)) es)))
	 (('pow x y)  (if (and (integer? y)  (positive? y))
			  (if (> y 1)  (let ((tmp (gensym "x")))
					 `(let ((,tmp ,x)) (* . ,(list-tabulate (inexact->exact y) (lambda (i) tmp)))))
			      x)
			    expr))
	 ((s . es)    (if (symbol? s)  (cons (if (member s builtin-fns) s (python-name s)) (map (lambda (x) (rhsexpr/python x)) es)) expr))
	 (id          (if (symbol? id) (python-name id) id))))

(define (python-state-name n s)
  (python-name (s+ n s)))

(define-syntax pp
  (syntax-rules ()
    ((pp indent val ...) (ppf indent (quasiquote val) ...))))


(define tuple/python   (doc:block 2 (doc:text "(") (doc:text ")")))
(define dict/python    (doc:block 2 (doc:text "{") (doc:text "}")))
(define group/python   (doc:block 2 (doc:text "(") (doc:text ")")))
(define block/python   (doc:block 2 (doc:empty) (doc:empty)))

(define (stmt/python x) 
  (match x
	 (($ doc 'DocCons _ ($ doc 'DocText "")) x)
	 (else  (doc:cons x (doc:text "")))))


(define (ifthen/python c e1 e2)
  (doc:nest 
   2 (doc:connect
      (doc:connect (doc:group (doc:connect (doc:text "if") c))
		   (doc:connect (doc:nest 2 e1)
				(doc:nest 2 (doc:connect (doc:text "else") e2))))
      (doc:text "end"))))

(define (letblk/python e1 e2)
  (cond ((equal? e1 (doc:empty)) (doc:group (doc:nest 2 e2)))
	((equal? e2 (doc:empty)) (doc:group (doc:nest 2 e1)))
	(else (doc:connect (doc:group (doc:nest 2 (stmt/python e1)))
			   (doc:group (doc:nest 2 e2))))))
	

(define (format-op/python indent op args)
  (let ((op1 (doc:text (->string op))))
    (if (null? args) op1
	(match args
	       ((x)      (doc:concat (list op1 x)))
	       ((x y)    (doc:concat (intersperse (list x op1 y) (doc:space))))
	       ((x y z)  (doc:concat (intersperse (list x op1 y op1 z) (doc:space))))
	       (lst      (let* ((n   (length lst))
				(n/2 (inexact->exact (round (/ n 2)))))
			   (doc:concat 
			    (intersperse 
			     (list (format-op/python indent op (take lst n/2 )) op1 
				   (format-op/python indent op (drop lst n/2 )))
			     (doc:space)))))))))


(define (format-fncall/python indent op args)
  (let ((op1 (doc:text (->string op))))
    (doc:cons op1 (group/python ((doc:list indent identity (lambda () (doc:text ", "))) args)))))


(define (format-dict/python indent args)
  (dict/python ((doc:list indent 
			  (lambda (x) 
			    (let ((k (car x)) (v (cdr x)))
			      (doc:cons (doc:text k) 
					(doc:cons (doc:text ": ") 
						  (doc:cons (doc:text v) 
							    (doc:empty))))))
			  (lambda () (doc:cons (doc:text ", ") 
					       (doc:cons (doc:break) 
							 (doc:empty)))))
		args)))

(define (format-tuple/python indent args)
  (tuple/python ((doc:list indent 
			  (lambda (v) (doc:text v))
			  (lambda () (doc:cons (doc:text ", ") 
					       (doc:cons (doc:break) 
							 (doc:empty)))))
		args)))


(define (name-normalize expr)
  (match expr 
	 (('if c t e)  `(if ,(name-normalize c) ,(name-normalize t) ,(name-normalize e)))
	 (('let bs e)
	  `(let ,(map (lambda (b) `(,(car b) ,(name-normalize (cadr b)))) bs) ,(name-normalize e)))
	 ((f . es) 
	  (cons f (map name-normalize es)))
	 ((? symbol? ) (python-name expr))
	 ((? atom? ) expr)))


(define (canonicalize-expr/python expr)
  (let ((subst-convert  (subst-driver (lambda (x) (and (symbol? x) x)) nemo:binding? identity nemo:bind nemo:subst-term)))
    (let* ((expr1 (if-convert expr))
	   (expr2 (subst-convert expr1 subst-empty))
	   (expr3 (let-lift expr2))
	   (expr4 (name-normalize expr3)))
      expr4)))


(define (format-expr/python indent expr . rest)  
  (let-optionals rest ((rv #f))
   (let ((indent+ (+ 2 indent)))
    (match expr
       (('let bindings body)
	(letblk/python
	 (fold-right 
	   (lambda (x ax)
	     (letblk/python
	      (match (second x)
		     (('if c t e)
		      (ifthen/python
		       (group/python (format-expr/python indent c))
		       (block/python (format-expr/python indent t (first x)))
		       (block/python (format-expr/python indent e (first x)))))
		     (else 
		      (stmt/python
		       (format-op/python indent+ " = "
					 (list (format-expr/python indent (first x) )
					       (format-expr/python indent (second x)))))))
	      ax))
	   (doc:empty) bindings)
	 (match body
		(('let _ _) (format-expr/python indent body rv))
		(else
		 (let ((body1 (doc:nest indent (format-expr/python indent body))))
		   (if rv (stmt/python (format-op/python indent " = " (list (format-expr/python indent+ rv ) body1)))
		       body1))))))
       
       (('if . rest) (error 'format-expr/python "invalid if statement " expr))

       ((op . rest)  
       (let ((op (case op ((pow)  '^)  ((ln) 'log) (else op))))
	 (let ((fe
		(if (member op python-ops)
		    (let ((mdiv?  (any (lambda (x) (match x (('* . _) #t) (('/ . _) #t) (else #f))) rest))
			  (mul?   (any (lambda (x) (match x (('* . _) #t) (else #f))) rest))
			  (plmin? (any (lambda (x) (match x (('+ . _) #t) (('- . _) #t) (else #f))) rest)))
		      (case op
			((/)  
			 (format-op/python indent op 
					  (map (lambda (x) 
						 (let ((fx (format-expr/python indent+ x)))
						   (if (or (symbol? x) (number? x)) fx
						       (if (or mul? plmin?) (group/python fx) fx)))) rest)))
			((*)  
			 (format-op/python indent op 
					  (map (lambda (x) 
						 (let ((fx (format-expr/python indent+ x)))
						   (if (or (symbol? x) (number? x)) fx
						       (if plmin? (group/python fx) fx)))) rest)))
			
			((^)  
			 (format-op/python indent op 
					  (map (lambda (x) 
						 (let ((fx (format-expr/python indent+ x)))
						   (if (or (symbol? x)  (number? x)) fx
						       (if (or mdiv? plmin?) (group/python fx) fx)))) rest)))
			
			(else
			 (format-op/python indent op 
					  (map (lambda (x) 
						 (let ((fx (format-expr/python indent+ x))) fx)) rest)))))
		    
		    (let ((op (case op ((neg) '-) (else op))))
		      (format-fncall/python indent op (map (lambda (x) (format-expr/python indent+ x)) rest))))))
	   (if rv 
	       (stmt/python (format-op/python indent " = " (list (format-expr/python indent+ rv ) fe)))
	       fe))))
      
      (else  (let ((fe (doc:text (->string expr))))
	       (if rv 
		   (stmt/python (format-op/python indent " = " (list (format-expr/python indent+ rv ) fe)))
		   fe)))))))

(define (doc->string x . rest)
  (let-optionals rest ((width 72))
   (sdoc->string (doc:format width x))))
	  
(define (expr->string/python x . rest)
  (let-optionals rest ((rv #f) (width 72))
     (doc->string
      (format-expr/python 2 x rv)
      width)))


(define (state-init n init)
  (let* ((init  (rhsexpr/python init))
	 (init1 (canonicalize-expr/python init)))
    (list (python-name n) init1)))


(define (asgn-eq n rhs)
  (let* ((fbody   (rhsexpr/python rhs))
	 (fbody1  (canonicalize-expr/python fbody)))
    (list (python-name n) fbody1)))


(define (reaction-eq n open transitions conserve)
  (match-let (((g cnode node-subs)  (transitions-graph n open transitions conserve python-state-name)))
    (let ((nodes ((g 'nodes))))
;      (if (< 2 (length nodes))
;	  (list (python-name n) `(abs (/ ,(python-state-name n open) 
;					 ,(sum (filter-map (lambda (x) (and (not (= (first x) (first cnode)))
;									    (python-state-name n (second x)))) nodes)))))
	  (list (python-name n) (python-state-name n open)))))

(define (reaction-transition-eqs n initial open transitions conserve power)
  (match-let (((g cnode node-subs)  (transitions-graph n open transitions conserve python-state-name)))
     (let* ((out-edges  (g 'out-edges))
	    (in-edges   (g 'in-edges))
	    (nodes      ((g 'nodes))))
       ;; generate differential equations for each state in the transitions system
      (let ((eqs    (fold (lambda (s ax) 
			    (if (and cnode (= (first cnode) (first s) )) ax
				(let* ((out   (out-edges (first s)))
				       (in    (in-edges (first s)))
				       (open? (eq? (second s) open))
				       (name  (python-name (lookup-def (second s) node-subs))))
				  (let* ((rhs1  (cond ((and (not (null? out)) (not (null? in)))
						       `(+ (neg ,(sum (map third out)))
							   ,(sum (map third in))))
						      ((and (not (null? out)) (null? in))
						       `(neg ,(sum (map third out))))
						      ((and (null? out) (not (null? in)))
						       (sum (map third in)))))
					 (fbody0 (rhsexpr/python rhs1))
					 (fbody1 (canonicalize-expr/python fbody0)))
				    (cons (list name fbody1) ax))
				  )))
			  (list) nodes)))
	eqs))))

	

(define (poset->asgn-eq-defs poset sys)
  (fold-right
   (lambda (lst ax)
     (fold  (lambda (x ax) 
	      (match-let (((i . n)  x))
			 (let ((en (hash-table-ref sys n)))
			   (if (nemo:quantity? en)
			       (cases nemo:quantity en
				      (ASGN  (name value rhs) (cons (asgn-eq name rhs) ax))
				      (else  ax))
			       ax))))
	    ax lst))
   (list) poset))

(define (poset->rate-eq-defs poset sys)
  (fold-right
   (lambda (lst ax)
     (fold  (lambda (x ax) 
	      (match-let (((i . n)  x))
			 (let ((en (hash-table-ref sys n)))
			   (if (nemo:quantity? en)
			       (cases nemo:quantity en

				      (REACTION  (name initial open transitions conserve power) 
						 (append (reaction-transition-eqs name initial open transitions 
										  conserve power) ax))
				      
				      (RATE (name initial rhs power)
					    (let ((fbody0  (rhsexpr/python rhs))
						  (dy      (python-name name) ))
					      (cons (list dy (canonicalize-expr/python fbody0)) ax)))

				      (else  ax))
			       ax))))
	    ax lst))
   (list) poset))


(define (poset->reaction-eq-defs poset sys)
  (fold-right
   (lambda (lst ax)
     (fold  (lambda (x ax) 
	      (match-let (((i . n)  x))
			 (let ((en (hash-table-ref sys n)))
			   (if (nemo:quantity? en)
			       (cases nemo:quantity en
				      (REACTION  (name initial open transitions conserve power) 
						 (cons (reaction-eq name open transitions conserve) ax))
				      (else  ax))
			       ax))))
	    ax lst))
   (list) poset))


(define (poset->init-defs poset sys)
  (fold-right
   (lambda (lst ax)
     (fold-right
      (lambda (x ax) 
	      (match-let (((i . n)  x))
			 (let ((en (hash-table-ref sys n)))
			   (if (nemo:quantity? en)
			       (cases nemo:quantity en
				      (REACTION  (name initial open transitions conserve power)
						 (if (nemo:rhs? initial)
						     (cons* (state-init name initial) 
							    (state-init (python-state-name name open) name) ax) 
						     ax))

				      (RATE  (name initial rhs power)
					     (if (and initial (nemo:rhs? initial))
						 (cons (state-init name initial) ax) 
						 ax))

				      (else  ax))
			       ax))))
	    ax lst))
   (list) poset))


(define (poset->state-conserve-eq-defs poset sys)
  (fold-right
   (lambda (lst ax)
     (fold  (lambda (x ax) 
	      (match-let (((i . n)  x))
			 (let ((en (hash-table-ref sys n)))
			   (if (nemo:quantity? en)
			       (cases nemo:quantity en
				      (REACTION (name initial open transitions conserve power)
						(if (and (list? conserve) (every nemo:conseq? conserve))
						    (cons (state-conseqs (python-name name) transitions conserve
									python-state-name) ax) 
						    ax))
				      (else  ax))
			       ax))))
	    ax lst))
   (list) poset))


(define (reaction-power sys n)
  (let ((en (hash-table-ref sys n)))
    (if (nemo:quantity? en)
	(cases nemo:quantity en
	       (REACTION  (name initial open transitions conserve power)  power)
	       (else  #f))  #f)))


(define (bucket-partition p lst)
  (let loop ((lst lst) (ax (list)))
    (if (null? lst) ax
	(let ((x (car lst)))
	  (let bkt-loop ((old-bkts ax) (new-bkts (list)))
	    (if (null? old-bkts) (loop (cdr lst) (cons (list x) new-bkts))
		(if (p x (caar old-bkts ))
		    (loop (cdr lst) (append (cdr old-bkts) (cons (cons x (car old-bkts)) new-bkts)))
		    (bkt-loop (cdr old-bkts) (cons (car old-bkts) new-bkts)))))))))

  
(define (make-define-fn)
  (lambda (indent n proc)
    (let ((lst (procedure-data proc))
	  (indent+ (+ 2 indent)))
      (let ((retval   (python-name (gensym "retval")))
	    (rt       (lookup-def 'rt lst))
	    (formals  (lookup-def 'formals lst))
	    (vars     (lookup-def 'vars lst))
	    (body     (lookup-def 'body lst)))
	(pp indent ,nl (function ,retval = ,(python-name n) (,(slp ", " vars)) ))
	(let* ((body0 (rhsexpr/python body))
	       (body1 (canonicalize-expr/python body0))
	       (lbs   (enum-bnds body1 (list))))
	  (pp indent+ ,(expr->string/python body1 retval))
	  (pp indent end))
	  ))))




(define (output-pyparams sysname i-params i-eqs
			 const-defs asgn-eq-defs init-eq-defs 
			 pool-ions perm-ions indent indent+)

    
    (let* ((init-eqs 
	    (append 
	     
	     const-defs
	     asgn-eq-defs
	     init-eq-defs
	     
	     (map (lambda (pool-ion)
		    (let ((n (third pool-ion))
			  (b (first pool-ion)))
		      (list n b)))
		  pool-ions)))

	   (init-dag 
	    (map (lambda (def)
		   (cons (first def) (enum-freevars (second def) '() '())))
		 init-eqs))

	   (init-order
	    (reverse
	     (topological-sort init-dag 
	       (lambda (x y) (string=? (->string x) (->string y)))))))

      (pp indent ,nl (Parameters = collections.namedtuple ("'Parameters'" "," []) ))

      (for-each (lambda (x)
		  (pp indent ,(expr->string/python (cadr x) (python-name (car x)))))
		const-defs)

      (pp indent ,nl)

      (let recur ((i-params i-params) 
		  (property-tuples '() ))

	(if (null? i-params)
	    
	    (for-each (lambda (t)
			(pp indent (,(car t) = ,(doc->string (format-tuple/python indent+ (cdr t))))))
		      property-tuples)

	    (let ((paramset (car i-params)))
	      (let ((alst (cdr paramset)))
		(let ((label (lookup-def 'label alst))
		      (maximal-permeability (lookup-def 'maximal-permeability alst))
		      (maximal-conductance  (lookup-def 'maximal-conductance alst))
		      (reversal-potential   (lookup-def 'reversal-potential alst)))
		  (recur (cdr i-params)
			 (let* ((property-tuples1 
				 (fold (lambda (l x ax)
					 (or (and x (cons `(,l ,(symbol->string (python-name x)) 
							       ,(s+ "'" (nmodl-name x) "'")) ax))
					     ax))
				       property-tuples
				       (list (s+ label "_MAXIMAL_PERMEABILITY")
					     (s+ label "_MAXIMAL_CONDUCTANCE")
					     (s+ label "_REVERSAL_POTENTIAL"))
				       (list maximal-permeability
					     maximal-conductance 
					     reversal-potential)
				       )))
			   property-tuples1))
		  )))
	    ))

    ))

(define (pyparams-translator1 sys . rest)
  (define (cid x)  (second x))
  (define (cn x)   (first x))
  (let-optionals rest ((mode 'multiple) (filename #f))
  (match-let ((($ nemo:quantity 'DISPATCH  dis) (hash-table-ref sys (nemo-intern 'dispatch))))
    (let ((imports  ((dis 'imports)  sys))
	  (exports  ((dis 'exports)  sys)))
      (let* ((indent      0)
	     (indent+     (+ 2 indent ))
	     (eval-const  (dis 'eval-const))
	     (sysname     (python-name ((dis 'sysname) sys)))
	     (prefix      sysname)
	     (filename    (or filename (s+ sysname ".py")))
	     (deps*       ((dis 'depgraph*) sys))
	     (consts      ((dis 'consts)  sys))
	     (asgns       ((dis 'asgns)   sys))
	     (states      ((dis 'states)  sys))
	     (reactions   ((dis 'reactions) sys))
	     (defuns      ((dis 'defuns)  sys))
	     (components  ((dis 'components) sys))
	     
	     (g             (match-let (((state-list asgn-list g) ((dis 'depgraph*) sys))) g))
	     (poset         (vector->list ((dis 'depgraph->bfs-dist-poset) g)))

	     (const-defs       (filter-map
				(lambda (nv)
				  (and (not (member (first nv) python-builtin-consts))
				       (let ((v1 (canonicalize-expr/python (second nv))))
					 (list (python-name (first nv)) v1))))
				consts))
	     
	     (gate-complex-info    (nemo:gate-complex-query sys))

	     (gate-complexes       (lookup-def 'gate-complexes gate-complex-info))
	     (perm-ions     (map (match-lambda ((comp i e erev) `(,comp ,(python-name i) ,(python-name e) ,erev)))
				 (lookup-def 'perm-ions gate-complex-info)))
	     (acc-ions      (map (match-lambda ((comp i in out) `(,comp ,@(map python-name (list i in out)))))
				 (lookup-def 'acc-ions gate-complex-info)))
	     (epools        (lookup-def 'pool-ions gate-complex-info))
	     (pool-ions     (map (lambda (lst) (map python-name lst)) epools))

	     (i-gates       (lookup-def 'i-gates gate-complex-info))

	     (capcomp       (any (match-lambda ((name 'membrane-capacitance id) (list name id)) (else #f)) components))
	     (mcap          (and capcomp (car ((dis 'component-exports) sys (cid capcomp)))))
		
	     (i-eqs+params
	      (filter-map
		     (lambda (gate-complex) 
		       
		       (let* ((label             (first gate-complex))
			      (n                 (second gate-complex))
			      (subcomps          ((dis 'component-subcomps) sys n))
			      (acc               (lookup-def 'accumulating-substance subcomps))
			      (perm              (lookup-def 'permeating-ion subcomps))
			      (permqs            (and perm ((dis 'component-exports) sys (cid perm))))
			      (pore              (lookup-def 'pore subcomps))
			      (permeability      (lookup-def 'permeability subcomps))
			      (gate              (lookup-def 'gate subcomps))
			      (sts               (and gate ((dis 'component-exports) sys (cid gate)))))
			 
			 (if (not (or pore permeability))
			     (nemo:error 'nemo:python-translator ": ion channel definition " label
					 "lacks any pore or permeability components"))


			 (cond ((and perm permeability gate)
				     (let* ((i     (python-name (s+ 'i (cn perm))))
					    (pmax  (cadr ((dis 'component-exports) sys (cid permeability))))
					    (pwrs  (map (lambda (n) (reaction-power sys n)) sts))
					    (sptms (map (lambda (st pwr) `(pow ,st ,pwr)) sts pwrs))
					    (gion  `(* ,pmax ,@sptms)))
					 (list i #f gion (python-name (s+ 'i_ label) )
					       `((label . ,label) 
						 (maximal-permeability . ,pmax) ))))

			       ((and perm pore gate)
				(case (cn perm)
				  ((non-specific)
				   (let* ((i     (python-name 'i))
					  (e     (car permqs))
					  (gmax  (car ((dis 'component-exports) sys (cid pore))))
					  (pwrs  (map (lambda (n) (reaction-power sys n)) sts))
					  (sptms (map (lambda (st pwr) `(pow ,st ,pwr)) sts pwrs))
					  (gion  `(* ,gmax ,@sptms)))
				     (list i e gion  (python-name (s+ 'i_ label))
					   `((label . ,label)
					     (maximal-conductance . ,gmax)
					     (reversal-potential . ,e))
					   )))

				  (else
				   (let* ((i     (python-name (s+ 'i (cn perm))))
					  (e     (car permqs))
					  (gmax  (car ((dis 'component-exports) sys (cid pore))))
					  (pwrs  (map (lambda (n) (reaction-power sys n)) sts))
					  (sptms (map (lambda (st pwr) `(pow ,st ,pwr)) sts pwrs))
					  (gion  `(* ,gmax ,@sptms)))
				     (list i e gion (python-name (s+ 'i_ label))
					   `((label . ,label)
					     (maximal-conductance . ,gmax)
					     (reversal-potential . ,e))
					   )))))
			       
			       ((and perm pore)
				(case (cn perm)
				  ((non-specific)
				   (let* ((i     (python-name 'i))
					  (e     (car permqs))
					  (gmax  (car ((dis 'component-exports) sys (cid pore)))))
				     (list i e gmax (python-name (s+ 'i_ label))
					   `((label . ,label)
					     (maximal-conductance . ,gmax)
					     (reversal-potential . ,e))
					   )))
				  (else
				   (nemo:error 'nemo:python-translator ": invalid ion channel definition " label))))
			       
			       ((and acc pore gate)
				(let* ((i     (python-name (s+ 'i (cn acc))))
				       (gmax  (car ((dis 'component-exports) sys (cid pore))))
				       (pwrs  (map (lambda (n) (reaction-power sys n)) sts))
				       (sptms (map (lambda (st pwr) `(pow ,st ,pwr)) sts pwrs))
				       (gion  `(* ,gmax ,@sptms)))
				  (list i #f gion  (python-name (s+ 'i_ label))
					   `((label . ,label)
					     (maximal-conductance . ,gmax)))
					))
			       (else (nemo:error 'nemo:python-translator ": invalid ion channel definition " label))
			       )))
		     gate-complexes))

	     (i-params (map (lambda (i-eq) (cons (car i-eq) (cadr (cdddr i-eq)))) i-eqs+params))
	     (i-eqs    (map (lambda (i-eq) (take i-eq 4)) i-eqs+params))

	     (i-names (delete-duplicates (map first i-eqs)))
		
	     (i-eqs  (fold  (lambda (i-gate ax) 
			      (let ((i-gate-var (first i-gate)))
				(cons (list (python-name 'i) #f i-gate-var (s+ 'i_ (second i-gate))) ax)))
			    i-eqs i-gates))

	     (i-bkts (bucket-partition (lambda (x y) (eq? (car x) (car y))) i-eqs))
	     
	     (i-eqs  (fold (lambda (b ax) 
			     (match b 
				    ((and ps ((i e gion ii) . rst))  
				     (let loop ((ps ps) (summands (list)) (eqs (list)))
				       (if (null? ps)
					   
					   (let* ((sum0  (sum summands))
						  (sum1  (rhsexpr/python sum0))
						  (sum2  (canonicalize-expr/python sum1)))
					     (append eqs (list (list i sum2)) ax))
					   
					   (match-let (((i e gion ii) (car ps)))
						      (loop (cdr ps) 
							    (cons ii summands) 
							    (let* ((expr0 (rhsexpr/python (if e `(* ,gion (- v ,e)) gion)))
								   (expr1 (canonicalize-expr/python expr0)))
							      (cons (list ii expr1) eqs)))))))
				    
				    ((i e gion ii)
				     (let* ((expr0  (rhsexpr/python (if e `(* ,gion (- v ,e)) gion)))
					    (expr1  (canonicalize-expr/python expr0)))
				       (cons (list i expr1) ax)))
				    
				    (else ax)))
			   (list) i-bkts))

	     (asgn-eq-defs     (poset->asgn-eq-defs poset sys))
	     
	     (rate-eq-defs       (reverse (poset->rate-eq-defs poset sys)))
	     
	     (reaction-eq-defs   (poset->reaction-eq-defs poset sys))

	     (init-eq-defs       (poset->init-defs poset sys))
	     
	     (conserve-eq-defs   (map (lambda (eq) (list 0 `(- ,(second eq) ,(first eq)))) 
				      (poset->state-conserve-eq-defs poset sys)))
	     
	     (v-eq    (if mcap 
			  (list 'v (rhsexpr/python `(/ (neg ,(sum i-names)) ,mcap)))
			  (list 'v 0.0)))
	     
	     (dfenv 
	      (map (lambda (x) (let ((n (first x)))
				 (list n (python-name (s+ "d_" n )))))
		   defuns))

	     )
	
	(for-each 
	 (lambda (a)
	   (let ((acc-ion   (car a)))
	     (if (assoc acc-ion perm-ions)
		 (nemo:error 'nemo:python-translator 
			     ": ion species " acc-ion " cannot be declared as both accumulating and permeating"))))
	 acc-ions)

	(for-each 
	 (lambda (p)
	   (let ((pool-ion  (car p)))
	     (if (assoc pool-ion perm-ions)
		 (nemo:error 'nemo:python-translator 
			     ": ion species " pool-ion " cannot be declared as both pool and permeating"))))
	 pool-ions)
	
	(let ((output 
	       (case mode
		 ((single)  (open-output-file filename #:append))
		 (else #f))))
	  
	     (let ((output1 (or output (open-output-file (s+ prefix "_params.py")))))

	       (with-output-to-port output1
		 (lambda ()
		   (output-pyparams sysname i-params i-eqs const-defs asgn-eq-defs init-eq-defs pool-ions perm-ions indent indent+)
		   (pp indent ,nl)))

	       (if (not output) (close-output-port output1)))

	   (if output (close-output-port output)))

	   )))))


(define (nemo:pyparams-translator syss . rest)
  (let-optionals rest ((filename #f))
    (close-output-port (open-output-file filename))
    (for-each 
     (lambda (sys)
       (apply pyparams-translator1 (cons sys (cons 'single rest))))
     syss)))


)