~ubuntu-branches/ubuntu/vivid/emscripten/vivid

1. The origin of this software must not be misrepresented; you must not claim that you wrote the original software. If you use this software in a product, an acknowledgment in the product documentation would be appreciated but is not required.

2. Altered source versions must be plainly marked as such, and must not be misrepresented as being the original software.

3. This notice may not be removed or altered from any source distribution.

#ifndef MINICL_TASK_SCHEDULER_H

#define MINICL_TASK_SCHEDULER_H

#include <assert.h>

#include "BulletMultiThreaded/PlatformDefinitions.h"

#include <stdlib.h>

#include "LinearMath/btAlignedObjectArray.h"

#include "MiniCLTask/MiniCLTask.h"

//just add your commands here, try to keep them globally unique for debugging purposes

#define CMD_SAMPLE_TASK_COMMAND 10

struct MiniCLKernel;

/// MiniCLTaskScheduler handles SPU processing of collision pairs.

/// When PPU issues a task, it will look for completed task buffers

/// PPU will do postprocessing, dependent on workunit output (not likely)

class MiniCLTaskScheduler

{

// track task buffers that are being used, and total busy tasks

btAlignedObjectArray<bool> m_taskBusy;

btAlignedObjectArray<MiniCLTaskDesc> m_spuSampleTaskDesc;

btAlignedObjectArray<const MiniCLKernel*> m_kernels;

int m_numBusyTasks;

// the current task and the current entry to insert a new work unit

int m_currentTask;

bool m_initialized;

void postProcess(int taskId, int outputSize);

class btThreadSupportInterface* m_threadInterface;

int m_maxNumOutstandingTasks;

public:

MiniCLTaskScheduler(btThreadSupportInterface* threadInterface, int maxNumOutstandingTasks);

~MiniCLTaskScheduler();

///call initialize in the beginning of the frame, before addCollisionPairToTask

void initialize();

void issueTask(int firstWorkUnit, int lastWorkUnit, MiniCLKernel* kernel);

///call flush to submit potential outstanding work to SPUs and wait for all involved SPUs to be finished

void flush();

class btThreadSupportInterface* getThreadSupportInterface()

{

return m_threadInterface;

}

int findProgramCommandIdByName(const char* programName) const;

int getMaxNumOutstandingTasks() const

{

return m_maxNumOutstandingTasks;

}

void registerKernel(MiniCLKernel* kernel)

{

m_kernels.push_back(kernel);

}

};

typedef void (*kernelLauncherCB)(MiniCLTaskDesc* taskDesc, int guid);

struct MiniCLKernel

100

{

101

MiniCLTaskScheduler* m_scheduler;

102

103

// int m_kernelProgramCommandId;

104

105

char m_name[MINI_CL_MAX_KERNEL_NAME];

106

unsigned int m_numArgs;

107

kernelLauncherCB m_launcher;

108

void* m_pCode;

109

void updateLauncher();

110

MiniCLKernel* registerSelf();

111

112

void* m_argData[MINI_CL_MAX_ARG];

113

int m_argSizes[MINI_CL_MAX_ARG];

114

};

115

116

117

#if defined(USE_LIBSPE2) && defined(__SPU__)

118

////////////////////MAIN/////////////////////////////

119

#include "../SpuLibspe2Support.h"

120

#include <spu_intrinsics.h>

121

#include <spu_mfcio.h>

122

#include <SpuFakeDma.h>

123

124

void * SamplelsMemoryFunc();

125

void SampleThreadFunc(void* userPtr,void* lsMemory);

126

127

//#define DEBUG_LIBSPE2_MAINLOOP

128

129

int main(unsigned long long speid, addr64 argp, addr64 envp)

130

{

131

printf("SPU is up \n");

132

133

ATTRIBUTE_ALIGNED128(btSpuStatus status);

134

ATTRIBUTE_ALIGNED16( SpuSampleTaskDesc taskDesc ) ;

135

unsigned int received_message = Spu_Mailbox_Event_Nothing;

136

bool shutdown = false;

137

138

cellDmaGet(&status, argp.ull, sizeof(btSpuStatus), DMA_TAG(3), 0, 0);

139

cellDmaWaitTagStatusAll(DMA_MASK(3));

140

141

status.m_status = Spu_Status_Free;

142

status.m_lsMemory.p = SamplelsMemoryFunc();

143

144

cellDmaLargePut(&status, argp.ull, sizeof(btSpuStatus), DMA_TAG(3), 0, 0);

145

cellDmaWaitTagStatusAll(DMA_MASK(3));

146

147

148

while (!shutdown)

149

{

150

received_message = spu_read_in_mbox();

151

152

153

154

switch(received_message)

155

{

156

case Spu_Mailbox_Event_Shutdown:

157

shutdown = true;

158

break;

159

case Spu_Mailbox_Event_Task:

160

// refresh the status

161

#ifdef DEBUG_LIBSPE2_MAINLOOP

162

printf("SPU recieved Task \n");

163

#endif //DEBUG_LIBSPE2_MAINLOOP

164

cellDmaGet(&status, argp.ull, sizeof(btSpuStatus), DMA_TAG(3), 0, 0);

165

cellDmaWaitTagStatusAll(DMA_MASK(3));

166

167

btAssert(status.m_status==Spu_Status_Occupied);

168

169

cellDmaGet(&taskDesc, status.m_taskDesc.p, sizeof(SpuSampleTaskDesc), DMA_TAG(3), 0, 0);

170

cellDmaWaitTagStatusAll(DMA_MASK(3));

171

172

SampleThreadFunc((void*)&taskDesc, reinterpret_cast<void*> (taskDesc.m_mainMemoryPtr) );

173

break;

174

case Spu_Mailbox_Event_Nothing:

175

default:

176

break;

177

}

178

179

// set to status free and wait for next task

180

status.m_status = Spu_Status_Free;

181

cellDmaLargePut(&status, argp.ull, sizeof(btSpuStatus), DMA_TAG(3), 0, 0);

182

cellDmaWaitTagStatusAll(DMA_MASK(3));

183

184

185

}

186

return 0;

187

}

188

//////////////////////////////////////////////////////

189

#endif

190

191

192

193

#endif // MINICL_TASK_SCHEDULER_H

194

Older »