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

1

import * as THREE from 'three';

1✔

2

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

1✔

3

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

1✔

4

import ShaderUtils from 'Renderer/Shader/ShaderUtils';

1✔

5

import CommonMaterial from 'Renderer/CommonMaterial';

1✔

6

import Gradients from 'Utils/Gradients';

1✔

7

1✔

8

export const PNTS_MODE = {

1✔

9

    COLOR: 0,

1✔

10

    INTENSITY: 1,

1✔

11

    CLASSIFICATION: 2,

1✔

12

    ELEVATION: 3,

1✔

13

    RETURN_NUMBER: 4,

1✔

14

    RETURN_TYPE: 5,

1✔

15

    RETURN_COUNT: 6,

1✔

16

    POINT_SOURCE_ID: 7,

1✔

17

    SCAN_ANGLE: 8,

1✔

18

    NORMAL: 9,

1✔

19

};

1✔

20

1✔

21

export const PNTS_SHAPE = {

1✔

22

    CIRCLE: 0,

1✔

23

    SQUARE: 1,

1✔

24

};

1✔

25

1✔

26

export const PNTS_SIZE_MODE = {

1✔

27

    VALUE: 0,

1✔

28

    ATTENUATED: 1,

1✔

29

};

1✔

30

1✔

31

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

1✔

32

1✔

33

/**

1✔

34

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

1✔

35

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

1✔

36

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

1✔

37

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

1✔

38

 * integer codes in the files.

1✔

39

1✔

40

 * @property {object} category - category classification,

1✔

41

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

1✔

42

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

1✔

43

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

1✔

44

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

1✔

45

*/

1✔

46

// eslint-disable-next-line

1✔

47

class /* istanbul ignore next */ Classification {}

1✔

48

1✔

49

export const ClassificationScheme = {

1✔

50

    DEFAULT: {

1✔

51

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

1✔

52

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

1✔

53

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

1✔

54

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

1✔

55

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

1✔

56

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

1✔

57

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

1✔

58

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

1✔

59

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

1✔

60

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

1✔

61

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

1✔

62

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

1✔

63

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

1✔

64

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

1✔

65

},

1✔

66

};

1✔

67

1✔

68

const DiscreteScheme = {

1✔

69

    DEFAULT: {

1✔

70

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

1✔

71

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

1✔

72

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

1✔

73

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

1✔

74

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

1✔

75

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

1✔

76

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

1✔

77

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

1✔

78

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

1✔

79

},

1✔

80

};

1✔

81

1✔

82

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

1✔

83

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

1✔

84

