Esempio n. 1
0
    iome = IOManager()

    # Read file with simulation data
    try:
        filename = sys.argv[1]
    except IndexError:
        filename = GD.file_resultdatafile

    iomc.open_file(filename=filename)

    # New file for eigen transformed data
    P = iomc.load_parameters()
    iome.create_file(P, filename=filename[:-5]+"_eigen.hdf5")

    # Iterate over all groups
    for groupid in iomc.get_group_ids():

        # Create the group if necessary
        if not groupid in iome.get_group_ids():
            iome.create_group(groupid=groupid)

        for blockid in iomc.get_block_ids(groupid=groupid):
            print("Computing eigentransformation of data in block '"+str(blockid)+"'")

            # Create the block if necessary
            if not blockid in iome.get_block_ids(groupid=groupid):
                iome.create_block(blockid=blockid, groupid=groupid)

            # See if we have a wavefunction
            if iomc.has_wavefunction(blockid=blockid):
                from EigentransformWavefunction import transform_wavefunction_to_eigen
# Read file with simulation data
iomc = IOManager()
iomc.open_file(inputfile)

iome = IOManager()
iome.create_file(outputfile)

# New file for eigen transformed data
P = iomc.load_parameters()

# Save the simulation parameters
iome.add_parameters()
iome.save_parameters(P)

# Iterate over all groups
for groupid in iomc.get_group_ids():

    # Create the group if necessary
    if groupid not in iome.get_group_ids():
        iome.create_group(groupid=groupid)

    for blockid in iomc.get_block_ids(groupid=groupid):
        print("Computing eigentransformation of data in block '{}'".format(blockid))

        # Create the block if necessary
        if blockid not in iome.get_block_ids(groupid=groupid):
            iome.create_block(blockid=blockid, groupid=groupid)

        # See if we have a wavefunction
        if iomc.has_wavefunction(blockid=blockid):
            from WaveBlocksND.Interface import EigentransformWavefunction