OpenMMBrookInterface.cpp 24.6 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
/* -------------------------------------------------------------------------- *
 *                                   OpenMM                                   *
 * -------------------------------------------------------------------------- *
 * This is part of the OpenMM molecular simulation toolkit originating from   *
 * Simbios, the NIH National Center for Physics-Based Simulation of           *
 * Biological Structures at Stanford, funded under the NIH Roadmap for        *
 * Medical Research, grant U54 GM072970. See https://simtk.org.               *
 *                                                                            *
 * Portions copyright (c) 2008 Stanford University and the Authors.           *
 * Authors: Peter Eastman, Mark Friedrichs                                    *
 * Contributors:                                                              *
 *                                                                            *
 * Permission is hereby granted, free of charge, to any person obtaining a    *
 * copy of this software and associated documentation files (the "Software"), *
 * to deal in the Software without restriction, including without limitation  *
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,   *
 * and/or sell copies of the Software, and to permit persons to whom the      *
 * Software is furnished to do so, subject to the following conditions:       *
 *                                                                            *
 * The above copyright notice and this permission notice shall be included in *
 * all copies or substantial portions of the Software.                        *
 *                                                                            *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR *
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,   *
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL    *
 * THE AUTHORS, CONTRIBUTORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,    *
 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR      *
 * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE  *
 * USE OR OTHER DEALINGS IN THE SOFTWARE.                                     *
 * -------------------------------------------------------------------------- */

#include <cmath>
#include <limits>
#include "OpenMMException.h"
#include <sstream>

Mark Friedrichs's avatar
Mark Friedrichs committed
37
38
39
40
#include "LangevinIntegrator.h"
#include "ReferencePlatform.h"
#include "internal/OpenMMContextImpl.h"

41
42
#include "BrookStreamImpl.h"
#include "OpenMMBrookInterface.h"
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
43
#include "gpu/kcommon.h"
44
45
46
47
48
49
50
51
52

using namespace OpenMM;
using namespace std;

/** 
 * OpenMMBrookInterface constructor
 * 
 */

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
53
OpenMMBrookInterface::OpenMMBrookInterface( int streamWidth ) : _particleStreamWidth(streamWidth){
54
55
56
57
58
59
60

// ---------------------------------------------------------------------------------------

   // static const std::string methodName      = "OpenMMBrookInterface::OpenMMBrookInterface";

// ---------------------------------------------------------------------------------------

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
61
   _numberOfParticles                       = 0;
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
62

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
63
64
65
   _triggerForceKernel                      = NULL;
   _triggerEnergyKernel                     = NULL;

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
66
67
68
69
   _positions                               = NULL;
   _velocities                              = NULL;
   _forces                                  = NULL;

70
71
72
73
74
75
76
77
78
   _refForceField                           = NULL;
   _log                                     = NULL;

   _refForceField                           = NULL;
   _refSystem                               = NULL;
   _refOpenMMContext                        = NULL;
   _referencePlatform                       = NULL;
   _refVerletIntegrator                     = NULL;

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
79
80
81
82
83
   _particleStreamSize                      = -1;

   for( int ii = 0; ii < LastBondForce; ii++ ){
      _bondParameters[ii] = NULL;
   }
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
}   

/** 
 * OpenMMBrookInterface destructor
 * 
 */

OpenMMBrookInterface::~OpenMMBrookInterface( ){

// ---------------------------------------------------------------------------------------

   // static const std::string methodName      = "OpenMMBrookInterface::OpenMMBrookInterface";

// ---------------------------------------------------------------------------------------

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
99
100
101
   for( int ii = 0; ii < LastBondForce; ii++ ){
      delete _bondParameters[ii];
   }
102
103
104
105
106
107
108
109
110
111
112
113

   // deleted w/ kernel delete? If activated, program crashes

   //delete _refForceField;
/*
   delete _refSystem;
   delete _refOpenMMContext;
   delete _referencePlatform;
   delete _refVerletIntegrator;
*/
}

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
114
115
116
117
118
119
120
121
122
123
124
/** 
 * Get number of particles
 * 
 * @return   number of particles
 *
 */

int OpenMMBrookInterface::getNumberOfParticles( void ) const {
   return _numberOfParticles;
}

