iTowns/itowns | Build 9065871326 | 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

    const data = texture.image.data;

1✔

120

    const width = texture.image.width;

1✔

121

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

1✔

122

1✔

123

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

1✔

124

        let color;

1✔

125

        let opacity;

1✔

126

        let visible = true;

1✔

127

1✔

128

        if (scheme[i]) {

1✔

129

            color = scheme[i].color;

1✔

130

            visible = scheme[i].visible;

1✔

131

            opacity = scheme[i].opacity;

1✔

132

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

1✔

133

            color = scheme[i % nbClass].color;

1✔

134

            visible = scheme[i % nbClass].visible;

1✔

135

            opacity = scheme[i % nbClass].opacity;

1✔

136

        } else if (scheme.DEFAULT) {

1✔

137

            color = scheme.DEFAULT.color;

1✔

138

            visible = scheme.DEFAULT.visible;

1✔

139

            opacity = scheme.DEFAULT.opacity;

1✔

140

        } else {

1✔

141

            color = white;

1✔

142

            opacity = 1.0;

1✔

143

1✔

144

1✔

145

        const j = 4 * i;

1✔

146

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

1✔

147

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

1✔

148

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

1✔

149

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

1✔

150

1✔

151

        texture.userData.transparent = texture.userData.transparent || opacity < 1;

1✔

152

1✔

153

    texture.needsUpdate = true;

1✔

154

1✔

155

1✔

156

function setUniformProperty(object, property, initValue) {

1✔

157

    object.uniforms[property] = new THREE.Uniform(initValue);

1✔

158

    Object.defineProperty(object, property, {

1✔

159

        get: () => object.uniforms[property].value,

1✔

160

        set: (value) => {

1✔

161

            switch (value) {

1✔

162

                case '2':

1✔

163

                    object.transparent = object.uniforms.classificationTexture.value.userData.transparent;

1✔

164

                    break;

1✔

165

                case '4':

1✔

166

                case '5':

1✔

167

                case '6':

1✔

168

                case '7':

1✔

169

                    object.transparent = object.uniforms.discreteTexture.value.userData.transparent;

1✔

170

                    break;

1✔

171

                default:

1✔

172

                    object.transparent = false;

1✔

173

1✔

174

            if (object.uniforms[property].value != value) {

1✔

175

                object.uniforms[property].value = value;

1✔

176

1✔

177

},

1✔

178

});

1✔

179

1✔

180

1✔

181

class PointsMaterial extends THREE.ShaderMaterial {

1✔

182

/**

1✔

183

     * @class      PointsMaterial

1✔

184

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

1✔

185

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

1✔

186

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

1✔

187

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

1✔

188

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

1✔

189

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

1✔

190

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

1✔

191

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

1✔

192

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

1✔

193

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

1✔

194

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

1✔

195

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

1✔

196

     * @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✔

197

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

1✔

198

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

1✔

199

1✔

200

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

1✔

201

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

1✔

202

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

1✔

203

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

1✔

204

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

1✔

205

1✔

206

     * @example

1✔

207

     * // change color category classification

1✔

208

     * const pointMaterial = new PointsMaterial();

1✔

209

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

1✔

210

     * pointMaterial.recomputeClassification();

1✔

211

*/

1✔

212

    constructor(options = {}) {

1✔

213

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

1✔

214

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

1✔

215

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

1✔

216

        const oiMaterial = options.orientedImageMaterial;

1✔

217

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

1✔

218

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

1✔

219

        const size = options.size || 0;

1✔

220

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

1✔

221

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

1✔

222

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

1✔

223

        const minAttenuatedSize = options.minAttenuatedSize || 3;

1✔

224

        const maxAttenuatedSize = options.maxAttenuatedSize || 10;

1✔

225

        let gradients = Gradients;

1✔

226

        if (options.gradient) {

1✔

227

            gradients = {

1✔

228

                ...options.gradient,

1✔

229

                ...Gradients,

1✔

230

};

1✔

231

1✔

232

1✔

233

        delete options.intensityRange;

1✔

234

        delete options.elevationRange;

1✔

235

        delete options.angleRange;

1✔

236

        delete options.orientedImageMaterial;

1✔

237

        delete options.classification;

1✔

238

        delete options.discreteScheme;

1✔

239

        delete options.size;

1✔

240

        delete options.mode;

1✔

241

        delete options.shape;

1✔

242

        delete options.sizeMode;

1✔

243

        delete options.minAttenuatedSize;

1✔

244

        delete options.maxAttenuatedSize;

1✔

245

        delete options.gradient;

1✔

246

1✔

247

        super(options);

1✔

248

        this.gradients = gradients;

1✔

249

        this.gradientTexture = new THREE.CanvasTexture();

1✔

250

1✔

251

        this.vertexShader = PointsVS;

1✔

252

1✔

253

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

1✔

254

1✔

255

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

1✔

256

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

1✔

257

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

8✔

258

8✔

259

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

8✔

260

        setUniformProperty(this, 'mode', mode);

8✔

261

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

8✔

262

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

8✔

263

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

8✔

264

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

8✔

265

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

8✔

266

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

8✔

267

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

1✔

268

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

16✔

269

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

16✔

270

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

16✔

271

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

4,096✔

272

168✔

273

        // add classification texture to apply classification lut.

168✔

274

        const data = new Uint8Array(256 * 4);

4,096✔

275

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

2,488✔

276

        texture.needsUpdate = true;

2,488✔

277

        texture.magFilter = THREE.NearestFilter;

3,928✔

278

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

1,440✔

279

1,440✔

281

        const dataLUT = new Uint8Array(256 * 4);

4,096✔

282

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

4,096✔

283

        textureLUT.needsUpdate = true;

4,096✔

284

        textureLUT.magFilter = THREE.NearestFilter;

4,096✔

286

4,096✔

287

        // Classification and other discrete values scheme

16✔

288

        this.classificationScheme = classificationScheme;

1✔

289

        this.discreteScheme = discreteScheme;

8✔

290

8✔

291

        // Update classification and discrete Texture

8✔

292

        this.recomputeClassification();

8✔

293

        this.recomputeDiscreteTexture();

8✔

294

1✔

298

1✔

301

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

1✔

303

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

1✔

304

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

1✔

305

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

1✔

306

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

1✔

307

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

1✔

308

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

1✔

309

            this.defines.USE_TEXTURES_PROJECTIVE = true;

1✔

310

            this.defines.USE_BASE_MATERIAL = true;

1✔

311

            // three loop unrolling of ShaderMaterial only supports integer

1✔

312

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

1✔

313

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

1✔

314

        } else {

1✔

315

            this.fragmentShader = PointsFS;

1✔

316

1✔

317

1✔

318

        if (__DEBUG__) {

1✔

319

            this.defines.DEBUG = 1;

1✔

320

1✔

321

1✔

322

1✔

323

    recomputeClassification() {

1✔

324

        recomputeTexture(this.classificationScheme, this.classificationTexture, 32);

1✔

325

        this.dispatchEvent({

1✔

326

            type: 'material_property_changed',

1✔

327

            target: this.uniforms,

1✔

328

});

1✔

329

1✔

330

1✔

331

    recomputeDiscreteTexture() {

1✔

332

        recomputeTexture(this.discreteScheme, this.discreteTexture);

1✔

333

        this.dispatchEvent({

1✔

334

            type: 'material_property_changed',

1✔

335

            target: this.uniforms,

1✔

336

});

1✔

337

1✔

338

1✔

339

    copy(source) {

1✔

340

        super.copy(source);

1✔

341

        if (source.uniforms.projectiveTextureAlphaBorder) {

1✔

342

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

1✔

343

            // It needs a reference to oriented image material.

1✔

344

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

1✔

345

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

1✔

346

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

1✔

347

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

1✔

348

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

1✔

349

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

1✔

350

1✔

351

        return this;

1✔

352

1✔

353

1✔

354

    enablePicking(picking) {

1✔

355

        this.picking = picking;

1✔

356

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

8✔

357

8✔

358

1✔

359

    update(source) {

8✔

360

        this.visible = source.visible;

8✔

361

        this.opacity = source.opacity;

8✔

362

        this.transparent = source.transparent;

8✔

363

        this.size = source.size;

8✔

364

        this.mode = source.mode;

8✔

365

        this.shape = source.shape;

8✔

366

        this.sizeMode = source.sizeMode;

8✔

367

        this.minAttenuatedSize = source.minAttenuatedSize;

8✔

368

        this.maxAttenuatedSize = source.maxAttenuatedSize;

8✔

369

        this.picking = source.picking;

8✔

370

        this.scale = source.scale;

8✔

371

        this.overlayColor.copy(source.overlayColor);

8✔

372

        this.intensityRange.copy(source.intensityRange);

8✔

373

        this.elevationRange.copy(source.elevationRange);

8✔

375

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

8✔

376

        return this;

8✔

377

8✔

378

8✔

379

    set gradient(value) {

8✔

380

        this.gradientTexture = generateGradientTexture(value);

8✔

381

8✔

382

8✔

383

8✔

iTowns / itowns / 9065871326

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	const data = texture.image.data;	1✔
120	const width = texture.image.width;	1✔
121	if (!nbClass) { nbClass = Object.keys(scheme).length; }	1✔
122		1✔
123	for (let i = 0; i < width; i++) {	1✔
124	let color;	1✔
125	let opacity;	1✔
126	let visible = true;	1✔
127		1✔
128	if (scheme[i]) {	1✔
129	color = scheme[i].color;	1✔
130	visible = scheme[i].visible;	1✔
131	opacity = scheme[i].opacity;	1✔
132	} else if (scheme[i % nbClass]) {	1✔
133	color = scheme[i % nbClass].color;	1✔
134	visible = scheme[i % nbClass].visible;	1✔
135	opacity = scheme[i % nbClass].opacity;	1✔
136	} else if (scheme.DEFAULT) {	1✔
137	color = scheme.DEFAULT.color;	1✔
138	visible = scheme.DEFAULT.visible;	1✔
139	opacity = scheme.DEFAULT.opacity;	1✔
140	} else {	1✔
141	color = white;	1✔
142	opacity = 1.0;	1✔
143	}	1✔
144		1✔
145	const j = 4 * i;	1✔
146	data[j + 0] = parseInt(255 * color.r, 10);	1✔
147	data[j + 1] = parseInt(255 * color.g, 10);	1✔
148	data[j + 2] = parseInt(255 * color.b, 10);	1✔
149	data[j + 3] = visible ? parseInt(255 * opacity, 10) : 0;	1✔
150		1✔
151	texture.userData.transparent = texture.userData.transparent \|\| opacity < 1;	1✔
152	}	1✔
153	texture.needsUpdate = true;	1✔
154	}	1✔
155		1✔
156	function setUniformProperty(object, property, initValue) {	1✔
157	object.uniforms[property] = new THREE.Uniform(initValue);	1✔
158	Object.defineProperty(object, property, {	1✔
159	get: () => object.uniforms[property].value,	1✔
160	set: (value) => {	1✔
161	switch (value) {	1✔
162	case '2':	1✔
163	object.transparent = object.uniforms.classificationTexture.value.userData.transparent;	1✔
164	break;	1✔
165	case '4':	1✔
166	case '5':	1✔
167	case '6':	1✔
168	case '7':	1✔
169	object.transparent = object.uniforms.discreteTexture.value.userData.transparent;	1✔
170	break;	1✔
171	default:	1✔
172	object.transparent = false;	1✔
173	}	1✔
174	if (object.uniforms[property].value != value) {	1✔
175	object.uniforms[property].value = value;	1✔
176	}	1✔
177	},	1✔
178	});	1✔
179	}	1✔
180		1✔
181	class PointsMaterial extends THREE.ShaderMaterial {	1✔
182	/**	1✔
183	* @class PointsMaterial	1✔
184	* @param {object} [options={}] The options	1✔
185	* @param {number} [options.size=0] size point	1✔
186	* @param {number} [options.mode=PNTS_MODE.COLOR] display mode.	1✔
187	* @param {number} [options.mode=PNTS_SHAPE.CIRCLE] rendered points shape.	1✔
188	* @param {THREE.Vector4} [options.overlayColor=new THREE.Vector4(0, 0, 0, 0)] overlay color.	1✔
189	* @param {THREE.Vector2} [options.intensityRange=new THREE.Vector2(1, 65536)] intensity range.	1✔
190	* @param {THREE.Vector2} [options.elevationRange=new THREE.Vector2(0, 1000)] elevation range.	1✔
191	* @param {THREE.Vector2} [options.angleRange=new THREE.Vector2(-90, 90)] scan angle range.	1✔
192	* @param {Scheme} [options.classification] LUT for point classification colorization.	1✔
193	* @param {Scheme} [options.discreteScheme] LUT for other discret point values colorization.	1✔
194	* @param {string} [options.gradient] Descrition of the gradient to use for continuous point values.	1✔
195	* (Default value will be the 'SPECTRAL' gradient from Utils/Gradients)	1✔
196	* @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✔
197	* @param {number} [options.minAttenuatedSize=3] minimum scale used by 'ATTENUATED' size mode	1✔
198	* @param {number} [options.maxAttenuatedSize=10] maximum scale used by 'ATTENUATED' size mode	1✔
199	*	1✔
200	* @property {Scheme} classificationScheme - Color scheme for point classification values.	1✔
201	* @property {Scheme} discreteScheme - Color scheme for all other discrete values.	1✔
202	* @property {object} gradients - Descriptions of all available gradients.	1✔
203	* @property {object} gradient - Description of the gradient to use for display.	1✔
204	* @property {THREE.CanvasTexture} gradientTexture - The texture generate from the choosen gradient.	1✔
205	*	1✔
206	* @example	1✔
207	* // change color category classification	1✔
208	* const pointMaterial = new PointsMaterial();	1✔
209	* pointMaterial.classification[3].color.setStyle('red');	1✔
210	* pointMaterial.recomputeClassification();	1✔
211	*/	1✔
212	constructor(options = {}) {	1✔
213	const intensityRange = options.intensityRange \|\| new THREE.Vector2(1, 65536);	1✔
214	const elevationRange = options.elevationRange \|\| new THREE.Vector2(0, 1000);	1✔
215	const angleRange = options.angleRange \|\| new THREE.Vector2(-90, 90);	1✔
216	const oiMaterial = options.orientedImageMaterial;	1✔
217	const classificationScheme = options.classification \|\| ClassificationScheme.DEFAULT;	1✔
218	const discreteScheme = options.discreteScheme \|\| DiscreteScheme.DEFAULT;	1✔
219	const size = options.size \|\| 0;	1✔
220	const mode = options.mode \|\| PNTS_MODE.COLOR;	1✔
221	const shape = options.shape \|\| PNTS_SHAPE.CIRCLE;	1✔
222	const sizeMode = size === 0 ? PNTS_SIZE_MODE.ATTENUATED : (options.sizeMode \|\| PNTS_SIZE_MODE.VALUE);	1✔
223	const minAttenuatedSize = options.minAttenuatedSize \|\| 3;	1✔
224	const maxAttenuatedSize = options.maxAttenuatedSize \|\| 10;	1✔
225	let gradients = Gradients;	1✔
226	if (options.gradient) {	1✔
227	gradients = {	1✔
228	...options.gradient,	1✔
229	...Gradients,	1✔
230	};	1✔
231	}	1✔
232		1✔
233	delete options.intensityRange;	1✔
234	delete options.elevationRange;	1✔
235	delete options.angleRange;	1✔
236	delete options.orientedImageMaterial;	1✔
237	delete options.classification;	1✔
238	delete options.discreteScheme;	1✔
239	delete options.size;	1✔
240	delete options.mode;	1✔
241	delete options.shape;	1✔
242	delete options.sizeMode;	1✔
243	delete options.minAttenuatedSize;	1✔
244	delete options.maxAttenuatedSize;	1✔
245	delete options.gradient;	1✔
246		1✔
247	super(options);	1✔
248	this.gradients = gradients;	1✔
249	this.gradientTexture = new THREE.CanvasTexture();	1✔
250		1✔
251	this.vertexShader = PointsVS;	1✔
252		1✔
253	const scale = options.scale \|\| 0.05 * 0.5 / Math.tan(1.0 / 2.0); // autosizing scale	1✔
254		1✔
255	CommonMaterial.setDefineMapping(this, 'PNTS_MODE', PNTS_MODE);	1✔
256	CommonMaterial.setDefineMapping(this, 'PNTS_SHAPE', PNTS_SHAPE);	1✔
257	CommonMaterial.setDefineMapping(this, 'PNTS_SIZE_MODE', PNTS_SIZE_MODE);	8✔
258		8✔
259	CommonMaterial.setUniformProperty(this, 'size', size);	8✔
260	setUniformProperty(this, 'mode', mode);	8✔
261	CommonMaterial.setUniformProperty(this, 'shape', shape);	8✔
262	CommonMaterial.setUniformProperty(this, 'picking', false);	8✔
263	CommonMaterial.setUniformProperty(this, 'opacity', this.opacity);	8✔
264	CommonMaterial.setUniformProperty(this, 'overlayColor', options.overlayColor \|\| new THREE.Vector4(0, 0, 0, 0));	8✔
265	CommonMaterial.setUniformProperty(this, 'intensityRange', intensityRange);	8✔
266	CommonMaterial.setUniformProperty(this, 'elevationRange', elevationRange);	8✔
267	CommonMaterial.setUniformProperty(this, 'angleRange', angleRange);	1✔
268	CommonMaterial.setUniformProperty(this, 'sizeMode', sizeMode);	16✔
269	CommonMaterial.setUniformProperty(this, 'scale', scale);	16✔
270	CommonMaterial.setUniformProperty(this, 'minAttenuatedSize', minAttenuatedSize);	16✔
271	CommonMaterial.setUniformProperty(this, 'maxAttenuatedSize', maxAttenuatedSize);	4,096✔
272		168✔
273	// add classification texture to apply classification lut.	168✔
274	const data = new Uint8Array(256 * 4);	4,096✔
275	const texture = new THREE.DataTexture(data, 256, 1, THREE.RGBAFormat);	2,488✔
276	texture.needsUpdate = true;	2,488✔
277	texture.magFilter = THREE.NearestFilter;	3,928✔
278	CommonMaterial.setUniformProperty(this, 'classificationTexture', texture);	1,440✔
279		1,440✔
280	// add texture to applying the discrete lut.	1,440!
281	const dataLUT = new Uint8Array(256 * 4);	4,096✔
282	const textureLUT = new THREE.DataTexture(dataLUT, 256, 1, THREE.RGBAFormat);	4,096✔
283	textureLUT.needsUpdate = true;	4,096✔
284	textureLUT.magFilter = THREE.NearestFilter;	4,096✔
285	CommonMaterial.setUniformProperty(this, 'discreteTexture', textureLUT);	4,096!
286		4,096✔
287	// Classification and other discrete values scheme	16✔
288	this.classificationScheme = classificationScheme;	1✔
289	this.discreteScheme = discreteScheme;	8✔
290		8✔
291	// Update classification and discrete Texture	8✔
292	this.recomputeClassification();	8✔
293	this.recomputeDiscreteTexture();	8✔
294		1✔
295	// Gradient texture for continuous values	1!
UNCOV 296	this.gradient = Object.values(gradients)[0];	×
297	CommonMaterial.setUniformProperty(this, 'gradientTexture', this.gradientTexture);	1!
298		1✔
299	if (oiMaterial) {	1!
UNCOV 300	this.uniforms.projectiveTextureAlphaBorder = oiMaterial.uniforms.projectiveTextureAlphaBorder;	×
301	this.uniforms.projectiveTextureDistortion = oiMaterial.uniforms.projectiveTextureDistortion;	1✔
302	this.uniforms.projectiveTextureMatrix = oiMaterial.uniforms.projectiveTextureMatrix;	1!
303	this.uniforms.projectiveTexture = oiMaterial.uniforms.projectiveTexture;	1✔
304	this.uniforms.mask = oiMaterial.uniforms.mask;	1✔
305	this.uniforms.boostLight = oiMaterial.uniforms.boostLight;	1✔
306	this.defines.ORIENTED_IMAGES_COUNT = oiMaterial.defines.ORIENTED_IMAGES_COUNT;	1✔
307	this.defines.USE_DISTORTION = oiMaterial.defines.USE_DISTORTION;	1✔
308	this.defines.DEBUG_ALPHA_BORDER = oiMaterial.defines.DEBUG_ALPHA_BORDER;	1✔
309	this.defines.USE_TEXTURES_PROJECTIVE = true;	1✔
310	this.defines.USE_BASE_MATERIAL = true;	1✔
311	// three loop unrolling of ShaderMaterial only supports integer	1✔
312	// bounds, see https://github.com/mrdoob/three.js/issues/28020	1✔
313	this.fragmentShader = ShaderUtils.unrollLoops(PointsFS, this.defines);	1✔
314	} else {	1✔
315	this.fragmentShader = PointsFS;	1✔
316	}	1✔
317		1✔
318	if (__DEBUG__) {	1✔
319	this.defines.DEBUG = 1;	1✔
320	}	1✔
321	}	1✔
322		1✔
323	recomputeClassification() {	1✔
324	recomputeTexture(this.classificationScheme, this.classificationTexture, 32);	1✔
325	this.dispatchEvent({	1✔
326	type: 'material_property_changed',	1✔
327	target: this.uniforms,	1✔
328	});	1✔
329	}	1✔
330		1✔
331	recomputeDiscreteTexture() {	1✔
332	recomputeTexture(this.discreteScheme, this.discreteTexture);	1✔
333	this.dispatchEvent({	1✔
334	type: 'material_property_changed',	1✔
335	target: this.uniforms,	1✔
336	});	1✔
337	}	1✔
338		1✔
339	copy(source) {	1✔
340	super.copy(source);	1✔
341	if (source.uniforms.projectiveTextureAlphaBorder) {	1✔
342	// Don't copy oriented image because, it's a link to oriented image material.	1✔
343	// It needs a reference to oriented image material.	1✔
344	this.uniforms.projectiveTextureAlphaBorder = source.uniforms.projectiveTextureAlphaBorder;	1✔
345	this.uniforms.projectiveTextureDistortion = source.uniforms.projectiveTextureDistortion;	1✔
346	this.uniforms.projectiveTextureMatrix = source.uniforms.projectiveTextureMatrix;	1✔
347	this.uniforms.projectiveTexture = source.uniforms.projectiveTexture;	1✔
348	this.uniforms.mask = source.uniforms.mask;	1✔
349	this.uniforms.boostLight = source.uniforms.boostLight;	1✔
350	}	1✔
351	return this;	1✔
352	}	1✔
353		1✔
354	enablePicking(picking) {	1✔
355	this.picking = picking;	1✔
356	this.blending = picking ? THREE.NoBlending : THREE.NormalBlending;	8✔
357	}	8✔
358		1✔
359	update(source) {	8✔
360	this.visible = source.visible;	8✔
361	this.opacity = source.opacity;	8✔
362	this.transparent = source.transparent;	8✔
363	this.size = source.size;	8✔
364	this.mode = source.mode;	8✔
365	this.shape = source.shape;	8✔
366	this.sizeMode = source.sizeMode;	8✔
367	this.minAttenuatedSize = source.minAttenuatedSize;	8✔
368	this.maxAttenuatedSize = source.maxAttenuatedSize;	8✔
369	this.picking = source.picking;	8✔
370	this.scale = source.scale;	8✔
371	this.overlayColor.copy(source.overlayColor);	8✔
372	this.intensityRange.copy(source.intensityRange);	8✔
373	this.elevationRange.copy(source.elevationRange);	8✔
374	this.angleRange.copy(source.angleRange);	8!
375	Object.assign(this.defines, source.defines);	8✔
376	return this;	8✔
377	}	8✔
378		8✔
379	set gradient(value) {	8✔
380	this.gradientTexture = generateGradientTexture(value);	8✔
381	}	8✔
382	}	8✔
383		8✔
384	export default PointsMaterial;	8!

iTowns / itowns / 9065871326

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