glaville / mcmas

Blame view

src/mior/MiorSimulation2.java 3.86 KB
edit raw normal view history
1b1e928cc   glaville   initial import of... 
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
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
163
164
165
166
167
168
169
170
 
  package mior;
  /*
  import java.util.Random;
  import java.util.concurrent.ExecutionException;
  
  import javax.swing.JFrame;
  
  import mior.model.IMiorModel;
  import mior.view.MiorCanvas;
  
  import org.jocl.Pointer;
  import org.jocl.Sizeof;
  import ocl.OCL;
  import ocl.OCLCommandQueue;
  import ocl.OCLContext;
  import ocl.OCLEvent;
  import ocl.OCLKernel;
  import ocl.OCLMem;
  import ocl.OCLProgram;
  
  public class MiorSimulation2 implements IMiorModel {
  	
  	private MiorMM  [] mmList;
  	private StructBuffer<MiorMM> mmBuffer;
  	
  	private MiorOM  [] omList;
  	private StructBuffer<MiorOM> omBuffer;
  	
  	private int     [] associations;
  	private long       associationsSize;
  	private OCLMem     associationsMem;
  	
  	private MiorWorld  world;
  	private StructBuffer<MiorWorld> worldBuffer;
  	
  	private OCLContext context;
  	private OCLCommandQueue queue;
  	
  	public MiorSimulation2() {
  		initializeOpenCL();
  		
  		this.world = new MiorWorld();
  		this.worldBuffer = new StructBuffer<MiorWorld>(context, world);
  		
  		this.associations = new int[world.nbMM * world.nbOM];
  		this.associationsSize = associations.length * Sizeof.cl_int;
  		this.associationsMem = context.createBuffer(associations.length, OCL.INT, OCL.MEM_RW);
  		
  		initializeMM();
  		initializeOM();
  		
  		computeTopography();
  		
  		releaseOpenCL();
  	}
  	
  	public void computeTopography() {
  		OCLProgram program = context.createProgram(MiorKernel.topographyKernel);
  		OCLKernel topographyKernel = program.createKernel("topography");
  		
  		topographyKernel.setArguments(mmBuffer.getMem(), omBuffer.getMem(), associationsMem, worldBuffer.getMem());
  		
  		OCLEvent event = queue.enqueueKernel(
  				topographyKernel, 2,
  				new long[] { 0, 0 },
  				new long[] { mmList.length, omList.length },
  				new long[] { 1, 1 }
  		);
  		
  		try {
  			event.get();
  		} catch (InterruptedException e) {
  			// TODO Auto-generated catch block
  			e.printStackTrace();
  		} catch (ExecutionException e) {
  			// TODO Auto-generated catch block
  			e.printStackTrace();
  		}
  		
  		queue.blockingReadBuffer(associationsMem, Pointer.to(associations), 0, associationsSize);
  		
  		/*
  		for (int i = 0; i < world.nbMM; i++) {
  			for (int j = 0; j < world.nbOM; j++) {
  				//System.out.print(", " + associations[i * world.nbMM + j]);
  				double distance = Math.sqrt(Math.pow(mmList[i].x - omList[j].x, 2) + Math.pow(mmList[i].y - omList[j].y, 2));
  				
  				if (distance <= world.radius) {
  					System.out.print(".");
  				}
  			}
  			System.out.println();
  		}*/
  /*	}
  	
  	private void initializeMM() {
  		Random rand = new Random();
  		
  		this.mmList = new MiorMM[world.nbMM];
  		
  		for (int i = 0; i < world.nbMM; i++) {
  			mmList[i] = new MiorMM(
  					rand.nextInt(world.width),
  					rand.nextInt(world.width)
  			);
  		}
  		
  		this.mmBuffer = new StructBuffer<MiorMM>(context, mmList);
  	}
  	private void initializeOM() {
  		Random rand = new Random();
  		
  		this.omList = new MiorOM[world.nbOM];
  		
  		for (int i = 0; i < world.nbOM; i++) {
  			omList[i] = new MiorOM(
  					rand.nextInt(world.width),
  					rand.nextInt(world.width)
  			);
  		}
  		
  		this.omBuffer = new StructBuffer<MiorOM>(context, omList);
  	}
  	
  	private void initializeOpenCL() {
  		this.context = new OCLContext();
  		this.queue = context.createCommandQueue();
  	}
  	
  	private void releaseOpenCL() {
  		associationsMem.release();
  		worldBuffer.release();
  		mmBuffer.release();
  		omBuffer.release();
  		queue.release();
  		context.release();
  	}
  	
  	@Override
  	public MiorOM[] getOMList() {
  		return omList;
  	}
  	
  	@Override
  	public MiorMM[] getMMList() {
  		return mmList;
  	}
  	
  	@Override
  	public MiorWorld getWorld() {
  		return world;
  	}
  	
  	@Override
  	public int[] getAssociations() {
  		return associations;
  	}
  	
  	public static void main(String[] args) {
  		MiorSimulation2 simu = new MiorSimulation2();
  		MiorCanvas canvas = new MiorCanvas(simu);
  		
  		JFrame frame = new JFrame("Mior CL");
  		frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
  		frame.setContentPane(canvas);
  		frame.pack();
  		frame.setVisible(true);
  	}
  }
  */