Mark Friedrichs's avatar
Mark Friedrichs committed
125
126
127
128
129
130
131
132
133
134
135
136
/** 
 * Set number of particles
 * 
 * @param numberOfParticles number of particles
 *
 */

int OpenMMBrookInterface::setNumberOfParticles( int numberOfParticles ){
   _numberOfParticles = numberOfParticles;
   return BrookCommon::DefaultReturnValue;
}

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
/** 
 * Get particle stream width
 * 
 * @return    particle stream width
 *
 */

int OpenMMBrookInterface::getParticleStreamWidth( void ) const {
   return _particleStreamWidth;
}

/** 
 * Get particle stream size
 * 
 * @return    particle stream size
 *
 */

int OpenMMBrookInterface::getParticleStreamSize( void ) const {
   if( _particleStreamSize < 0 && _numberOfParticles > 0 ){
       OpenMMBrookInterface* localThis = const_cast<OpenMMBrookInterface* const>(this);
      localThis->_particleStreamSize = BrookPlatform::getStreamSize( _numberOfParticles, _particleStreamWidth, NULL );
   }
   return _particleStreamSize;
}

163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
/** 
 * Get log file reference
 * 
 * @return  log file reference
 *
 */

FILE* OpenMMBrookInterface::getLog( void ) const {
   return _log;
}

/** 
 * Set log file reference
 * 
 * @param  log file reference
 *
 * @return  DefaultReturnValue
 *
 */

int OpenMMBrookInterface::setLog( FILE* log ){
Mark Friedrichs's avatar
Mark Friedrichs committed
184
185
186
187
   _log          = log;
   _brookBonded.setLog( log );
   _brookNonBonded.setLog( log );
   _brookGbsa.setLog( log );
188
189
190
   return BrookCommon::DefaultReturnValue;
}

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
191
192
193
194
195
196
197
198
199
200
201
202
203
/** 
 * Get BrookBondParameters at specified index
 * 
 * @param   index
 *
 * @return  BrookBondParameters* object
 *
 */

BrookBondParameters* OpenMMBrookInterface::_getBondParameters( BondParameterIndices index ) const {
   return _bondParameters[index];
}

204
205
206
207
208
209
210
211
212
213
214
/** 
 * Set BrookBondParameters at specified index
 * 
 * @param index
 * @param brookBondParameters brookBondParameters for BrookBondParameters
 *
 * @return  DefaultReturnValue
 *
 */

int OpenMMBrookInterface::_setBondParameters( BondParameterIndices index, BrookBondParameters* brookBondParameters ){
Mark Friedrichs's avatar
Mark Friedrichs committed
215
216
217
218
219

   if( brookBondParameters && brookBondParameters->getNumberOfBonds() > 0 ){
      _brookBonded.setIsActive( 1 );
   }

220
   _bondParameters[index] = brookBondParameters;
Mark Friedrichs's avatar
Mark Friedrichs committed
221
222
223
224
225
226
227
228

   if( !brookBondParameters || brookBondParameters->getNumberOfBonds() < 1 ){
      int isActive = 0;
      for( int ii = 0; ii < LastBondForce && !isActive; ii++ ){
         isActive = (_bondParameters[ii] != NULL && brookBondParameters->getNumberOfBonds() > 0) ? 1 : 0;
      }
       _brookBonded.setIsActive( isActive );
   }
229
230
231
   return BrookCommon::DefaultReturnValue;
}

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
232
233
234
235
236
237
238
239
240
241
242
/** 
 * Get BrookBondParameters for harmonic bond force
 * 
 * @return brookBondParameters for harmonic bond force
 *
 */

BrookBondParameters* OpenMMBrookInterface::getHarmonicBondForceParameters( void ) const {
   return _getBondParameters( HarmonicBondIndex );
}

243
244
245
246
247
248
249
250
251
252
/** 
 * Set BrookBondParameters for harmonic bond force
 * 
 * @param brookBondParameters brookBondParameters for harmonic bond force
 *
 * @return  DefaultReturnValue
 *
 */

int OpenMMBrookInterface::setHarmonicBondForceParameters( BrookBondParameters* brookBondParameters ){
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
253
   _brookBonded.setupCompleted( 0 );
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
254
255
256
257
258
259
260
261
262
263
264
265
   return _setBondParameters( HarmonicBondIndex, brookBondParameters );
}

