LoadTerrain.py

import os
from TextureHolder import TextureHolder
import LinAlgOps
import Image
from Road import *
from RenderTexture import *

CONSTANTS = "constants.conf"

class LoadTerrain:
    X_FACTOR = 1
    Y_FACTOR = 1
    Z_FACTOR = 1
    SEA_LEVEL = 4
    counter = 1


    def __init__(self, filename, convert, tex_holder):
        f = open(CONSTANTS, 'r')
        lines = f.read().split('\n')
        self.MAP_SIZE = int(lines[1].split()[1])
        self.roadc = int(lines[4].split()[1])
        f.close()
        
        #self.road_range1 = (self.MAP_SIZE/2)-roadc
        #self.road_range2 = (self.MAP_SIZE/2)+roadc
        
        self.filename = filename
        self.convert = convert
        self.tex_holder = tex_holder
        self.im = Image.open(filename)
        self.convert.set_dimensions(self.im.size[0], self.im.size[1])
        self.images = []
        self.center = 0
        self.index  = 0  

    def load(self):
        '''Load a heightmap bitmap file into the Terrain.
        Store the terrain as a list of triangle strips.
        '''      
        heights = []
        
        curr_pix = self.MAP_SIZE/2

        # Down the rows
        for x in range(self.convert.heightmap_x):
            col = []
            # Across the columns
            for z in range(self.convert.heightmap_z):
                pix = (self.im.getpixel((x, z)) / 255.) * self.convert.open_gl_scale[1]
                col.append(pix)
            heights.append(col)


       # r = Road()
       # heights2 = r.create_yroad(heights,(0,0), self.roadc)
        #heights2 = self.create_road(heights)
        return heights             
                
    def init_createRenderList(self, heights, textname):
        #rend = RenderTexture(heights, (self.X_FACTOR, self.Y_FACTOR, self.Z_FACTOR))
        rend = RenderTexture(heights, self.convert, self.tex_holder)
        tex_file_name = rend.initial_run(self.filename.split('/')[-1])
        #print "start norms"
        face_norms, vert_norms = LinAlgOps.calc_face_normals(heights, self.convert)
        #print "end norms"
        #return (tex_file_name, 0, 0)
        return (tex_file_name, face_norms, vert_norms)

    def createRenderList(self, heights, textname):
        #rend = RenderTexture(heights, (self.X_FACTOR, self.Y_FACTOR, self.Z_FACTOR))
        rend = RenderTexture(heights, self.convert, self.tex_holder)
        tex_file_name = rend.run(self.filename.split('/')[-1])
        #print "start norms"
        face_norms, vert_norms = LinAlgOps.calc_face_normals(heights, self.convert)
        #print "end norms"
        #return (tex_file_name, 0, 0)
        return (tex_file_name, face_norms, vert_norms)