Fortraneries/Fraktalism/voxelize.f90

204 lines
6.3 KiB
Fortran

!-----------------------------------------------------
! VOXELIZE
! ========
! this is the main program, see also mkvoxvidz.sh
! showvoxels.pov and vox2inc.awk
!-----------------------------------------------------
program voxelize
use fraktals
implicit none
integer, parameter :: DIMC = 180
integer, dimension(:,:,:), allocatable :: cube
type(t_point3d), dimension(:), allocatable :: points
integer :: errcode, foo, argc
integer :: ix, iy, iz
integer :: nbr_points, maxcube
double precision, dimension(4) :: KA, KB, KM
double precision :: dmaxcube, delta
character(200) :: filename, string
write(0, *) "--- start of voxelize"
argc = IARGC()
if (2 .NE. argc) then
STOP ": VOXELIZE NEED PARAMETERS !"
endif
call getarg(1, filename)
call getarg(2, string) ; read (string, *) delta
write(0, "( ' --- delta = ', F11.6)") delta
allocate (cube(DIMC,DIMC,DIMC), stat=errcode)
if (0 .NE. errcode) then
STOP " : NO ENOUGH MEMORY FOR CUBE"
endif
nbr_points = 7000000
allocate(points(nbr_points), stat=errcode)
if (0 .NE. errcode) then
STOP " : NO ENOUGH MEMORY FOR POINTS"
endif
KA(1) = -1.3402 ; KA(2) = 1.5245
KA(3) = 1.0966 ; KA(4) = -2.3423
KB(1) = -1.2100 ; KB(2) = 1.3685
KB(3) = 1.3237 ; KB(4) = -2.3992
call interp4dp(KA, KB, KM, delta)
write(0, "(' --- coefs = ', 4F11.6)") KM
call compute_pickover(points, KM)
call clear_cube(cube)
!
! and now, we loop over all the pre-computed
! points of the attractor
!
do foo=1, nbr_points
call fcoor2icoor(points(foo)%x, ix)
call fcoor2icoor(points(foo)%y, iy)
call fcoor2icoor(points(foo)%z, iz)
cube(ix,iy,iz) = cube(ix,iy,iz) + 1
enddo
maxcube = MAXVAL(cube)
dmaxcube = DBLE(maxcube)
write(0, *) "--- maxval(cube) = ", maxcube
call spit_cube_as_union(filename, cube, &
maxcube/2000, dble(9000.00))
write(0, *) "--- end of voxelize"
!-----------------------------------------------------
contains
!-----------------------------------------------------
! or maybe, we can write a function ?
subroutine fcoor2icoor(in, out)
double precision, intent(in) :: in
integer, intent(out) :: out
double precision :: invalue
integer :: outvalue
invalue = (in + 2.0) / 2.0
outvalue = int(invalue * real(DIMC/2))
! add molly-guard here
out = outvalue
if (outvalue .LT. 1) out = 1
if (outvalue .GE. DIMC) out = DIMC-1
end subroutine
!------------------------------------------------------------
! USELESS USE OF LOOPS !
subroutine clear_cube(cube)
type(integer), dimension(:,:,:), intent(out) :: cube
integer :: i, j, k
do i=lbound(cube, 1), ubound(cube, 1)
do j=lbound(cube, 2), ubound(cube, 2)
do k=lbound(cube, 3), ubound(cube, 3)
cube(i, j, k) = 0
enddo
enddo
enddo
end subroutine
!------------------------------------------------------------
subroutine print_cube(cube, points, scaling)
type(integer), dimension(:,:,:), intent(in) :: cube
type(t_point3d), dimension(:), intent(in) :: points
double precision, intent(in) :: scaling
integer :: foo
do foo=1, nbr_points
call fcoor2icoor(points(foo)%x, ix)
call fcoor2icoor(points(foo)%y, iy)
call fcoor2icoor(points(foo)%z, iz)
print *, ix, iy, iz, &
cube(ix,iy,iz), &
DBLE(cube(ix,iy,iz)) / scaling
enddo
end subroutine
!------------------------------------------------------------
! generation Povray STL source file !
subroutine spit_cube_as_union(fname, voxels, limit, scaling)
character(*), intent(in) :: fname
type(integer), dimension(:,:,:), intent(in) :: voxels
integer, intent(in) :: limit
double precision, intent(in) :: scaling
integer :: fd, errcode
integer :: ix, iy, iz, maxv
integer :: nbrvox = 0
double precision :: bx, by, bz, valeur
character(200) :: chaine
! molly-guard
maxv = limit
if (maxv .LT. 2) maxv = 2
open (newunit=fd, file=trim(fname), &
status='replace', &
action='write', iostat=errcode)
if (0 .NE. errcode) then
STOP ' : SPIT UNION, FAIL OPEN OUTPUT FILE'
endif
write(fd, *) "// generated file, don't touch it bastard !"
write(fd, *) "// version 2.09"
write(fd, *) "#declare DIMC = ", DIMC, ";"
write(fd, *) "#declare Voxels = object {"
write(fd, *) "union {"
bx = 0.0 ; by = 0.0 ; bz = 0.0
do ix=lbound(voxels,1), ubound(voxels,1)
do iy=lbound(voxels,2), ubound(voxels,2)
do iz=lbound(voxels,3), ubound(voxels,3)
if (cube(ix,iy,iz) .LT. maxv) then
! print *, "foo = ", foo, cube(ix,iy,iz)
cycle ! REDO FROM START
endif
nbrvox = nbrvox + 1
bx = bx + dble(ix)
by = by + dble(iy)
bz = bz + dble(iz)
valeur = DBLE(cube(ix,iy,iz)) / scaling
! XXX
if (valeur .GT. 1.5) then
valeur = 1.5
endif
write(chaine, "( 'translate <', I4, ',', I4, ',', I4, '> ' )") &
ix, iy, iz
write(unit=fd, &
fmt="( 'object { VOXEL scale ', F11.6, 1X, A, ' }' )", &
iostat=errcode) &
valeur, trim(chaine)
enddo
enddo
enddo
write(fd, *) "} }"
write(fd, *) "// limit = ", limit
write(fd, "( '#declare NbrVox = ', I9, ';' )") nbrvox
write(fd, "( '#declare Bary_X = ', F11.6, ';' )") bx / dble(nbrvox)
write(fd, "( '#declare Bary_Y = ', F11.6, ';' )") by / dble(nbrvox)
write(fd, "( '#declare Bary_Z = ', F11.6, ';' )") bz / dble(nbrvox)
close(fd)
end subroutine
!-----------------------------------------------------
end program voxelize
!-----------------------------------------------------