/** 
 * Get BrookBondParameters for harmonic angle force
 * 
 * @return brookBondParameters for harmonic angle force
 *
 */

BrookBondParameters* OpenMMBrookInterface::getHarmonicAngleForceParameters( void ) const {
   return _getBondParameters( HarmonicAngleIndex );
266
267
268
269
270
271
272
273
274
275
276
277
}

/** 
 * Set BrookBondParameters for harmonic angle force
 * 
 * @param brookBondParameters brookBondParameters for harmonic angle force
 *
 * @return  DefaultReturnValue
 *
 */

int OpenMMBrookInterface::setHarmonicAngleForceParameters( BrookBondParameters* brookBondParameters ){
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
278
   _brookBonded.setupCompleted( 0 );
279
280
281
282
   return _setBondParameters( HarmonicAngleIndex, brookBondParameters );
}

/** 
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
283
 * Get BrookBondParameters for periodic torsion force
284
 * 
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
285
286
287
288
289
290
291
292
293
294
295
296
 * @return brookBondParameters for periodic torsion force
 *
 */

BrookBondParameters* OpenMMBrookInterface::getPeriodicTorsionForceParameters( void ) const {
   return _getBondParameters( PeriodicTorsionForceIndex );
}

/** 
 * Set BrookBondParameters for periodic torsion force
 * 
 * @param brookBondParameters brookBondParameters for periodic torsion force
297
298
299
300
301
302
 *
 * @return  DefaultReturnValue
 *
 */

int OpenMMBrookInterface::setPeriodicTorsionForceParameters( BrookBondParameters* brookBondParameters ){
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
303
   _brookBonded.setupCompleted( 0 );
304
305
306
307
   return _setBondParameters( PeriodicTorsionForceIndex, brookBondParameters );
}

/** 
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
308
309
310
311
312
313
314
315
316
317
318
319
 * Get BrookBondParameters for rb torsion force
 * 
 * @return brookBondParameters for rb torsion force
 *
 */

BrookBondParameters* OpenMMBrookInterface::getRBTorsionForceParameters( void ) const {
   return _getBondParameters( RbTorsionForceIndex );
}

/** 
 * Set BrookBondParameters for RB torsion force
320
 * 
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
321
 * @param brookBondParameters brookBondParameters for RB torsion force
322
323
324
325
326
327
 *
 * @return  DefaultReturnValue
 *
 */

int OpenMMBrookInterface::setRBTorsionForceParameters( BrookBondParameters* brookBondParameters ){
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
328
   _brookBonded.setupCompleted( 0 );
329
330
331
   return _setBondParameters( RbTorsionForceIndex, brookBondParameters );
}

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
332
333
334
335
336
337
338
339
340
341
342
/** 
 * Get BrookBondParameters for LJ 14 force
 * 
 * @return brookBondParameters for LJ 14 force
 *
 */

BrookBondParameters* OpenMMBrookInterface::getNonBonded14ForceParameters( void ) const {
   return _getBondParameters( LJ14Index );
}

343
344
345
346
347
348
349
350
351
/** 
 * Set BrookBondParameters for LJ 14 force
 * 
 * @param brookBondParameters brookBondParameters for LJ 14 force
 *
 * @return  DefaultReturnValue
 *
 */

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
352
int OpenMMBrookInterface::setNonBonded14ForceParameters( BrookBondParameters* brookBondParameters ){
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
353
   _brookBonded.setupCompleted( 0 );
354
355
356
357
   return _setBondParameters( LJ14Index, brookBondParameters );
}

/** 
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
358
 * Get positions stream
359
 * 
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
360
361
 * @return particle positions 
 *
362
363
 */
    
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
364
365
366
BrookStreamImpl* OpenMMBrookInterface::getParticlePositions( void ){
   return _positions;
}
367

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
368
369
370
371
372
373
374
375
376
377
378
379
380
/** 
 * Set positions stream
 * 
 * @param positions Brook stream containing particle positions
 *
 * @return  DefaultReturnValue
 *
 */
    
