DataMap.py

"""

"""
import numpy as np
import PIL
from PIL import Image
from array import *
import sys
import math
import os

from TextureMap import *
from THREE.textures.DataTexture import *
from Config import *
import pickle


class DataMap(TextureMap):
    """
    Data storage for textures
    """
    def __init__(self, size, datamaps=None):
        name = None
        if  isinstance(size, str):
            name = size
            size = 0
        super(DataMap, self).__init__(size, 0, 4, np.float32)

        if name is not None:
            self.load(datamaps, name)

    def setXY(self, x, y, normal, z):
        super().setXY(x, y, (normal.x, normal.y, normal.z, z))

    def save(self, datamaps, name):
        size = self.data.size * self.data.dtype.itemsize
        datamaps.save(self.data, size, name)

    def load(self, datamaps, name):
        self.data = datamaps.load(name)
        self.size = int(math.sqrt(self.data.size/4))

    def average(self):
        z = 0
        s = self.data.size
        data = self.data
        for p in range(0, s, 4):
            z += data[p + 3]
        z /= (self.size ** 2)

        return z

    def DataTexture(self):
        dt = DataTexture(self.data, self.size, self.size)
        #dt.magFilter = LinearFilter
        #dt.minFilter = LinearFilter
        return dt