cnl.py

Go to the documentation of this file.

from biogeme import *
from mev import *


def cnl_avail(V,availability,nests,choice) :
    Gi = {}
    Gidict = {}
    for k in V:
        Gidict[k] = []
    for m in nests:
        biosumlist = []
        for i,a in m[1].items():
            biosumlist.append(Elem({0:0,1:a**(m[0]) * exp(m[0] * (V[i]))},availability[i] != 0))
        biosum = bioMultSum(biosumlist)        
        for i,a in m[1].items():
            Gidict[i].append(Elem({0:0,1:(biosum**((1.0/m[0])-1.0)) * (a**m[0]) * exp((m[0]-1.0)*(V[i]))},availability[i] != 0))
    for k in V:
        Gi[k] = bioMultSum(Gidict[k])
    P = mev(V,Gi,availability,choice) 
    return P



def logcnl_avail(V,availability,nests,choice) :
    Gi = {}
    Gidict = {}
    for k in V:
        Gidict[k] = []
    for m in nests:
        biosumlist = []
        for i,a in m[1].items():
            biosumlist.append(Elem({0:0,1:a**(m[0]) * exp(m[0] * (V[i]))},availability[i] != 0))
        biosum = bioMultSum(biosumlist)
        for i,a in m[1].items():
            Gidict[i].append(Elem({0:0,1:(biosum**((1.0/m[0])-1.0)) * (a**m[0]) * exp((m[0]-1.0)*(V[i]))},availability[i] != 0))
    for k in V:
        Gi[k] = bioMultSum(Gidict[k])
    logP = logmev(V,Gi,availability,choice) 
    return logP



def cnlmu(V,availability,nests,choice,bmu) :
    Gi = {}
    Gidict = {}
    for k in V:
        Gidict[k] = []
    for m in nests:
        biosumdict = []
        for i,a in m[1].items():
            biosumdict.append(Elem({0:0,1:a**(m[0]/bmu) * exp(m[0] * (V[i]))},availability[i] != 0))
        biosum = bioMultSum(biosumdict)
        for i,a in m[1].items():
            Gidict[i].append(Elem({0:0,1:bmu * (biosum**((bmu/m[0])-1.0)) * (a**(m[0]/bmu)) * exp((m[0]-1.0)*(V[i]))},availability[i] != 0))
    for k in V:
        Gi[k] = bioMultSum(Gidict[k])
    P = mev(V,Gi,availability,choice) 
    return P


def logcnlmu(V,availability,nests,choice,bmu) :
    Gi = {}
    Gidict = {}
    for k in V:
        Gidict[k] = []
    for m in nests:
        biosumlist = [] 
        for i,a in m[1].items():
            biosumlist.append(Elem({0:0,1:a**(m[0]/bmu) * exp(m[0] * (V[i]))},availability[i] != 0))
        biosum = bioMultSum(biosumlist)
        for i,a in m[1].items():
            Gidict[i].append(Elem({0:0,1:bmu * (biosum**((bmu/m[0])-1.0)) * (a**(m[0]/bmu)) * exp((m[0]-1.0)*(V[i]))},availability[i] != 0))
    for k in V:
        Gi[k] = bioMultSum(Gidict[k])
    logP = logmev(V,Gi,availability,choice) 
    return logP