int OpenMMBrookInterface::setParticlePositions( BrookStreamImpl* positions ){
   _positions = positions;
   return DefaultReturnValue;
}
381

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
382
383
384
385
386
387
388
389
390
391
/** 
 * Get velocities stream
 * 
 * @return particle velocities
 *
 */
    
BrookStreamImpl* OpenMMBrookInterface::getParticleVelocities( void ){
   return _velocities;
}
392

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
393
394
395
396
397
398
399
400
401
402
403
404
405
/** 
 * Set velocities stream
 * 
 * @param velocities Brook stream containing particle velocities
 *
 * @return  DefaultReturnValue
 *
 */
    
int OpenMMBrookInterface::setParticleVelocities( BrookStreamImpl* velocities ){
   _velocities = velocities;
   return DefaultReturnValue;
}
406

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
407
408
409
410
411
412
413
414
415
416
/** 
 * Get forces stream
 * 
 * @return ParticleForces
 *
 */
    
BrookStreamImpl* OpenMMBrookInterface::getParticleForces( void ){
   return _forces;
}
417

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
418
419
420
421
422
423
424
425
426
427
428
429
/** 
 * Set forces stream
 * 
 * @param forces Brook stream containing particle forces
 *
 * @return  DefaultReturnValue
 *
 */
    
int OpenMMBrookInterface::setParticleForces( BrookStreamImpl* forces ){
   _forces = forces;
   return DefaultReturnValue;
430
431
}

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
/** 
 * Set trigger Force Kernel
 *
 * @param triggerForceKernel kernel to calculate force
 *
 */
    
void OpenMMBrookInterface::setTriggerForceKernel( void* triggerForceKernel ){
   _triggerForceKernel = triggerForceKernel;
}
    
/** 
 * Get trigger Force Kernel
 *
 * @return triggerForceKernel kernel to calculate force
 *
 */
    
void* OpenMMBrookInterface::getTriggerForceKernel( void ) const {
   return _triggerForceKernel;
}

/** 
 * Set trigger Energy Kernel
 *
 * @param triggerEnergyKernel kernel to calculate force
 *
 */
    
void OpenMMBrookInterface::setTriggerEnergyKernel( void* triggerEnergyKernel ){
   _triggerEnergyKernel = triggerEnergyKernel;
}
    
/** 
 * Get trigger Energy Kernel
 *
 * @return triggerEnergyKernel kernel to calculate force
 *
 */
    
void* OpenMMBrookInterface::getTriggerEnergyKernel( void ) const {
   return _triggerEnergyKernel;
}

Mark Friedrichs's avatar
Mark Friedrichs committed
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
/** 
 * Get Brook non bonded
 *
 * @return BrookNonBonded reference
 *
 */
    
BrookNonBonded& OpenMMBrookInterface::getBrookNonBonded( void ){
   return _brookNonBonded;
}

/** 
 * Get Brook GBSA
 *
 * @return BrookGbsa reference
 *
 */
    
BrookGbsa& OpenMMBrookInterface::getBrookGbsa( void ){
   return _brookGbsa;
}

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
/** 
 * Zero forces
 * 
 * @param context     context
 *
 */

void OpenMMBrookInterface::zeroForces( OpenMMContextImpl& context ){

// ---------------------------------------------------------------------------------------

   //static const std::string methodName      = "OpenMMBrookInterface::zeroForces";

// ---------------------------------------------------------------------------------------

   // zero forces

   BrookStreamImpl* forces  = getParticleForces();
   kzerof3( forces->getBrookStream() );

   // ---------------------------------------------------------------------------------------
}

