ParticleFilterVisualizer.py 文件源码

def main(argv):
    dataFolder = ''
    particlesFile = ''
    numberOfParticles = 0
    outputFile = "animation"

    usage = 'ParticleFilterVisualizer.py -f <dataFolder> -p <particleFile> -n <numberOfParticles> [-o <outputFileName>]'

    try:
        opts, args = getopt.getopt(argv,"hf:p:n:o:",["dataFolder=","particleFile=","numberParticles=" ,"outPutFile="])
    except getopt.GetoptError:
        print(usage)
        sys.exit(2)

    if len(opts) is 0:
        print(usage)
        sys.exit()

    for opt, arg in opts:
        if opt == '-h':
            print(usage)
            sys.exit()
        elif opt in ("-f", "--dataFolder"):
            dataFolder = arg
        elif opt in ("-p", "--particleFile"):
            particlesFile = arg
        elif opt in ("-n", "--numberOfParticles"):
            numberOfParticles = arg
        elif opt in ("-o", "--outputFile"):
            outputFile = arg
        else:
            print(usage)
            sys.exit(2)

    plotData(dataFolder, particlesFile, numberOfParticles, outputFile)