vtkModel.js 34 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084
  1. import * as d3 from "d3-scale";
  2. import { formatDefaultLocale } from "d3-format";
  3. import "@kitware/vtk.js/Rendering/Profiles/Geometry";
  4. import vtkFullScreenRenderWindow from "@kitware/vtk.js/Rendering/Misc/FullScreenRenderWindow";
  5. import vtkGenericRenderWindow from '@kitware/vtk.js/Rendering/Misc/GenericRenderWindow';
  6. import vtkActor from "@kitware/vtk.js/Rendering/Core/Actor";
  7. import vtkMapper from "@kitware/vtk.js/Rendering/Core/Mapper";
  8. import vtkScalarBarActor from "@kitware/vtk.js/Rendering/Core/ScalarBarActor";
  9. import vtkColorTransferFunction from "@kitware/vtk.js/Rendering/Core/ColorTransferFunction";
  10. import vtkLookupTable from '@kitware/vtk.js/Common/Core/LookupTable';
  11. import { Representation } from "@kitware/vtk.js/Rendering/Core/Property/Constants";
  12. import vtkPolyData from "@kitware/vtk.js/Common/DataModel/PolyData";
  13. import vtkLight from "@kitware/vtk.js/Rendering/Core/Light";
  14. import { throttle } from '@kitware/vtk.js/macros';
  15. import {
  16. FieldDataTypes,
  17. FieldAssociations,
  18. } from '@kitware/vtk.js/Common/DataModel/DataSet/Constants';
  19. import vtkSphereSource from '@kitware/vtk.js/Filters/Sources/SphereSource';
  20. import vtkLineSource from '@kitware/vtk.js/Filters/Sources/LineSource';
  21. import vtkSphereMapper from '@kitware/vtk.js/Rendering/Core/SphereMapper';
  22. import vtk2DShape from '@kitware/vtk.js/Filters/Sources/Arrow2DSource';
  23. import vtkArrowSource from '@kitware/vtk.js/Filters/Sources/ArrowSource';
  24. import vtkArrow2DSource from '@kitware/vtk.js/Filters/Sources/Arrow2DSource';
  25. import vtkConeSource from '@kitware/vtk.js/Filters/Sources/ConeSource';
  26. import vtkAxesActor from '@kitware/vtk.js/Rendering/Core/AxesActor';
  27. import vtkAnnotatedCubeActor from '@kitware/vtk.js/Rendering/Core/AnnotatedCubeActor';
  28. import vtkOrientationMarkerWidget from '@kitware/vtk.js/Interaction/Widgets/OrientationMarkerWidget';
  29. import vtkCamera from '@kitware/vtk.js/Rendering/Core/Camera';
  30. import vtkPLYReader from '@kitware/vtk.js/IO/Geometry/PLYReader';
  31. import vtkTexture from '@kitware/vtk.js/Rendering/Core/Texture';
  32. import vtkMatrixBuilder from '@kitware/vtk.js/Common/Core/MatrixBuilder';
  33. import axios from 'axios';
  34. export class VtkModel {
  35. constructor() {
  36. //当前显示 模版 还是结果
  37. this.isJg = false;
  38. /**
  39. * 选择点线
  40. */
  41. this.isSelectNode = false;
  42. /**
  43. * 选择管道
  44. */
  45. this.isSelectPipe = false;
  46. //管道节点选择对象
  47. this.selectObj = null;
  48. //X,Y
  49. this.epageX=null;
  50. this.epageY=null;
  51. //结果选择的点数据
  52. this.selectJgPointId = 0;
  53. this.validNodes = [];//节点数据
  54. this.pipes = [];//管道数据
  55. this.renderWindowWith = null;
  56. this.renderer = null;
  57. this.renderWindow = null;
  58. //选取
  59. this.lastActor = null;
  60. this.nodeActors = []; //节点actor 数据
  61. this.pipeActors = []; //管道actor 数据
  62. this.interactor = null;
  63. this.apiSpecificRenderWindow = null;
  64. this.hardwareSelector = null;
  65. //模板数据
  66. this.polyData = vtkPolyData.newInstance();
  67. this.mapper = vtkMapper.newInstance();
  68. this.actor = vtkActor.newInstance();
  69. this.actor.getProperty().setRepresentation(Representation.SURFACE); //线
  70. this.actor.getProperty().setColor(WHITE);
  71. this.mapper.setInputData(this.polyData);
  72. this.actor.setMapper(this.mapper);
  73. //点数据
  74. this.actor2 = vtkActor.newInstance();
  75. this.actor2.getProperty().setRepresentation(Representation.POINTS); //点
  76. this.actor2.getProperty().setColor(WHITE);
  77. this.actor2.getProperty().setPointSize(2);
  78. this.actor2.setMapper(this.mapper);
  79. //结果数据渲染
  80. this.jgActor = vtkActor.newInstance();
  81. this.jgMapper = vtkMapper.newInstance();
  82. this.jgActor.getProperty().setRepresentation(Representation.SURFACE); //面
  83. this.scalarBarActor = vtkScalarBarActor.newInstance();
  84. this.scalarBarActor.setGenerateTicks(this.generateTicks(10));
  85. this.scalarBarActor.setDrawAboveRangeSwatch(true);
  86. this.scalarBarActor.setDrawNanAnnotation(false);
  87. this.scalarBarActor.setDrawAboveRangeSwatch(false);
  88. this.scalarBarActor.setBoxPosition([1, 0]);
  89. // 修改设条颜色
  90. const ctf = vtkColorTransferFunction.newInstance();
  91. ctf.addRGBPoint(0.0,0, 1, 154/255.0);
  92. ctf.addRGBPoint(1.0, 1.0, 165.0/255.0, 0.0);
  93. ctf.addRGBPoint(2.0, 230/255.0, 0.0, 92.0/225.0);
  94. // ctf.addRGBPoint(0.0, 0.0, 0.0, 1.0);
  95. // ctf.addRGBPoint(1.0, 0.0, 1.0, 0.5);
  96. // ctf.addRGBPoint(2.0, 0.0, 1.0, 0.0);
  97. // ctf.addRGBPoint(3.0, 1.0, 0.5, 0.0);
  98. // ctf.addRGBPoint(4.0, 1.0, 0.0, 0.0);
  99. this.jgMapper.setLookupTable(ctf);
  100. this.jgActor.setMapper(this.jgMapper);
  101. const lut = this.jgMapper.getLookupTable();
  102. this.scalarBarActor.setScalarsToColors(lut);
  103. //告警数据渲染
  104. this.gjActor = vtkActor.newInstance();
  105. this.gjMapper = vtkMapper.newInstance();
  106. this.gjActor.getProperty().setRepresentation(Representation.SURFACE); //面
  107. this.gjscalarBarActor = vtkScalarBarActor.newInstance();
  108. this.gjscalarBarActor.setGenerateTicks(this.generateTicks(2));
  109. this.gjscalarBarActor.setDrawAboveRangeSwatch(true);
  110. this.gjscalarBarActor.setDrawNanAnnotation(false);
  111. this.gjscalarBarActor.setDrawAboveRangeSwatch(false);
  112. this.gjscalarBarActor.setBoxPosition([1, 0]);
  113. this.gjMapper.setLookupTable(vtkLookupTable.newInstance());
  114. const gjlut = this.gjMapper.getLookupTable();
  115. gjlut.setNumberOfColors(parseInt(2, 10));
  116. // hueRange: [0.0, 0.66667], //色调范围
  117. // saturationRange: [1.0, 1.0],//饱和度
  118. // valueRange: [1.0, 1.0],
  119. // alphaRange: [1.0, 1.0], //透明度
  120. gjlut.setAlphaRange([1,1]);
  121. gjlut.setValueRange([0.5,1]);
  122. // gjlut.setHueRange([0.66667,0.0]);//蓝色到红色
  123. gjlut.setHueRange([0.3375,0.0]);//绿色到红色
  124. // this.gjscalarBarActor.setScalarsToColors(gjlut);
  125. this.gjMapper.setInterpolateScalarsBeforeMapping(true);//有无渐变
  126. this.gjActor.setMapper(this.gjMapper);
  127. this.ways = [];
  128. this.wayActors = [];
  129. //管道位置数据
  130. const sphereSource = vtkSphereSource.newInstance({
  131. center: [0, 0, 0],
  132. radius: 4.0,
  133. });
  134. const pipsiteMapper = vtkMapper.newInstance();
  135. this.actorpipsite = vtkActor.newInstance();
  136. pipsiteMapper.setInputConnection(sphereSource.getOutputPort());
  137. this.actorpipsite.setMapper(pipsiteMapper);
  138. // this.actorpipsite.setPosition(parseFloat(node.x), parseFloat(node.y), parseFloat(node.z));
  139. this.actorpipsite.getProperty().setColor(WHITE);
  140. this.jcdActors = []; //检测点数据
  141. this.jcreader = vtkPLYReader.newInstance();
  142. fetchJSFileAsArrayBuffer('src/assets/3d/jiancd.ply').then(arrayBuffer => {
  143. // 读取PLY数据
  144. this.jcreader.parseAsArrayBuffer(arrayBuffer);
  145. });
  146. this.pnameActors =[];
  147. this.soureActors =[];
  148. this.soureReader = vtkPLYReader.newInstance();
  149. fetchJSFileAsArrayBuffer('src/assets/3d/fire.ply').then(arrayBuffer => {
  150. // 读取PLY数据
  151. this.soureReader.parseAsArrayBuffer(arrayBuffer);
  152. });
  153. }
  154. xyz_back(){
  155. let cam =this.renderer.getActiveCamera();
  156. cam.setViewUp([0,1,0]);
  157. cam.setPosition(495544.15625, 5403554.25, 9989.55990464236);//相机位置;
  158. this.renderer.resetCamera();
  159. this.renderWindow.render()
  160. }
  161. x_up(){
  162. let cam =this.renderer.getActiveCamera();
  163. cam.setViewUp([0,0,1]);
  164. cam.setPosition(495544.15625+5000, 5403554.25, 467.62999725341797);//相机位置;
  165. this.renderer.resetCamera();
  166. this.renderWindow.render();
  167. }
  168. x_down(){
  169. let cam =this.renderer.getActiveCamera();
  170. cam.setViewUp([0,0,1]);
  171. cam.setPosition(495544.15625+-5000, 5403554.25, 467.62999725341797);//相机位置
  172. this.renderer.resetCamera();
  173. this.renderWindow.render();
  174. }
  175. y_up(){
  176. let cam =this.renderer.getActiveCamera();
  177. cam.setViewUp([0,0,1]);
  178. cam.setPosition(495544.15625, 5403554.25+5000, 467.62999725341797);//相机位置
  179. this.renderer.resetCamera();
  180. this.renderWindow.render();
  181. }
  182. y_down(){
  183. let cam =this.renderer.getActiveCamera();
  184. cam.setViewUp([0,0,1]);
  185. cam.setPosition(495544.15625, 5403554.25-5000, 467.62999725341797);//相机位置
  186. this.renderer.resetCamera();
  187. this.renderWindow.render();
  188. }
  189. z_up(){
  190. let cam =this.renderer.getActiveCamera();
  191. cam.setViewUp([0,1,0]);
  192. cam.setPosition(495544.15625, 5403554.25, 467.62999725341797+5000);//相机位置
  193. this.renderer.resetCamera();
  194. this.renderWindow.render();
  195. }
  196. z_down(){
  197. let cam =this.renderer.getActiveCamera();
  198. cam.setViewUp([0,1,0]);
  199. cam.setPosition(495544.15625, 5403554.25, 467.62999725341797-5000);//相机位置
  200. this.renderer.resetCamera();
  201. this.renderWindow.render();
  202. }
  203. //巷道名称
  204. initPname() {
  205. fetch('src/assets/3d/pipname.json')
  206. .then(response => response.json()).then(data => {
  207. let pipnames = data;
  208. pipnames.forEach((pipname)=>{
  209. let pnameReader = vtkPLYReader.newInstance();
  210. fetchJSFileAsArrayBuffer('src/assets/3d/'+pipname.name+'.ply').then(arrayBuffer => {
  211. // 读取PLY数据
  212. pnameReader.parseAsArrayBuffer(arrayBuffer);
  213. });
  214. const mapper = vtkMapper.newInstance();
  215. mapper.setInputConnection(pnameReader.getOutputPort());
  216. const actor = vtkActor.newInstance();
  217. actor.setPosition(pipname.x,pipname.y,pipname.z);
  218. actor.getProperty().setColor(TEXTCOLOR);
  219. actor.setMapper(mapper);
  220. actor.setScale(20,20,20);//放大
  221. this.pnameActors.push(actor);
  222. // this.renderer.addActor(actor);
  223. });
  224. }).catch(error => {
  225. console.error('Error fetching JSON:', error);
  226. });
  227. }
  228. //显示巷道名称
  229. showPname(){
  230. this.pnameActors.forEach((actor)=>{
  231. this.renderer.addActor(actor);
  232. });
  233. this.renderWindow.render();
  234. }
  235. //隐藏巷道名称
  236. removePname(){
  237. this.pnameActors.forEach((actor)=>{
  238. this.renderer.removeActor(actor);
  239. });
  240. this.renderWindow.render();
  241. }
  242. //清空灾源
  243. delSoures(){
  244. this.removeSoure();
  245. this.soureActors=[];
  246. }
  247. //添加一个灾源点
  248. addSoures(pid,site){
  249. let pipe =this.pipeByPid(pid);
  250. let point1 = this.pointByPipeNodeId(pipe.snId);
  251. let point2 = this.pointByPipeNodeId(pipe.enId);
  252. let py = site/(1-site);
  253. let pipsite_point=[
  254. (point1[0]+py*point2[0])/(1+py),
  255. (point1[1]+py*point2[1])/(1+py),
  256. (point1[2]+py*point2[2])/(1+py),
  257. ]
  258. this.initSoure(pipsite_point[0],pipsite_point[1],pipsite_point[2]);
  259. }
  260. initSoure(x,y,z) {
  261. const mapper = vtkMapper.newInstance();
  262. mapper.setInputConnection(this.soureReader.getOutputPort());
  263. const actor = vtkActor.newInstance();
  264. actor.setPosition(x,y,z);
  265. actor.getProperty().setColor(SOURECOLOR);
  266. actor.setMapper(mapper);
  267. // 旋转actor
  268. actor.rotateY(45);// 绕y轴旋转
  269. actor.setScale(20,20,20);//放大
  270. this.soureActors.push(actor);
  271. // this.renderer.addActor(actor);
  272. }
  273. //显示灾源点
  274. showSoure(){
  275. this.soureActors.forEach((actor)=>{
  276. this.renderer.addActor(actor);
  277. });
  278. this.renderWindow.render();
  279. }
  280. //隐藏灾源点
  281. removeSoure(){
  282. this.soureActors.forEach((actor)=>{
  283. this.renderer.removeActor(actor);
  284. });
  285. this.renderWindow.render();
  286. }
  287. //初始化监测点
  288. initCheckNodes(checkNodes){
  289. checkNodes.forEach((checkNode) => {
  290. let pipe =this.pipeByPid(checkNode.pid);
  291. let site =checkNode.site;
  292. let point1 = this.pointByPipeNodeId(pipe.snId);
  293. let point2 = this.pointByPipeNodeId(pipe.enId);
  294. let py = site/(1-site);
  295. let pipsite_point=[
  296. (point1[0]+py*point2[0])/(1+py),
  297. (point1[1]+py*point2[1])/(1+py),
  298. (point1[2]+py*point2[2])/(1+py),
  299. ]
  300. this.initjc(pipsite_point[0],pipsite_point[1],pipsite_point[2]);
  301. });
  302. }
  303. initjc(x,y,z) {
  304. const mapper = vtkMapper.newInstance();
  305. mapper.setInputConnection(this.jcreader.getOutputPort());
  306. const actor = vtkActor.newInstance();
  307. actor.setPosition(x,y,z);
  308. actor.getProperty().setColor(WHITE);
  309. actor.setMapper(mapper);
  310. actor.setScale(5,5,5);//放大
  311. this.jcdActors.push(actor);
  312. }
  313. //显示监测点
  314. showjc(){
  315. this.jcdActors.forEach((actor)=>{
  316. this.renderer.addActor(actor);
  317. });
  318. this.renderWindow.render();
  319. }
  320. //隐藏监测点
  321. removejc(){
  322. this.jcdActors.forEach((actor)=>{
  323. this.renderer.removeActor(actor);
  324. });
  325. this.renderWindow.render();
  326. }
  327. //显示所选择巷道位置
  328. addpipsite(pipe,site){
  329. let point1 = this.pointByPipeNodeId(pipe.snId);
  330. let point2 = this.pointByPipeNodeId(pipe.enId);
  331. let py = site/(1-site);
  332. let pipsite_point=[
  333. (point1[0]+py*point2[0])/(1+py),
  334. (point1[1]+py*point2[1])/(1+py),
  335. (point1[2]+py*point2[2])/(1+py),
  336. ]
  337. this.actorpipsite.setPosition(pipsite_point);
  338. this.renderer.addActor(this.actorpipsite);
  339. this.renderWindow.render();
  340. }
  341. //移除所选择的巷道位置
  342. clearpipsite(){
  343. this.renderer.removeActor(this.actorpipsite);
  344. }
  345. waysInit(ways) {
  346. this.renderWindowWith.resize();
  347. this.ways = ways;
  348. this.ways.forEach((way) => {
  349. let nodes1 = [];
  350. let nodes2 = [];
  351. for (let index = 0; index < way.nrows.length - 1; index++) {
  352. nodes1.push(way.nrows[index]);
  353. }
  354. for (let index = 1; index < way.nrows.length; index++) {
  355. nodes2.push(way.nrows[index]);
  356. }
  357. for (let index = 0; index < nodes2.length; index++) {
  358. const lineSource = vtkLineSource.newInstance();
  359. let snode = nodes1[index];
  360. let enode = nodes2[index];
  361. let point1 = [parseFloat(snode.x), parseFloat(snode.y), parseFloat(snode.z)];
  362. let point2 = [parseFloat(enode.x), parseFloat(enode.y), parseFloat(enode.z)];
  363. lineSource.setPoint1(point1);
  364. lineSource.setPoint2(point2);
  365. lineSource.setResolution(12);
  366. const actor = vtkActor.newInstance();
  367. const mapper = vtkMapper.newInstance();
  368. actor.setMapper(mapper);
  369. actor.getProperty().setLineWidth(3);//设置线宽
  370. if(way.safety=="safety"){
  371. actor.getProperty().setColor(GREEN);
  372. }else if(way.safety=="danger"){
  373. actor.getProperty().setColor(REA);
  374. }
  375. const initialValues = {
  376. base: 0,
  377. center: point2,
  378. height: 9,
  379. radius: 5,
  380. resolution: 8,
  381. direction: [
  382. point2[0]-point1[0],
  383. point2[1]-point1[1],
  384. point2[2]-point1[2],
  385. ]
  386. };
  387. const vtk2d = vtkConeSource.newInstance(initialValues);
  388. const mapper2 = vtkMapper.newInstance();
  389. mapper2.setInputConnection(vtk2d.getOutputPort());
  390. const actor2 = vtkActor.newInstance();
  391. actor2.setMapper(mapper2);
  392. if(way.safety=="safety"){
  393. actor2.getProperty().setColor(GREEN);
  394. }else if(way.safety=="danger"){
  395. actor2.getProperty().setColor(REA);
  396. }
  397. this.wayActors.push(actor2);
  398. mapper.setInputConnection(lineSource.getOutputPort());
  399. this.wayActors.push(actor);
  400. }
  401. });
  402. this.wayActors.forEach((wayAcotor) => {
  403. this.renderer.addActor(wayAcotor);
  404. });
  405. // vtkmodel.renderer.resetCamera();
  406. vtkmodel.renderWindow.render();
  407. }
  408. initMode(rootContainer) {
  409. this.renderWindowWith = vtkGenericRenderWindow.newInstance();
  410. this.renderWindowWith.setContainer(rootContainer);
  411. this.renderer = this.renderWindowWith.getRenderer();
  412. this.renderer.setBackground(0.0, 0.0, 0.0, 0.0)
  413. this.renderWindow = this.renderWindowWith.getRenderWindow();
  414. //选取
  415. this.lastActor = null;
  416. this.nodeActors = []; //节点actor 数据
  417. this.pipeActors = []; //管道actor 数据
  418. this.interactor = this.renderer.getRenderWindow().getInteractor();
  419. this.apiSpecificRenderWindow = this.interactor.getView();
  420. this.hardwareSelector = this.apiSpecificRenderWindow.getSelector();
  421. this.hardwareSelector.setCaptureZValues(true);
  422. // this.hardwareSelector.setFieldAssociation(FieldAssociations.FIELD_ASSOCIATION_POINTS);
  423. this.hardwareSelector.setFieldAssociation(FieldAssociations.FIELD_ASSOCIATION_CELLS);
  424. // document.addEventListener("mousemove", throttleMouseHandler);
  425. document.addEventListener('contextmenu', throttleMouseHandler,function(e){
  426. e.preventDefault();
  427. // throttleMouseHandler
  428. })
  429. // //箭头
  430. this.arrowActors = [];
  431. // const viewColors = [
  432. // [1, 0, 0], // sagittal
  433. // [0, 1, 0], // coronal
  434. // [0, 0, 1], // axial
  435. // [0.5, 0.5, 0.5], // 3D
  436. // ];
  437. // // create axes
  438. // const axes = vtkAnnotatedCubeActor.newInstance();
  439. // axes.setDefaultStyle({
  440. // text: '+X',
  441. // fontStyle: 'bold',
  442. // fontFamily: 'Arial',
  443. // fontColor: 'black',
  444. // fontSizeScale: (res) => res / 2,
  445. // faceColor: createRGBStringFromRGBValues(viewColors[0]),
  446. // faceRotation: 0,
  447. // edgeThickness: 0.1,
  448. // edgeColor: 'black',
  449. // resolution: 400,
  450. // });
  451. // // axes.setXPlusFaceProperty({ text: '+X' });
  452. // axes.setXMinusFaceProperty({
  453. // text: '-X',
  454. // faceColor: createRGBStringFromRGBValues(viewColors[0]),
  455. // faceRotation: 90,
  456. // fontStyle: 'italic',
  457. // });
  458. // axes.setYPlusFaceProperty({
  459. // text: '+Y',
  460. // faceColor: createRGBStringFromRGBValues(viewColors[1]),
  461. // fontSizeScale: (res) => res / 4,
  462. // });
  463. // axes.setYMinusFaceProperty({
  464. // text: '-Y',
  465. // faceColor: createRGBStringFromRGBValues(viewColors[1]),
  466. // fontColor: 'white',
  467. // });
  468. // axes.setZPlusFaceProperty({
  469. // text: '+Z',
  470. // faceColor: createRGBStringFromRGBValues(viewColors[2]),
  471. // });
  472. // axes.setZMinusFaceProperty({
  473. // text: '-Z',
  474. // faceColor: createRGBStringFromRGBValues(viewColors[2]),
  475. // faceRotation: 45,
  476. // });
  477. // // const vtkAxesActor = vtkAxesActor.newInstance();
  478. // // create orientation widget
  479. // this.orientationWidget = vtkOrientationMarkerWidget.newInstance({
  480. // actor: axes,
  481. // interactor: this.renderWindow.getInteractor(),
  482. // });
  483. // this.orientationWidget.setEnabled(true);
  484. // this.orientationWidget.setViewportCorner(
  485. // vtkOrientationMarkerWidget.Corners.BOTTOM_LEFT
  486. // );
  487. // this.orientationWidget.setViewportSize(0.15);
  488. // this.orientationWidget.setMinPixelSize(100);
  489. // this.orientationWidget.setMaxPixelSize(300);
  490. }
  491. generateTicks(numberOfTicks) {
  492. return (helper) => {
  493. const lastTickBounds = helper.getLastTickBounds();
  494. // compute tick marks for axes
  495. const scale = d3
  496. .scaleLinear()
  497. .domain([0.0, 1.0])
  498. .range([lastTickBounds[0], lastTickBounds[1]]);
  499. const samples = scale.ticks(numberOfTicks);
  500. const ticks = samples.map((tick) => scale(tick));
  501. // Replace minus "\u2212" with hyphen-minus "\u002D" so that parseFloat() works
  502. formatDefaultLocale({ minus: "\u002D" });
  503. const format = scale.tickFormat(ticks[0], ticks[ticks.length - 1], numberOfTicks);
  504. const tickStrings = ticks
  505. .map(format)
  506. .map((tick) => Number(parseFloat(tick).toPrecision(12)).toPrecision()); // d3 sometimes adds unwanted whitespace
  507. helper.setTicks(ticks);
  508. helper.setTickStrings(tickStrings);
  509. };
  510. }
  511. //添加箭头
  512. addArrow(velocitys) {
  513. this.clearArrow();
  514. this.arrowActors = [];
  515. velocitys.forEach(velocity => {
  516. velocity.points.forEach(point => {
  517. // const initialValues = {
  518. // base:0,
  519. // shape: 'arrow4points',
  520. // center: [point.x,point.y,point.z],
  521. // direction:[
  522. // velocity.vector[2],
  523. // velocity.vector[0],
  524. // velocity.vector[1]
  525. // ],
  526. // height: 10.0,
  527. // thickness: 1.5,
  528. // width: 3,
  529. // };
  530. // const vtk2d = vtk2DShape.newInstance(initialValues);
  531. const initialValues = {
  532. // base: 0,
  533. center: [point.x, point.y, point.z],
  534. // height: 10,
  535. // radius: 3,
  536. // resolution: 6,
  537. height: 10.0,
  538. radius:2,
  539. resolution: 6,
  540. direction: [
  541. // velocity.vector[2],
  542. // velocity.vector[0],
  543. // velocity.vector[1]
  544. velocity.vector[0],
  545. velocity.vector[1],
  546. velocity.vector[2]
  547. ],
  548. capping: true,
  549. };
  550. const vtk2d = vtkConeSource.newInstance(initialValues);
  551. const mapper = vtkMapper.newInstance();
  552. mapper.setInputConnection(vtk2d.getOutputPort());
  553. const actor = vtkActor.newInstance();
  554. actor.setMapper(mapper);
  555. actor.getProperty().setColor(REA);
  556. this.arrowActors.push(actor);
  557. // this.renderer.addActor(actor);
  558. })
  559. });
  560. for (let index = 0; index < this.arrowActors.length; index=index+30) {
  561. let actor =this.arrowActors[index];
  562. this.renderer.addActor(actor);
  563. };
  564. this.renderWindow.render();
  565. }
  566. clearArrow() {
  567. if (this.arrowActors) {
  568. this.arrowActors.forEach(actor => {
  569. this.renderer.removeActor(actor);
  570. });
  571. }
  572. }
  573. modelInit(validNodes, pipes) {
  574. this.renderWindowWith.resize();
  575. this.validNodes = validNodes;
  576. this.pipes = pipes;
  577. this.modelCreate();
  578. }
  579. modelCreate() {
  580. console.log("modelInit..");
  581. const points = this.polyData.getPoints();
  582. const lines = this.polyData.getLines();
  583. //无节点actor
  584. this.validNodes.forEach((node) => {
  585. points.insertNextPoint(parseFloat(node.x), parseFloat(node.y), parseFloat(node.z));
  586. });
  587. //无管道actor
  588. this.pipes.forEach((pipe) => {
  589. let sid = this.indexIdByPipeNodeId(pipe.snId);
  590. let eid = this.indexIdByPipeNodeId(pipe.enId);
  591. lines.insertNextCell([sid, eid]);
  592. });
  593. //有节点actors
  594. const sphereSource = vtkSphereSource.newInstance({
  595. center: [0, 0, 0],
  596. radius: 4.0,
  597. });
  598. const mapper = vtkMapper.newInstance();
  599. mapper.setInputConnection(sphereSource.getOutputPort());
  600. this.validNodes.forEach((node) => {
  601. const actor = vtkActor.newInstance();
  602. actor.setMapper(mapper);
  603. actor.setPosition(parseFloat(node.x), parseFloat(node.y), parseFloat(node.z));
  604. actor.getProperty().setColor(WHITE);
  605. const nodeActor = {};
  606. nodeActor.node = node;
  607. nodeActor.actor = actor;
  608. this.nodeActors.push(nodeActor);
  609. });
  610. //有管道actors
  611. this.pipes.forEach((pipe) => {
  612. const lineSource = vtkLineSource.newInstance();
  613. let point1 = this.pointByPipeNodeId(pipe.snId);
  614. let point2 = this.pointByPipeNodeId(pipe.enId);
  615. lineSource.setPoint1(point1);
  616. lineSource.setPoint2(point2);
  617. lineSource.setResolution(12);
  618. const actor = vtkActor.newInstance();
  619. const mapper = vtkMapper.newInstance();
  620. actor.setMapper(mapper);
  621. actor.getProperty().setLineWidth(3);//设置线宽
  622. actor.getProperty().setColor(WHITE);
  623. mapper.setInputConnection(lineSource.getOutputPort());
  624. const pipeActor = {};
  625. pipeActor.pipe = pipe;
  626. pipeActor.actor = actor;
  627. this.pipeActors.push(pipeActor);
  628. });
  629. this.modelShow();
  630. this.renderer.resetCamera();
  631. const cam = this.renderer.getActiveCamera();
  632. console.log("透视或平行相机", cam.getParallelProjection());
  633. // cam.setParallelProjection(true);
  634. // console.log("透视或平行相机",cam.getParallelProjection());
  635. console.log("水平视角使用", cam.getUseHorizontalViewAngle());
  636. // cam.setUseHorizontalViewAngle(true);
  637. console.log("平行位置投影比例", cam.getParallelScale());
  638. // cam.setParallelScale(600);
  639. console.log("窗口中心位置", cam.getWindowCenter());//[0, 0]
  640. // cam.setWindowCenter([1,-1]);
  641. console.log("相机视角", cam.getViewAngle());//30
  642. // cam.setViewAngle(90);
  643. console.log("摄像头位置", cam.getPosition());//[495544.15625, 5403554.25, 9989.55990464236]
  644. // cam.setPosition(495544.15625, 5403554.25, 9989.55990464236);
  645. console.log("相机焦点", cam.getFocalPoint());// [495544.15625, 5403554.25, 467.62999725341797]
  646. // cam.setFocalPoint(495544.15625, 5403554.25, 467.62999725341797);
  647. console.log("相机方向", cam.getViewUp());//[0,1,0]
  648. // cam.setViewUp([0,1,1]);
  649. console.log("近远平面", cam.getClippingRange());// [9141.491986233214, 9913.53967078369]
  650. // cam.setClippingRange(9241.491986233214,9300.53967078369);
  651. //设置环境光和光照
  652. const light = vtkLight.newInstance();
  653. light.setColor(1.0, 1.0, 1.0); // 白色环境光
  654. light.setIntensity(3.0); // 强度为1.0
  655. this.renderer.addLight(light);
  656. // this.renderer.resetCamera();
  657. // cam.setPosition(0, 0, 9989);
  658. //相机焦点 495544.15625 5403554.25 467.62999725341797
  659. // cam.setPosition( 495544.15625+500, 5403554.25, 467.62999725341797);
  660. // cam.setPosition( 495544.15625-500, 5403554.25, 467.62999725341797);
  661. // cam.setViewUp([0,0,1]);
  662. // this.renderer.updateCamera();
  663. }
  664. modelClearShow() {
  665. this.renderer.removeActor(this.actor);
  666. this.renderer.removeActor(this.actor2);
  667. this.nodeActors.forEach((nodeActor) => {
  668. this.renderer.removeActor(nodeActor.actor);
  669. });
  670. this.pipeActors.forEach((pipeActor) => {
  671. this.renderer.removeActor(pipeActor.actor);
  672. });
  673. }
  674. wayClearShow(){
  675. this.wayActors.forEach((wayAcotor) => {
  676. this.renderer.removeActor(wayAcotor);
  677. });
  678. }
  679. modelShow() {
  680. this.modelClearShow();
  681. if (!this.isSelectNode) {//不选择节点
  682. this.renderer.addActor(this.actor2);
  683. }
  684. if (!this.isSelectPipe) {//不选择管道
  685. this.renderer.addActor(this.actor);
  686. }
  687. if (this.isSelectNode) {//选择节点
  688. this.nodeActors.forEach((nodeActor) => {
  689. this.renderer.addActor(nodeActor.actor);
  690. });
  691. }
  692. if (this.isSelectPipe) {//选择管道
  693. this.pipeActors.forEach((pipeActor) => {
  694. this.renderer.addActor(pipeActor.actor);
  695. });
  696. }
  697. // this.addjc(495530.903,5403828.821,432.40);
  698. // this.addpname(495530.903,5403829.821,432.40);
  699. // this.addSoure(495530.903,5403829.821,432.40);
  700. // this.addpname(495530.903,5403829.821,432.40);
  701. this.initPname();
  702. this.renderer.addActor(this.axesActor);
  703. // this.rotateY();
  704. }
  705. selectNodes() {
  706. this.isSelectNode = true;
  707. this.isSelectPipe = false;
  708. this.modelShow();
  709. document.addEventListener("mousemove", throttleMouseHandler);
  710. this.renderWindow.render();
  711. }
  712. selectPipes() {
  713. this.isSelectNode = false;
  714. this.isSelectPipe = true;
  715. document.addEventListener("mousemove", throttleMouseHandler);
  716. this.modelShow();
  717. this.renderWindow.render();
  718. }
  719. selectNoting() {
  720. this.isSelectNode = false;
  721. this.isSelectPipe = false;
  722. this.modelShow();
  723. this.renderWindow.render();
  724. }
  725. /**
  726. * 根据id 获取对应的编号
  727. */
  728. indexIdByPipeNodeId(nid) {
  729. for (let index = 0; index < this.validNodes.length; index++) {
  730. const node = this.validNodes[index];
  731. if (node.id == nid) {
  732. return index;
  733. }
  734. }
  735. return 0;
  736. }
  737. pipeByPid(pid) {
  738. console.log("pipeByPid",this.pipes.length)
  739. for (let index = 0; index < this.pipes.length; index++) {
  740. const pipe = this.pipes[index];
  741. if (pipe.id == pid) {
  742. return pipe;
  743. }
  744. }
  745. }
  746. pointByPipeNodeId(nid) {
  747. for (let index = 0; index < this.validNodes.length; index++) {
  748. const node = this.validNodes[index];
  749. if (node.id == nid) {
  750. return [parseFloat(node.x), parseFloat(node.y), parseFloat(node.z)];
  751. }
  752. }
  753. return 0;
  754. }
  755. clearAllShowGj() {
  756. this.renderer.addActor(this.gjActor);
  757. this.renderer.addActor(this.gjscalarBarActor);
  758. this.renderer.removeActor(this.jgActor);
  759. this.renderer.removeActor(this.scalarBarActor);
  760. this.isSelectNode = false;
  761. this.isSelectPipe = false;
  762. this.wayClearShow();
  763. this.modelClearShow();
  764. this.isJg = true; //显示结果
  765. document.removeEventListener("mousemove", throttleMouseHandler);
  766. }
  767. clearModeAddJg() {
  768. this.renderer.addActor(this.jgActor);
  769. this.renderer.addActor(this.scalarBarActor);
  770. this.isSelectNode = false;
  771. this.isSelectPipe = false;
  772. this.renderer.removeActor(this.gjActor);
  773. this.renderer.removeActor(this.gjscalarBarActor)
  774. this.wayClearShow();
  775. this.modelClearShow();
  776. this.isJg = true; //显示结果
  777. document.removeEventListener("mousemove", throttleMouseHandler);
  778. }
  779. clearJgAddMode() {
  780. this.renderer.removeActor(this.scalarBarActor);
  781. this.renderer.removeActor(this.jgActor);
  782. this.renderer.removeActor(this.gjActor);
  783. this.renderer.removeActor(this.gjscalarBarActor)
  784. this.clearArrow();
  785. this.modelShow();
  786. this.wayClearShow();
  787. this.isJg = false; //不显示结果
  788. }
  789. clearAll() {
  790. this.renderer.removeActor(this.jgActor);
  791. this.renderer.removeActor(this.scalarBarActor);
  792. this.renderer.removeActor(this.gjActor);
  793. this.renderer.removeActor(this.gjscalarBarActor)
  794. this.clearArrow();
  795. this.modelClearShow();
  796. this.wayClearShow();
  797. this.renderWindow.render();
  798. }
  799. //选中指定点
  800. selectByNodeId(nid) {
  801. console.log("selectByNodeId", nid);
  802. if (this.isSelectNode) {
  803. if (this.lastActor != null) {
  804. this.lastActor.getProperty().setColor(WHITE);
  805. }
  806. this.nodeActors.forEach(nodeActor => {
  807. if (nid == nodeActor.node.id) {
  808. nodeActor.actor.getProperty().setColor(GREEN);
  809. this.renderWindow.render();
  810. this.lastActor = nodeActor.actor;
  811. this.selectObj = nodeActor.node;
  812. return;
  813. }
  814. });
  815. }
  816. }
  817. //选中指定管道
  818. selectByPipeId(pid) {
  819. console.log("selectByPipeId", pid);
  820. if (this.isSelectPipe) {
  821. if (this.lastActor != null) {
  822. this.lastActor.getProperty().setColor(WHITE);
  823. }
  824. this.pipeActors.forEach(pipeActor => {
  825. if (pid == pipeActor.pipe.id) {
  826. pipeActor.actor.getProperty().setColor(GREEN);
  827. this.renderWindow.render();
  828. this.lastActor = pipeActor.actor;
  829. this.selectObj = pipeActor.node;
  830. return;
  831. }
  832. });
  833. }
  834. }
  835. }
  836. function createRGBStringFromRGBValues(rgb) {
  837. if (rgb.length !== 3) {
  838. return 'rgb(0, 0, 0)';
  839. }
  840. return `rgb(${(rgb[0] * 255).toString()}, ${(rgb[1] * 255).toString()}, ${(
  841. rgb[2] * 255
  842. ).toString()})`;
  843. }
  844. function pickOnMouseEvent(event) {
  845. if (vtkmodel.interactor.isAnimating()) {
  846. // We should not do picking when interacting with the scene
  847. return;
  848. }
  849. const [x, y] = eventToWindowXY(event);
  850. // console.log([x,y]);
  851. // if(!vtkmodel.getPointData){//没有数据
  852. // return ;
  853. // }
  854. // console.log(vtkmodel.hardwareSelector.getPointData());
  855. // if(!vtkmodel.renderer.getActors().size==0){
  856. // return;
  857. // }
  858. vtkmodel.hardwareSelector
  859. .getSourceDataAsync(vtkmodel.renderer, x, y, x, y)
  860. .then((result) => {
  861. if (result) {
  862. processSelections(result.generateSelection(x, y, x, y));
  863. // processSelections(result.generateSelection(x, y, x, y));
  864. } else {
  865. // processSelections(null);
  866. }
  867. });
  868. }
  869. function eventToWindowXY(event) {
  870. // We know we are full screen => window.innerXXX
  871. // Otherwise we can use pixel device ratio or else...
  872. var left = event.pageX ;
  873. var top = event.pageY ;
  874. vtkmodel.epageX=left;
  875. vtkmodel.epageY=top;
  876. const { clientX, clientY } = event;
  877. const [width, height] = vtkmodel.apiSpecificRenderWindow.getSize();
  878. const x = Math.round((width * clientX) / window.innerWidth);
  879. const y = Math.round(height * (1 - clientY / window.innerHeight)); // Need to flip Y
  880. // console.log(x,y)
  881. return [x, y];
  882. }
  883. function processSelections(selections) {
  884. if (!selections || selections.length === 0) {//没有选择
  885. if (vtkmodel.lastActor == null) {
  886. } else {
  887. // vtkmodel.lastActor.getProperty().setColor(WHITE);
  888. // vtkmodel.renderWindow.render();
  889. }
  890. // vtkmodel.lastActor = null;
  891. return;
  892. }
  893. const {
  894. worldPosition: rayHitWorldPosition,
  895. compositeID,
  896. prop,
  897. propID,
  898. attributeID,
  899. } = selections[0].getProperties();
  900. if (vtkmodel.lastActor != null) {
  901. vtkmodel.lastActor.getProperty().setColor(WHITE);
  902. vtkmodel.renderWindow.render();
  903. }
  904. vtkmodel.nodeActors.forEach((nodeActor) => {
  905. if (prop == nodeActor.actor) {
  906. vtkmodel.selectObj = nodeActor.node;
  907. prop.getProperty().setColor(GREEN);
  908. vtkmodel.lastActor = prop;
  909. // console.log(vtkmodel.selectObj );
  910. }
  911. });
  912. vtkmodel.pipeActors.forEach((pipeActor) => {
  913. if (prop == pipeActor.actor) {
  914. vtkmodel.selectObj = pipeActor.pipe;
  915. prop.getProperty().setColor(GREEN);
  916. vtkmodel.lastActor = prop;
  917. // console.log(vtkmodel.selectObj );
  918. }
  919. });
  920. if (vtkmodel.isJg) {//结果数据选择cell
  921. if (vtkmodel.hardwareSelector.getFieldAssociation() ===
  922. FieldAssociations.FIELD_ASSOCIATION_POINTS) {//选择点
  923. } else {
  924. //选择Cell
  925. const input = prop.getMapper().getInputData();
  926. if (!input.getCells()) {
  927. input.buildCells();
  928. // return;
  929. }
  930. const cellPoints = input.getCellPoints(attributeID);
  931. if (cellPoints) {
  932. const pointIds = cellPoints.cellPointIds;
  933. // Find the closest cell point, and use that as cursor position
  934. const points = Array.from(pointIds).map((pointId) =>
  935. input.getPoints().getPoint(pointId)
  936. );
  937. const scalarDataArray = input.getPointData().getScalars().getData();
  938. console.log(scalarDataArray[pointIds[0]]);//节点标量数据
  939. // pointIds.forEach(pointId=>{
  940. // console.log(input.getPointData().getScalars().getName());
  941. // console.log(scalarDataArray[pointId]);//节点标量数据
  942. // });
  943. vtkmodel.selectJgPointId = pointIds[0];
  944. }
  945. }
  946. } else {
  947. vtkmodel.renderWindow.render();
  948. }
  949. }
  950. // 异步函数,用于读取JS文件并转换为ArrayBuffer
  951. async function fetchJSFileAsArrayBuffer(url) {
  952. try {
  953. // 使用fetch API异步加载JS文件
  954. const response = await fetch(url, {
  955. method: 'GET',
  956. responseType: 'arraybuffer' // 指定响应类型为arraybuffer
  957. });
  958. // 确保响应成功
  959. if (!response.ok) {
  960. throw new Error(`HTTP error! status: ${response.status}`);
  961. }
  962. // 获取ArrayBuffer对象
  963. const arrayBuffer = await response.arrayBuffer();
  964. // 可选:验证转换后的ArrayBuffer内容
  965. const decoder = new TextDecoder(); // 使用TextDecoder将ArrayBuffer解码为字符串
  966. const content = decoder.decode(arrayBuffer);
  967. // console.log('File content:', content);
  968. return arrayBuffer;
  969. } catch (error) {
  970. console.error('Error fetching the file:', error);
  971. throw error; // 重新抛出错误以便上层处理
  972. }
  973. }
  974. const throttleMouseHandler = throttle(pickOnMouseEvent, 20);
  975. const WHITE = [1, 0.2, 1];
  976. const GREEN = [0.1, 0.8, 0.1];
  977. const REA = [1, 0, 0];
  978. const JCDCOLOR = [0.5, 0, 0];
  979. const SOURECOLOR = [0.5, 0.5, 0];
  980. const TEXTCOLOR = [1, 1, 1];
  981. const vtkmodel = new VtkModel();
  982. export { vtkmodel };