generate_heat_distribution.py
c371a820134dmaster

generate_heat_distribution.py
View Options

			##!/usr/bin/env python3
			## -- coding: utf-8 --
			#"""
			#Created on Mon Dec 10 09:57:09 2018
			#
			#@author: masc
			#"""
			import numpy as np
			import argparse
			import math
			import matplotlib.pyplot as plt

			parser = argparse.ArgumentParser()
			parser.add_argument("number", help="total number of particles", type=int)
			parser.add_argument("radius_heat_source", help="radius of the heat source")
			parser.add_argument("filename", help="name of generated input file")
			args = parser.parse_args()

			nb_part = args.number
			size = int(math.sqrt(nb_part))
			radius = float(args.radius_heat_source)


			domain = [[-1, 1], [-1, 1]]
			heat_distribution = np.zeros((size, size))
			for i in range(1, size+1):
			for j in range(0, size+1):
			x = domain[0][0] + (domain[0][1] - domain[0][0]) / (size+1) * i
			y = domain[1][0] + (domain[1][1] - domain[1][0]) / (size+1) * j
			if ( (x2 + y2) < radius ):
			heat_distribution[i, j] = 1

			fig = plt.figure()
			plt.imshow(heat_distribution)
			fig.suptitle('Heat distribution ')
			plt.show()

			np.savetxt(args.filename, heat_distribution, delimiter=" ")