function generateGradientTexture(gradient) {

1✔

85

    const size = 64;

1✔

86

1✔

87

    // create canvas

1✔

88

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

1✔

89

    canvas.width = size;

1✔

90

    canvas.height = size;

1✔

91

1✔

92

    // get context

1✔

93

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

1✔

94

1✔

95

    // draw gradient

1✔

96

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

1✔

97

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

1✔

98

1✔

99

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

1✔

100

        const step = gradient[i];

1✔

101

1✔

102

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

1✔

103

1✔

104

1✔

105

    context.fillStyle = ctxGradient;

1✔

106

    context.fill();

1✔

107

1✔

108

    const texture = new THREE.CanvasTexture(canvas);

1✔

109

    texture.needsUpdate = true;

1✔

110

1✔

111

    texture.minFilter = THREE.LinearFilter;

1✔

112

    texture.wrap = THREE.RepeatWrapping;

1✔

113

    texture.repeat = 2;

1✔

114

1✔

115

    return texture;

1✔

116

1✔

117

1✔

118

function recomputeTexture(scheme, texture, nbClass) {

1✔

119

    let needTransparency;

1✔

120

    const data = texture.image.data;

1✔

121

    const width = texture.image.width;

1✔

122

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

1✔

123

1✔

124

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

1✔

125

        let color;

1✔

126

        let opacity;

1✔

127

        let visible = true;

1✔

128

1✔

129

        if (scheme[i]) {

1✔

130

            color = scheme[i].color;

1✔

131

            visible = scheme[i].visible;

1✔

132

            opacity = scheme[i].opacity;

1✔

133

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

1✔

134

            color = scheme[i % nbClass].color;

1✔

135

            visible = scheme[i % nbClass].visible;

1✔

136

            opacity = scheme[i % nbClass].opacity;

1✔

137

        } else if (scheme.DEFAULT) {

1✔

138

            color = scheme.DEFAULT.color;

1✔

139

            visible = scheme.DEFAULT.visible;

1✔

140

            opacity = scheme.DEFAULT.opacity;

1✔

141

        } else {

1✔

142

            color = white;

1✔

143

            opacity = 1.0;

1✔

144

1✔

145

1✔

146

        const j = 4 * i;

1✔

147

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

1✔

148

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

1✔

149

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

1✔

150

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

1✔

151

1✔

152

        needTransparency = needTransparency || opacity < 1;

1✔

153

1✔

154

    texture.needsUpdate = true;

1✔

155

    return needTransparency;

1✔

156

1✔

157

1✔

158

class PointsMaterial extends THREE.ShaderMaterial {

1✔

159

/**

1✔

160

     * @class      PointsMaterial

1✔

161

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

1✔

162

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

1✔

163

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

1✔

164

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

1✔

165

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

1✔

166

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

1✔

167

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

1✔

168

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

1✔

169

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

1✔

170

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

1✔

171

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

1✔

172

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

1✔

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.

1✔

174

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

1✔

175

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

1✔

176

1✔

177

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

1✔

178

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

1✔

179

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

1✔

180

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

1✔

181

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

1✔

182

1✔

183

     * @example

1✔

184

     * // change color category classification

1✔

185

     * const pointMaterial = new PointsMaterial();

1✔

186

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

1✔

187

     * pointMaterial.recomputeClassification();

1✔

188

*/

1✔

189

    constructor(options = {}) {

1✔

190

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

1✔

191

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

1✔

192

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

1✔

193

        const oiMaterial = options.orientedImageMaterial;

1✔

194

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

1✔

195

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

1✔

196

        const size = options.size || 0;

1✔

197

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

1✔

198

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

1✔

199

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

1✔

200

        const minAttenuatedSize = options.minAttenuatedSize || 3;

1✔

201

        const maxAttenuatedSize = options.maxAttenuatedSize || 10;

1✔

202

        let gradients = Gradients;

1✔

203

        if (options.gradient) {

1✔

204

            gradients = {

1✔

205

                ...options.gradient,

1✔

206

                ...Gradients,

1✔

207

};

1✔

208

1✔

209

1✔

210

        delete options.intensityRange;

1✔

211

        delete options.elevationRange;

1✔

212

        delete options.angleRange;

1✔

213

        delete options.orientedImageMaterial;

1✔

214

        delete options.classification;

1✔

215

        delete options.discreteScheme;

1✔

216

        delete options.size;

1✔

217

        delete options.mode;

1✔

218

        delete options.shape;

1✔

219

        delete options.sizeMode;

1✔

220

        delete options.minAttenuatedSize;

1✔

221

        delete options.maxAttenuatedSize;

1✔

222

        delete options.gradient;

1✔

223

1✔

224

        super(options);

1✔

225

        this.userData.needTransparency = {};

1✔

226

        this.gradients = gradients;

1✔

227

        this.gradientTexture = new THREE.CanvasTexture();

1✔

228

1✔

229

        this.vertexShader = PointsVS;

1✔

230

1✔

231

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

1✔

232

1✔

233

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

1✔

234

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

1✔

235

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

1✔

236

1✔

237

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

1✔

238

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

1✔

239

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

1✔

240

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

1✔

241

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

1✔

242

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

1✔

243

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

1✔

244

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

1✔

245

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

1✔

246

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

8✔

247

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

8✔

248

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

8✔

249

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

8✔

250

8✔

251

        // add classification texture to apply classification lut.

8✔

252

        const data = new Uint8Array(256 * 4);

8✔

253

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

8✔

254

        texture.needsUpdate = true;

8✔

255

        texture.magFilter = THREE.NearestFilter;

8✔

256

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

8✔

257

8✔

258

        // add texture to applying the discrete lut.

8✔

259

        const dataLUT = new Uint8Array(256 * 4);

1✔

260

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

16✔

261

        textureLUT.needsUpdate = true;

16✔

262

        textureLUT.magFilter = THREE.NearestFilter;

16✔

263

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

16✔

264

4,096✔

265

        // Classification and other discrete values scheme

4,096✔

266

        this.classificationScheme = classificationScheme;

168✔

267

        this.discreteScheme = discreteScheme;

4,096✔

268

3,928✔

269

        // Update classification and discrete Texture

3,928✔

270

        this.recomputeClassification();

2,488✔

271

        this.recomputeDiscreteTexture();

2,488✔

272

2,488✔

273

        // Gradient texture for continuous values

3,928✔

274

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

1,440✔

278

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

4,096✔

279

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

4,096✔

281

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

1✔

282

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

1✔

283

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

1✔

284

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

1✔

285

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

1✔

286

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

1✔

287

            this.defines.USE_TEXTURES_PROJECTIVE = true;

1✔

288

            this.defines.USE_BASE_MATERIAL = true;

1✔

289

            // three loop unrolling of ShaderMaterial only supports integer

1✔

290

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

1✔

291

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

1✔

292

        } else {

1✔

293

            this.fragmentShader = PointsFS;

1✔

294

1✔

295

1✔

296

        if (__DEBUG__) {

1✔

297

            this.defines.DEBUG = 1;

1✔

298

1✔

299

1✔

300

1✔

301

    recomputeClassification() {

1✔

302

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

1✔

303

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

1✔

304

        this.dispatchEvent({

1✔

305

            type: 'material_property_changed',

1✔

306

            target: this.uniforms,

1✔

307

});

1✔

308

1✔

309

1✔

310

    recomputeDiscreteTexture() {

1✔

311

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

1✔

312

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

1✔

313

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

1✔

314

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

1✔

315

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

1✔

316

        this.dispatchEvent({

1✔

317

            type: 'material_property_changed',

1✔

318

            target: this.uniforms,

1✔

319

});

1✔

320

1✔

321

1✔

322

    copy(source) {

1✔

323

        super.copy(source);

1✔

324

        if (source.uniforms.projectiveTextureAlphaBorder) {

1✔

325

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

1✔

326

            // It needs a reference to oriented image material.

1✔

327

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

1✔

328

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

1✔

329

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

8✔

330

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

8✔

331

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

8✔

332

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

8✔

333

8✔

334

        return this;

8✔

335

8✔

336

8✔

337

    enablePicking(picking) {

8✔

338

        this.picking = picking;

8✔

339

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

8✔

340

8✔

341

8✔

342

    update(source) {

8✔

343

        this.visible = source.visible;

8✔

344

        this.opacity = source.opacity;

8✔

345

        this.transparent = source.transparent;

8✔

346

        this.size = source.size;

8✔

347

        this.mode = source.mode;

8✔

349

        this.sizeMode = source.sizeMode;

8✔

350

        this.minAttenuatedSize = source.minAttenuatedSize;

8✔

351

        this.maxAttenuatedSize = source.maxAttenuatedSize;

8✔

352

        this.picking = source.picking;

8✔

355

        this.intensityRange.copy(source.intensityRange);

8✔

356

        this.elevationRange.copy(source.elevationRange);

8✔

357

        this.angleRange.copy(source.angleRange);

8✔

358

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

8✔

359

        return this;

8✔

360

8✔

361

8✔

362

    set gradient(value) {

8✔

363

        this.gradientTexture = generateGradientTexture(value);

8✔

364

8✔

365

8✔

366

8✔

367

export default PointsMaterial;

8✔

iTowns / itowns / 9175345751

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

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

iTowns / itowns / 9175345751

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