cosolver_symmetrize.m
b3f94e22e87cllechot

cosolver_symmetrize.m
View Options

			function cosolver_symmetrize(resfile)
			% cosolver symmetrize symmetrizes the self-species
			% collision matrix /Caapj/Ciipj in the H5 resfile into the
			% new dataset /Caapj/Ciipj_full

			dataset = '/Caapj/Ciipj';
			Pmax = h5readatt(resfile,dataset,'Pmaxi')
			Jmax= h5readatt(resfile,dataset,'Jmaxi')
			numb = @(pp,jj) (Jmax + 1)*pp + 1 + jj

			% Get Ciipj
			Ciipj_mpi_balanced = h5read(resfile,'/Caapj/Ciipj');


			% Get rows labels
			rows_mpi_balanced= h5read(resfile,'/Caapj/rows_i');

			new_Ciipj = zeros(numb(Pmax,Jmax),numb(Pmax,Jmax));

			for irow=1:length(rows_mpi_balanced)
			row_tmp = rows_mpi_balanced(irow);
			new_Ciipj(row_tmp,:) = Ciipj_mpi_balanced(irow,:);
			end

			% Create full matrix
			Ciipj_full = new_Ciipj + transpose(new_Ciipj);% - diag(new_Ciipj);
			% remove diagonal as counted twice in the previous step
			for irow=1:numb(Pmax,Jmax)
			Ciipj_full(irow,irow) = Ciipj_full(irow,irow)/2;
			end

			% Create new dataset
			dim = double(numb(Pmax,Jmax));
			fprintf('Create new dataset "/Caapj/Ciipj_full"')
			h5create(resfile,'/Caapj/Ciipj_full',[dim dim])
			h5write(resfile,'/Caapj/Ciipj_full',Ciipj_full)
			end