521
/** 
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
522
 * Compute forces
523
 * 
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
524
525
 * @param context     context
 *
526
527
 */

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
528
void OpenMMBrookInterface::computeForces( OpenMMContextImpl& context ){
529
530
531
532
533

// ---------------------------------------------------------------------------------------

   static const std::string methodName      = "OpenMMBrookInterface::executeForces";

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
534
   static       int printOn                 = 0;
535
536
   static const int MaxErrorMessages        = 2;
   static       int ErrorMessages           = 0;
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
537
   FILE* log;
538
539
540

// ---------------------------------------------------------------------------------------

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
541
542
543
544
545
(void) fprintf( stderr, "%s done nonbonded & bonded=%d completed=%d\n", methodName.c_str(),
                _brookBonded.isActive(), _brookBonded.isSetupCompleted() ); 
(void) fflush( stderr );

setLog( stderr );
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
546
   printOn = (printOn && getLog()) ? printOn : 0;
547

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
548
549
   BrookStreamImpl* positions = getParticlePositions();
   BrookStreamImpl* forces    = getParticleForces();
550

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
551
552
553
   // info

   //if( printOn > 1 ){
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
554
   if( 0 ){
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
555
556
557
      printForcesToFile( context );
   }

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
558
559
560
561
(void) fprintf( stderr, "%s x1 done nonbonded & bonded=%d completed=%d\n", methodName.c_str(),
                _brookBonded.isActive(), _brookBonded.isSetupCompleted() ); 
(void) fflush( stderr );

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
562
563
   // nonbonded forces

Mark Friedrichs's avatar
Mark Friedrichs committed
564
565
   if( _brookNonBonded.isActive() ){
      _brookNonBonded.computeForces( *positions, *forces );
566
567
568
569
570
571
   }

// ---------------------------------------------------------------------------------------

   // bonded forces

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
572
573
   if( printOn ){
      log = getLog();
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
574
      (void) fprintf( log, "%s done nonbonded: bonded=%d completed=%d\n", methodName.c_str(),
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
575
                      _brookBonded.isActive(), _brookBonded.isSetupCompleted() ); 
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
576
      (void) fflush( log );
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
577
578
   }

Mark Friedrichs's avatar
Mark Friedrichs committed
579
   if( _brookBonded.isActive() ){
580

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
581
582
      // perform setup first time through

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
583
      if( _brookBonded.isSetupCompleted() == 0 ){
Mark Friedrichs's avatar
Mark Friedrichs committed
584
         _brookBonded.setup( getNumberOfParticles(), getHarmonicBondForceParameters(), getHarmonicAngleForceParameters(),
Mark Friedrichs's avatar
Latest  
Mark Friedrichs committed
585
586
                             getPeriodicTorsionForceParameters(), getRBTorsionForceParameters(),
                             getNonBonded14ForceParameters(),
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
587
                             getParticleStreamWidth(), getParticleStreamSize() );
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
588
589
      }

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
590
(void) fprintf( stderr, "%s done nonbonded & bonded=%d completed=%d\n", methodName.c_str(),
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
591
                _brookBonded.isActive(), _brookBonded.isSetupCompleted() ); 
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
592
(void) fflush( stderr );
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
593

Mark Friedrichs's avatar
Mark Friedrichs committed
594
      _brookBonded.computeForces( *positions, *forces );
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
595
596
597
   
      // diagnostics
   
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
598
      if( printOn ){
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
599
600
601
         FILE* log = stderr;
         static int step           = 0;
         static const int stopStep = 10;
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
602
         (void) fprintf( log, "%s done bonded computeForces\n", methodName.c_str() );
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
603
         (void) fflush( log );
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
604
605
606
607
608
609
610
611
612
613
   
   /*
         (void) fprintf( getLog(), "\nFinal NB & bonded forces" );
         BrookStreamInternal* brookStreamInternalF   = forceStream.getBrookStreamImpl();
         brookStreamInternalF->printToFile( getLog() );
         void* dataV = brookStreamInternalF->getData(1);
         float* data = (float*) dataV;
         (void) fprintf( getLog(), "\nFinal NB & bonded forces RAW\n" );
         for( int ii = 0; ii < _brookNonBonded->getNumberOfParticles()*3; ii += 3 ){
            (void) fprintf( getLog(), "%d [%.6e %.6e %.6e]\n", ii, data[ii], data[ii+1], data[ii+2] );
614
         }
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
615
   */
616
   
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
617
618
619
         if( step++ >= stopStep ){
            exit(0);
         }
620
      }
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
621

622
623
   }

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
624

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
625
   // GBSA OBC forces
626

Mark Friedrichs's avatar
Mark Friedrichs committed
627
628
   if( _brookGbsa.isActive() ){
      _brookGbsa.computeForces( *positions, *forces );
629
630
   }

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
631
632
633
(void) fprintf( stderr, "%s done computeForces\n", methodName.c_str() );
(void) fflush( stderr );

634
635
   // ---------------------------------------------------------------------------------------
}
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
636

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
/** 
 * Print forces to file 
 * 
 * @param context     context
 *
 */

