from XdmfCore import *

if __name__ == "__main__":

    values1 = [0, 1, 2, 3]
    array1 = XdmfArray.New()
    array1.insertAsInt8(0, values1)
    print array1.getValuesString()
    print str(array1.getArrayType()) + " ?= " + str(XdmfArrayType.Int8())
    assert array1.getArrayType() == XdmfArrayType.Int8()

    array2 = XdmfArray.New()
    array2.insertAsInt16(0, values1)
    print array2.getValuesString()
    print str(array2.getArrayType()) + " ?= " + str(XdmfArrayType.Int16())
    assert array2.getArrayType() == XdmfArrayType.Int16()

    array3 = XdmfArray.New()
    array3.insertAsInt32(0, values1[2:4])
    print array3.getValuesString()
    print str(array3.getArrayType()) + " ?= " + str(XdmfArrayType.Int32())
    assert array3.getArrayType() == XdmfArrayType.Int32()

    # Stride
    array4 = XdmfArray.New()
    array4.insertAsInt64(0, values1[0:4:2])
    print array4.getValuesString()
    print str(array4.getArrayType()) + " ?= " + str(XdmfArrayType.Int64())
    assert array4.getArrayType() == XdmfArrayType.Int64()

    # Stride on List
    array4a = XdmfArray.New()
    array4a.insertAsInt64(0, values1, 2, 4, 2, 1);
    print array4a.getValuesString()
    print str(array4a.getArrayType()) + " ?= " + str(XdmfArrayType.Int64())
    assert array4a.getArrayType() == XdmfArrayType.Int64()

    # Reverse
    array5 = XdmfArray.New()
    array5.insertAsFloat32(0, values1[::-1])
    print array5.getValuesString()
    print str(array5.getArrayType()) + " ?= " + str(XdmfArrayType.Float32())
    assert array5.getArrayType() == XdmfArrayType.Float32()

    array6 = XdmfArray.New()
    array6.insertAsFloat64(0, values1)
    print array6.getValuesString()
    print str(array6.getArrayType()) + " ?= " + str(XdmfArrayType.Float64())
    assert array6.getArrayType() == XdmfArrayType.Float64()

    array7 = XdmfArray.New()
    array7.insertAsUInt8(0, values1)
    print array7.getValuesString()
    print str(array7.getArrayType()) + " ?= " + str(XdmfArrayType.UInt8())
    assert array7.getArrayType() == XdmfArrayType.UInt8()

    array8 = XdmfArray.New()
    array8.insertAsUInt16(0, values1)
    print array8.getValuesString()
    print str(array8.getArrayType()) + " ?= " + str(XdmfArrayType.UInt16())
    assert array8.getArrayType() == XdmfArrayType.UInt16()

    array9 = XdmfArray.New()
    array9.insertAsUInt32(0, values1)
    print array9.getValuesString()
    print str(array9.getArrayType()) + " ?= " + str(XdmfArrayType.UInt32())
    assert array9.getArrayType() == XdmfArrayType.UInt32()

    stringArray = XdmfArray.New()
    stringArray.insertValueAsString(0, "foo")
    strings = ["bar", "car", "cellar"]
    print str(stringArray.getArrayType()) + " ?= " + str(XdmfArrayType.String())
    assert stringArray.getArrayType() == XdmfArrayType.String()
    stringArray.insertAsString(0, strings)
    print str(stringArray.getSize()) + " ?= " + str(3)
    print stringArray.getValueAsString(0) + " ?= " + "bar"
    assert stringArray.getSize() == 3
    assert stringArray.getValueAsString(0) == "bar"
    stringArray.pushBackAsString("dog")
    print str(stringArray.getSize()) + " ?= " + str(4)
    print stringArray.getValueAsString(3) + " ?= " + "dog"
    assert stringArray.getSize() == 4
    assert stringArray.getValueAsString(3) == "dog"
    stringArray.resizeAsString(5, "")
    print str(stringArray.getSize()) + " ?= " + str(5)
    print stringArray.getValueAsString(4) + " ?= " + ""
    assert stringArray.getSize() == 5
    assert stringArray.getValueAsString(4) == ""
    print stringArray.getValuesString()

    # Retrieving data from XdmfArray
    sourceArray = XdmfArray.New()
    for i in range(0,20):
      sourceArray.pushBackAsInt32(i)
    output = []
    sourceArray.getValues(0, output, sourceArray.getSize())
    print output
    for i in range(0,20):
      assert(output[i] == i)
    output2 = []
    sourceArray.getValues(0, output2, sourceArray.getSize()/2, 2, 2)
    print output2
    for i in range(0,len(output2)):
      if (i % 2 == 0):
        assert(output2[i] == i)
      else:
        assert(output2[i] == 0)