!> Collapses D-dimensional output to D-1 view by line-of-sight integration ! for now: only for non-stretched cartesian cases, with LOS along coordinate ! writes out as either csv or vti file (collapse_type) module mod_collapse use mod_global_parameters use mod_comm_lib, only: mpistop implicit none contains !============================================================================= subroutine write_collapsed use mod_global_parameters ! Writes a collapsed view of the data integrated over one grid-direction. ! E.g. column density maps. ! Uses flat interpolation throughout. ! by Oliver Porth ! 6.Nov 2013 integer :: idir logical, save :: firstcollapse=.true. !----------------------------------------------------------------------------- if (firstcollapse) then firstcollapse=.false. end if do idir=1, ndim if (collapse(idir)) call put_collapse(idir) end do collapsenext=collapsenext+1 end subroutine write_collapsed !============================================================================= subroutine put_collapse(dir) use mod_forest, only: Morton_start, Morton_stop, sfc_to_igrid use mod_slice, only:alloc_subnode, dealloc_subnode use mod_global_parameters integer, intent(in) :: dir ! .. local .. integer :: jgrid, igrid, Morton_no double precision,dimension(0:nw+nwauxio) :: normconv !----------------------------------------------------------------------------- if(.not.slab) call mpistop("collapse only for slab cartesian cases") call allocate_collapsed(dir) jgrid=0 do Morton_no=Morton_start(mype),Morton_stop(mype) igrid=sfc_to_igrid(Morton_no) jgrid=jgrid+1 call alloc_subnode(jgrid,dir,nwauxio) call collapse_subnode(igrid,jgrid,dir,normconv) call integrate_subnode(igrid,jgrid,dir) end do ! Reduce to head-node: if (mype==0) then call MPI_REDUCE(MPI_IN_PLACE,collapsedData,size(collapsedData),& MPI_DOUBLE_PRECISION,MPI_SUM,0,icomm,ierrmpi) else call MPI_REDUCE(collapsedData,collapsedData,size(collapsedData),& MPI_DOUBLE_PRECISION,MPI_SUM,0,icomm,ierrmpi) end if call MPI_BARRIER(icomm, ierrmpi) select case(collapse_type) case('vti') call output_collapsed_vti(dir,normconv) case('csv') call output_collapsed_csv(dir,normconv) case('default') call mpistop("Unknown filetype for collapsed views") end select ! If we need the subnodes later, remove deallocation here: do jgrid=1,Morton_stop(mype)-Morton_start(mype)+1 call dealloc_subnode(jgrid) end do deallocate(collapsedData) end subroutine put_collapse !============================================================================= subroutine output_collapsed_csv(dir,normconv) use mod_global_parameters use mod_calculate_xw integer, intent(in) :: dir double precision,dimension(0:nw+nwauxio),intent(in):: normconv character(len=1024) :: filename, outfilehead, line logical :: fileopen integer :: iw character(len=name_len) :: wnamei(1:nw+nwauxio),xandwnamei(1:ndim+nw+nwauxio) integer, dimension(ndim) :: myshape integer :: ix1,ix2 double precision :: dxdim1,dxdim2, xdimmin1,& xdimmin2,xdimmax1,xdimmax2 !----------------------------------------------------------------------------- if (mype==0) then ! Get coordinates: select case(dir) case (1) dxdim1 = dx(2,collapseLevel) dxdim2 = dx(3,collapseLevel) xdimmin1=xprobmin2;xdimmax1=xprobmax2; xdimmin2=xprobmin3;xdimmax2=xprobmax3; case (2) dxdim1 = dx(1,collapseLevel) dxdim2 = dx(3,collapseLevel) xdimmin1=xprobmin1;xdimmax1=xprobmax1; xdimmin2=xprobmin3;xdimmax2=xprobmax3; case (3) dxdim1 = dx(1,collapseLevel) dxdim2 = dx(2,collapseLevel) xdimmin1=xprobmin1;xdimmax1=xprobmax1; xdimmin2=xprobmin2;xdimmax2=xprobmax2; case default dxdim1 = 1 dxdim2 = 1 xdimmin1=xprobmin2;xdimmax1=xprobmax2; xdimmin2=xprobmin3;xdimmax2=xprobmax3; call mpistop("slice direction not clear in output_collapsed_csv") end select inquire(unitcollapse,opened=fileopen) if(.not.fileopen)then ! generate filename: write(filename,"(a,i1.1,a,i1.1,a,i4.4,a)") trim(base_filename) // '_d',& dir,'_l',collapseLevel,'_n',collapsenext,'.csv' open(unitcollapse,file=filename,status='unknown',form='formatted') end if ! get and write the header: call getheadernames(wnamei,xandwnamei,outfilehead) line='' do iw=1,ndim+nw+nwauxio-1 if (iw .eq. dir) cycle line = trim(line)//trim(xandwnamei(iw))//', ' end do line = trim(line)//trim(xandwnamei(ndim+nw+nwauxio)) write(unitcollapse,'(a)')trim(line) myshape = shape(collapsedData) do ix2 = 1,myshape(2) do ix1 = 1,myshape(1) ! following assumes uniform cartesian grid write(unitcollapse,'(200(1pe20.12))') dxdim1*dble(ix1)+xdimmin1,& dxdim2*dble(ix2)+xdimmin2, (collapsedData(ix1,ix2,iw)*normconv(iw),iw=1,& nw+nwauxio) enddo enddo close(unitcollapse) end if end subroutine output_collapsed_csv !============================================================================= subroutine output_collapsed_vti(dir,normconv) use mod_global_parameters use mod_calculate_xw integer, intent(in) :: dir double precision,dimension(0:nw+nwauxio),intent(in):: normconv character(len=1024) :: filename, outfilehead, line logical :: fileopen integer :: iw character(len=name_len) :: wnamei(1:nw+nwauxio),xandwnamei(1:ndim+nw+nwauxio) integer, dimension(ndim) :: myshape integer :: ix1,ix2 double precision :: dxdim1,dxdim2, xdimmin1,& xdimmin2,xdimmax1,xdimmax2 double precision :: origin(1:3), spacing(1:3) integer :: wholeExtent(1:6),& size_single, length, size_length integer*8 :: offset character :: buf !----------------------------------------------------------------------------- if (mype==0) then offset=0 size_single=4 size_length=4 ! Get coordinates: select case(dir) case (1) dxdim1 = dx(2,1)*2.0d0**(1-collapseLevel) dxdim2 = dx(3,1)*2.0d0**(1-collapseLevel) xdimmin1=xprobmin2;xdimmax1=xprobmax2; xdimmin2=xprobmin3;xdimmax2=xprobmax3; case (2) dxdim1 = dx(1,1)*2.0d0**(1-collapseLevel) dxdim2 = dx(3,1)*2.0d0**(1-collapseLevel) xdimmin1=xprobmin1;xdimmax1=xprobmax1; xdimmin2=xprobmin3;xdimmax2=xprobmax3; case (3) dxdim1 = dx(1,1)*2.0d0**(1-collapseLevel) dxdim2 = dx(2,1)*2.0d0**(1-collapseLevel) xdimmin1=xprobmin1;xdimmax1=xprobmax1; xdimmin2=xprobmin2;xdimmax2=xprobmax2; case default dxdim1 = 1 dxdim2 = 1 xdimmin1=xprobmin2;xdimmax1=xprobmax2; xdimmin2=xprobmin3;xdimmax2=xprobmax3; call mpistop("slice direction not clear in output_collapsed_vti") end select origin=0 spacing=zero wholeExtent=0 myshape = shape(collapsedData) length = myshape(1)*myshape(2) length = length*size_single wholeExtent(1*2)=myshape(1); wholeExtent(2*2)=myshape(2); spacing(1) = dxdim1; spacing(2) = dxdim2; origin(1) = xdimmin1; origin(2) = xdimmin2; inquire(unitcollapse,opened=fileopen) if(.not.fileopen)then ! generate filename: write(filename,"(a,i1.1,a,i1.1,a,i4.4,a)") trim(base_filename)//'_d',dir,& '_l',collapseLevel,'_n',collapsenext,'.vti' open(unitcollapse,file=filename,status='unknown',form='formatted') end if ! get the header: call getheadernames(wnamei,xandwnamei,outfilehead) ! generate xml header write(unitcollapse,'(a)')'' write(unitcollapse,'(a)',advance='no') '' else write(unitcollapse,'(a)')' version="0.1" byte_order="BigEndian">' endif ! following corresponds to uniform cartesian grid write(unitcollapse,'(a,3(1pe14.6),a,6(i10),a,3(1pe14.6),a)')& ' ' write(unitcollapse,'(a)')'' write(unitcollapse,'(2a)')'' write(unitcollapse,*) real(global_time*time_convert_factor) write(unitcollapse,'(a)')'' write(unitcollapse,'(a)')'' ! we write one VTK PIECE write(unitcollapse,'(a,6(i10),a)') '' write(unitcollapse,'(a)')'' do iw=1,nw+nwauxio if(iw<=nw) then if(.not.w_write(iw)) cycle endif write(unitcollapse,'(a,a,a,i16,a)')'' offset = offset + length + size_length enddo write(unitcollapse,'(a)')'' write(unitcollapse,'(a)')'' write(unitcollapse,'(a)')'' write(unitcollapse,'(a)')'' close(unitcollapse) open(unitcollapse,file=filename,access='stream',form='unformatted',& position='append') buf='_' write(unitcollapse) TRIM(buf) do iw=1,nw+nwauxio if(iw<=nw) then if(.not.w_write(iw)) cycle endif write(unitcollapse) length write(unitcollapse) ((real(collapsedData(ix1,ix2,iw)*normconv(iw)),ix1=1,& myshape(1)),ix2=1,myshape(2)) enddo close(unitcollapse) open(unitcollapse,file=filename,status='unknown',form='formatted',& position='append') write(unitcollapse,'(a)')'' write(unitcollapse,'(a)')'' close(unitcollapse) end if end subroutine output_collapsed_vti !============================================================================= subroutine allocate_collapsed(dir) use mod_global_parameters integer, intent(in) :: dir integer :: dim1,dim2,dim3 integer :: nx1,nx2,nx3 !----------------------------------------------------------------------------- ! allocate array for the collapsed data: ! number of cells per grid. nx1=ixMhi1-ixMlo1+1;nx2=ixMhi2-ixMlo2+1;nx3=ixMhi3-ixMlo3+1; dim1=ng1(1)*2**(collapselevel-1)*nx1 dim2=ng2(1)*2**(collapselevel-1)*nx2 dim3=ng3(1)*2**(collapselevel-1)*nx3 select case(dir) case (1) allocate(collapsedData(dim2,dim3,nw+nwauxio)) case (2) allocate(collapsedData(dim1,dim3,nw+nwauxio)) case (3) allocate(collapsedData(dim1,dim2,nw+nwauxio)) case default call mpistop("slice direction not clear in allocate_collapsed") end select collapsedData = zero end subroutine allocate_collapsed !============================================================================= subroutine integrate_subnode(igrid,jgrid,dir) use mod_forest, only: tree_node_ptr, igrid_to_node use mod_global_parameters integer, intent(in) :: igrid, jgrid, dir ! .. local .. type(tree_node_ptr) :: tree integer :: nx1,nx2,nx3 integer :: ig1,ig2,ig3, level, ig1targetmin,& ig2targetmin,ig3targetmin, ig1targetmax,ig2targetmax,ig3targetmax integer :: ix1targetmin,ix1targetmax,& ix2targetmin,ix2targetmax,ix3targetmin,ix3targetmax, idim1targetmin,& idim1targetmax,idim2targetmin,idim2targetmax,idim3targetmin,idim3targetmax integer :: ixMdimlo1,ixMdimlo2,ixMdimlo3,& ixMdimhi1,ixMdimhi2,ixMdimhi3, ix1orig,ix2orig,ix3orig, ix1,ix2,ix3 integer :: igdim1,igdim2 !----------------------------------------------------------------------------- tree%node => igrid_to_node(igrid, mype)%node ig1 = tree%node%ig1; ig2 = tree%node%ig2; ig3 = tree%node%ig3; level = tree%node%level ! number of cells per grid. nx1=ixMhi1-ixMlo1+1;nx2=ixMhi2-ixMlo2+1;nx3=ixMhi3-ixMlo3+1; ! assumes uniform cartesian grid with factor 2 refinement per dimension if (level > collapseLevel) then ig1targetmin = int(dble(ig1-1)/2.0d0**(level-collapseLevel))+1 ig1targetmax = ig1targetmin else if (level < collapseLevel) then ig1targetmin = int(2.0d0**(collapseLevel-level))*(ig1-1)+1 ig1targetmax = int(2.0d0**(collapseLevel-level))*ig1 else ig1targetmin = ig1 ig1targetmax = ig1 end if ix1targetmin = nx1*(ig1targetmin-1)+1 ix1targetmax = nx1*ig1targetmax if (level > collapseLevel) then ig2targetmin = int(dble(ig2-1)/2.0d0**(level-collapseLevel))+1 ig2targetmax = ig2targetmin else if (level < collapseLevel) then ig2targetmin = int(2.0d0**(collapseLevel-level))*(ig2-1)+1 ig2targetmax = int(2.0d0**(collapseLevel-level))*ig2 else ig2targetmin = ig2 ig2targetmax = ig2 end if ix2targetmin = nx2*(ig2targetmin-1)+1 ix2targetmax = nx2*ig2targetmax if (level > collapseLevel) then ig3targetmin = int(dble(ig3-1)/2.0d0**(level-collapseLevel))+1 ig3targetmax = ig3targetmin else if (level < collapseLevel) then ig3targetmin = int(2.0d0**(collapseLevel-level))*(ig3-1)+1 ig3targetmax = int(2.0d0**(collapseLevel-level))*ig3 else ig3targetmin = ig3 ig3targetmax = ig3 end if ix3targetmin = nx3*(ig3targetmin-1)+1 ix3targetmax = nx3*ig3targetmax select case(dir) case (1) igdim1=(ig2-1)*nx2 igdim2=(ig3-1)*nx3 idim1targetmin=ix2targetmin;idim1targetmax=ix2targetmax; idim2targetmin=ix3targetmin;idim2targetmax=ix3targetmax; ixMdimlo1=ixMlo2;ixMdimhi1=ixMhi2; ixMdimlo2=ixMlo3;ixMdimhi2=ixMhi3; case (2) igdim1=(ig1-1)*nx1 igdim2=(ig3-1)*nx3 idim1targetmin=ix1targetmin;idim1targetmax=ix1targetmax; idim2targetmin=ix3targetmin;idim2targetmax=ix3targetmax; ixMdimlo1=ixMlo1;ixMdimhi1=ixMhi1; ixMdimlo2=ixMlo3;ixMdimhi2=ixMhi3; case (3) igdim1=(ig1-1)*nx1 igdim2=(ig2-1)*nx2 idim1targetmin=ix1targetmin;idim1targetmax=ix1targetmax; idim2targetmin=ix2targetmin;idim2targetmax=ix2targetmax; ixMdimlo1=ixMlo1;ixMdimhi1=ixMhi1; ixMdimlo2=ixMlo2;ixMdimhi2=ixMhi2; case default call mpistop("slice direction not clear in integrate_subnode") end select if (level >= collapseLevel) then do ix1orig = ixMdimlo1,ixMdimhi1 do ix2orig = ixMdimlo2,ixMdimhi2 ix1 = int(dble(ix1orig-nghostcells+igdim1-1)*2.0d0**(collapseLevel-level))+1 ix2 = int(dble(ix2orig-nghostcells+igdim2-1)*2.0d0**(collapseLevel-level))+1 collapsedData(ix1,ix2,1:nw+nwauxio) = collapsedData(ix1,ix2,& 1:nw+nwauxio) + ps_sub(jgrid)%w(ix1orig,ix2orig,& 1:nw+nwauxio) / 2.0d0**(2*(level-collapseLevel)) end do end do else do ix1 = idim1targetmin,idim1targetmax do ix2 = idim2targetmin,idim2targetmax ix1orig = int(dble(ix1-idim1targetmin)/2.0d0**(collapseLevel-level))+1+& nghostcells ix2orig = int(dble(ix2-idim2targetmin)/2.0d0**(collapseLevel-level))+1+& nghostcells collapsedData(ix1,ix2,1:nw+nwauxio) = collapsedData(ix1,ix2,& 1:nw+nwauxio) + ps_sub(jgrid)%w(ix1orig,ix2orig,1:nw+nwauxio) enddo enddo end if end subroutine integrate_subnode !============================================================================= subroutine collapse_subnode(igrid,jgrid,dir,normconv) use mod_usr_methods, only: usr_aux_output use mod_global_parameters use mod_physics, only: phys_to_primitive use mod_calculate_xw integer, intent(in) :: igrid, jgrid, dir double precision,dimension(0:nw+nwauxio),intent(out) :: normconv ! .. local .. integer :: ix, iw double precision :: dx1,dx2,dx3 double precision, dimension(ixMlo1-1:ixMhi1,ixMlo2-1:ixMhi2,ixMlo3-1:ixMhi3,& ndim) :: xC_TMP, xC double precision, dimension(ixMlo1:ixMhi1,ixMlo2:ixMhi2,ixMlo3:ixMhi3,& ndim) :: xCC_TMP, xCC double precision, dimension(ixMlo1-1:ixMhi1,ixMlo2-1:ixMhi2,ixMlo3-1:ixMhi3,& nw+nwauxio) :: wC_TMP double precision, dimension(ixMlo1:ixMhi1,ixMlo2:ixMhi2,ixMlo3:ixMhi3,& nw+nwauxio) :: wCC_TMP integer :: ixCmin1,ixCmin2,ixCmin3,ixCmax1,ixCmax2,ixCmax3,& ixCCmin1,ixCCmin2,ixCCmin3,ixCCmax1,ixCCmax2,ixCCmax3 !----------------------------------------------------------------------------- ps_sub(jgrid)%w=zero dx1=rnode(rpdx1_,igrid);dx2=rnode(rpdx2_,igrid);dx3=rnode(rpdx3_,igrid); call calc_x(igrid,xC,xCC) call calc_grid(unitslice,igrid,xC,xCC,xC_TMP,xCC_TMP,wC_TMP,wCC_TMP,normconv,& ixCmin1,ixCmin2,ixCmin3,ixCmax1,ixCmax2,ixCmax3,ixCCmin1,ixCCmin2,ixCCmin3,& ixCCmax1,ixCCmax2,ixCCmax3,.true.) select case (dir) case (1) if(slab_uniform) then do ix=ixMlo1,ixMhi1 ps_sub(jgrid)%w(ixMlo2:ixMhi2,ixMlo3:ixMhi3,& 1:nw+nwauxio) = ps_sub(jgrid)%w(ixMlo2:ixMhi2,ixMlo3:ixMhi3,& 1:nw+nwauxio) + wCC_TMP(ix,ixMlo2:ixMhi2,ixMlo3:ixMhi3,& 1:nw+nwauxio) * dx1 end do else do iw=1,nw+nwauxio do ix=ixMlo1,ixMhi1 ps_sub(jgrid)%w(ixMlo2:ixMhi2,ixMlo3:ixMhi3,& iw) = ps_sub(jgrid)%w(ixMlo2:ixMhi2,ixMlo3:ixMhi3,iw) + wCC_TMP(ix,& ixMlo2:ixMhi2,ixMlo3:ixMhi3,iw) * ps(igrid)%dx(ix,ixMlo2:ixMhi2,& ixMlo3:ixMhi3,1) end do end do end if ps_sub(jgrid)%x(ixMlo2:ixMhi2,ixMlo3:ixMhi3,1:ndim) = ps(igrid)%x(ixMlo1,& ixMlo2:ixMhi2,ixMlo3:ixMhi3,1:ndim) case (2) if(slab_uniform) then do ix=ixMlo2,ixMhi2 ps_sub(jgrid)%w(ixMlo1:ixMhi1,ixMlo3:ixMhi3,& 1:nw+nwauxio) = ps_sub(jgrid)%w(ixMlo1:ixMhi1,ixMlo3:ixMhi3,& 1:nw+nwauxio) + wCC_TMP(ixMlo1:ixMhi1,ix,ixMlo3:ixMhi3,& 1:nw+nwauxio) * dx2 end do else do iw=1,nw+nwauxio do ix=ixMlo2,ixMhi2 ps_sub(jgrid)%w(ixMlo1:ixMhi1,ixMlo3:ixMhi3,& iw) = ps_sub(jgrid)%w(ixMlo1:ixMhi1,ixMlo3:ixMhi3,& iw) + wCC_TMP(ixMlo1:ixMhi1,ix,ixMlo3:ixMhi3,& iw) * ps(igrid)%dx(ixMlo1:ixMhi1,ix,ixMlo3:ixMhi3,2) end do end do endif ps_sub(jgrid)%x(ixMlo1:ixMhi1,ixMlo3:ixMhi3,& 1:ndim) = ps(igrid)%x(ixMlo1:ixMhi1,ixMlo2,ixMlo3:ixMhi3,1:ndim) case (3) if(slab_uniform) then do ix=ixMlo3,ixMhi3 ps_sub(jgrid)%w(ixMlo1:ixMhi1,ixMlo2:ixMhi2,& 1:nw+nwauxio) = ps_sub(jgrid)%w(ixMlo1:ixMhi1,ixMlo2:ixMhi2,& 1:nw+nwauxio) + wCC_TMP(ixMlo1:ixMhi1,ixMlo2:ixMhi2,ix,& 1:nw+nwauxio) * dx3 end do else do iw=1,nw+nwauxio do ix=ixMlo3,ixMhi3 ps_sub(jgrid)%w(ixMlo1:ixMhi1,ixMlo2:ixMhi2,& iw) = ps_sub(jgrid)%w(ixMlo1:ixMhi1,ixMlo2:ixMhi2,& iw) + wCC_TMP(ixMlo1:ixMhi1,ixMlo2:ixMhi2,ix,& iw) * ps(igrid)%dx(ixMlo1:ixMhi1,ixMlo2:ixMhi2,ix,3) end do end do endif ps_sub(jgrid)%x(ixMlo1:ixMhi1,ixMlo2:ixMhi2,& 1:ndim) = ps(igrid)%x(ixMlo1:ixMhi1,ixMlo2:ixMhi2,ixMlo3,1:ndim) case default print*, 'subnode, dir: ', dir call mpistop("slice direction not clear in collapse_subnode") end select end subroutine collapse_subnode !============================================================================= end module mod_collapse