void OpenMMBrookInterface::printForcesToFile( OpenMMContextImpl& context ){

// ---------------------------------------------------------------------------------------

   static const std::string methodName      = "OpenMMBrookInterface::printForcesToFile";
   float zero                               = 0.0f;
   static int step                          = 0;

// ---------------------------------------------------------------------------------------

   // first step only?

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
656
   if( step > 20 ){
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
      return;
   }

   std::stringstream fileNameBaseS;
   //fileNameBase << "Brook_" << context.getTime() << "_"; 
   fileNameBaseS << "Brook_" << step++ << "_"; 
   std::string fileNameBase = fileNameBaseS.str();

   // create vector of streams for output

   BrookStreamImpl* positions = getParticlePositions();
   BrookStreamImpl* forces    = getParticleForces();

   std::vector<BrookStreamImpl*> streams;
   streams.push_back( positions );
   streams.push_back( forces );

// ---------------------------------------------------------------------------------------

   // nonbonded forces

   if( _brookNonBonded.isActive() ){
      forces->fillWithValue( &zero );
      _brookNonBonded.computeForces( *positions, *forces );
      std::string fileName = fileNameBase + "NonBonded.txt";
      printStreamsToFile( fileName, streams );
   }

// ---------------------------------------------------------------------------------------

   // bonded forces

   if( _brookBonded.isActive() ){

      // perform setup first time through

      if( _brookBonded.isSetupCompleted() == 0 ){

         _brookBonded.setup( getNumberOfParticles(), getHarmonicBondForceParameters(), getHarmonicAngleForceParameters(),
                             getPeriodicTorsionForceParameters(), getRBTorsionForceParameters(),
                             getNonBonded14ForceParameters(),
                             getParticleStreamWidth(), getParticleStreamSize() );
      }

      forces->fillWithValue( &zero );
      _brookBonded.computeForces( *positions, *forces );
      std::string fileName = fileNameBase + "Bonded.txt";
      printStreamsToFile( fileName, streams );
   
   }

// ---------------------------------------------------------------------------------------

   // GBSA OBC forces

   if( _brookGbsa.isActive() ){

      forces->fillWithValue( &zero );

      _brookGbsa.computeForces( *positions, *forces );

      std::string fileName = fileNameBase + "Obc.txt";
      printStreamsToFile( fileName, streams );

   }
   forces->fillWithValue( &zero );

// ---------------------------------------------------------------------------------------

   // all forces

   if( 1 ){

      if( _brookNonBonded.isActive() ){
         _brookNonBonded.computeForces( *positions, *forces );
      }

      if( _brookBonded.isActive() ){
         _brookBonded.computeForces( *positions, *forces );
      }

      if( _brookGbsa.isActive() ){
         _brookGbsa.computeForces( *positions, *forces );
      }

      std::string fileName = fileNameBase + "AllF.txt";
      printStreamsToFile( fileName, streams );
   }

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
746
747
(void) fprintf( stderr, "%s done computeForces\n", methodName.c_str() );
(void) fflush( stderr );
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
748
749
750
751
752
   forces->fillWithValue( &zero );

   // ---------------------------------------------------------------------------------------
}

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
753
754
755
756
/** 
 * Compute energy
 * 
 * @param context     context
Mark Friedrichs's avatar
Mark Friedrichs committed
757
 * @param system      system reference
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
758
759
760
 *
 */

Mark Friedrichs's avatar
Mark Friedrichs committed
761
double OpenMMBrookInterface::computeEnergy( OpenMMContextImpl& context, System& system ){
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
762
763
764

// ---------------------------------------------------------------------------------------

Mark Friedrichs's avatar
Mark Friedrichs committed
765
   //static const std::string methodName      = "OpenMMBrookInterface::computeEnergy";
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
766
767
768

// ---------------------------------------------------------------------------------------

Mark Friedrichs's avatar
Mark Friedrichs committed
769
770
771
772
773
774
775
776
777
778
779
780
   // We don't currently have GPU kernels to calculate energy, so instead we have the reference
   // platform do it.  This is VERY slow.
    
   LangevinIntegrator integrator(0.0, 1.0, 0.0);
   ReferencePlatform platform;
   OpenMMContext refContext( system, integrator, platform );

   const Stream& positions = context.getPositions();
   double* posData         = new double[positions.getSize()*3];
   positions.saveToArray(posData);
   vector<Vec3> pos(positions.getSize());

Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
781
   for( unsigned int ii = 0; ii < pos.size(); ii++ ){
Mark Friedrichs's avatar
Mark Friedrichs committed
782
783
784
785
786
787
      pos[ii] = Vec3(posData[3*ii], posData[3*ii+1], posData[3*ii+2]);
   }
   delete[] posData;
   refContext.setPositions(pos);

   return refContext.getState(State::Energy).getPotentialEnergy();
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
788
789

}
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835

