View Javadoc
1   // ******************************************************************************
2   //
3   // Title:       Force Field X.
4   // Description: Force Field X - Software for Molecular Biophysics.
5   // Copyright:   Copyright (c) Michael J. Schnieders 2001-2024.
6   //
7   // This file is part of Force Field X.
8   //
9   // Force Field X is free software; you can redistribute it and/or modify it
10  // under the terms of the GNU General Public License version 3 as published by
11  // the Free Software Foundation.
12  //
13  // Force Field X is distributed in the hope that it will be useful, but WITHOUT
14  // ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
15  // FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
16  // details.
17  //
18  // You should have received a copy of the GNU General Public License along with
19  // Force Field X; if not, write to the Free Software Foundation, Inc., 59 Temple
20  // Place, Suite 330, Boston, MA 02111-1307 USA
21  //
22  // Linking this library statically or dynamically with other modules is making a
23  // combined work based on this library. Thus, the terms and conditions of the
24  // GNU General Public License cover the whole combination.
25  //
26  // As a special exception, the copyright holders of this library give you
27  // permission to link this library with independent modules to produce an
28  // executable, regardless of the license terms of these independent modules, and
29  // to copy and distribute the resulting executable under terms of your choice,
30  // provided that you also meet, for each linked independent module, the terms
31  // and conditions of the license of that module. An independent module is a
32  // module which is not derived from or based on this library. If you modify this
33  // library, you may extend this exception to your version of the library, but
34  // you are not obligated to do so. If you do not wish to do so, delete this
35  // exception statement from your version.
36  //
37  // ******************************************************************************
38  package ffx.numerics.atomic;
39  
40  import edu.rit.pj.IntegerForLoop;
41  import edu.rit.pj.ParallelRegion;
42  import edu.rit.pj.ParallelTeam;
43  import edu.rit.pj.reduction.SharedDoubleArray;
44  import java.util.logging.Level;
45  import java.util.logging.Logger;
46  
47  /**
48   * PJDoubleArray implements the AtomicDoubleArray interface using the Parallel Java class
49   * SharedDoubleArray.
50   *
51   * <p>SharedDoubleArray is multiple thread safe and uses lock-free atomic compare-and-set.
52   *
53   * <p>Note: Class SharedDoubleArray is implemented using class
54   * java.util.concurrent.atomic.AtomicLongArray. Each double array element is stored as a long whose
55   * bit pattern is the same as the double value.
56   *
57   * @author Michael J. Schnieders
58   * @since 1.0
59   */
60  public class PJDoubleArray implements AtomicDoubleArray {
61  
62    private static final Logger logger = Logger.getLogger(PJDoubleArray.class.getName());
63  
64    private SharedDoubleArray array;
65    private int size;
66  
67    /**
68     * Constructor for PJDoubleArray.
69     *
70     * @param size the size of the array.
71     */
72    public PJDoubleArray(int size) {
73      this.size = size;
74      array = new SharedDoubleArray(size);
75    }
76  
77    /** {@inheritDoc} */
78    @Override
79    public void add(int threadID, int index, double value) {
80      array.getAndAdd(index, value);
81    }
82  
83    /** {@inheritDoc} */
84    @Override
85    public void alloc(int size) {
86      this.size = size;
87      if (array.length() < size) {
88        array = new SharedDoubleArray(size);
89      }
90    }
91  
92    /** {@inheritDoc} */
93    @Override
94    public double get(int index) {
95      return array.get(index);
96    }
97  
98    /**
99     * Reduction is handled atomically by the PJ SharedDoubleArray, so this method does nothing.
100    */
101   @Override
102   public void reduce(int lb, int ub) {
103     // Nothing to do.
104   }
105 
106   /**
107    * Reduction is handled atomically by the PJ SharedDoubleArray, so this method does nothing.
108    */
109   @Override
110   public void reduce(ParallelTeam parallelTeam, int lb, int ub) {
111     // Nothing to do.
112   }
113 
114   /** {@inheritDoc} */
115   @Override
116   public void reset(int threadID, int lb, int ub) {
117     for (int i = lb; i <= ub; i++) {
118       array.set(i, 0.0);
119     }
120   }
121 
122   /** {@inheritDoc} */
123   @Override
124   public void reset(ParallelTeam parallelTeam, int lb, int ub) {
125     try {
126       parallelTeam.execute(
127           new ParallelRegion() {
128             @Override
129             public void run() throws Exception {
130               execute(
131                   lb,
132                   ub,
133                   new IntegerForLoop() {
134                     @Override
135                     public void run(int first, int last) {
136                       reset(getThreadIndex(), first, last);
137                     }
138                   });
139             }
140           });
141     } catch (Exception e) {
142       logger.log(Level.WARNING, " Exception resetting an PJDoubleArray", e);
143     }
144   }
145 
146   /** {@inheritDoc} */
147   @Override
148   public void scale(int threadID, int index, double value) {
149     double current = array.get(index);
150     array.getAndSet(index, current * value);
151   }
152 
153   /** {@inheritDoc} */
154   @Override
155   public void set(int threadID, int index, double value) {
156     array.getAndSet(index, value);
157   }
158 
159   /** {@inheritDoc} */
160   @Override
161   public int size() {
162     return size;
163   }
164 
165   /** {@inheritDoc} */
166   @Override
167   public void sub(int threadID, int index, double value) {
168     array.getAndAdd(index, -value);
169   }
170 }