Exemplo n.º 1
0
    def getPlotData(self, plottype, plotfilename, doubleaxis, converty,
                    converty2, normalized, desvest):
        exdata = ExtractData(plottype, plotfilename, doubleaxis, desvest)
        #xdata, ydata, xdata2, ydata2
        plotdata = exdata.get_data()

        #print("convert; ", converty, " ", converty2)

        #TODO: smoothing data

        if converty:
            for i in range(len(plotdata[1])):
                plotdata[1][i] = np.array(
                    plotdata[1][i]).astype(float) * float(converty)

        if converty2:
            for i in range(len(plotdata[3])):
                plotdata[3][i] = np.array(
                    plotdata[3][i]).astype(float) * float(converty2)

        if normalized == "Normalized by max":
            for i in range(len(plotdata[1])):
                maxvalue = max(plotdata[1][i])
                for j in range(len(plotdata[1][i])):
                    plotdata[1][i][j] = plotdata[1][i][j] / maxvalue
        elif normalized == "Normalized by first":
            for i in range(len(plotdata[1])):
                firstvalue = plotdata[1][i][0]
                for j in range(len(plotdata[1][i])):
                    plotdata[1][i][j] = plotdata[1][i][j] / firstvalue

        return plotdata