/* 
 * Print contents of object to file
 *
 * @param fileName     file name
 * @param streams      streams to print
 *
 * @return DefaultReturnValue
 *
 * */

int OpenMMBrookInterface::printStreamsToFile( std::string fileName, std::vector<BrookStreamImpl*>& streams ){

// ---------------------------------------------------------------------------------------

   static const std::string methodName      = "OpenMMBrookInterface::printStreamsToFile";

// ---------------------------------------------------------------------------------------

   FILE* filePtr = fopen( fileName.c_str(), "w" );
   if( !filePtr ){
      (void) fprintf( stderr, "%s coud not open file=<%s>\n", methodName.c_str(), fileName.c_str() );
      (void) fflush( stderr );
      return ErrorReturnValue;
   }

   // gather arrays, widths for eah stream, and set index for each stream
   // also set minimum of stream sizes

   int minIndex    = 10000000;
   float** arrays  = new float*[streams.size()];
   float** sums    = new float*[streams.size()];
   int* widths     = new int[streams.size()];
   int* indices    = new int[streams.size()];
   for( unsigned int ii = 0; ii < streams.size(); ii++ ){
      BrookStreamImpl* stream  = streams[ii];
      void* dataArrayV         = stream->getData( 1 );
      arrays[ii]               = (float*) dataArrayV;
      widths[ii]               =  stream->getWidth();
      indices[ii]              = 0;
      sums[ii]                 = new float[4];
      sums[ii][0] = sums[ii][1] = sums[ii][2] = sums[ii][3] = 0.0f;
      if( minIndex > stream->getStreamSize() ){
         minIndex = stream->getStreamSize();
      }
   }
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
836
   minIndex = minIndex > getNumberOfParticles() ? getNumberOfParticles() : minIndex;
Mark Friedrichs's avatar
Mods  
Mark Friedrichs committed
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900

   // sum columns 

   for( int ii = 0; ii < minIndex; ii++ ){
      for( unsigned int kk = 0; kk < streams.size(); kk++ ){
         for( int jj = 0; jj < widths[kk]; jj++ ){
            sums[kk][jj] += arrays[kk][indices[kk]++];
         }
      }
   }

   // reinitialize indices

   for( unsigned int kk = 0; kk < streams.size(); kk++ ){
      indices[kk] = 0;
   }

   // show column sums

   for( unsigned int kk = 0; kk < streams.size(); kk++ ){
      (void) fprintf( filePtr, "Sms " );
      for( int jj = 0; jj < widths[kk]; jj++ ){
         (void) fprintf( filePtr, "%12.5e ", sums[kk][jj] );
      }
   }
   (void) fprintf( filePtr, "\n" );

   for( int ii = 0; ii < minIndex; ii++ ){
      (void) fprintf( filePtr, "%d ", ii );

      // streams

      for( unsigned int kk = 0; kk < streams.size(); kk++ ){

//         (void) fprintf( filePtr, "[ " );
 
         // ii elements of stream kk

         for( int jj = 0; jj < widths[kk]; jj++ ){
            (void) fprintf( filePtr, "%12.5e ", arrays[kk][indices[kk]++] );
         }

 //        (void) fprintf( filePtr, " ]", ii );
      }
      (void) fprintf( filePtr, "\n", ii );
   }


   // cleanup

   (void) fclose( filePtr );

   delete[] arrays;
   delete[] widths;
   delete[] indices;
   for( int ii = 0; ii < 4; ii++ ){
      delete[] sums[ii];
   }
   delete[] sums;

   return DefaultReturnValue;

}