webgl_loader_pdb.html 7.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322
  1. <!DOCTYPE html>
  2. <html lang="en">
  3. <head>
  4. <title>three.js webgl - molecules</title>
  5. <meta charset="utf-8">
  6. <meta name="viewport" content="width=device-width, user-scalable=no, minimum-scale=1.0, maximum-scale=1.0">
  7. <style>
  8. body {
  9. margin: 0;
  10. font-family: Arial;
  11. overflow: hidden;
  12. }
  13. a {
  14. color: #ffffff;
  15. }
  16. #info {
  17. position: absolute;
  18. top: 0px;
  19. width: 100%;
  20. color: #ffffff;
  21. padding: 5px;
  22. font-family: Monospace;
  23. font-size: 13px;
  24. font-weight: bold;
  25. text-align: center;
  26. z-index: 1;
  27. }
  28. #menu {
  29. position: absolute;
  30. bottom: 20px;
  31. width: 100%;
  32. text-align: center;
  33. padding: 0;
  34. margin: 0;
  35. }
  36. button {
  37. color: rgb(255,255,255);
  38. background: transparent;
  39. border: 0px;
  40. padding: 5px 10px;
  41. cursor: pointer;
  42. }
  43. button:hover {
  44. background-color: rgba(0,255,255,0.5);
  45. }
  46. button:active {
  47. color: #000000;
  48. background-color: rgba(0,255,255,1);
  49. }
  50. .label {
  51. text-shadow: -1px 1px 1px rgb(0,0,0);
  52. margin-left: 25px;
  53. }
  54. </style>
  55. </head>
  56. <body>
  57. <script src="../build/three.js"></script>
  58. <script src="js/controls/TrackballControls.js"></script>
  59. <script src="js/loaders/PDBLoader.js"></script>
  60. <script src="js/renderers/CSS2DRenderer.js"></script>
  61. <div id="container"></div>
  62. <div id="info"><a href="http://threejs.org" target="_blank" rel="noopener">three.js webgl</a> - molecules</div>
  63. <div id="menu"></div>
  64. <script>
  65. var camera, scene, renderer, labelRenderer;
  66. var controls;
  67. var root;
  68. var MOLECULES = {
  69. "Ethanol": "ethanol.pdb",
  70. "Aspirin": "aspirin.pdb",
  71. "Caffeine": "caffeine.pdb",
  72. "Nicotine": "nicotine.pdb",
  73. "LSD": "lsd.pdb",
  74. "Cocaine": "cocaine.pdb",
  75. "Cholesterol": "cholesterol.pdb",
  76. "Lycopene": "lycopene.pdb",
  77. "Glucose": "glucose.pdb",
  78. "Aluminium oxide": "Al2O3.pdb",
  79. "Cubane": "cubane.pdb",
  80. "Copper": "cu.pdb",
  81. "Fluorite": "caf2.pdb",
  82. "Salt": "nacl.pdb",
  83. "YBCO superconductor": "ybco.pdb",
  84. "Buckyball": "buckyball.pdb",
  85. "Graphite": "graphite.pdb"
  86. };
  87. var loader = new THREE.PDBLoader();
  88. var offset = new THREE.Vector3();
  89. var menu = document.getElementById( 'menu' );
  90. init();
  91. animate();
  92. function init() {
  93. scene = new THREE.Scene();
  94. scene.background = new THREE.Color( 0x050505 );
  95. camera = new THREE.PerspectiveCamera( 70, window.innerWidth / window.innerHeight, 1, 5000 );
  96. camera.position.z = 1000;
  97. scene.add( camera );
  98. var light = new THREE.DirectionalLight( 0xffffff, 0.8 );
  99. light.position.set( 1, 1, 1 );
  100. scene.add( light );
  101. var light = new THREE.DirectionalLight( 0xffffff, 0.5 );
  102. light.position.set( -1, -1, 1 );
  103. scene.add( light );
  104. root = new THREE.Group();
  105. scene.add( root );
  106. //
  107. renderer = new THREE.WebGLRenderer( { antialias: true } );
  108. renderer.setPixelRatio( window.devicePixelRatio );
  109. renderer.setSize( window.innerWidth, window.innerHeight );
  110. document.getElementById( 'container' ).appendChild( renderer.domElement );
  111. labelRenderer = new THREE.CSS2DRenderer();
  112. labelRenderer.setSize( window.innerWidth, window.innerHeight );
  113. labelRenderer.domElement.style.position = 'absolute';
  114. labelRenderer.domElement.style.top = '0';
  115. labelRenderer.domElement.style.pointerEvents = 'none';
  116. document.getElementById( 'container' ).appendChild( labelRenderer.domElement );
  117. //
  118. controls = new THREE.TrackballControls( camera, renderer.domElement );
  119. controls.minDistance = 500;
  120. controls.maxDistance = 2000;
  121. //
  122. loadMolecule( 'models/molecules/caffeine.pdb' );
  123. createMenu();
  124. //
  125. window.addEventListener( 'resize', onWindowResize, false );
  126. }
  127. //
  128. function generateButtonCallback( url ) {
  129. return function ( event ) {
  130. loadMolecule( url );
  131. }
  132. }
  133. function createMenu() {
  134. for ( var m in MOLECULES ) {
  135. var button = document.createElement( 'button' );
  136. button.innerHTML = m;
  137. menu.appendChild( button );
  138. var url = 'models/molecules/' + MOLECULES[ m ];
  139. button.addEventListener( 'click', generateButtonCallback( url ), false );
  140. }
  141. }
  142. //
  143. function loadMolecule( url ) {
  144. while ( root.children.length > 0 ) {
  145. var object = root.children[ 0 ];
  146. object.parent.remove( object );
  147. }
  148. loader.load( url, function ( pdb ) {
  149. var geometryAtoms = pdb.geometryAtoms;
  150. var geometryBonds = pdb.geometryBonds;
  151. var json = pdb.json;
  152. var boxGeometry = new THREE.BoxBufferGeometry( 1, 1, 1 );
  153. var sphereGeometry = new THREE.IcosahedronBufferGeometry( 1, 2 );
  154. geometryAtoms.computeBoundingBox();
  155. geometryAtoms.boundingBox.getCenter( offset ).negate();
  156. geometryAtoms.translate( offset.x, offset.y, offset.z );
  157. geometryBonds.translate( offset.x, offset.y, offset.z );
  158. var positions = geometryAtoms.getAttribute( 'position' );
  159. var colors = geometryAtoms.getAttribute( 'color' );
  160. var position = new THREE.Vector3();
  161. var color = new THREE.Color();
  162. for ( var i = 0; i < positions.count; i ++ ) {
  163. position.x = positions.getX( i );
  164. position.y = positions.getY( i );
  165. position.z = positions.getZ( i );
  166. color.r = colors.getX( i );
  167. color.g = colors.getY( i );
  168. color.b = colors.getZ( i );
  169. var material = new THREE.MeshPhongMaterial( { color: color } );
  170. var object = new THREE.Mesh( sphereGeometry, material );
  171. object.position.copy( position );
  172. object.position.multiplyScalar( 75 );
  173. object.scale.multiplyScalar( 25 );
  174. root.add( object );
  175. var atom = json.atoms[ i ];
  176. var text = document.createElement( 'div' );
  177. text.className = 'label';
  178. text.style.color = 'rgb(' + atom[ 3 ][ 0 ] + ',' + atom[ 3 ][ 1 ] + ',' + atom[ 3 ][ 2 ] + ')';
  179. text.textContent = atom[ 4 ];
  180. var label = new THREE.CSS2DObject( text );
  181. label.position.copy( object.position );
  182. root.add( label );
  183. }
  184. positions = geometryBonds.getAttribute( 'position' );
  185. var start = new THREE.Vector3();
  186. var end = new THREE.Vector3();
  187. for ( var i = 0; i < positions.count; i += 2 ) {
  188. start.x = positions.getX( i );
  189. start.y = positions.getY( i );
  190. start.z = positions.getZ( i );
  191. end.x = positions.getX( i + 1 );
  192. end.y = positions.getY( i + 1 );
  193. end.z = positions.getZ( i + 1 );
  194. start.multiplyScalar( 75 );
  195. end.multiplyScalar( 75 );
  196. var object = new THREE.Mesh( boxGeometry, new THREE.MeshPhongMaterial( 0xffffff ) );
  197. object.position.copy( start );
  198. object.position.lerp( end, 0.5 );
  199. object.scale.set( 5, 5, start.distanceTo( end ) );
  200. object.lookAt( end );
  201. root.add( object );
  202. }
  203. render();
  204. } );
  205. }
  206. //
  207. function onWindowResize() {
  208. camera.aspect = window.innerWidth / window.innerHeight;
  209. camera.updateProjectionMatrix();
  210. renderer.setSize( window.innerWidth, window.innerHeight );
  211. labelRenderer.setSize( window.innerWidth, window.innerHeight );
  212. render();
  213. }
  214. function animate() {
  215. requestAnimationFrame( animate );
  216. controls.update();
  217. var time = Date.now() * 0.0004;
  218. root.rotation.x = time;
  219. root.rotation.y = time * 0.7;
  220. render();
  221. }
  222. function render() {
  223. renderer.render( scene, camera );
  224. labelRenderer.render( scene, camera );
  225. }
  226. </script>
  227. </body>
  228. </html>