tests/testAtomic.cpp

   1 /*
   2         Copyright (C) 2004-2008 Grame
   3
   4         This program is free software; you can redistribute it and/or modify
   5         it under the terms of the GNU General Public License as published by
   6         the Free Software Foundation; either version 2 of the License, or
   7         (at your option) any later version.
   8
   9         This program is distributed in the hope that it will be useful,
  10         but WITHOUT ANY WARRANTY; without even the implied warranty of
  11         MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12         GNU General Public License for more details.
  13
  14         You should have received a copy of the GNU General Public License
  15         along with this program; if not, write to the Free Software
  16         Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17
  18 */
  19
  20 #include <iostream>
  21 #include <unistd.h>
  22
  23 #include "JackAtomicState.h"
  24 #include "JackPosixThread.h"
  25
  26 using namespace Jack;
  27
  28 #define SIZE 1024
  29
  30 struct TestState {
  31
  32         long fTable[SIZE];
  33         long fReadCounter;
  34         long fWriteCounter;
  35
  36         TestState()
  37         {
  38                 for (int i = 0; i < SIZE; i++) {
  39                         fTable[i] = 0;
  40                 }
  41                 fReadCounter = 0;
  42                 fWriteCounter = 0;
  43         }
  44         virtual ~TestState()
  45         {}
  46
  47         void Write()
  48         {
  49         fWriteCounter++;
  50                 for (int i = 0; i < SIZE; i++) {
  51                         fTable[i] = fTable[i] + 10;
  52                 }
  53         }
  54
  55         bool Read()
  56         {
  57                 int val = fTable[0];
  58                 fReadCounter++;
  59                 for (int i = 0; i < SIZE; i++) {
  60                         if (fTable[i] != val) {
  61                                 printf("Read error fReadCounter %ld, i %ld, curVal %ld, oldVal %ld\n", fReadCounter, i, fTable[i], val);
  62                                 return false;
  63                         }
  64                 }
  65                 return true;
  66         }
  67
  68         bool ReadCopy(long* result)
  69         {
  70                 int val = fTable[0];
  71                 fReadCounter++;
  72                 for (int i = 0; i < SIZE; i++) {
  73                         result[i] = fTable[i];
  74                         if (fTable[i] != val) {
  75                                 //printf("ReadCopy error fReadCounter %ld, i %ld, curVal %ld, oldVal %ld\n", fReadCounter, i, fTable[i], val);
  76                                 return false;
  77                         }
  78                 }
  79                 return true;
  80         }
  81
  82         bool Check(long* result)
  83         {
  84                 int val = result[0];
  85                 for (int i = 0; i < SIZE; i++) {
  86                         if (result[i] != val) {
  87                                 printf("Check error fReadCounter %ld, i %ld, curVal %ld, oldVal %ld\n", fReadCounter, i, fTable[i], val);
  88                                 return false;
  89                         }
  90                 }
  91                 return true;
  92         }
  93
  94         int GetVal()
  95         {
  96                 return fTable[10];
  97         }
  98
  99 };
 100
 101 /*!
 102 \brief The state wrapped with the 2 state atomic class.
 103 */
 104
 105 class TestStateUser : public JackAtomicState<TestState> {
 106
 107         public:
 108
 109                 TestStateUser(){}
 110                 virtual ~TestStateUser(){}
 111
 112                 void TestWriteMethod()
 113                 {
 114                         TestState* state = WriteNextStateStart();
 115                         state->Write();
 116                         state->Write();
 117                         state->Write();
 118                         WriteNextStateStop();
 119                 }
 120
 121                 void TestReadMethod()
 122                 {
 123                         TestState* state;
 124                         int fCount = 0;
 125                         long result[SIZE];
 126             UInt16 cur_index;
 127             UInt16 next_index;
 128                         do {
 129                 cur_index = GetCurrentIndex();
 130                 fCount++;
 131                                 state = ReadCurrentState();
 132                                 bool res = state->ReadCopy(result);
 133                             next_index = GetCurrentIndex();
 134                 if (!res)
 135                     printf("TestReadMethod fCount %ld cur %ld next %ld\n", fCount, cur_index, next_index);
 136                 }while (cur_index != next_index);
 137                         state->Check(result);
 138                 }
 139
 140                 void TestReadRTMethod1()
 141                 {
 142                         TestState* state = TrySwitchState();
 143                         bool res = state->Read();
 144                 }
 145
 146                 void TestReadRTMethod2()
 147                 {
 148                         TestState* state = ReadCurrentState();
 149                         state->Read();
 150                 }
 151 };
 152
 153 /*!
 154 \brief Base class for reader/writer threads.
 155 */
 156
 157 struct TestThread : public JackRunnableInterface {
 158
 159         JackPosixThread* fThread;
 160         TestStateUser* fObject;
 161
 162         TestThread(TestStateUser* state):fObject(state)
 163         {
 164                 fThread = new JackPosixThread(this);
 165                 int res = fThread->Start();
 166         }
 167
 168         virtual ~TestThread()
 169         {
 170                 fThread->Kill();
 171                 delete fThread;
 172         }
 173
 174 };
 175
 176 /*!
 177 \brief "Real-time" reader thread.
 178 */
 179
 180 struct RTReaderThread : public TestThread {
 181
 182         RTReaderThread(TestStateUser* state):TestThread(state)
 183         {}
 184
 185         bool Execute()
 186         {
 187                 fObject->TestReadRTMethod1();
 188
 189                 for (int i = 0; i < 5; i++) {
 190                         fObject->TestReadRTMethod2();
 191                 }
 192
 193                 usleep(50);
 194                 return true;
 195         }
 196 };
 197
 198 /*!
 199 \brief Non "Real-time" reader thread.
 200 */
 201
 202 struct ReaderThread : public TestThread {
 203
 204         ReaderThread(TestStateUser* state):TestThread(state)
 205         {}
 206
 207         bool Execute()
 208         {
 209                  fObject->TestReadMethod();
 210          usleep(56);
 211                  return true;
 212         }
 213 };
 214
 215 /*!
 216 \brief Writer thread.
 217 */
 218
 219 struct WriterThread : public TestThread {
 220
 221         WriterThread(TestStateUser* state):TestThread(state)
 222         {}
 223
 224         bool Execute()
 225         {
 226                 fObject->TestWriteMethod();
 227                 usleep(75);
 228                 return true;
 229         }
 230 };
 231
 232
 233 int main(int argc, char * const argv[])
 234 {
 235         char c;
 236
 237         printf("Test concurrent access to a TestState data structure protected with the 2 state JackAtomicState class\n");
 238
 239         TestStateUser fObject;
 240         WriterThread writer(&fObject);
 241         RTReaderThread readerRT1(&fObject);
 242         ReaderThread reader1(&fObject);
 243
 244         /*
 245         ReaderThread reader2(&fObject);
 246         ReaderThread reader3(&fObject);
 247         ReaderThread reader4(&fObject);
 248         ReaderThread reader5(&fObject);
 249         ReaderThread reader6(&fObject);
 250         */
 251
 252         while ((c = getchar()) != 'q') {}
 253         return 1;
 254 }
 255
 256