Changeset 19044 for issm/trunk-jpl/src/m/plot/export_gl.py

Timestamp:

01/28/15 12:21:22 (10 years ago)

Author:

dlcheng

Message:

Updated export_gl scripts now working

File:

• issm/trunk-jpl/src/m/plot/export_gl.py (modified) (4 diffs)

issm/trunk-jpl/src/m/plot/export_gl.py

@@ -1 @@
-def export_gl(md,varargin):
-        return;
-        templist=plotoptions(varargin[:]); 
+from plotoptions import plotoptions
+from checkplotoptions import checkplotoptions
+from model import model
+import numpy as np
+import math
+from writejsfile import writejsfile
+
+def export_gl(md,*varargin):
+        class ResultObj(object):
+            def __getattr__(self, attr):
+                return self.__dict__.get(attr)
+        print ('getting options')
+        templist=plotoptions(varargin); 
         optionslist=templist.list;
         options=optionslist[1];
         options=checkplotoptions(md,options);
-
+        
         #Setup unique directory in present dir: 
-        directory=getfieldvalue(options,'directory','./');
-        databasename=getfieldvalue(options,'database','webgl');
-
+        print ('setting directory')
+        directory=options.getfieldvalue('directory','./');
+        databasename=options.getfieldvalue('database','Pig');
+        print(directory +databasename)  
         #scaling factor: 
-        scaling_factor=getfieldvalue(options,'scaling_factor',50);
+        print ('setting scaling factor')
+        scaling_factor=options.getfieldvalue('scaling_factor',50);
 
         #Deal with title: 
-        if options['title']:
-                title=getfieldvalue(options,'title');
+        print ('setting title')
+        if options.exist('title'):
+                title=options.getfieldvalue('title');
         else:
                 title='';
 
         #initialize model: 
+        print ('initializing model')
         model.title=title;
-        model.initialZoomFactor=getfieldvalue(options,'zoom',-.25);
+        model.initialZoomFactor=options.getfieldvalue('zoom',-.25);
 
         #Deal with contour {{{
-        contour_lat1=md.mesh.lat(md.mesh.segments[:,1]);
-        contour_lat2=md.mesh.lat(md.mesh.segments[:,2]);
-        contour_long1=md.mesh.long(md.mesh.segments[:,1]);
-        contour_long2=md.mesh.long(md.mesh.segments[:,2]);
-        contour_surface1=md.geometry.surface(md.mesh.segments[:,1]);
-        contour_surface2=md.geometry.surface(md.mesh.segments[:,2]);
+        print ('getting contour')
+        contour_lat1=md.mesh.lat.take(md.mesh.segments[:,0])
+        contour_lat2=md.mesh.lat.take(md.mesh.segments[:,1]);
+        contour_long1=md.mesh.long.take(md.mesh.segments[:,0]);
+        contour_long2=md.mesh.long.take(md.mesh.segments[:,1]);
+        contour_surface1=md.geometry.surface.take(md.mesh.segments[:,0]);
+        contour_surface2=md.geometry.surface.take(md.mesh.segments[:,1]);
 
-        R1=6371000*ones(len(contour_surface1),1)+scaling_factor*contour_surface1;
-        R2=6371000*ones(len(contour_surface2),1)+scaling_factor*contour_surface2;
+        R1=6371000*np.ones(len(contour_surface1))+scaling_factor*contour_surface1;
+        R2=6371000*np.ones(len(contour_surface2))+scaling_factor*contour_surface2;
 
-        contourx1 = R1 * cosd(contour_lat1) * cosd(contour_long1);
-        contoury1 = R1 * cosd(contour_lat1) * sind(contour_long1);
-        contourz1 = R1 * sind(contour_lat1);
+        contourx1 = map(lambda r, lat, long: r * math.cos(math.radians(lat)) * math.cos(math.radians(long)), R1, contour_lat1, contour_long1);
+        contoury1 = map(lambda r, lat, long: r * math.cos(math.radians(lat)) * math.sin(math.radians(long)), R1, contour_lat1, contour_long1);
+        contourz1 = map(lambda r, lat: r * math.sin(math.radians(lat)), R1, contour_lat1);
         
-        contourx2 = R2 * cosd(contour_lat2) * cosd(contour_long2);
-        contoury2 = R2 * cosd(contour_lat2) * sind(contour_long2);
-        contourz2 = R2 * sind(contour_lat2);
-
+        contourx2 = map(lambda r, lat, long: r * math.cos(math.radians(lat)) * math.cos(math.radians(long)), R2, contour_lat2, contour_long2);
+        contoury2 = map(lambda r, lat, long: r * math.cos(math.radians(lat)) * math.sin(math.radians(long)), R2, contour_lat2, contour_long2);
+        contourz2 = map(lambda r, lat: r * math.sin(math.radians(lat)), R2, contour_lat2);
 
         model.contourx1=contourx1;
@@ -52 +66 @@
 
         #}}}
-#Deal with mesh and results {{{
-        
-        
-        lat=md.mesh.lat;
-        long=md.mesh.long;
+        #Deal with mesh and results {{{
+        print ('getting mesh')
         surface=md.geometry.surface;
         numberofelements=md.mesh.numberofelements;
         numberofvertices=md.mesh.numberofvertices;
-
-        R=6371000*ones(numberofvertices,1)+scaling_factor*surface;
-
-        x = R * cosd(lat) * cosd(long);
-        y = R * cosd(lat) * sind(long);
-        z = R * sind(lat);
-
-
+        R=6371000*np.ones(len(md.mesh.lat))+scaling_factor*surface.flatten();
+        
+        x = map(lambda r, lat, long: r * math.cos(math.radians(lat)) * math.cos(math.radians(long)), R, md.mesh.lat,md.mesh.long);
+        y = map(lambda r, lat, long: r * math.cos(math.radians(lat)) * math.sin(math.radians(long)), R, md.mesh.lat,md.mesh.long);
+        z = map(lambda r, lat: r * math.sin(math.radians(lat)), R, md.mesh.lat);
+        
         #Deal with triangulation: 
+        print('getting triangulation')
         model.index=md.mesh.elements;
         model.x=x;
@@ -75 +85 @@
         model.surface=surface;
         
+        results = [""]
+        
         #Deal with data: 
-        results=struct([]);
+        print('getting data')
         for i in xrange(1,len(optionslist)):
                 options=optionslist[i]; 
                 options=checkplotoptions(md,options);
-                data=getfieldvalue(options,'data');
+                data=options.getfieldvalue('data');
+                results.append(ResultObj())
                 results[i].data=data;
-                results[i].caxis=getfieldvalue(options,'caxis',[min(data), max(data)]);
+                results[i].caxis=options.getfieldvalue('caxis',[min(data), max(data)]);
 
-                label=getfieldvalue(options,'label','');
-                if strcmpi(label,''):
+                label=options.getfieldvalue('label','');
+                if label=='':
                         #create generic label: 
-                        label=['data', num2str(i)];
+                        label=['data', str(i)];
                 results[i].label=label;
 
-                shortlabel=getfieldvalue(options,'shortlabel','');
-                if strcmpi(shortlabel,''):
+                shortlabel=options.getfieldvalue('shortlabel','');
+                if shortlabel=='':
                         #create generic short label: 
-                        shortlabel=['data', num2str(i)];
+                        shortlabel=['data', str(i)];
                 results[i].shortlabel=shortlabel;
-                
+
                 if len(data[2])>1:
-                        time_range=getfieldvalue(options,'time_range',[0, 100]);
+                        time_range=options.getfieldvalue('time_range',[0, 100]);
                         results[i].time_range=time_range;
 
-                unit=getfieldvalue(options,'unit','');
-                if strcmpi(unit,''):
+                unit=options.getfieldvalue('unit','');
+                if unit=='':
                         #create generic unit: 
                         unit='SI';
@@ -108 +121 @@
         
         #Write model to javascript database file: 
+        print('writing to file')
         writejsfile(directory + databasename + '.js',model,databasename);
 #}}}