Home

Context Navigation

source: palm/trunk/SOURCE/init_particles.f90 @ 17

Last change on this file since 17 was 16, checked in by raasch, 18 years ago
bugfixes in advec_particles and init_particles
Property svn:keywords set to `Id`
File size: 20.3 KB

Line
1	SUBROUTINE init_particles
2
3	!------------------------------------------------------------------------------!
4	! Actual revisions:
5	! -----------------
6	! Bugfix: MPI_REAL in MPI_ALLREDUCE replaced by MPI_INTEGER
7	!
8	! Former revisions:
9	! -----------------
10	! $Id: init_particles.f90 16 2007-02-15 13:16:47Z raasch $
11	! RCS Log replace by Id keyword, revision history cleaned up
12	!
13	! Revision 1.24 2007/02/11 13:00:17 raasch
14	! Bugfix: allocation of tail_mask and new_tail_id in case of restart-runs
15	! Bugfix: __ was missing in a cpp-directive
16	!
17	! Revision 1.1 1999/11/25 16:22:38 raasch
18	! Initial revision
19	!
20	!
21	! Description:
22	! ------------
23	! This routine initializes a set of particles and their attributes (position,
24	! radius, ..). Advection of these particles is carried out by advec_particles,
25	! plotting is done in data_output_dvrp.
26	!------------------------------------------------------------------------------!
27	#if defined( __particles )
28
29	USE arrays_3d
30	USE control_parameters
31	USE grid_variables
32	USE indices
33	USE particle_attributes
34	USE pegrid
35	USE random_function_mod
36
37
38	IMPLICIT NONE
39
40	CHARACTER (LEN=10) :: particle_binary_version, version_on_file
41
42	INTEGER :: i, j, n, nn
43	#if defined( __parallel )
44	INTEGER, DIMENSION(3) :: blocklengths, displacements, types
45	#endif
46	LOGICAL :: uniform_particles_l
47	REAL :: factor, pos_x, pos_y, pos_z, value
48
49
50	#if defined( __parallel )
51	!
52	!-- Define MPI derived datatype for FORTRAN datatype particle_type (see module
53	!-- particle_attributes). Integer length is 4 byte, Real is 8 byte (=> total
54	!-- length 100, nevertheless, 120 bytes are needed on T3E since integer seems
55	!-- to be 8 bytes long there)
56	blocklengths(1) = 18; blocklengths(2) = 4; blocklengths(3) = 1
57	#if defined( __t3eb )
58	displacements(1) = 0; displacements(2) = 144; displacements(3) = 176
59	#else
60	displacements(1) = 0; displacements(2) = 144; displacements(3) = 160
61	#endif
62	types(1) = MPI_REAL
63	types(2) = MPI_INTEGER
64	types(3) = MPI_UB
65	CALL MPI_TYPE_STRUCT( 3, blocklengths, displacements, types, &
66	mpi_particle_type, ierr )
67	CALL MPI_TYPE_COMMIT( mpi_particle_type, ierr )
68	#endif
69
70	!
71	!-- Check the number of particle groups.
72	IF ( number_of_particle_groups > max_number_of_particle_groups ) THEN
73	PRINT*, '+++ WARNING: init_particles: ', &
74	'max_number_of_particle_groups =', &
75	max_number_of_particle_groups
76	PRINT*, '+++ number_of_particle_groups reset to ', &
77	max_number_of_particle_groups
78	number_of_particle_groups = max_number_of_particle_groups
79	ENDIF
80
81	!
82	!-- Set default start positions, if necessary
83	IF ( psl(1) == 9999999.9 ) psl(1) = -0.5 * dx
84	IF ( psr(1) == 9999999.9 ) psr(1) = ( nx + 0.5 ) * dx
85	IF ( pss(1) == 9999999.9 ) pss(1) = -0.5 * dy
86	IF ( psn(1) == 9999999.9 ) psn(1) = ( ny + 0.5 ) * dy
87	IF ( psb(1) == 9999999.9 ) psb(1) = zu(nz/2)
88	IF ( pst(1) == 9999999.9 ) pst(1) = psb(1)
89
90	IF ( pdx(1) == 9999999.9 .OR. pdx(1) == 0.0 ) pdx(1) = dx
91	IF ( pdy(1) == 9999999.9 .OR. pdy(1) == 0.0 ) pdy(1) = dy
92	IF ( pdz(1) == 9999999.9 .OR. pdz(1) == 0.0 ) pdz(1) = zu(2) - zu(1)
93
94	DO j = 2, number_of_particle_groups
95	IF ( psl(j) == 9999999.9 ) psl(j) = psl(j-1)
96	IF ( psr(j) == 9999999.9 ) psr(j) = psr(j-1)
97	IF ( pss(j) == 9999999.9 ) pss(j) = pss(j-1)
98	IF ( psn(j) == 9999999.9 ) psn(j) = psn(j-1)
99	IF ( psb(j) == 9999999.9 ) psb(j) = psb(j-1)
100	IF ( pst(j) == 9999999.9 ) pst(j) = pst(j-1)
101	IF ( pdx(j) == 9999999.9 .OR. pdx(j) == 0.0 ) pdx(j) = pdx(j-1)
102	IF ( pdy(j) == 9999999.9 .OR. pdy(j) == 0.0 ) pdy(j) = pdy(j-1)
103	IF ( pdz(j) == 9999999.9 .OR. pdz(j) == 0.0 ) pdz(j) = pdz(j-1)
104	ENDDO
105
106	!
107	!-- For the first model run of a possible job chain initialize the
108	!-- particles, otherwise read the particle data from file.
109	IF ( TRIM( initializing_actions ) == 'read_restart_data' &
110	.AND. read_particles_from_restartfile ) THEN
111
112	!
113	!-- Read particle data from previous model run.
114	!-- First open the input unit.
115	IF ( myid_char == '' ) THEN
116	OPEN ( 90, FILE='PARTICLE_RESTART_DATA_IN'//myid_char, &
117	FORM='UNFORMATTED' )
118	ELSE
119	OPEN ( 90, FILE='PARTICLE_RESTART_DATA_IN/'//myid_char, &
120	FORM='UNFORMATTED' )
121	ENDIF
122
123	!
124	!-- First compare the version numbers
125	READ ( 90 ) version_on_file
126	particle_binary_version = '3.0'
127	IF ( TRIM( version_on_file ) /= TRIM( particle_binary_version ) ) THEN
128	IF ( myid == 0 ) THEN
129	PRINT*, '+++ init_particles: version mismatch concerning data ', &
130	'from prior run'
131	PRINT*, ' version on file = "', TRIM( version_on_file ),&
132	'"'
133	PRINT*, ' version in program = "', &
134	TRIM( particle_binary_version ), '"'
135	ENDIF
136	CALL local_stop
137	ENDIF
138
139	!
140	!-- Read some particle parameters and the size of the particle arrays,
141	!-- allocate them and read their contents.
142	READ ( 90 ) bc_par_b, bc_par_lr, bc_par_ns, bc_par_t, &
143	maximum_number_of_particles, maximum_number_of_tailpoints, &
144	maximum_number_of_tails, number_of_initial_particles, &
145	number_of_particles, number_of_particle_groups, &
146	number_of_tails, particle_groups, time_prel, &
147	time_write_particle_data, uniform_particles
148
149	IF ( number_of_initial_particles /= 0 ) THEN
150	ALLOCATE( initial_particles(1:number_of_initial_particles) )
151	READ ( 90 ) initial_particles
152	ENDIF
153
154	ALLOCATE( prt_count(nzb:nzt+1,nys-1:nyn+1,nxl-1:nxr+1), &
155	prt_start_index(nzb:nzt+1,nys-1:nyn+1,nxl-1:nxr+1), &
156	particle_mask(maximum_number_of_particles), &
157	particles(maximum_number_of_particles) )
158
159	READ ( 90 ) prt_count, prt_start_index
160	READ ( 90 ) particles
161
162	IF ( use_particle_tails ) THEN
163	ALLOCATE( particle_tail_coordinates(maximum_number_of_tailpoints,5, &
164	maximum_number_of_tails), &
165	new_tail_id(maximum_number_of_tails), &
166	tail_mask(maximum_number_of_tails) )
167	READ ( 90 ) particle_tail_coordinates
168	ENDIF
169
170	CLOSE ( 90 )
171
172	ELSE
173
174	!
175	!-- Allocate particle arrays and set attributes of the initial set of
176	!-- particles, which can be also periodically released at later times.
177	!-- Also allocate array for particle tail coordinates, if needed.
178	ALLOCATE( prt_count(nzb:nzt+1,nys-1:nyn+1,nxl-1:nxr+1), &
179	prt_start_index(nzb:nzt+1,nys-1:nyn+1,nxl-1:nxr+1), &
180	particle_mask(maximum_number_of_particles), &
181	particles(maximum_number_of_particles) )
182
183	!
184	!-- Initialize all particles with dummy values (otherwise errors may
185	!-- occur within restart runs). The reason for this is still not clear
186	!-- and may be presumably caused by errors in the respective user-interface.
187	particles = particle_type( 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, &
188	0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, &
189	0, 0, 0, 0 )
190	particle_groups = particle_groups_type( 0.0, 0.0, 0.0, 0.0 )
191
192	!
193	!-- Set the default particle size used for dvrp plots
194	IF ( dvrp_psize == 9999999.9 ) dvrp_psize = 0.2 * dx
195
196	!
197	!-- Set values for the density ratio and radius for all particle
198	!-- groups, if necessary
199	IF ( density_ratio(1) == 9999999.9 ) density_ratio(1) = 0.0
200	IF ( radius(1) == 9999999.9 ) radius(1) = 0.0
201	DO i = 2, number_of_particle_groups
202	IF ( density_ratio(i) == 9999999.9 ) THEN
203	density_ratio(i) = density_ratio(i-1)
204	ENDIF
205	IF ( radius(i) == 9999999.9 ) radius(i) = radius(i-1)
206	ENDDO
207
208	DO i = 1, number_of_particle_groups
209	IF ( density_ratio(i) /= 0.0 .AND. radius(i) == 0 ) THEN
210	IF ( myid == 0 ) THEN
211	PRINT*, '+++ init_particles: particle group #', i, 'has a', &
212	'density ratio /= 0 but radius = 0'
213	ENDIF
214	CALL local_stop
215	ENDIF
216	particle_groups(i)%density_ratio = density_ratio(i)
217	particle_groups(i)%radius = radius(i)
218	ENDDO
219
220	!
221	!-- Calculate particle positions and store particle attributes, if
222	!-- particle is situated on this PE
223	n = 0
224
225	DO i = 1, number_of_particle_groups
226
227	pos_z = psb(i)
228
229	DO WHILE ( pos_z <= pst(i) )
230
231	pos_y = pss(i)
232
233	DO WHILE ( pos_y <= psn(i) )
234
235	IF ( pos_y >= ( nys - 0.5 ) * dy .AND. &
236	pos_y < ( nyn + 0.5 ) * dy ) THEN
237
238	pos_x = psl(i)
239
240	DO WHILE ( pos_x <= psr(i) )
241
242	IF ( pos_x >= ( nxl - 0.5 ) * dx .AND. &
243	pos_x < ( nxr + 0.5 ) * dx ) THEN
244
245	DO j = 1, particles_per_point
246
247	n = n + 1
248	IF ( n > maximum_number_of_particles ) THEN
249	PRINT*,'+++ init_particles: number of initial', &
250	' particles (', n, ') exceeds'
251	PRINT*,' maximum_number_of_particles (', &
252	maximum_number_of_particles, ') on PE ', &
253	myid
254	#if defined( __parallel )
255	CALL MPI_ABORT( comm2d, 9999, ierr )
256	#else
257	CALL local_stop
258	#endif
259	ENDIF
260	particles(n)%x = pos_x
261	particles(n)%y = pos_y
262	particles(n)%z = pos_z
263	particles(n)%age = 0.0
264	particles(n)%dt_sum = 0.0
265	particles(n)%dvrp_psize = dvrp_psize
266	particles(n)%e_m = 0.0
267	particles(n)%speed_x = 0.0
268	particles(n)%speed_x_sgs = 0.0
269	particles(n)%speed_y = 0.0
270	particles(n)%speed_y_sgs = 0.0
271	particles(n)%speed_z = 0.0
272	particles(n)%speed_z_sgs = 0.0
273	particles(n)%origin_x = pos_x
274	particles(n)%origin_y = pos_y
275	particles(n)%origin_z = pos_z
276	particles(n)%radius = particle_groups(i)%radius
277	particles(n)%weight_factor =initial_weighting_factor
278	particles(n)%color = 1
279	particles(n)%group = i
280	particles(n)%tailpoints = 0
281	IF ( use_particle_tails .AND. &
282	MOD( n, skip_particles_for_tail ) == 0 ) THEN
283	number_of_tails = number_of_tails + 1
284	!
285	!-- This is a temporary provisional setting (see
286	!-- further below!)
287	particles(n)%tail_id = number_of_tails
288	ELSE
289	particles(n)%tail_id = 0
290	ENDIF
291
292	ENDDO
293
294	ENDIF
295
296	pos_x = pos_x + pdx(i)
297
298	ENDDO
299
300	ENDIF
301
302	pos_y = pos_y + pdy(i)
303
304	ENDDO
305
306	pos_z = pos_z + pdz(i)
307
308	ENDDO
309
310	ENDDO
311
312	number_of_initial_particles = n
313	number_of_particles = n
314
315	!
316	!-- Calculate the number of particles and tails of the total domain
317	#if defined( __parallel )
318	CALL MPI_ALLREDUCE( number_of_particles, total_number_of_particles, 1, &
319	MPI_INTEGER, MPI_SUM, comm2d, ierr )
320	CALL MPI_ALLREDUCE( number_of_tails, total_number_of_tails, 1, &
321	MPI_INTEGER, MPI_SUM, comm2d, ierr )
322	#else
323	total_number_of_particles = number_of_particles
324	total_number_of_tails = number_of_tails
325	#endif
326
327	!
328	!-- Set a seed value for the random number generator to be exclusively
329	!-- used for the particle code. The generated random numbers should be
330	!-- different on the different PEs.
331	iran_part = iran_part + myid
332
333	!
334	!-- User modification of initial particles
335	CALL user_init_particles
336
337	!
338	!-- Store the initial set of particles for release at later times
339	IF ( number_of_initial_particles /= 0 ) THEN
340	ALLOCATE( initial_particles(1:number_of_initial_particles) )
341	initial_particles(1:number_of_initial_particles) = &
342	particles(1:number_of_initial_particles)
343	ENDIF
344
345	!
346	!-- Add random fluctuation to particle positions
347	IF ( random_start_position ) THEN
348
349	iran = iran + myid ! Random positions should be different on
350	! different PEs
351
352	DO n = 1, number_of_initial_particles
353	IF ( psl(particles(n)%group) /= psr(particles(n)%group) ) THEN
354	particles(n)%x = particles(n)%x + &
355	( random_function( iran ) - 0.5 ) * &
356	pdx(particles(n)%group)
357	IF ( particles(n)%x <= ( nxl - 0.5 ) * dx ) THEN
358	particles(n)%x = ( nxl - 0.4999999999 ) * dx
359	ELSEIF ( particles(n)%x >= ( nxr + 0.5 ) * dx ) THEN
360	particles(n)%x = ( nxr + 0.4999999999 ) * dx
361	ENDIF
362	ENDIF
363	IF ( pss(particles(n)%group) /= psn(particles(n)%group) ) THEN
364	particles(n)%y = particles(n)%y + &
365	( random_function( iran ) - 0.5 ) * &
366	pdy(particles(n)%group)
367	IF ( particles(n)%y <= ( nys - 0.5 ) * dy ) THEN
368	particles(n)%y = ( nys - 0.4999999999 ) * dy
369	ELSEIF ( particles(n)%y >= ( nyn + 0.5 ) * dy ) THEN
370	particles(n)%y = ( nyn + 0.4999999999 ) * dy
371	ENDIF
372	ENDIF
373	IF ( psb(particles(n)%group) /= pst(particles(n)%group) ) THEN
374	particles(n)%z = particles(n)%z + &
375	( random_function( iran ) - 0.5 ) * &
376	pdz(particles(n)%group)
377	ENDIF
378	ENDDO
379	ENDIF
380
381	!
382	!-- Sort particles in the sequence the gridboxes are stored in the memory
383	CALL sort_particles
384
385	!
386	!-- Open file for statistical informations about particle conditions
387	IF ( write_particle_statistics ) THEN
388	CALL check_open( 80 )
389	WRITE ( 80, 8000 ) current_timestep_number, simulated_time, &
390	number_of_initial_particles, &
391	maximum_number_of_particles
392	CALL close_file( 80 )
393	ENDIF
394
395	!
396	!-- Check if particles are really uniform in color and radius (dvrp_size)
397	!-- (uniform_particles is preset TRUE)
398	IF ( uniform_particles ) THEN
399	IF ( number_of_initial_particles == 0 ) THEN
400	uniform_particles_l = .TRUE.
401	ELSE
402	n = number_of_initial_particles
403	IF ( MINVAL( particles(1:n)%dvrp_psize ) == &
404	MAXVAL( particles(1:n)%dvrp_psize ) .AND. &
405	MINVAL( particles(1:n)%color ) == &
406	MAXVAL( particles(1:n)%color ) ) THEN
407	uniform_particles_l = .TRUE.
408	ELSE
409	uniform_particles_l = .FALSE.
410	ENDIF
411	ENDIF
412
413	#if defined( __parallel )
414	CALL MPI_ALLREDUCE( uniform_particles_l, uniform_particles, 1, &
415	MPI_LOGICAL, MPI_LAND, comm2d, ierr )
416	#else
417	uniform_particles = uniform_particles_l
418	#endif
419
420	ENDIF
421
422	!
423	!-- Set the beginning of the particle tails and their age
424	IF ( use_particle_tails ) THEN
425	!
426	!-- Choose the maximum number of tails significantly larger than the
427	!-- one initially required
428	factor = 10.0
429	value = number_of_tails
430	DO WHILE ( value / 10.0 >= 1.0 )
431	factor = factor * 10.0
432	value = value / 10.0
433	ENDDO
434	maximum_number_of_tails = factor * INT( value )
435
436	ALLOCATE( particle_tail_coordinates(maximum_number_of_tailpoints,5, &
437	maximum_number_of_tails), &
438	new_tail_id(maximum_number_of_tails), &
439	tail_mask(maximum_number_of_tails) )
440
441	particle_tail_coordinates = 0.0
442	minimum_tailpoint_distance = minimum_tailpoint_distance**2
443	number_of_initial_tails = number_of_tails
444
445	nn = 0
446	DO n = 1, number_of_particles
447	!
448	!-- Only for those particles marked above with a provisional tail_id
449	!-- tails will be created. Particles now get their final tail_id.
450	IF ( particles(n)%tail_id /= 0 ) THEN
451
452	nn = nn + 1
453	particles(n)%tail_id = nn
454
455	particle_tail_coordinates(1,1,nn) = particles(n)%x
456	particle_tail_coordinates(1,2,nn) = particles(n)%y
457	particle_tail_coordinates(1,3,nn) = particles(n)%z
458	particle_tail_coordinates(1,4,nn) = particles(n)%color
459	particles(n)%tailpoints = 1
460	IF ( minimum_tailpoint_distance /= 0.0 ) THEN
461	particle_tail_coordinates(2,1,nn) = particles(n)%x
462	particle_tail_coordinates(2,2,nn) = particles(n)%y
463	particle_tail_coordinates(2,3,nn) = particles(n)%z
464	particle_tail_coordinates(2,4,nn) = particles(n)%color
465	particle_tail_coordinates(1:2,5,nn) = 0.0
466	particles(n)%tailpoints = 2
467	ENDIF
468
469	ENDIF
470	ENDDO
471	ENDIF
472
473	!
474	!-- Plot initial positions of particles (only if particle advection is
475	!-- switched on from the beginning of the simulation (t=0))
476	IF ( particle_advection_start == 0.0 ) CALL data_output_dvrp
477
478	ENDIF
479
480	!
481	!-- Check boundary condition and set internal variables
482	SELECT CASE ( bc_par_b )
483
484	CASE ( 'absorb' )
485	ibc_par_b = 1
486
487	CASE ( 'reflect' )
488	ibc_par_b = 2
489
490	CASE DEFAULT
491	IF ( myid == 0 ) THEN
492	PRINT*,'+++ init_particles: unknown boundary condition ', &
493	'bc_par_b = "', TRIM( bc_par_b ), '"'
494	ENDIF
495	CALL local_stop
496
497	END SELECT
498	SELECT CASE ( bc_par_t )
499
500	CASE ( 'absorb' )
501	ibc_par_t = 1
502
503	CASE ( 'reflect' )
504	ibc_par_t = 2
505
506	CASE DEFAULT
507	IF ( myid == 0 ) THEN
508	PRINT*,'+++ init_particles: unknown boundary condition ', &
509	'bc_par_t = "', TRIM( bc_par_t ), '"'
510	ENDIF
511	CALL local_stop
512
513	END SELECT
514	SELECT CASE ( bc_par_lr )
515
516	CASE ( 'cyclic' )
517	ibc_par_lr = 0
518
519	CASE ( 'absorb' )
520	ibc_par_lr = 1
521
522	CASE ( 'reflect' )
523	ibc_par_lr = 2
524
525	CASE DEFAULT
526	IF ( myid == 0 ) THEN
527	PRINT*,'+++ init_particles: unknown boundary condition ', &
528	'bc_par_lr = "', TRIM( bc_par_lr ), '"'
529	ENDIF
530	CALL local_stop
531
532	END SELECT
533	SELECT CASE ( bc_par_ns )
534
535	CASE ( 'cyclic' )
536	ibc_par_ns = 0
537
538	CASE ( 'absorb' )
539	ibc_par_ns = 1
540
541	CASE ( 'reflect' )
542	ibc_par_ns = 2
543
544	CASE DEFAULT
545	IF ( myid == 0 ) THEN
546	PRINT*,'+++ init_particles: unknown boundary condition ', &
547	'bc_par_ns = "', TRIM( bc_par_ns ), '"'
548	ENDIF
549	CALL local_stop
550
551	END SELECT
552	!
553	!-- Formats
554	8000 FORMAT (I6,1X,F7.2,4X,I6,71X,I6)
555
556	#endif
557	END SUBROUTINE init_particles

Note: See TracBrowser for help on using the repository browser.

Download in other formats:

| Impressum | ©Leibniz Universität Hannover |