def test_from_file():
vw = View.from_file('./tests/assets/view.vf')
assert vw.identifier == 'view'
assert list(vw.position) == [0, 0, 0]
assert list(vw.direction) == [0, 0, 1]
assert list(vw.up_vector) == [0, 1, 0]
assert vw.h_size == 29.341
assert vw.v_size == 32.204
assert vw.shift == -0.500
assert vw.lift == -0.500
assert vw.fore_clip == 100
assert vw.aft_clip == 200
def from_view_file(cls: Camera, file_path: str) -> Camera:
"""Create a Camera object from a radiance view file.
Args:
file_path: A valid path to a radiance view file with .vf extension.
Returns:
A Camera object.
"""
view_file = Path(file_path)
if view_file.is_file() and view_file.as_posix()[-3:] == '.vf':
return Camera.from_view(view=View.from_file(view_file.as_posix()))
else:
raise FileNotFoundError('Radiance view file not found.')
def split_view(view, count, skip_overture, octree, rad_params, folder,
log_file):
"""Split a radiance view file into smaller views based on count.
\b
Args:
view: Full path to input sensor view file.
count: Maximum number of sensors in new files. The number will be rounded to
closest round number for each file. For example if the input file has 21
sensors and input count is set to 5 this command will generate 4 files where
the first three files will have 5 sensors and the last file will have 6.
"""
try:
# split the view into smaller views
view_obj = View.from_file(view)
views = view_obj.grid(y_div_count=count)
views_info = []
for c, v in enumerate(views):
name = '%s_%04d' % (view_obj.identifier, c)
path = '%s.vf' % name
full_path = os.path.join(folder, path)
v.to_file(folder, path, mkdir=True)
views_info.append({
'name': name,
'path': path,
'full_path': full_path
})
# create the ambient cache file if specified
amb_file = os.path.basename(view).replace('.vf', '.amb')
if not skip_overture:
options = RpictOptions()
if rad_params:
options.update_from_string(rad_params.strip())
# overwrite default image size to be small for the ambient cache (64 x 64)
options.x = 64
options.y = 64
options.af = amb_file
# create command and run it to get the .amb file
assert octree is not None, \
'Octree must be specified for an overture calculation.'
out_file = os.path.basename(view).replace('.vf', '.unf')
rpict = Rpict(options=options,
output=out_file,
octree=octree,
view=view)
env = None
if folders.env != {}:
env = folders.env
env = dict(os.environ, **env) if env else None
rpict.run(env=env, cwd=folder)
os.remove(os.path.join(folder, out_file))
else: # write a dummy ambient file so that queenbee does not crash
write_to_file_by_name(folder, amb_file, '')
# record all of the view files that were generated
log_file.write(json.dumps(views_info))
except Exception:
_logger.exception('Failed to split view file.')
sys.exit(1)
else:
sys.exit(0)
def merge_view(input_folder, base_name, extension, scale_factor, folder, name):
"""Merge several radiance HDR image files into a single file.
This command will also perform an anti-aliasing operation on the output and
replace the view information in the header of the merged file if a single .vf
file is found within the root of the input-folder.
\b
Args:
input_folder: Input folder.
base_name: File base name. All of the files must start with base name and
continue with _ and an integer values.
extension: File extention. [Default: .unf]
"""
try:
# get all of the files in the folder with the given extension
pattern = r'{}_\d+{}'.format(base_name, extension)
images = sorted(f for f in os.listdir(input_folder)
if re.match(pattern, f))
if len(images) == 0:
raise ValueError('Found no files to merge.')
name = name or base_name
# get the new dir name as view name might be group/name
dirname = os.path.dirname(os.path.normpath(os.path.join(folder, name)))
if dirname and not os.path.exists(dirname):
os.makedirs(dirname)
temp_output = os.path.join(dirname, name + '_temp.HDR')
output_file = os.path.join(dirname, name + '.HDR')
# set up the pcompos command
in_dirname = os.path.normpath(input_folder)
pcompos = Pcompos(output=temp_output)
pcompos.input = [os.path.join(in_dirname, img) for img in images]
pcompos.options.a = 1
# setup the pfilt command to perform anti-aliasing on the output
pfilt = Pfilt(input=temp_output)
pfilt.options.r = 0.6
if scale_factor != 1:
pfilt.options.x = '/{}'.format(scale_factor)
pfilt.options.y = '/{}'.format(scale_factor)
# search for a single .vf in the folder and, if it's found, grab the info
views = sorted(f for f in os.listdir(input_folder)
if f.endswith('.vf'))
if len(views) == 1: # replace the header with the info in the view
view_obj = View.from_file(os.path.join(input_folder, views[0]))
getinfo = Getinfo(output=output_file)
getinfo.options.a = 'VIEW= {}'.format(view_obj)
pfilt.pipe_to = getinfo
else: # just let the output of pfilt be the final output
pfilt.output = output_file
# run the commands in series
env = None
if folders.env != {}:
env = folders.env
env = dict(os.environ, **env) if env else None
for r_cmd in (pcompos, pfilt):
r_cmd.run(env)
except Exception:
_logger.exception('Failed to merge image files.')
sys.exit(1)
else:
sys.exit(0)
