css3d_molecules.html 6.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337
  1. <!DOCTYPE html>
  2. <html>
  3. <head>
  4. <meta charset="utf-8">
  5. <meta name="viewport" content="width=device-width, user-scalable=no, minimum-scale=1.0, maximum-scale=1.0">
  6. <title>three.js css3d - molecules</title>
  7. <style>
  8. html, body {
  9. height: 100%;
  10. }
  11. body {
  12. background-color: #000000;
  13. margin: 0;
  14. font-family: Arial;
  15. overflow: hidden;
  16. }
  17. a {
  18. color: #ffffff;
  19. }
  20. #info {
  21. position: absolute;
  22. width: 100%;
  23. color: #ffffff;
  24. padding: 5px;
  25. font-family: Monospace;
  26. font-size: 13px;
  27. font-weight: bold;
  28. text-align: center;
  29. z-index: 100;
  30. }
  31. #menu {
  32. position: absolute;
  33. bottom: 20px;
  34. width: 100%;
  35. text-align: center;
  36. padding: 0;
  37. margin: 0;
  38. }
  39. button {
  40. color: rgb(255,255,255);
  41. background: transparent;
  42. border: 0px;
  43. padding: 5px 10px;
  44. cursor: pointer;
  45. }
  46. button:hover {
  47. background-color: rgba(0,255,255,0.5);
  48. }
  49. button:active {
  50. color: #000000;
  51. background-color: rgba(0,255,255,1);
  52. }
  53. </style>
  54. </head>
  55. <body>
  56. <script src="../build/three.min.js"></script>
  57. <script src="js/controls/TrackballControls.js"></script>
  58. <script src="js/renderers/CSS3DRenderer.js"></script>
  59. <script src="js/loaders/PDBLoader.js"></script>
  60. <div id="container"></div>
  61. <div id="info"><a href="http://threejs.org" target="_blank">three.js css3d</a> - molecules</div>
  62. <div id="menu"></div>
  63. <script>
  64. var camera, scene, renderer;
  65. var controls;
  66. var objects = [];
  67. var MOLECULES = {
  68. "Ethanol" :"ethanol.pdb",
  69. "Aspirin" :"aspirin.pdb",
  70. "Caffeine" :"caffeine.pdb",
  71. "Nicotine" :"nicotine.pdb",
  72. "LSD" :"lsd.pdb",
  73. "Cocaine" :"cocaine.pdb",
  74. "Cholesterol" :"cholesterol.pdb",
  75. "Lycopene" :"lycopene.pdb",
  76. "Glucose" : "glucose.pdb",
  77. "Aluminium oxide":"Al2O3.pdb",
  78. "Cubane" :"cubane.pdb",
  79. "Copper" :"cu.pdb",
  80. "Fluorite" :"caf2.pdb",
  81. "Salt" :"nacl.pdb",
  82. "YBCO superconductor":"ybco.pdb",
  83. "Buckyball" : "buckyball.pdb",
  84. //"Diamond" :"diamond.pdb",
  85. "Graphite" :"graphite.pdb"
  86. };
  87. var loader = new THREE.PDBLoader();
  88. var colorSpriteMap = {};
  89. var baseSprite = document.createElement( 'img' );
  90. var menu = document.getElementById( "menu" );
  91. init();
  92. animate();
  93. function init() {
  94. camera = new THREE.PerspectiveCamera( 50, window.innerWidth / window.innerHeight, 1, 5000 );
  95. camera.position.z = 1800;
  96. scene = new THREE.Scene();
  97. //
  98. renderer = new THREE.CSS3DRenderer();
  99. renderer.setSize( window.innerWidth, window.innerHeight );
  100. renderer.domElement.style.position = 'absolute';
  101. renderer.domElement.style.top = 0;
  102. document.getElementById( 'container' ).appendChild( renderer.domElement );
  103. //
  104. controls = new THREE.TrackballControls( camera, renderer.domElement );
  105. controls.rotateSpeed = 0.5;
  106. controls.addEventListener( 'change', render );
  107. //
  108. baseSprite.onload = function () {
  109. loadMolecule( "models/molecules/caffeine.pdb" );
  110. function generateButtonCallback( url ) {
  111. return function ( event ) {
  112. loadMolecule( url );
  113. }
  114. }
  115. for ( var m in MOLECULES ) {
  116. var button = document.createElement( 'button' );
  117. button.innerHTML = m;
  118. menu.appendChild( button );
  119. var url = "models/molecules/" + MOLECULES[ m ];
  120. button.addEventListener( 'click', generateButtonCallback( url ), false );
  121. }
  122. };
  123. baseSprite.src = 'textures/sprites/ball.png';
  124. //
  125. window.addEventListener( 'resize', onWindowResize, false );
  126. }
  127. //
  128. function parseColor( colorString ) {
  129. // default to white opaque
  130. var r, g, b, a;
  131. r = g = b = a = 1;
  132. var matchRGBA = reRGBA.exec( colorString );
  133. if ( matchRGBA ) {
  134. r = parseInt( matchRGBA[ 1 ] ) / 255;
  135. g = parseInt( matchRGBA[ 2 ] ) / 255;
  136. b = parseInt( matchRGBA[ 3 ] ) / 255;
  137. a = parseFloat( matchRGBA[ 4 ] );
  138. } else {
  139. var matchRGB = reRGB.exec( colorString );
  140. if ( matchRGB ) {
  141. r = parseInt( matchRGB[ 1 ] ) / 255;
  142. g = parseInt( matchRGB[ 2 ] ) / 255;
  143. b = parseInt( matchRGB[ 3 ] ) / 255;
  144. }
  145. }
  146. return { "r": r, "g": g, "b": b, "a": a };
  147. }
  148. function colorify( ctx, width, height, color, a ) {
  149. var r = color.r;
  150. var g = color.g;
  151. var b = color.b;
  152. var imageData = ctx.getImageData( 0, 0, width, height );
  153. var data = imageData.data;
  154. for ( var y = 0; y < height; y ++ ) {
  155. for ( var x = 0; x < width; x ++ ) {
  156. var index = ( y * width + x ) * 4;
  157. data[ index ] *= r;
  158. data[ index + 1 ] *= g;
  159. data[ index + 2 ] *= b;
  160. data[ index + 3 ] *= a;
  161. }
  162. }
  163. ctx.putImageData( imageData, 0, 0 );
  164. }
  165. function imageToCanvas( image ) {
  166. var width = image.width;
  167. var height = image.height;
  168. var canvas = document.createElement( 'canvas' );
  169. canvas.width = width;
  170. canvas.height = height;
  171. var context = canvas.getContext( '2d' );
  172. context.drawImage( image, 0, 0, width, height );
  173. return canvas;
  174. }
  175. //
  176. function loadMolecule( url ) {
  177. //console.log( "loading", url );
  178. for ( var i = 0; i < objects.length; i ++ ) {
  179. scene.remove( objects[ i ] );
  180. renderer.cameraElement.removeChild( objects[ i ].element );
  181. }
  182. objects = [];
  183. loader.load( url, function ( geometry ) {
  184. THREE.GeometryUtils.center( geometry );
  185. for ( var i = 0; i < geometry.vertices.length; i ++ ) {
  186. var position = geometry.vertices[ i ];
  187. var color = geometry.colors[ i ];
  188. var element = geometry.elements[ i ];
  189. if ( !colorSpriteMap[ element ] ) {
  190. var canvas = imageToCanvas( baseSprite );
  191. var context = canvas.getContext( '2d' );
  192. colorify( context, canvas.width, canvas.height, color, 1 );
  193. var dataUrl = canvas.toDataURL();
  194. colorSpriteMap[ element ] = dataUrl;
  195. }
  196. colorSprite = colorSpriteMap[ element ];
  197. var atom = document.createElement( 'img' );
  198. atom.src = colorSprite;
  199. var object = new THREE.CSS3DObject( atom );
  200. object.position.copy( position );
  201. object.position.multiplyScalar( 50 );
  202. object.billboard = true;
  203. scene.add( object );
  204. objects.push( object );
  205. }
  206. //console.log( "atoms", geometry.vertices.length );
  207. render();
  208. } );
  209. }
  210. //
  211. function onWindowResize() {
  212. camera.aspect = window.innerWidth / window.innerHeight;
  213. camera.updateProjectionMatrix();
  214. renderer.setSize( window.innerWidth, window.innerHeight );
  215. render();
  216. }
  217. function animate() {
  218. requestAnimationFrame( animate );
  219. controls.update();
  220. }
  221. function render() {
  222. renderer.render( scene, camera );
  223. }
  224. </script>
  225. </body>
  226. </html>