<div dir="ltr">I have created an unstructured grid data set that uses the SetTimeStep method to get the data for each time step. What I am having trouble doing to adding an array with calculated information and then writing the file out.<div><br></div><div>see below.<br><div><br><div><br></div><div>I create a variable TAWSS</div><div><br></div><div><div>TAWSS = vtk.vtkFloatArray()</div><div>TAWSS.SetNumberOfComponents(1)</div><div>TAWSS.SetNumberOfValues(calc2.GetOutput().GetNumberOfCells())</div><div>TAWSS.SetName("TAWSS")</div><div><br></div><div>then I calculate the average over the whole dataset, where calc2 is a vtkArrayCalculator() earlier in the pipeline.</div><div><br></div><div><div>for i in range(N):</div><div>    reader.SetTimeStep(i)</div><div>    print("Time step {0} for average calc".format(i))</div><div>    reader.Update()</div><div>    calc2.Update()</div><div>    wss_temp = calc2.GetOutput().GetCellData().GetArray("WSS")</div><div>    if ( i == 0 ):</div><div>        TAWSS.DeepCopy(wss_temp)</div><div>    else:</div><div>        array_avg(TAWSS, wss_temp, reader.GetOutput().GetNumberOfCells(), float(i+1.0))</div><div><br></div><div>    calc2.GetOutput().GetCellData().AddArray(TAWSS)</div></div><div><br></div><div>this doesn't add TAWSS to the outfile.</div><div><br></div><div><div>writer = vtk.vtkXMLUnstructuredGridWriter()</div><div>writer.SetFileName("calc_test.vtu")</div><div>writer.SetNumberOfTimeSteps(N)</div><div>writer.SetInputConnection(calc2.GetOutputPort())</div><div>writer.Start()</div><div><br></div><div>for i in range(N):</div><div>    reader.SetTimeStep(i)</div><div>    print("Writing time step {0}".format(i))</div><div>    calc2.GetOutput().GetCellData().AddArray(TAWSS)</div><div>    reader.Update()</div><div>    calc2.Update()</div><div><br></div><div>    writer.WriteNextTime(reader.GetTimeStep())</div><div>writer.Stop()</div></div><div><br></div>-- <br><div class="gmail_signature">Kurt Sansom<br></div>
</div></div></div></div>