]>
Commit | Line | Data |
---|---|---|
37b09b91 | 1 | /************************************************************************** |
2 | * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * | |
3 | * * | |
4 | * Author: The ALICE Off-line Project. * | |
5 | * Contributors are mentioned in the code where appropriate. * | |
6 | * * | |
7 | * Permission to use, copy, modify and distribute this software and its * | |
8 | * documentation strictly for non-commercial purposes is hereby granted * | |
9 | * without fee, provided that the above copyright notice appears in all * | |
10 | * copies and that both the copyright notice and this permission notice * | |
11 | * appear in the supporting documentation. The authors make no claims * | |
12 | * about the suitability of this software for any purpose. It is * | |
13 | * provided "as is" without express or implied warranty. * | |
14 | **************************************************************************/ | |
15 | ||
16 | /* $Id$ */ | |
17 | ||
6196c437 | 18 | |
19 | #include "TriangleSet.h" | |
d54c094e | 20 | #include "RGBAPalette.h" |
6196c437 | 21 | |
37b09b91 | 22 | #include <TMath.h> |
6196c437 | 23 | #include <TVector3.h> |
24 | #include <TRandom3.h> | |
25 | #include <TVirtualPad.h> | |
26 | #include <TVirtualViewer3D.h> | |
27 | #include <TBuffer3D.h> | |
28 | #include <TBuffer3DTypes.h> | |
6196c437 | 29 | |
30 | using namespace Reve; | |
31 | ||
9e1efcba | 32 | //______________________________________________________________________ |
33 | // TriangleSet | |
34 | // | |
35 | // Made from a list of vertices and a list of triangles (triplets of | |
36 | // vertex indices). | |
37 | // | |
38 | // If input is composed from triangles with direct vertex coordinates | |
39 | // one should consider finding all occurences of the same vertex | |
40 | // and specifying it only once. | |
6196c437 | 41 | |
42 | ClassImp(TriangleSet) | |
43 | ||
44 | TriangleSet::TriangleSet(Int_t nv, Int_t nt, Bool_t norms, Bool_t cols) : | |
45 | RenderElement(fColor), | |
46 | TNamed("TriangleSet", 0), | |
3c67f72c | 47 | fNVerts (nv), fVerts(0), |
48 | fNTrings (nt), fTrings(0), fTringNorms(0), fTringCols(0), | |
49 | fColor (2), | |
50 | fHMTrans () | |
6196c437 | 51 | { |
52 | fVerts = new Float_t[3*fNVerts]; | |
53 | fTrings = new Int_t [3*fNTrings]; | |
54 | fTringNorms = (norms) ? new Float_t[3*fNTrings] : 0; | |
55 | fTringCols = (cols) ? new UChar_t[3*fNTrings] : 0; | |
56 | } | |
57 | ||
58 | TriangleSet::~TriangleSet() | |
59 | { | |
60 | delete [] fVerts; | |
61 | delete [] fTrings; | |
62 | delete [] fTringNorms; | |
63 | delete [] fTringCols; | |
64 | } | |
65 | ||
66 | /**************************************************************************/ | |
67 | ||
68 | void TriangleSet::GenerateTriangleNormals() | |
69 | { | |
70 | if (fTringNorms == 0) fTringNorms = new Float_t[3*fNTrings]; | |
71 | ||
72 | TVector3 e1, e2, n; | |
73 | Float_t *N = fTringNorms; | |
74 | Int_t *T = fTrings; | |
75 | for(Int_t t=0; t<fNTrings; ++t, N+=3, T+=3) | |
76 | { | |
77 | Float_t* v0 = Vertex(T[0]); | |
78 | Float_t* v1 = Vertex(T[1]); | |
79 | Float_t* v2 = Vertex(T[2]); | |
80 | e1.SetXYZ(v1[0]-v0[0], v1[1]-v0[1], v1[2]-v0[2]); | |
81 | e2.SetXYZ(v2[0]-v0[0], v2[1]-v0[1], v2[2]-v0[2]); | |
82 | n = e1.Cross(e2); | |
83 | n.SetMag(1); | |
84 | n.GetXYZ(N); | |
85 | } | |
86 | } | |
87 | ||
88 | void TriangleSet::GenerateRandomColors() | |
89 | { | |
90 | if (fTringCols == 0) fTringCols = new UChar_t[3*fNTrings]; | |
91 | ||
92 | TRandom r; | |
93 | r.SetSeed(0); | |
94 | UChar_t *C = fTringCols; | |
95 | for(Int_t t=0; t<fNTrings; ++t, C+=3) | |
96 | { | |
97 | C[0] = (UChar_t) r.Uniform(60, 255); | |
98 | C[1] = (UChar_t) r.Uniform(60, 255); | |
99 | C[2] = (UChar_t) r.Uniform(60, 255); | |
100 | } | |
101 | } | |
102 | ||
64dae94d | 103 | void TriangleSet::GenerateZNormalColors(Float_t fac, Int_t min, Int_t max, |
104 | Bool_t interp, Bool_t wrap) | |
105 | { | |
106 | if (fTringCols == 0) fTringCols = new UChar_t[3*fNTrings]; | |
107 | if (fTringNorms == 0) GenerateTriangleNormals(); | |
108 | ||
d54c094e | 109 | RGBAPalette pal(min, max, interp, wrap); |
64dae94d | 110 | UChar_t *C = fTringCols; |
111 | Float_t *N = fTringNorms; | |
112 | for(Int_t t=0; t<fNTrings; ++t, C+=3, N+=3) | |
113 | { | |
114 | Int_t v = TMath::Nint(fac * N[2]); | |
9e1efcba | 115 | pal.ColorFromValue(v, C, kFALSE); |
64dae94d | 116 | } |
117 | gPad->Modified(); gPad->Update(); | |
118 | } | |
119 | ||
6196c437 | 120 | /**************************************************************************/ |
121 | ||
122 | void TriangleSet::ComputeBBox() | |
123 | { | |
124 | if (fNVerts <= 0) { | |
125 | BBoxZero(); | |
126 | return; | |
127 | } | |
128 | ||
129 | BBoxInit(); | |
130 | Float_t* v = fVerts; | |
131 | for (Int_t i=0; i<fNVerts; ++i, v += 3) | |
132 | BBoxCheckPoint(v); | |
133 | } | |
134 | ||
135 | void TriangleSet::Paint(Option_t* ) | |
136 | { | |
137 | TBuffer3D buffer(TBuffer3DTypes::kGeneric); | |
138 | ||
139 | // Section kCore | |
140 | buffer.fID = this; | |
141 | buffer.fColor = fColor; | |
142 | buffer.fTransparency = 0; | |
143 | fHMTrans.SetBuffer3D(buffer); | |
144 | buffer.SetSectionsValid(TBuffer3D::kCore); | |
145 | ||
146 | // We fill kCore on first pass and try with viewer | |
147 | Int_t reqSections = gPad->GetViewer3D()->AddObject(buffer); | |
148 | if (reqSections == TBuffer3D::kNone) { | |
149 | return; | |
150 | } | |
151 | ||
152 | Error("TriangleSet::Paint", "only direct OpenGL rendering supported."); | |
153 | } | |
154 | ||
155 | /**************************************************************************/ | |
156 | ||
157 | #include <stdio.h> | |
158 | ||
159 | TriangleSet* TriangleSet::ReadTrivialFile(const char* file) | |
160 | { | |
161 | FILE* f = fopen(file, "r"); | |
8f40a17c | 162 | if (f == 0) { |
163 | ::Error("TriangleSet::ReadTrivialFile", Form("file '%s' not found.", file)); | |
164 | return 0; | |
165 | } | |
6196c437 | 166 | |
167 | Int_t nv, nt; | |
168 | fscanf(f, "%d %d", &nv, &nt); | |
169 | ||
170 | TriangleSet* ts = new TriangleSet(nv, nt); | |
171 | ||
172 | Float_t *V = ts->Vertex(0); | |
173 | for (Int_t i=0; i<nv; ++i) { | |
174 | fscanf(f, "%f %f %f", V++, V++, V++); | |
175 | } | |
176 | ||
177 | Int_t *T = ts->Triangle(0); | |
178 | for (Int_t i=0; i<nt; ++i) { | |
179 | fscanf(f, "%d %d %d", T++, T++, T++); | |
180 | } | |
181 | ||
182 | fclose(f); | |
183 | ||
184 | return ts; | |
185 | } |