iTowns/itowns | Build 9174695781 | src/Renderer/PointsMaterial.js | Coveralls

1

import * as THREE from 'three';

2✔

2

import PointsVS from 'Renderer/Shader/PointsVS.glsl';

2✔

3

import PointsFS from 'Renderer/Shader/PointsFS.glsl';

2✔

4

import ShaderUtils from 'Renderer/Shader/ShaderUtils';

2✔

5

import CommonMaterial from 'Renderer/CommonMaterial';

2✔

6

import Gradients from 'Utils/Gradients';

2✔

7

2✔

8

export const PNTS_MODE = {

2✔

9

    COLOR: 0,

2✔

10

    INTENSITY: 1,

2✔

11

    CLASSIFICATION: 2,

2✔

12

    ELEVATION: 3,

2✔

13

    RETURN_NUMBER: 4,

2✔

14

    RETURN_TYPE: 5,

2✔

15

    RETURN_COUNT: 6,

2✔

16

    POINT_SOURCE_ID: 7,

2✔

17

    SCAN_ANGLE: 8,

2✔

18

    NORMAL: 9,

2✔

19

};

2✔

20

2✔

21

export const PNTS_SHAPE = {

2✔

22

    CIRCLE: 0,

2✔

23

    SQUARE: 1,

2✔

24

};

2✔

25

2✔

26

export const PNTS_SIZE_MODE = {

2✔

27

    VALUE: 0,

2✔

28

    ATTENUATED: 1,

2✔

29

};

2✔

30

2✔

31

const white = new THREE.Color(1.0,  1.0,  1.0);

2✔

32

2✔

33

/**

2✔

34

 * Every lidar point can have a classification assigned to it that defines

2✔

35

 * the type of object that has reflected the laser pulse. Lidar points can be

2✔

36

 * classified into a number of categories including bare earth or ground,

2✔

37

 * top of canopy, and water. The different classes are defined using numeric

2✔

38

 * integer codes in the files.

2✔

39

2✔

40

 * @property {object} category - category classification,

2✔

41

 * @property {boolean} category.visible - category visibility,

2✔

42

 * @property {string} category.name - category name,

2✔

43

 * @property {THREE.Color} category.color - category color,

2✔

44

 * @property {number} category.opacity - category opacity,

2✔

45

*/

2✔

46

// eslint-disable-next-line

2✔

47

class /* istanbul ignore next */ Classification {}

2✔

48

2✔

49

export const ClassificationScheme = {

2✔

50

    DEFAULT: {

2✔

51

        0: { visible: true, name: 'never classified', color: new THREE.Color(0.5,  0.5,  0.5), opacity: 1.0 },

2✔

52

        1: { visible: true, name: 'unclassified', color: new THREE.Color(0.5,  0.5,  0.5), opacity: 1.0 },

2✔

53

        2: { visible: true, name: 'ground', color: new THREE.Color(0.63, 0.32, 0.18), opacity: 1.0 },

2✔

54

        3: { visible: true, name: 'low vegetation', color: new THREE.Color(0.0,  1.0,  0.0), opacity: 1.0 },

2✔

55

        4: { visible: true, name: 'medium vegetation', color: new THREE.Color(0.0,  0.8,  0.0), opacity: 1.0 },

2✔

56

        5: { visible: true, name: 'high vegetation', color: new THREE.Color(0.0,  0.6,  0.0), opacity: 1.0 },

2✔

57

        6: { visible: true, name: 'building', color: new THREE.Color(1.0,  0.66, 0.0), opacity: 1.0 },

2✔

58

        7: { visible: true, name: 'low point(noise)', color: new THREE.Color(1.0,  0.0,  1.0), opacity: 1.0 },

2✔

59

        8: { visible: true, name: 'key-point', color: new THREE.Color(1.0,  0.0,  0.0), opacity: 1.0 },

2✔

60

        9: { visible: true, name: 'water', color: new THREE.Color(0.0,  0.0,  1.0), opacity: 1.0 },

2✔

61

        10: { visible: true, name: 'rail', color: new THREE.Color(0.8,  0.8,  1.0), opacity: 1.0 },

2✔

62

        11: { visible: true, name: 'road Surface', color: new THREE.Color(0.4,  0.4,  0.7), opacity: 1.0 },

2✔

63

        12: { visible: true, name: 'overlap', color: new THREE.Color(1.0,  1.0,  0.0), opacity: 1.0 },

2✔

64

        DEFAULT: { visible: true, name: 'default', color: new THREE.Color(0.3, 0.6, 0.6), opacity: 1.0 },

2✔

65

},

2✔

66

};

2✔

67

2✔

68

const DiscreteScheme = {

2✔

69

    DEFAULT: {

2✔

70

        0: { visible: true, name: '0', color: new THREE.Color('rgb(67, 99, 216)'), opacity: 1.0 },

2✔

71

        1: { visible: true, name: '1', color: new THREE.Color('rgb(60, 180, 75);'), opacity: 1.0 },

2✔

72

        2: { visible: true, name: '2', color: new THREE.Color('rgb(255, 255, 25)'), opacity: 1.0 },

2✔

73

        3: { visible: true, name: '3', color: new THREE.Color('rgb(145, 30, 180)'), opacity: 1.0 },

2✔

74

        4: { visible: true, name: '4', color: new THREE.Color('rgb(245, 130, 49)'), opacity: 1.0 },

2✔

75

        5: { visible: true, name: '5', color: new THREE.Color('rgb(230, 25, 75)'), opacity: 1.0 },

2✔

76

        6: { visible: true, name: '6', color: new THREE.Color('rgb(66, 212, 244)'), opacity: 1.0 },

2✔

77

        7: { visible: true, name: '7', color: new THREE.Color('rgb(240, 50, 230)'), opacity: 1.0 },

2✔

78

        DEFAULT: { visible: true, name: 'default', color: white, opacity: 1.0 },

2✔

79

},

2✔

80

};

