source: trunk/packages/vizservers/vtkvis/Renderer.h @ 3695

Last change on this file since 3695 was 3695, checked in by ldelgass, 11 years ago

Add protocol for setting glyphs tesselation quality
  • Property svn:eol-style set to native
File size: 33.3 KB
Line 
1/* -*- mode: c++; c-basic-offset: 4; indent-tabs-mode: nil -*- */
2/*
3 * Copyright (C) 2004-2012  HUBzero Foundation, LLC
4 *
5 * Author: Leif Delgass <ldelgass@purdue.edu>
6 */
7
8#ifndef VTKVIS_RENDERER_H
9#define VTKVIS_RENDERER_H
10
11#include <string>
12#include <vector>
13#include <tr1/unordered_map>
14#include <typeinfo>
15
16#include <vtkSmartPointer.h>
17#ifdef USE_CUSTOM_AXES
18#include "vtkRpCubeAxesActor.h"
19#else
20#include <vtkCubeAxesActor.h>
21#endif
22#include <vtkScalarBarActor.h>
23#include <vtkRenderer.h>
24#include <vtkRenderWindow.h>
25#include <vtkUnsignedCharArray.h>
26
27#include "ColorMap.h"
28#include "Types.h"
29#include "DataSet.h"
30#include "Arc.h"
31#include "Arrow.h"
32#include "Box.h"
33#include "Cone.h"
34#include "Contour2D.h"
35#include "Contour3D.h"
36#include "Cutplane.h"
37#include "Cylinder.h"
38#include "Disk.h"
39#include "Glyphs.h"
40#include "Group.h"
41#include "HeightMap.h"
42#include "LIC.h"
43#include "Line.h"
44#include "Molecule.h"
45#include "Outline.h"
46#include "PolyData.h"
47#include "Polygon.h"
48#include "PseudoColor.h"
49#include "Sphere.h"
50#include "Streamlines.h"
51#include "Volume.h"
52#include "Warp.h"
53#include "Trace.h"
54
55// Controls if TGA format is sent to client
56//#define RENDER_TARGA
57#define TARGA_BYTES_PER_PIXEL 3
58
59namespace VtkVis {
60
61/**
62 * \brief VTK Renderer
63 */
64class Renderer
65{
66public:
67    Renderer();
68    virtual ~Renderer();
69
70    enum AxisRangeMode {
71        RANGE_AUTO = 0,
72        RANGE_SCALE_BOUNDS,
73        RANGE_EXPLICIT
74    };
75
76    enum AxesFlyMode {
77        FLY_OUTER_EDGES = 0,
78        FLY_CLOSEST_TRIAD,
79        FLY_FURTHEST_TRIAD,
80        FLY_STATIC_EDGES,
81        FLY_STATIC_TRIAD
82    };
83
84    enum AxesTickPosition {
85        TICKS_INSIDE,
86        TICKS_OUTSIDE,
87        TICKS_BOTH
88    };
89
90    enum CameraMode {
91        PERSPECTIVE,
92        ORTHO,
93        IMAGE
94    };
95
96    enum Aspect {
97        ASPECT_NATIVE,
98        ASPECT_SQUARE,
99        ASPECT_WINDOW
100    };
101
102    enum LegendType {
103        LEGEND_SCALAR,
104        LEGEND_VECTOR_MAGNITUDE,
105        LEGEND_VECTOR_X,
106        LEGEND_VECTOR_Y,
107        LEGEND_VECTOR_Z
108    };
109
110    typedef std::string DataSetId;
111    typedef std::string ColorMapId;
112    typedef std::string FieldId;
113
114    // Data sets
115
116    void addDataSet(const DataSetId& id);
117
118    void deleteDataSet(const DataSetId& id);
119
120    DataSet *getDataSet(const DataSetId& id);
121
122    void getDataSetNames(std::vector<std::string>& names);
123
124    bool setData(const DataSetId& id, char *data, int nbytes);
125
126    bool setDataFile(const DataSetId& id, const char *filename);
127
128    bool setDataSetActiveScalars(const DataSetId& id, const char *scalarName);
129
130    bool setDataSetActiveVectors(const DataSetId& id, const char *vectorName);
131
132    bool getScalarValueAtPixel(const DataSetId& id, int x, int y, double *value);
133
134    bool getScalarValue(const DataSetId& id, double x, double y, double z, double *value);
135
136    bool getVectorValueAtPixel(const DataSetId& id, int x, int y, double vector[3]);
137
138    bool getVectorValue(const DataSetId& id, double x, double y, double z, double vector[3]);
139
140    void setDataSetOpacity(const DataSetId& id, double opacity);
141
142    void setDataSetVisibility(const DataSetId& id, bool state);
143
144    void setUseCumulativeDataRange(bool state, bool onlyVisible = false);
145
146    bool getUseCumulativeRange();
147
148    bool setCumulativeDataRange(double *range, const char *name,
149                                DataSet::DataAttributeType type,
150                                int numComponents,
151                                int component = -1);
152
153    bool getCumulativeDataRange(double *range, const char *name,
154                                int numComponents,
155                                int component = -1);
156
157    bool getCumulativeDataRange(double *range, const char *name,
158                                DataSet::DataAttributeType type,
159                                int numComponents,
160                                int component = -1);
161
162    // Render window
163
164    /// Get the VTK render window object this Renderer uses
165    vtkRenderWindow *getRenderWindow()
166    {
167        return _renderWindow;
168    }
169
170    void setWindowSize(int width, int height);
171
172    int getWindowWidth() const;
173
174    int getWindowHeight() const;
175
176    // Lights
177
178    int addLight(float pos[3]);
179
180    vtkLight *getLight(int lightIdx);
181
182    void setLightSwitch(int lightIdx, bool state);
183
184    // Camera controls
185
186    void setViewAngle(int height);
187
188    /// Return the VTK camera object this Renderer uses
189    vtkCamera *getVtkCamera()
190    {
191        if (_renderer != NULL)
192            return _renderer->GetActiveCamera();
193        else
194            return NULL;
195    }
196
197    bool isCameraMaximized()
198    {
199        return (_cameraZoomRatio == 1.0 &&
200                _cameraPan[0] == 0.0 &&
201                _cameraPan[1] == 0.0);
202    }
203
204    void getImageCameraSizes(int *imgWidthPx, int *imgHeightPx,
205                             int *_pxOffsetX = NULL, int *_pxOffsetY = NULL);
206
207    double getImageCameraAspect();
208
209    void setCameraMode(CameraMode mode);
210
211    CameraMode getCameraMode() const;
212
213    void setCameraAspect(Aspect aspect);
214
215    void resetVtkCamera(double *bounds = NULL);
216
217    void resetCamera(bool resetOrientation = true);
218
219    void resetCameraClippingRange();
220
221    bool setCameraZoomRegionPixels(int x, int y, int width, int height);
222
223    bool setCameraZoomRegion(double x, double y, double width, double height);
224
225    void getCameraZoomRegion(double xywh[4]) const;
226
227    void getScreenWorldCoords(double xywh[4]) const;
228
229    void rotateCamera(double yaw, double pitch, double roll);
230
231    void setCameraOrientation(const double quat[4], bool absolute = true);
232
233    void panCamera(double x, double y, bool absolute = true);
234
235    void zoomCamera(double z, bool absolute = true);
236
237    void setCameraOrientationAndPosition(const double position[3],
238                                         const double focalPoint[3],
239                                         const double viewUp[3]);
240
241    void getCameraOrientationAndPosition(double position[3],
242                                         double focalPoint[3],
243                                         double viewUp[3]);
244
245    void eventuallyResetCamera()
246    {
247        _needsCameraReset = true;
248    }
249
250    bool needsCameraReset()
251    {
252        return _needsCameraReset;
253    }
254
255    void eventuallyResetCameraClippingRange()
256    {
257        _needsCameraClippingRangeReset = true;
258    }
259
260    bool needsCameraClippingRangeReset()
261    {
262        return _needsCameraClippingRangeReset;
263    }
264
265    // Rendering an image
266
267    void setBackgroundColor(float color[3]);
268
269    void setClipPlane(Axis axis, double ratio, int direction);
270
271    void setUseTwoSidedLighting(bool state);
272
273    void setUseDepthPeeling(bool state);
274
275    void eventuallyRender();
276
277    bool render();
278
279    void getRenderedFrame(vtkUnsignedCharArray *imgData);
280
281    // Axes
282
283    void setAxesOrigin(double x, double y, double z, bool useCustom = true);
284
285    void setAxesAutoBounds(bool state);
286
287    void setAxesBounds(double min, double max);
288
289    void setAxesAutoRange(bool state);
290
291    void setAxisBounds(Axis axis, double min, double max);
292
293    void setAxesScale(double scale);
294
295    void setAxesRange(double min, double max);
296
297    void setAxesLabelPowerScaling(int xPow, int yPow, int zPow, bool useCustom = true);
298
299    void setAxisAutoBounds(Axis axis, bool state);
300
301    void setAxisAutoRange(Axis axis, bool state);
302
303    void setAxisScale(Axis axis, double scale);
304
305    void setAxisRange(Axis axis, double min, double max);
306
307    void setAxesFlyMode(AxesFlyMode mode);
308
309    void setAxesVisibility(bool state);
310
311    void setAxesGridVisibility(bool state);
312
313    void setAxesInnerGridVisibility(bool state);
314
315    void setAxesGridpolysVisibility(bool state);
316
317    void setAxesLabelVisibility(bool state);
318
319    void setAxesTickVisibility(bool state);
320
321    void setAxesMinorTickVisibility(bool state);
322
323    void setAxesTickPosition(AxesTickPosition pos);
324
325    void setAxesColor(double color[3], double opacity = 1.0);
326
327    void setAxesTitleColor(double color[3], double opacity = 1.0);
328
329    void setAxesLabelColor(double color[3], double opacity = 1.0);
330
331    void setAxesLinesColor(double color[3], double opacity = 1.0);
332
333    void setAxesGridlinesColor(double color[3], double opacity = 1.0);
334
335    void setAxesInnerGridlinesColor(double color[3], double opacity = 1.0);
336
337    void setAxesGridpolysColor(double color[3], double opacity = 1.0);
338
339    void setAxesLabelScaling(bool autoScale, int xpow, int ypow, int zpow);
340
341    void setAxesPixelFontSize(double screenSize);
342
343    void setAxesTitleFont(const char *fontName);
344
345    void setAxesTitleFontSize(int sz);
346
347    void setAxesTitleOrientation(double orientation);
348
349    void setAxesLabelFont(const char *fontName);
350
351    void setAxesLabelFontSize(int sz);
352
353    void setAxesLabelOrientation(double orientation);
354
355    void setAxesLabelFormat(const char *format);
356
357    void setAxisVisibility(Axis axis, bool state);
358
359    void setAxisGridVisibility(Axis axis, bool state);
360
361    void setAxisInnerGridVisibility(Axis axis, bool state);
362
363    void setAxisGridpolysVisibility(Axis axis, bool state);
364
365    void setAxisLabelVisibility(Axis axis, bool state);
366
367    void setAxisTickVisibility(Axis axis, bool state);
368
369    void setAxisMinorTickVisibility(Axis axis, bool state);
370
371    void setAxisColor(Axis axis, double color[3], double opacity = 1.0);
372
373    void setAxisTitleColor(Axis axis, double color[3], double opacity = 1.0);
374
375    void setAxisLabelColor(Axis axis, double color[3], double opacity = 1.0);
376
377    void setAxisLinesColor(Axis axis, double color[3], double opacity = 1.0);
378
379    void setAxisGridlinesColor(Axis axis, double color[3], double opacity = 1.0);
380
381    void setAxisInnerGridlinesColor(Axis axis, double color[3], double opacity = 1.0);
382
383    void setAxisGridpolysColor(Axis axis, double color[3], double opacity = 1.0);
384
385    void setAxisTitle(Axis axis, const char *title);
386
387    void setAxisUnits(Axis axis, const char *units);
388
389    void setAxisTitleFont(Axis axis, const char *fontName);
390
391    void setAxisTitleFontSize(Axis axis, int sz);
392
393    void setAxisTitleOrientation(Axis axis, double orientation);
394
395    void setAxisLabelFont(Axis axis, const char *fontName);
396
397    void setAxisLabelFontSize(Axis axis, int sz);
398
399    void setAxisLabelOrientation(Axis axis, double orientation);
400
401    void setAxisLabelFormat(Axis axis, const char *format);
402
403    void eventuallyResetAxes()
404    {
405        _needsAxesReset = true;
406    }
407
408    bool needsAxesReset()
409    {
410        return _needsAxesReset;
411    }
412
413    // Colormaps
414
415    void addColorMap(const ColorMapId& id, ColorMap *colorMap);
416
417    void deleteColorMap(const ColorMapId& id);
418
419    ColorMap *getColorMap(const ColorMapId& id);
420
421    void setColorMapNumberOfTableEntries(const ColorMapId& id, int numEntries);
422
423    bool renderColorMap(const ColorMapId& id,
424                        const DataSetId& dataSetID,
425                        LegendType legendType,
426                        const char *fieldName,
427                        DataSet::DataAttributeType type,
428                        std::string& title,
429                        double range[2],
430                        int width, int height,
431                        bool opaque,
432                        int numLabels,
433                        vtkUnsignedCharArray *imgData);
434
435    bool renderColorMap(const ColorMapId& id,
436                        const DataSetId& dataSetID,
437                        LegendType legendType,
438                        const char *fieldName,
439                        std::string& title,
440                        double range[2],
441                        int width, int height,
442                        bool opaque,
443                        int numLabels,
444                        vtkUnsignedCharArray *imgData);
445
446    bool renderColorMap(const ColorMapId& id,
447                        const DataSetId& dataSetID,
448                        LegendType legendType,
449                        std::string& title,
450                        double range[2],
451                        int width, int height,
452                        bool opaque,
453                        int numLabels,
454                        vtkUnsignedCharArray *imgData);
455
456    // Generic GraphicsObject methods
457
458    GraphicsObject *getGenericGraphicsObject(const DataSetId& id);
459
460    template<class T>
461    T *getGraphicsObject(const DataSetId& id);
462
463    template<class T>
464    bool addGraphicsObject(const DataSetId& id);
465
466    template<class T>
467    void deleteGraphicsObject(const DataSetId& id);
468
469    template<class T>
470    void deleteAllGraphicsObjects();
471
472    template<class T>
473    void mergeGraphicsObjectBounds(double *bounds, bool onlyVisible);
474
475    template<class T>
476    void mergeGraphicsObjectUnscaledBounds(double *bounds, bool onlyVisible);
477
478    template<class T>
479    void updateGraphicsObjectFieldRanges();
480
481    template<class T>
482    void setGraphicsObjectClippingPlanes(vtkPlaneCollection *planes);
483
484    template<class T>
485    void setGraphicsObjectAspect(double aspectRatio);
486
487    template<class T>
488    void setGraphicsObjectInterpolateBeforeMapping(const DataSetId& id, bool state);
489
490    template<class T>
491    void setGraphicsObjectColorMap(const DataSetId& id, const ColorMapId& colorMapId);
492
493    template<class T>
494    void updateGraphicsObjectColorMap(ColorMap *cmap);
495
496    template<class T>
497    bool graphicsObjectColorMapUsed(ColorMap *cmap);
498
499    template<class T>
500    void setGraphicsObjectVolumeSlice(const DataSetId& id, Axis axis, double ratio);
501
502    //   Prop/Prop3D properties
503
504    template<class T>
505    void setGraphicsObjectOrientation(const DataSetId& id, double quat[4]);
506
507    template<class T>
508    void setGraphicsObjectOrientation(const DataSetId& id, double angle, double axis[3]);
509
510    template<class T>
511    void setGraphicsObjectOrigin(const DataSetId& id, double origin[3]);
512
513    template<class T>
514    void setGraphicsObjectPosition(const DataSetId& id, double pos[3]);
515
516    template<class T>
517    void setGraphicsObjectAspect(const DataSetId& id, double aspect);
518
519    template<class T>
520    void setGraphicsObjectScale(const DataSetId& id, double scale[3]);
521
522    template<class T>
523    void setGraphicsObjectTransform(const DataSetId& id, vtkMatrix4x4 *trans);
524
525    template<class T>
526    void setGraphicsObjectVisibility(const DataSetId& id, bool state);
527
528    //   Actor properties
529
530    template<class T>
531    void setGraphicsObjectColor(const DataSetId& id, float color[3]);
532
533    template<class T>
534    void setGraphicsObjectCullFace(const DataSetId& id, GraphicsObject::CullFace state);
535
536    template<class T>
537    void setGraphicsObjectCulling(const DataSetId& id, bool state);
538
539    template<class T>
540    void setGraphicsObjectEdgeVisibility(const DataSetId& id, bool state);
541
542    template<class T>
543    void setGraphicsObjectEdgeColor(const DataSetId& id, float color[3]);
544
545    template<class T>
546    void setGraphicsObjectEdgeWidth(const DataSetId& id, float edgeWidth);
547
548    template<class T>
549    void setGraphicsObjectAmbient(const DataSetId& id, double coeff);
550
551    template<class T>
552    void setGraphicsObjectDiffuse(const DataSetId& id, double coeff);
553
554    template<class T>
555    void setGraphicsObjectShadingModel(const DataSetId& id, GraphicsObject::ShadingModel state);
556
557    template<class T>
558    void setGraphicsObjectSpecular(const DataSetId& id, double coeff, double power);
559
560    template<class T>
561    void setGraphicsObjectLighting(const DataSetId& id, bool state);
562
563    template<class T>
564    void setGraphicsObjectOpacity(const DataSetId& id, double opacity);
565
566    template<class T>
567    void setGraphicsObjectPointSize(const DataSetId& id, float size);
568
569    template<class T>
570    void setGraphicsObjectWireframe(const DataSetId& id, bool state);
571
572    // For Shapes:
573
574    template<class T>
575    void setGraphicsObjectFlipNormals(const DataSetId& id, bool state);
576
577    // Arcs
578
579    bool addArc(const DataSetId& id, double center[3], double pt1[3], double pt2[3]);
580
581    void setArcResolution(const DataSetId& id, int res);
582
583    // Arrows
584
585    bool addArrow(const DataSetId& id, double tipRadius, double shaftRadius, double tipLength, bool flipNormals = false);
586
587    void setArrowResolution(const DataSetId& id, int resTip, int resShaft);
588
589    // Boxes
590
591    bool addBox(const DataSetId& id, double xLen, double yLen, double zLen, bool flipNormals = false);
592
593    // Cones
594
595    bool addCone(const DataSetId& id, double radius, double height, bool cap, bool flipNormals = false);
596
597    void setConeResolution(const DataSetId& id, int res);
598
599    // 2D Contour plots
600
601    bool addContour2D(const DataSetId& id, int numContours);
602
603    bool addContour2D(const DataSetId& id, const std::vector<double>& contours);
604
605    void setContour2DContourField(const DataSetId& id, const char *fieldName);
606
607    void setContour2DNumContours(const DataSetId& id, int numContours);
608
609    void setContour2DContourList(const DataSetId& id, const std::vector<double>& contours);
610
611    void setContour2DColorMode(const DataSetId& id,
612                               Contour2D::ColorMode mode,
613                               const char *name, double range[2] = NULL);
614
615    void setContour2DColorMode(const DataSetId& id,
616                               Contour2D::ColorMode mode,
617                               DataSet::DataAttributeType type,
618                               const char *name, double range[2] = NULL);
619
620    // 3D Contour (isosurface) plots
621
622    bool addContour3D(const DataSetId& id, int numContours);
623
624    bool addContour3D(const DataSetId& id, const std::vector<double>& contours);
625
626    void setContour3DContourField(const DataSetId& id, const char *fieldName);
627
628    void setContour3DNumContours(const DataSetId& id, int numContours);
629
630    void setContour3DContourList(const DataSetId& id, const std::vector<double>& contours);
631
632    void setContour3DColorMode(const DataSetId& id,
633                               Contour3D::ColorMode mode,
634                               const char *name, double range[2] = NULL);
635
636    void setContour3DColorMode(const DataSetId& id,
637                               Contour3D::ColorMode mode,
638                               DataSet::DataAttributeType type,
639                               const char *name, double range[2] = NULL);
640
641    // Cutplanes
642
643    void setCutplaneCloudStyle(const DataSetId& id,
644                               Cutplane::CloudStyle style);
645
646    void setCutplaneOutlineVisibility(const DataSetId& id, bool state);
647
648    void setCutplaneSliceVisibility(const DataSetId& id, Axis axis, bool state);
649
650    void setCutplaneColorMode(const DataSetId& id,
651                              Cutplane::ColorMode mode,
652                              const char *name, double range[2] = NULL);
653
654    void setCutplaneColorMode(const DataSetId& id,
655                              Cutplane::ColorMode mode,
656                              DataSet::DataAttributeType type,
657                              const char *name, double range[2] = NULL);
658
659    // Cylinders
660
661    bool addCylinder(const DataSetId& id, double radius, double height, bool cap, bool flipNormals = false);
662
663    void setCylinderCapping(const DataSetId& id, bool state);
664
665    void setCylinderResolution(const DataSetId& id, int res);
666
667    // Disks
668
669    bool addDisk(const DataSetId& id, double innerRadius, double outerRadius, bool flipNormals = false);
670
671    void setDiskResolution(const DataSetId& id, int resRadial, int resCircum);
672
673    // Glyphs
674
675    bool addGlyphs(const DataSetId& id, Glyphs::GlyphShape shape);
676
677    void setGlyphsShape(const DataSetId& id, Glyphs::GlyphShape shape);
678
679    void setGlyphsOrientMode(const DataSetId& id, bool state, const char *name);
680
681    void setGlyphsQuality(const DataSetId& id, double quality);
682
683    void setGlyphsColorMode(const DataSetId& id,
684                            Glyphs::ColorMode mode,
685                            const char *name,
686                            double range[2] = NULL);
687
688    void setGlyphsScalingMode(const DataSetId& id,
689                              Glyphs::ScalingMode mode,
690                              const char *name,
691                              double range[2] = NULL);
692
693    void setGlyphsNormalizeScale(const DataSetId& id, bool normalize);
694
695    void setGlyphsScaleFactor(const DataSetId& id, double scale);
696
697    // Groups
698
699    bool addGroup(const DataSetId& id, const std::vector<Group::NodeId>& nodeList);
700
701    void addChildrenToGroup(const DataSetId& id, const std::vector<Group::NodeId>& nodeList);
702
703    void removeChildrenFromGroup(const DataSetId& id, const std::vector<Group::NodeId>& nodeList);
704
705    // Height maps
706
707    bool addHeightMap(const DataSetId& id, int numContours, double heightScale);
708
709    bool addHeightMap(const DataSetId& id, const std::vector<double>& contours, double heightScale);
710
711    void setHeightMapHeightScale(const DataSetId& id, double scale);
712
713    void setHeightMapCloudStyle(const DataSetId& id,
714                                HeightMap::CloudStyle style);
715
716    void setHeightMapNumContours(const DataSetId& id, int numContours);
717
718    void setHeightMapContourList(const DataSetId& id, const std::vector<double>& contours);
719
720    void setHeightMapContourSurfaceVisibility(const DataSetId& id, bool state);
721
722    void setHeightMapContourLineVisibility(const DataSetId& id, bool state);
723
724    void setHeightMapContourLineColorMapEnabled(const DataSetId& id, bool mode);
725
726    void setHeightMapContourEdgeColor(const DataSetId& id, float color[3]);
727
728    void setHeightMapContourEdgeWidth(const DataSetId& id, float edgeWidth);
729
730    void setHeightMapColorMode(const DataSetId& id,
731                               HeightMap::ColorMode mode,
732                               const char *name, double range[2] = NULL);
733
734    void setHeightMapColorMode(const DataSetId& id,
735                               HeightMap::ColorMode mode,
736                               DataSet::DataAttributeType type,
737                               const char *name, double range[2] = NULL);
738
739
740    // Lines
741
742    bool addLine(const DataSetId& id, double pt1[3], double pt2[3]);
743
744    // Molecules
745
746    void setMoleculeAtomQuality(const DataSetId& id, double quality);
747
748    void setMoleculeBondQuality(const DataSetId& id, double quality);
749
750    void setMoleculeAtomRadiusScale(const DataSetId& id, double scale);
751
752    void setMoleculeBondRadiusScale(const DataSetId& id, double scale);
753
754    void setMoleculeAtomScaling(const DataSetId& id, Molecule::AtomScaling scaling);
755
756    void setMoleculeAtomVisibility(const DataSetId& id, bool state);
757
758    void setMoleculeAtomLabelField(const DataSetId& id, const char *fieldName);
759
760    void setMoleculeAtomLabelVisibility(const DataSetId& id, bool state);
761
762    void setMoleculeBondVisibility(const DataSetId& id, bool state);
763
764    void setMoleculeBondStyle(const DataSetId& id, Molecule::BondStyle style);
765
766    void setMoleculeBondColorMode(const DataSetId& id, Molecule::BondColorMode mode);
767
768    void setMoleculeBondColor(const DataSetId& id, float color[3]);
769
770    void setMoleculeColorMode(const DataSetId& id,
771                              Molecule::ColorMode mode,
772                              const char *name, double range[2] = NULL);
773
774    void setMoleculeColorMode(const DataSetId& id,
775                              Molecule::ColorMode mode,
776                              DataSet::DataAttributeType type,
777                              const char *name, double range[2] = NULL);
778
779    // N-sided Regular Polygons
780
781    bool addPolygon(const DataSetId& id, int numSides, double center[3], double normal[3], double radius);
782
783    // PolyData meshes
784
785    void setPolyDataCloudStyle(const DataSetId& id,
786                               PolyData::CloudStyle style);
787
788    // Color-mapped surfaces
789
790    void setPseudoColorCloudStyle(const DataSetId& id,
791                                  PseudoColor::CloudStyle style);
792
793    void setPseudoColorColorMode(const DataSetId& id,
794                                 PseudoColor::ColorMode mode,
795                                 const char *name, double range[2] = NULL);
796
797    void setPseudoColorColorMode(const DataSetId& id,
798                                 PseudoColor::ColorMode mode,
799                                 DataSet::DataAttributeType type,
800                                 const char *name, double range[2] = NULL);
801
802    // Spheres
803
804    bool addSphere(const DataSetId& id, double center[3], double radius, bool flipNormals = false);
805
806    void setSphereSection(const DataSetId& id,
807                          double thetaStart, double thetaEnd,
808                          double phiStart, double phiEnd);
809
810    void setSphereResolution(const DataSetId& id, int thetaRes, int phiRes);
811
812    // Streamlines
813
814    void setStreamlinesColorMode(const DataSetId& id,
815                                 Streamlines::ColorMode mode,
816                                 const char *name, double range[2] = NULL);
817
818    void setStreamlinesColorMode(const DataSetId& id,
819                                 Streamlines::ColorMode mode,
820                                 DataSet::DataAttributeType type,
821                                 const char *name, double range[2] = NULL);
822
823    void setStreamlinesLength(const DataSetId& id, double length);
824
825    void setStreamlinesNumberOfSeedPoints(const DataSetId& id, int numPoints);
826
827    void setStreamlinesSeedColor(const DataSetId& id, float color[3]);
828
829    void setStreamlinesSeedPointSize(const DataSetId& id, float size);
830
831    void setStreamlinesSeedToMeshPoints(const DataSetId& id,
832                                        int maxPoints = 500);
833
834    void setStreamlinesSeedToFilledMesh(const DataSetId& id, int numPoints);
835
836    bool setStreamlinesSeedToMeshPoints(const DataSetId& id,
837                                        char *data, size_t nbytes,
838                                        int maxPoints = 500);
839
840    bool setStreamlinesSeedToFilledMesh(const DataSetId& id,
841                                        char *data, size_t nbytes,
842                                        int numPoints);
843
844    void setStreamlinesSeedToRake(const DataSetId& id,
845                                  double start[3], double end[3],
846                                  int numPoints);
847
848    void setStreamlinesSeedToDisk(const DataSetId& id,
849                                  double center[3], double normal[3],
850                                  double radius, double innerRadius,
851                                  int numPoints);
852
853    void setStreamlinesSeedToPolygon(const DataSetId& id,
854                                     double center[3], double normal[3],
855                                     double angle, double radius,
856                                     int numSides);
857
858    void setStreamlinesSeedToFilledPolygon(const DataSetId& id,
859                                           double center[3], double normal[3],
860                                           double angle, double radius,
861                                           int numSides, int numPoints);
862
863    void setStreamlinesSeedVisibility(const DataSetId& id, bool state);
864
865    void setStreamlinesTerminalSpeed(const DataSetId& id, double speed);
866
867    void setStreamlinesTypeToLines(const DataSetId& id);
868
869    void setStreamlinesTypeToTubes(const DataSetId& id, int numSides, double radius);
870
871    void setStreamlinesTypeToRibbons(const DataSetId& id, double width, double angle);
872
873    // Volumes
874
875    void setVolumeSampleDistance(const DataSetId& id, double distance);
876
877    // Warps
878
879    void setWarpCloudStyle(const DataSetId& id,
880                           Warp::CloudStyle style);
881
882    void setWarpColorMode(const DataSetId& id,
883                          Warp::ColorMode mode,
884                          const char *name, double range[2] = NULL);
885
886    void setWarpColorMode(const DataSetId& id,
887                          Warp::ColorMode mode,
888                          DataSet::DataAttributeType type,
889                          const char *name, double range[2] = NULL);
890
891
892    void setWarpWarpScale(const DataSetId& id, double scale);
893
894private:
895    typedef std::tr1::unordered_map<DataSetId, DataSet *> DataSetHashmap;
896    typedef std::tr1::unordered_map<FieldId, double *> FieldRangeHashmap;
897    typedef std::tr1::unordered_map<ColorMapId, ColorMap *> ColorMapHashmap;
898
899    typedef std::tr1::unordered_map<DataSetId, Arc *> ArcHashmap;
900    typedef std::tr1::unordered_map<DataSetId, Arrow *> ArrowHashmap;
901    typedef std::tr1::unordered_map<DataSetId, Box *> BoxHashmap;
902    typedef std::tr1::unordered_map<DataSetId, Cone *> ConeHashmap;
903    typedef std::tr1::unordered_map<DataSetId, Contour2D *> Contour2DHashmap;
904    typedef std::tr1::unordered_map<DataSetId, Contour3D *> Contour3DHashmap;
905    typedef std::tr1::unordered_map<DataSetId, Cutplane *> CutplaneHashmap;
906    typedef std::tr1::unordered_map<DataSetId, Cylinder *> CylinderHashmap;
907    typedef std::tr1::unordered_map<DataSetId, Disk *> DiskHashmap;
908    typedef std::tr1::unordered_map<DataSetId, Glyphs *> GlyphsHashmap;
909    typedef std::tr1::unordered_map<DataSetId, Group *> GroupHashmap;
910    typedef std::tr1::unordered_map<DataSetId, HeightMap *> HeightMapHashmap;
911    typedef std::tr1::unordered_map<DataSetId, LIC *> LICHashmap;
912    typedef std::tr1::unordered_map<DataSetId, Line *> LineHashmap;
913    typedef std::tr1::unordered_map<DataSetId, Molecule *> MoleculeHashmap;
914    typedef std::tr1::unordered_map<DataSetId, Outline *> OutlineHashmap;
915    typedef std::tr1::unordered_map<DataSetId, PolyData *> PolyDataHashmap;
916    typedef std::tr1::unordered_map<DataSetId, Polygon *> PolygonHashmap;
917    typedef std::tr1::unordered_map<DataSetId, PseudoColor *> PseudoColorHashmap;
918    typedef std::tr1::unordered_map<DataSetId, Sphere *> SphereHashmap;
919    typedef std::tr1::unordered_map<DataSetId, Streamlines *> StreamlinesHashmap;
920    typedef std::tr1::unordered_map<DataSetId, Volume *> VolumeHashmap;
921    typedef std::tr1::unordered_map<DataSetId, Warp *> WarpHashmap;
922
923    void _setCameraZoomRegionPixels(int x, int y, int width, int height);
924
925    void _setCameraZoomRegion(double x, double y, double width, double height);
926
927    //static void printCameraInfo(vtkCamera *camera);
928
929    static void setCameraFromMatrix(vtkCamera *camera, vtkMatrix4x4 &mat);
930
931    static void mergeBounds(double *boundsDest, const double *bounds1, const double *bounds2);
932
933    template<class T>
934    std::tr1::unordered_map<DataSetId, T *>& getGraphicsObjectHashmap();
935
936    void setObjectAspects(double aspectRatio);
937
938    void collectBounds(double *bounds, bool onlyVisible);
939
940    void collectUnscaledBounds(double *bounds, bool onlyVisible);
941
942    void collectDataRanges();
943
944    void collectDataRanges(double *range, const char *name,
945                           DataSet::DataAttributeType type,
946                           int component, bool onlyVisible);
947
948    void clearFieldRanges();
949
950    void clearUserFieldRanges();
951
952    void initFieldRanges();
953
954    void updateFieldRanges();
955
956    void updateColorMap(ColorMap *cmap);
957
958    bool colorMapUsed(ColorMap *cmap);
959
960    void computeDisplayToWorld(double x, double y, double z, double worldPt[4]);
961
962    void computeWorldToDisplay(double x, double y, double z, double displayPt[3]);
963
964    void computeScreenWorldCoords();
965
966    bool is2D(const double bounds[6],
967              PrincipalPlane *plane,
968              double *offset) const;
969
970    void initCamera(bool initCameraMode = false);
971
972    void sceneBoundsChanged();
973
974    void initAxes();
975
976    void resetAxes(double bounds[6] = NULL);
977
978    void setAxesBounds(double bounds[6] = NULL);
979
980    void setAxesRanges();
981
982    void computeAxesScale();
983
984    void setCameraClippingPlanes();
985
986    bool _needsRedraw;
987    bool _needsAxesReset;
988    bool _needsCameraClippingRangeReset;
989    bool _needsCameraReset;
990
991    int _windowWidth, _windowHeight;
992    CameraMode _cameraMode;
993    Aspect _cameraAspect;
994    double _imgWorldOrigin[2];
995    double _imgWorldDims[2];
996    double _imgWindowWorldDims[2];
997    double _userImgWorldOrigin[2];
998    double _userImgWorldDims[2];
999    PrincipalPlane _imgCameraPlane;
1000    double _imgCameraOffset;
1001    double _screenWorldCoords[4];
1002    double _cameraOrientation[4];
1003    double _cameraZoomRatio;
1004    double _cameraPan[2];
1005    float _bgColor[3];
1006    bool _useCumulativeRange;
1007    bool _cumulativeRangeOnlyVisible;
1008
1009    FieldRangeHashmap _scalarPointDataRange;
1010    FieldRangeHashmap _vectorPointDataRange;
1011    FieldRangeHashmap _vectorCompPointDataRange[3];
1012    FieldRangeHashmap _scalarCellDataRange;
1013    FieldRangeHashmap _vectorCellDataRange;
1014    FieldRangeHashmap _vectorCompCellDataRange[3];
1015
1016    FieldRangeHashmap _userScalarPointDataRange;
1017    FieldRangeHashmap _userVectorPointDataRange;
1018    FieldRangeHashmap _userVectorCompPointDataRange[3];
1019    FieldRangeHashmap _userScalarCellDataRange;
1020    FieldRangeHashmap _userVectorCellDataRange;
1021    FieldRangeHashmap _userVectorCompCellDataRange[3];
1022
1023    bool _axesAutoBounds[3];
1024    double _axesUserBounds[6];
1025    AxisRangeMode _axesRangeMode[3];
1026    double _axesScale[3];
1027
1028    ColorMapHashmap _colorMaps;
1029    DataSetHashmap _dataSets;
1030    ArcHashmap _arcs;
1031    ArrowHashmap _arrows;
1032    BoxHashmap _boxes;
1033    ConeHashmap _cones;
1034    Contour2DHashmap _contour2Ds;
1035    Contour3DHashmap _contour3Ds;
1036    CutplaneHashmap _cutplanes;
1037    CylinderHashmap _cylinders;
1038    DiskHashmap _disks;
1039    GlyphsHashmap _glyphs;
1040    GroupHashmap _groups;
1041    HeightMapHashmap _heightMaps;
1042    LICHashmap _lics;
1043    LineHashmap _lines;
1044    MoleculeHashmap _molecules;
1045    OutlineHashmap _outlines;
1046    PolyDataHashmap _polyDatas;
1047    PolygonHashmap _polygons;
1048    PseudoColorHashmap _pseudoColors;
1049    SphereHashmap _spheres;
1050    StreamlinesHashmap _streamlines;
1051    VolumeHashmap _volumes;
1052    WarpHashmap _warps;
1053
1054    vtkSmartPointer<vtkPlane> _cameraClipPlanes[4];
1055    vtkSmartPointer<vtkPlane> _userClipPlanes[6];
1056    vtkSmartPointer<vtkPlaneCollection> _activeClipPlanes;
1057#ifdef USE_CUSTOM_AXES
1058    vtkSmartPointer<vtkRpCubeAxesActor> _cubeAxesActor;
1059#else
1060    vtkSmartPointer<vtkCubeAxesActor> _cubeAxesActor;
1061#endif
1062    vtkSmartPointer<vtkScalarBarActor> _scalarBarActor;
1063    vtkSmartPointer<vtkRenderer> _renderer;
1064    vtkSmartPointer<vtkRenderer> _legendRenderer;
1065    vtkSmartPointer<vtkRenderWindow> _renderWindow;
1066    vtkSmartPointer<vtkRenderWindow> _legendRenderWindow;
1067};
1068
1069}
1070
1071#endif
Note: See TracBrowser for help on using the repository browser.