#!/bin/env python

from __future__ import print_function
from Xdmf import *

def Expression(*args) :
    e = ''
    for arg in args :
        if hasattr(arg, 'GetTagName') :
            e += arg.GetTagName() + ' '
        else :
            e += arg + ' '
    return XdmfExpr(e)


if __name__ == '__main__' :
    a1 = XdmfArray()
    a1.SetNumberType(XDMF_FLOAT32_TYPE)
    a1.SetNumberOfElements(20)
    a1.Generate(1, 20)
    a2 = XdmfArray()
    a2.SetNumberType(XDMF_INT32_TYPE)
    a2.SetNumberOfElements(5)
    a2.Generate(2, 10)
    print ('a1 Values = ', a1.GetValues())
    print ('a1[2:10] = ' + Expression(a1 , '[ 2:10 ]').GetValues())
    print ('a2 Values = ', a2.GetValues())
    print ('a1[a2] = ' + Expression(a1 , '[', a2, ']').GetValues())
    print ('a1 + a2 = ' + Expression(a1 , ' + ', a2).GetValues())
    print ('a1 * a2 = ' + Expression(a1 , ' * ', a2).GetValues())
    a2.SetNumberType(XDMF_FLOAT32_TYPE)
    a2.SetNumberOfElements(20)
    a2.Generate(21, 40)
    print ('a2 Values = ', a2.GetValues())
    print ('a1 , a2 (Interlace) = ' + Expression(a1 , ' , ', a2).GetValues())
    print ('a1 , a2, a1 (Interlace) = ' + Expression(a1 , ' , ', a2, ' , ', a1).GetValues())
    print ('a1 ; a2 (Concat) = ' + Expression(a1 , ' ; ', a2).GetValues())
    print ('where(a1 > 10) = ' + Expression('Where( ', a1 , ' > 10)').GetValues())
    print ('a2[where(a1 > 10)] = ' + Expression(a2, '[Where( ', a1 , ' > 10)]').GetValues())