2✔

81

2✔

82

// Taken from Potree. Copyright (c) 2011-2020, Markus Schütz All rights reserved.

2✔

83

// https://github.com/potree/potree/blob/develop/src/materials/PointCloudMaterial.js

2✔

84

function generateGradientTexture(gradient) {

2✔

85

    const size = 64;

2✔

86

2✔

87

    // create canvas

2✔

88

    const canvas = document.createElement('canvas');

2✔

89

    canvas.width = size;

2✔

90

    canvas.height = size;

2✔

91

2✔

92

    // get context

2✔

93

    const context = canvas.getContext('2d');

2✔

94

2✔

95

    // draw gradient

2✔

96

    context.rect(0, 0, size, size);

2✔

97

    const ctxGradient = context.createLinearGradient(0, 0, size, size);

2✔

98

2✔

99

    for (let i = 0; i < gradient.length; i++) {

2✔

100

        const step = gradient[i];

2✔

101

2✔

102

        ctxGradient.addColorStop(step[0], `#${step[1].getHexString()}`);

2✔

103

2✔

104

2✔

105

    context.fillStyle = ctxGradient;

2✔

106

    context.fill();

2✔

107

2✔

108

    const texture = new THREE.CanvasTexture(canvas);

2✔

109

    texture.needsUpdate = true;

2✔

110

2✔

111

    texture.minFilter = THREE.LinearFilter;

2✔

112

    texture.wrap = THREE.RepeatWrapping;

2✔

113

    texture.repeat = 2;

2✔

114

2✔

115

    return texture;

2✔

116

2✔

117

2✔

118

function recomputeTexture(scheme, texture, nbClass) {

2✔

119

    let needTransparency;

2✔

120

    const data = texture.image.data;

2✔

121

    const width = texture.image.width;

2✔

122

    if (!nbClass) { nbClass = Object.keys(scheme).length; }

2✔

123

2✔

124

    for (let i = 0; i < width; i++) {

2✔

125

        let color;

2✔

126

        let opacity;

2✔

127

        let visible = true;

2✔

128

2✔

129

        if (scheme[i]) {

2✔

130

            color = scheme[i].color;

2✔

131

            visible = scheme[i].visible;

2✔

132

            opacity = scheme[i].opacity;

2✔

133

        } else if (scheme[i % nbClass]) {

2✔

134

            color = scheme[i % nbClass].color;

2✔

135

            visible = scheme[i % nbClass].visible;

2✔

136

            opacity = scheme[i % nbClass].opacity;

2✔

137

        } else if (scheme.DEFAULT) {

2✔

138

            color = scheme.DEFAULT.color;

2✔

139

            visible = scheme.DEFAULT.visible;

2✔

140

            opacity = scheme.DEFAULT.opacity;

2✔

141

        } else {

2✔

142

            color = white;

2✔

143

            opacity = 1.0;

2✔

144

2✔

145

2✔

146

        const j = 4 * i;

2✔

147

        data[j + 0] = parseInt(255 * color.r, 10);

2✔

148

        data[j + 1] = parseInt(255 * color.g, 10);

2✔

149

        data[j + 2] = parseInt(255 * color.b, 10);

2✔

150

        data[j + 3] = visible ? parseInt(255 * opacity, 10) : 0;

2✔

151

2✔

152

        needTransparency = needTransparency || opacity < 1;

2✔

153

2✔

154

    texture.needsUpdate = true;

2✔

155

    return needTransparency;

2✔

156

2✔

157

2✔

158

class PointsMaterial extends THREE.ShaderMaterial {

2✔

159

/**

2✔

160

     * @class      PointsMaterial

2✔

161

     * @param      {object}  [options={}]  The options

2✔

162

     * @param      {number}  [options.size=0]  size point

2✔

163

     * @param      {number}  [options.mode=PNTS_MODE.COLOR]  display mode.

2✔

164

     * @param      {number}  [options.mode=PNTS_SHAPE.CIRCLE]  rendered points shape.

2✔

165

     * @param      {THREE.Vector4}  [options.overlayColor=new THREE.Vector4(0, 0, 0, 0)]  overlay color.

2✔

166

     * @param      {THREE.Vector2}  [options.intensityRange=new THREE.Vector2(1, 65536)]  intensity range.

2✔

167

     * @param      {THREE.Vector2}  [options.elevationRange=new THREE.Vector2(0, 1000)]  elevation range.

2✔

168

     * @param      {THREE.Vector2}  [options.angleRange=new THREE.Vector2(-90, 90)]  scan angle range.

2✔

169

     * @param      {Scheme}  [options.classification]  LUT for point classification colorization.

2✔

170

     * @param      {Scheme}  [options.discreteScheme]  LUT for other discret point values colorization.

2✔

171

     * @param      {string}  [options.gradient]  Descrition of the gradient to use for continuous point values.

2✔

172

     *                          (Default value will be the 'SPECTRAL' gradient from Utils/Gradients)

2✔

173

     * @param      {number}  [options.sizeMode=PNTS_SIZE_MODE.VALUE]  point cloud size mode. Only 'VALUE' or 'ATTENUATED' are possible. VALUE use constant size, ATTENUATED compute size depending on distance from point to camera.

2✔

174

     * @param      {number}  [options.minAttenuatedSize=3]  minimum scale used by 'ATTENUATED' size mode

2✔

175

     * @param      {number}  [options.maxAttenuatedSize=10]  maximum scale used by 'ATTENUATED' size mode

2✔

176

2✔

177

     * @property {Scheme}  classificationScheme - Color scheme for point classification values.

2✔

178

     * @property {Scheme}  discreteScheme - Color scheme for all other discrete values.

2✔

179

     * @property {object}  gradients - Descriptions of all available gradients.

2✔

180

     * @property {object}  gradient - Description of the gradient to use for display.

2✔

181

     * @property {THREE.CanvasTexture}  gradientTexture - The texture generate from the choosen gradient.

2✔

182

2✔

183

     * @example

2✔

184

     * // change color category classification

2✔

185

     * const pointMaterial = new PointsMaterial();

2✔

186

     * pointMaterial.classification[3].color.setStyle('red');

2✔

187

     * pointMaterial.recomputeClassification();

2✔

188

*/

2✔

189

    constructor(options = {}) {

2✔

190

        const intensityRange = options.intensityRange || new THREE.Vector2(1, 65536);

2✔

191

        const elevationRange = options.elevationRange || new THREE.Vector2(0, 1000);

2✔

192

        const angleRange = options.angleRange || new THREE.Vector2(-90, 90);

2✔

193

        const oiMaterial = options.orientedImageMaterial;

2✔

194

        const classificationScheme = options.classification || ClassificationScheme.DEFAULT;

2✔

195

        const discreteScheme = options.discreteScheme || DiscreteScheme.DEFAULT;

2✔

196

        const size = options.size || 0;

2✔

197

        const mode = options.mode || PNTS_MODE.COLOR;

2✔

198

        const shape = options.shape || PNTS_SHAPE.CIRCLE;

2✔

199

        const sizeMode = size === 0 ? PNTS_SIZE_MODE.ATTENUATED : (options.sizeMode || PNTS_SIZE_MODE.VALUE);

2✔

200

        const minAttenuatedSize = options.minAttenuatedSize || 3;

2✔

201

        const maxAttenuatedSize = options.maxAttenuatedSize || 10;

2✔

202

        let gradients = Gradients;

2✔

203

        if (options.gradient) {

2✔

204

            gradients = {

2✔

205

                ...options.gradient,

2✔

206

                ...Gradients,

2✔

207

};

2✔

208

2✔

209

2✔

210

        delete options.intensityRange;

2✔

211

        delete options.elevationRange;

2✔

212

        delete options.angleRange;

2✔

213

        delete options.orientedImageMaterial;

2✔

214

        delete options.classification;

2✔

215

        delete options.discreteScheme;

2✔

216

        delete options.size;

2✔

217

        delete options.mode;

2✔

218

        delete options.shape;

2✔

219

        delete options.sizeMode;

2✔

220

        delete options.minAttenuatedSize;

2✔

221

        delete options.maxAttenuatedSize;

2✔

222

        delete options.gradient;

2✔

223

2✔

224

        super(options);

2✔

225

        this.userData.needTransparency = {};

2✔

226

        this.gradients = gradients;

2✔

227

        this.gradientTexture = new THREE.CanvasTexture();

2✔

228

2✔

229

        this.vertexShader = PointsVS;

2✔

230

2✔

231

        const scale = options.scale || 0.05 * 0.5 / Math.tan(1.0 / 2.0); // autosizing scale

2✔

232

2✔

233

        CommonMaterial.setDefineMapping(this, 'PNTS_MODE', PNTS_MODE);

2✔

234

        CommonMaterial.setDefineMapping(this, 'PNTS_SHAPE', PNTS_SHAPE);

2✔

235

        CommonMaterial.setDefineMapping(this, 'PNTS_SIZE_MODE', PNTS_SIZE_MODE);

2✔

236

2✔

237

        CommonMaterial.setUniformProperty(this, 'size', size);

2✔

238

        CommonMaterial.setUniformProperty(this, 'mode', mode);

2✔

239

        CommonMaterial.setUniformProperty(this, 'shape', shape);

2✔

240

        CommonMaterial.setUniformProperty(this, 'picking', false);

2✔

241

        CommonMaterial.setUniformProperty(this, 'opacity', this.opacity);

2✔

242

        CommonMaterial.setUniformProperty(this, 'overlayColor', options.overlayColor || new THREE.Vector4(0, 0, 0, 0));

2✔

243

        CommonMaterial.setUniformProperty(this, 'intensityRange', intensityRange);

2✔

244

        CommonMaterial.setUniformProperty(this, 'elevationRange', elevationRange);

2✔

245

        CommonMaterial.setUniformProperty(this, 'angleRange', angleRange);

2✔

246

        CommonMaterial.setUniformProperty(this, 'sizeMode', sizeMode);

16✔

247

        CommonMaterial.setUniformProperty(this, 'scale', scale);

16✔

248

        CommonMaterial.setUniformProperty(this, 'minAttenuatedSize', minAttenuatedSize);

16✔

249

        CommonMaterial.setUniformProperty(this, 'maxAttenuatedSize', maxAttenuatedSize);

16✔

250

16✔

251

        // add classification texture to apply classification lut.

16✔

252

        const data = new Uint8Array(256 * 4);

16✔

253

        const texture = new THREE.DataTexture(data, 256, 1, THREE.RGBAFormat);

16✔

254

        texture.needsUpdate = true;

16✔

255

        texture.magFilter = THREE.NearestFilter;

16✔

256

        CommonMaterial.setUniformProperty(this, 'classificationTexture', texture);

16✔

257

16✔

258

        // add texture to applying the discrete lut.

16✔

259

        const dataLUT = new Uint8Array(256 * 4);

2✔

260

        const textureLUT = new THREE.DataTexture(dataLUT, 256, 1, THREE.RGBAFormat);

32✔

261

        textureLUT.needsUpdate = true;

32✔

262

        textureLUT.magFilter = THREE.NearestFilter;

32✔

263

        CommonMaterial.setUniformProperty(this, 'discreteTexture', textureLUT);

32✔

264

8,192✔

265

        // Classification and other discrete values scheme

8,192✔

266

        this.classificationScheme = classificationScheme;

336✔

267

        this.discreteScheme = discreteScheme;

8,192✔

268

7,856✔

269

        // Update classification and discrete Texture

7,856✔

270

        this.recomputeClassification();

4,976✔

271

        this.recomputeDiscreteTexture();

4,976✔

272

4,976✔

273

        // Gradient texture for continuous values

7,856✔

274

        this.gradient = Object.values(gradients)[0];

2,880✔

278

            this.uniforms.projectiveTextureAlphaBorder = oiMaterial.uniforms.projectiveTextureAlphaBorder;

8,192✔

279

            this.uniforms.projectiveTextureDistortion = oiMaterial.uniforms.projectiveTextureDistortion;

8,192✔

281

            this.uniforms.projectiveTexture = oiMaterial.uniforms.projectiveTexture;

2✔

282

            this.uniforms.mask = oiMaterial.uniforms.mask;

2✔

283

            this.uniforms.boostLight = oiMaterial.uniforms.boostLight;

2✔

284

            this.defines.ORIENTED_IMAGES_COUNT = oiMaterial.defines.ORIENTED_IMAGES_COUNT;

2✔

285

            this.defines.USE_DISTORTION = oiMaterial.defines.USE_DISTORTION;

2✔

286

            this.defines.DEBUG_ALPHA_BORDER = oiMaterial.defines.DEBUG_ALPHA_BORDER;

2✔

287

            this.defines.USE_TEXTURES_PROJECTIVE = true;

2✔

288

            this.defines.USE_BASE_MATERIAL = true;

2✔

289

            // three loop unrolling of ShaderMaterial only supports integer

2✔

290

            // bounds, see https://github.com/mrdoob/three.js/issues/28020

2✔

291

            this.fragmentShader = ShaderUtils.unrollLoops(PointsFS, this.defines);

2✔

292

        } else {

2✔

293

            this.fragmentShader = PointsFS;

2✔

294

2✔

295

2✔

296

        if (__DEBUG__) {

2✔

297

            this.defines.DEBUG = 1;

2✔

298

2✔

299

2✔

300

2✔

301

    recomputeClassification() {

2✔

302

        const needTransparency = recomputeTexture(this.classificationScheme, this.classificationTexture, 32);

2✔

303

        this.userData.needTransparency[PNTS_MODE.CLASSIFICATION] = needTransparency;

2✔

304

        this.dispatchEvent({

2✔

305

            type: 'material_property_changed',

2✔

306

            target: this.uniforms,

2✔

307

});

2✔

308

2✔

309

2✔

310

    recomputeDiscreteTexture() {

2✔

311

        const needTransparency = recomputeTexture(this.discreteScheme, this.discreteTexture);

2✔

312

        this.userData.needTransparency[PNTS_MODE.RETURN_NUMBER] = needTransparency;

2✔

313

        this.userData.needTransparency[PNTS_MODE.RETURN_TYPE] = needTransparency;

2✔

314

        this.userData.needTransparency[PNTS_MODE.RETURN_COUNT] = needTransparency;

2✔

315

        this.userData.needTransparency[PNTS_MODE.POINT_SOURCE_ID] = needTransparency;

2✔

316

        this.dispatchEvent({

2✔

317

            type: 'material_property_changed',

2✔

318

            target: this.uniforms,

2✔

319

});

2✔

320

2✔

321

2✔

322

    copy(source) {

2✔

323

        super.copy(source);

2✔

324

        if (source.uniforms.projectiveTextureAlphaBorder) {

2✔

325

            // Don't copy oriented image because, it's a link to oriented image material.

2✔

326

            // It needs a reference to oriented image material.

2✔

327

            this.uniforms.projectiveTextureAlphaBorder = source.uniforms.projectiveTextureAlphaBorder;

2✔

328

            this.uniforms.projectiveTextureDistortion = source.uniforms.projectiveTextureDistortion;

2✔

329

            this.uniforms.projectiveTextureMatrix = source.uniforms.projectiveTextureMatrix;

16✔

330

            this.uniforms.projectiveTexture = source.uniforms.projectiveTexture;

16✔

331

            this.uniforms.mask = source.uniforms.mask;

16✔

332

            this.uniforms.boostLight = source.uniforms.boostLight;

16✔

333

16✔

334

        return this;

16✔

335

16✔

336

16✔

337

    enablePicking(picking) {

16✔

338

        this.picking = picking;

16✔

339

        this.blending = picking ? THREE.NoBlending : THREE.NormalBlending;

16✔

340

16✔

341

16✔

342

    update(source) {

16✔

343

        this.visible = source.visible;

16✔

344

        this.opacity = source.opacity;

16✔

345

        this.transparent = source.transparent;

16✔

346

        this.size = source.size;

16✔

347

        this.mode = source.mode;

16✔

349

        this.sizeMode = source.sizeMode;

16✔

350

        this.minAttenuatedSize = source.minAttenuatedSize;

16✔

351

        this.maxAttenuatedSize = source.maxAttenuatedSize;

16✔

352

        this.picking = source.picking;

16✔

355

        this.intensityRange.copy(source.intensityRange);

16✔

356

        this.elevationRange.copy(source.elevationRange);

16✔

357

        this.angleRange.copy(source.angleRange);

16✔

358

        Object.assign(this.defines, source.defines);

16✔

359

        return this;

16✔

360

16✔

361

16✔

362

    set gradient(value) {

16✔

363

        this.gradientTexture = generateGradientTexture(value);

16✔

364

16✔

365

16✔

366

16✔

367

export default PointsMaterial;

16✔

iTowns / itowns / 9174695781

Source File
Press 'n' to go to next uncovered line, 'b' for previous

1	import * as THREE from 'three';	2✔
2	import PointsVS from 'Renderer/Shader/PointsVS.glsl';	2✔
3	import PointsFS from 'Renderer/Shader/PointsFS.glsl';	2✔
4	import ShaderUtils from 'Renderer/Shader/ShaderUtils';	2✔
5	import CommonMaterial from 'Renderer/CommonMaterial';	2✔
6	import Gradients from 'Utils/Gradients';	2✔
7		2✔
8	export const PNTS_MODE = {	2✔
9	COLOR: 0,	2✔
10	INTENSITY: 1,	2✔
11	CLASSIFICATION: 2,	2✔
12	ELEVATION: 3,	2✔
13	RETURN_NUMBER: 4,	2✔
14	RETURN_TYPE: 5,	2✔
15	RETURN_COUNT: 6,	2✔
16	POINT_SOURCE_ID: 7,	2✔
17	SCAN_ANGLE: 8,	2✔
18	NORMAL: 9,	2✔
19	};	2✔
20		2✔
21	export const PNTS_SHAPE = {	2✔
22	CIRCLE: 0,	2✔
23	SQUARE: 1,	2✔
24	};	2✔
25		2✔
26	export const PNTS_SIZE_MODE = {	2✔
27	VALUE: 0,	2✔
28	ATTENUATED: 1,	2✔
29	};	2✔
30		2✔
31	const white = new THREE.Color(1.0, 1.0, 1.0);	2✔
32		2✔
33	/**	2✔
34	* Every lidar point can have a classification assigned to it that defines	2✔
35	* the type of object that has reflected the laser pulse. Lidar points can be	2✔
36	* classified into a number of categories including bare earth or ground,	2✔
37	* top of canopy, and water. The different classes are defined using numeric	2✔
38	* integer codes in the files.	2✔
39	*	2✔
40	* @property {object} category - category classification,	2✔
41	* @property {boolean} category.visible - category visibility,	2✔
42	* @property {string} category.name - category name,	2✔
43	* @property {THREE.Color} category.color - category color,	2✔
44	* @property {number} category.opacity - category opacity,	2✔
45	*/	2✔
46	// eslint-disable-next-line	2✔
47	class /* istanbul ignore next */ Classification {}	2✔
48		2✔
49	export const ClassificationScheme = {	2✔
50	DEFAULT: {	2✔
51	0: { visible: true, name: 'never classified', color: new THREE.Color(0.5, 0.5, 0.5), opacity: 1.0 },	2✔
52	1: { visible: true, name: 'unclassified', color: new THREE.Color(0.5, 0.5, 0.5), opacity: 1.0 },	2✔
53	2: { visible: true, name: 'ground', color: new THREE.Color(0.63, 0.32, 0.18), opacity: 1.0 },	2✔
54	3: { visible: true, name: 'low vegetation', color: new THREE.Color(0.0, 1.0, 0.0), opacity: 1.0 },	2✔
55	4: { visible: true, name: 'medium vegetation', color: new THREE.Color(0.0, 0.8, 0.0), opacity: 1.0 },	2✔
56	5: { visible: true, name: 'high vegetation', color: new THREE.Color(0.0, 0.6, 0.0), opacity: 1.0 },	2✔
57	6: { visible: true, name: 'building', color: new THREE.Color(1.0, 0.66, 0.0), opacity: 1.0 },	2✔
58	7: { visible: true, name: 'low point(noise)', color: new THREE.Color(1.0, 0.0, 1.0), opacity: 1.0 },	2✔
59	8: { visible: true, name: 'key-point', color: new THREE.Color(1.0, 0.0, 0.0), opacity: 1.0 },	2✔
60	9: { visible: true, name: 'water', color: new THREE.Color(0.0, 0.0, 1.0), opacity: 1.0 },	2✔
61	10: { visible: true, name: 'rail', color: new THREE.Color(0.8, 0.8, 1.0), opacity: 1.0 },	2✔
62	11: { visible: true, name: 'road Surface', color: new THREE.Color(0.4, 0.4, 0.7), opacity: 1.0 },	2✔
63	12: { visible: true, name: 'overlap', color: new THREE.Color(1.0, 1.0, 0.0), opacity: 1.0 },	2✔
64	DEFAULT: { visible: true, name: 'default', color: new THREE.Color(0.3, 0.6, 0.6), opacity: 1.0 },	2✔
65	},	2✔
66	};	2✔
67		2✔
68	const DiscreteScheme = {	2✔
69	DEFAULT: {	2✔
70	0: { visible: true, name: '0', color: new THREE.Color('rgb(67, 99, 216)'), opacity: 1.0 },	2✔
71	1: { visible: true, name: '1', color: new THREE.Color('rgb(60, 180, 75);'), opacity: 1.0 },	2✔
72	2: { visible: true, name: '2', color: new THREE.Color('rgb(255, 255, 25)'), opacity: 1.0 },	2✔
73	3: { visible: true, name: '3', color: new THREE.Color('rgb(145, 30, 180)'), opacity: 1.0 },	2✔
74	4: { visible: true, name: '4', color: new THREE.Color('rgb(245, 130, 49)'), opacity: 1.0 },	2✔
75	5: { visible: true, name: '5', color: new THREE.Color('rgb(230, 25, 75)'), opacity: 1.0 },	2✔
76	6: { visible: true, name: '6', color: new THREE.Color('rgb(66, 212, 244)'), opacity: 1.0 },	2✔
77	7: { visible: true, name: '7', color: new THREE.Color('rgb(240, 50, 230)'), opacity: 1.0 },	2✔
78	DEFAULT: { visible: true, name: 'default', color: white, opacity: 1.0 },	2✔
79	},	2✔
80	};	2✔
81		2✔
82	// Taken from Potree. Copyright (c) 2011-2020, Markus Schütz All rights reserved.	2✔
83	// https://github.com/potree/potree/blob/develop/src/materials/PointCloudMaterial.js	2✔
84	function generateGradientTexture(gradient) {	2✔
85	const size = 64;	2✔
86		2✔
87	// create canvas	2✔
88	const canvas = document.createElement('canvas');	2✔
89	canvas.width = size;	2✔
90	canvas.height = size;	2✔
91		2✔
92	// get context	2✔
93	const context = canvas.getContext('2d');	2✔
94		2✔
95	// draw gradient	2✔
96	context.rect(0, 0, size, size);	2✔
97	const ctxGradient = context.createLinearGradient(0, 0, size, size);	2✔
98		2✔
99	for (let i = 0; i < gradient.length; i++) {	2✔
100	const step = gradient[i];	2✔
101		2✔
102	ctxGradient.addColorStop(step[0], `#${step[1].getHexString()}`);	2✔
103	}	2✔
104		2✔
105	context.fillStyle = ctxGradient;	2✔
106	context.fill();	2✔
107		2✔
108	const texture = new THREE.CanvasTexture(canvas);	2✔
109	texture.needsUpdate = true;	2✔
110		2✔
111	texture.minFilter = THREE.LinearFilter;	2✔
112	texture.wrap = THREE.RepeatWrapping;	2✔
113	texture.repeat = 2;	2✔
114		2✔
115	return texture;	2✔
116	}	2✔
117		2✔
118	function recomputeTexture(scheme, texture, nbClass) {	2✔
119	let needTransparency;	2✔
120	const data = texture.image.data;	2✔
121	const width = texture.image.width;	2✔
122	if (!nbClass) { nbClass = Object.keys(scheme).length; }	2✔
123		2✔
124	for (let i = 0; i < width; i++) {	2✔
125	let color;	2✔
126	let opacity;	2✔
127	let visible = true;	2✔
128		2✔
129	if (scheme[i]) {	2✔
130	color = scheme[i].color;	2✔
131	visible = scheme[i].visible;	2✔
132	opacity = scheme[i].opacity;	2✔
133	} else if (scheme[i % nbClass]) {	2✔
134	color = scheme[i % nbClass].color;	2✔
135	visible = scheme[i % nbClass].visible;	2✔
136	opacity = scheme[i % nbClass].opacity;	2✔
137	} else if (scheme.DEFAULT) {	2✔
138	color = scheme.DEFAULT.color;	2✔
139	visible = scheme.DEFAULT.visible;	2✔
140	opacity = scheme.DEFAULT.opacity;	2✔
141	} else {	2✔
142	color = white;	2✔
143	opacity = 1.0;	2✔
144	}	2✔
145		2✔
146	const j = 4 * i;	2✔
147	data[j + 0] = parseInt(255 * color.r, 10);	2✔
148	data[j + 1] = parseInt(255 * color.g, 10);	2✔
149	data[j + 2] = parseInt(255 * color.b, 10);	2✔
150	data[j + 3] = visible ? parseInt(255 * opacity, 10) : 0;	2✔
151		2✔
152	needTransparency = needTransparency \|\| opacity < 1;	2✔
153	}	2✔
154	texture.needsUpdate = true;	2✔
155	return needTransparency;	2✔
156	}	2✔
157		2✔
158	class PointsMaterial extends THREE.ShaderMaterial {	2✔
159	/**	2✔
160	* @class PointsMaterial	2✔
161	* @param {object} [options={}] The options	2✔
162	* @param {number} [options.size=0] size point	2✔
163	* @param {number} [options.mode=PNTS_MODE.COLOR] display mode.	2✔
164	* @param {number} [options.mode=PNTS_SHAPE.CIRCLE] rendered points shape.	2✔
165	* @param {THREE.Vector4} [options.overlayColor=new THREE.Vector4(0, 0, 0, 0)] overlay color.	2✔
166	* @param {THREE.Vector2} [options.intensityRange=new THREE.Vector2(1, 65536)] intensity range.	2✔
167	* @param {THREE.Vector2} [options.elevationRange=new THREE.Vector2(0, 1000)] elevation range.	2✔
168	* @param {THREE.Vector2} [options.angleRange=new THREE.Vector2(-90, 90)] scan angle range.	2✔
169	* @param {Scheme} [options.classification] LUT for point classification colorization.	2✔
170	* @param {Scheme} [options.discreteScheme] LUT for other discret point values colorization.	2✔
171	* @param {string} [options.gradient] Descrition of the gradient to use for continuous point values.	2✔
172	* (Default value will be the 'SPECTRAL' gradient from Utils/Gradients)	2✔
173	* @param {number} [options.sizeMode=PNTS_SIZE_MODE.VALUE] point cloud size mode. Only 'VALUE' or 'ATTENUATED' are possible. VALUE use constant size, ATTENUATED compute size depending on distance from point to camera.	2✔
174	* @param {number} [options.minAttenuatedSize=3] minimum scale used by 'ATTENUATED' size mode	2✔
175	* @param {number} [options.maxAttenuatedSize=10] maximum scale used by 'ATTENUATED' size mode	2✔
176	*	2✔
177	* @property {Scheme} classificationScheme - Color scheme for point classification values.	2✔
178	* @property {Scheme} discreteScheme - Color scheme for all other discrete values.	2✔
179	* @property {object} gradients - Descriptions of all available gradients.	2✔
180	* @property {object} gradient - Description of the gradient to use for display.	2✔
181	* @property {THREE.CanvasTexture} gradientTexture - The texture generate from the choosen gradient.	2✔
182	*	2✔
183	* @example	2✔
184	* // change color category classification	2✔
185	* const pointMaterial = new PointsMaterial();	2✔
186	* pointMaterial.classification[3].color.setStyle('red');	2✔
187	* pointMaterial.recomputeClassification();	2✔
188	*/	2✔
189	constructor(options = {}) {	2✔
190	const intensityRange = options.intensityRange \|\| new THREE.Vector2(1, 65536);	2✔
191	const elevationRange = options.elevationRange \|\| new THREE.Vector2(0, 1000);	2✔
192	const angleRange = options.angleRange \|\| new THREE.Vector2(-90, 90);	2✔
193	const oiMaterial = options.orientedImageMaterial;	2✔
194	const classificationScheme = options.classification \|\| ClassificationScheme.DEFAULT;	2✔
195	const discreteScheme = options.discreteScheme \|\| DiscreteScheme.DEFAULT;	2✔
196	const size = options.size \|\| 0;	2✔
197	const mode = options.mode \|\| PNTS_MODE.COLOR;	2✔
198	const shape = options.shape \|\| PNTS_SHAPE.CIRCLE;	2✔
199	const sizeMode = size === 0 ? PNTS_SIZE_MODE.ATTENUATED : (options.sizeMode \|\| PNTS_SIZE_MODE.VALUE);	2✔
200	const minAttenuatedSize = options.minAttenuatedSize \|\| 3;	2✔
201	const maxAttenuatedSize = options.maxAttenuatedSize \|\| 10;	2✔
202	let gradients = Gradients;	2✔
203	if (options.gradient) {	2✔
204	gradients = {	2✔
205	...options.gradient,	2✔
206	...Gradients,	2✔
207	};	2✔
208	}	2✔
209		2✔
210	delete options.intensityRange;	2✔
211	delete options.elevationRange;	2✔
212	delete options.angleRange;	2✔
213	delete options.orientedImageMaterial;	2✔
214	delete options.classification;	2✔
215	delete options.discreteScheme;	2✔
216	delete options.size;	2✔
217	delete options.mode;	2✔
218	delete options.shape;	2✔
219	delete options.sizeMode;	2✔
220	delete options.minAttenuatedSize;	2✔
221	delete options.maxAttenuatedSize;	2✔
222	delete options.gradient;	2✔
223		2✔
224	super(options);	2✔
225	this.userData.needTransparency = {};	2✔
226	this.gradients = gradients;	2✔
227	this.gradientTexture = new THREE.CanvasTexture();	2✔
228		2✔
229	this.vertexShader = PointsVS;	2✔
230		2✔
231	const scale = options.scale \|\| 0.05 * 0.5 / Math.tan(1.0 / 2.0); // autosizing scale	2✔
232		2✔
233	CommonMaterial.setDefineMapping(this, 'PNTS_MODE', PNTS_MODE);	2✔
234	CommonMaterial.setDefineMapping(this, 'PNTS_SHAPE', PNTS_SHAPE);	2✔
235	CommonMaterial.setDefineMapping(this, 'PNTS_SIZE_MODE', PNTS_SIZE_MODE);	2✔
236		2✔
237	CommonMaterial.setUniformProperty(this, 'size', size);	2✔
238	CommonMaterial.setUniformProperty(this, 'mode', mode);	2✔
239	CommonMaterial.setUniformProperty(this, 'shape', shape);	2✔
240	CommonMaterial.setUniformProperty(this, 'picking', false);	2✔
241	CommonMaterial.setUniformProperty(this, 'opacity', this.opacity);	2✔
242	CommonMaterial.setUniformProperty(this, 'overlayColor', options.overlayColor \|\| new THREE.Vector4(0, 0, 0, 0));	2✔
243	CommonMaterial.setUniformProperty(this, 'intensityRange', intensityRange);	2✔
244	CommonMaterial.setUniformProperty(this, 'elevationRange', elevationRange);	2✔
245	CommonMaterial.setUniformProperty(this, 'angleRange', angleRange);	2✔
246	CommonMaterial.setUniformProperty(this, 'sizeMode', sizeMode);	16✔
247	CommonMaterial.setUniformProperty(this, 'scale', scale);	16✔
248	CommonMaterial.setUniformProperty(this, 'minAttenuatedSize', minAttenuatedSize);	16✔
249	CommonMaterial.setUniformProperty(this, 'maxAttenuatedSize', maxAttenuatedSize);	16✔
250		16✔
251	// add classification texture to apply classification lut.	16✔
252	const data = new Uint8Array(256 * 4);	16✔
253	const texture = new THREE.DataTexture(data, 256, 1, THREE.RGBAFormat);	16✔
254	texture.needsUpdate = true;	16✔
255	texture.magFilter = THREE.NearestFilter;	16✔
256	CommonMaterial.setUniformProperty(this, 'classificationTexture', texture);	16✔
257		16✔
258	// add texture to applying the discrete lut.	16✔
259	const dataLUT = new Uint8Array(256 * 4);	2✔
260	const textureLUT = new THREE.DataTexture(dataLUT, 256, 1, THREE.RGBAFormat);	32✔
261	textureLUT.needsUpdate = true;	32✔
262	textureLUT.magFilter = THREE.NearestFilter;	32✔
263	CommonMaterial.setUniformProperty(this, 'discreteTexture', textureLUT);	32✔
264		8,192✔
265	// Classification and other discrete values scheme	8,192✔
266	this.classificationScheme = classificationScheme;	336✔
267	this.discreteScheme = discreteScheme;	8,192✔
268		7,856✔
269	// Update classification and discrete Texture	7,856✔
270	this.recomputeClassification();	4,976✔
271	this.recomputeDiscreteTexture();	4,976✔
272		4,976✔
273	// Gradient texture for continuous values	7,856✔
274	this.gradient = Object.values(gradients)[0];	2,880✔
275	CommonMaterial.setUniformProperty(this, 'gradientTexture', this.gradientTexture);	2,880!
UNCOV 276		×
UNCOV 277	if (oiMaterial) {	×
278	this.uniforms.projectiveTextureAlphaBorder = oiMaterial.uniforms.projectiveTextureAlphaBorder;	8,192✔
279	this.uniforms.projectiveTextureDistortion = oiMaterial.uniforms.projectiveTextureDistortion;	8,192✔
280	this.uniforms.projectiveTextureMatrix = oiMaterial.uniforms.projectiveTextureMatrix;	32!
281	this.uniforms.projectiveTexture = oiMaterial.uniforms.projectiveTexture;	2✔
282	this.uniforms.mask = oiMaterial.uniforms.mask;	2✔
283	this.uniforms.boostLight = oiMaterial.uniforms.boostLight;	2✔
284	this.defines.ORIENTED_IMAGES_COUNT = oiMaterial.defines.ORIENTED_IMAGES_COUNT;	2✔
285	this.defines.USE_DISTORTION = oiMaterial.defines.USE_DISTORTION;	2✔
286	this.defines.DEBUG_ALPHA_BORDER = oiMaterial.defines.DEBUG_ALPHA_BORDER;	2✔
287	this.defines.USE_TEXTURES_PROJECTIVE = true;	2✔
288	this.defines.USE_BASE_MATERIAL = true;	2✔
289	// three loop unrolling of ShaderMaterial only supports integer	2✔
290	// bounds, see https://github.com/mrdoob/three.js/issues/28020	2✔
291	this.fragmentShader = ShaderUtils.unrollLoops(PointsFS, this.defines);	2✔
292	} else {	2✔
293	this.fragmentShader = PointsFS;	2✔
294	}	2✔
295		2✔
296	if (__DEBUG__) {	2✔
297	this.defines.DEBUG = 1;	2✔
298	}	2✔
299	}	2✔
300		2✔
301	recomputeClassification() {	2✔
302	const needTransparency = recomputeTexture(this.classificationScheme, this.classificationTexture, 32);	2✔
303	this.userData.needTransparency[PNTS_MODE.CLASSIFICATION] = needTransparency;	2✔
304	this.dispatchEvent({	2✔
305	type: 'material_property_changed',	2✔
306	target: this.uniforms,	2✔
307	});	2✔
308	}	2✔
309		2✔
310	recomputeDiscreteTexture() {	2✔
311	const needTransparency = recomputeTexture(this.discreteScheme, this.discreteTexture);	2✔
312	this.userData.needTransparency[PNTS_MODE.RETURN_NUMBER] = needTransparency;	2✔
313	this.userData.needTransparency[PNTS_MODE.RETURN_TYPE] = needTransparency;	2✔
314	this.userData.needTransparency[PNTS_MODE.RETURN_COUNT] = needTransparency;	2✔
315	this.userData.needTransparency[PNTS_MODE.POINT_SOURCE_ID] = needTransparency;	2✔
316	this.dispatchEvent({	2✔
317	type: 'material_property_changed',	2✔
318	target: this.uniforms,	2✔
319	});	2✔
320	}	2✔
321		2✔
322	copy(source) {	2✔
323	super.copy(source);	2✔
324	if (source.uniforms.projectiveTextureAlphaBorder) {	2✔
325	// Don't copy oriented image because, it's a link to oriented image material.	2✔
326	// It needs a reference to oriented image material.	2✔
327	this.uniforms.projectiveTextureAlphaBorder = source.uniforms.projectiveTextureAlphaBorder;	2✔
328	this.uniforms.projectiveTextureDistortion = source.uniforms.projectiveTextureDistortion;	2✔
329	this.uniforms.projectiveTextureMatrix = source.uniforms.projectiveTextureMatrix;	16✔
330	this.uniforms.projectiveTexture = source.uniforms.projectiveTexture;	16✔
331	this.uniforms.mask = source.uniforms.mask;	16✔
332	this.uniforms.boostLight = source.uniforms.boostLight;	16✔
333	}	16✔
334	return this;	16✔
335	}	16✔
336		16✔
337	enablePicking(picking) {	16✔
338	this.picking = picking;	16✔
339	this.blending = picking ? THREE.NoBlending : THREE.NormalBlending;	16✔
340	}	16✔
341		16✔
342	update(source) {	16✔
343	this.visible = source.visible;	16✔
344	this.opacity = source.opacity;	16✔
345	this.transparent = source.transparent;	16✔
346	this.size = source.size;	16✔
347	this.mode = source.mode;	16✔
348	this.shape = source.shape;	16!
349	this.sizeMode = source.sizeMode;	16✔
350	this.minAttenuatedSize = source.minAttenuatedSize;	16✔
351	this.maxAttenuatedSize = source.maxAttenuatedSize;	16✔
352	this.picking = source.picking;	16✔
353	this.scale = source.scale;	16!
UNCOV 354	this.overlayColor.copy(source.overlayColor);	×
355	this.intensityRange.copy(source.intensityRange);	16✔
356	this.elevationRange.copy(source.elevationRange);	16✔
357	this.angleRange.copy(source.angleRange);	16✔
358	Object.assign(this.defines, source.defines);	16✔
359	return this;	16✔
360	}	16✔
361		16✔
362	set gradient(value) {	16✔
363	this.gradientTexture = generateGradientTexture(value);	16✔
364	}	16✔
365	}	16✔
366		16✔
367	export default PointsMaterial;	16✔

iTowns / itowns / 9174695781

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous