VolumeLoader.java revision 5ee349cd7a75079b6bac1d8ace66455ecd1afb17
1/* 2 * Copyright (C) 2015 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17package com.example.android.rs.vr.loaders; 18 19import android.renderscript.RenderScript; 20import android.util.Log; 21 22import com.example.android.rs.vr.engine.Volume; 23 24import java.io.File; 25import java.io.FileReader; 26import java.io.FilenameFilter; 27import java.util.Arrays; 28import java.util.HashMap; 29import java.util.Properties; 30 31public class VolumeLoader { 32 private static final String LOGTAG = "VolumeLoader"; 33 HashMap<String, Properties> map = new HashMap<String, Properties>(); 34 File baseDir; 35 ProgressListener mListener; 36 37 public VolumeLoader(String dir) { 38 map.put(Mandelbulb.NAME,null); 39 map.put(Droid.NAME,null); 40 baseDir = new File(dir); 41 if (!baseDir.exists()) { 42 Log.e(LOGTAG, "Directory: \""+dir+"\" does not exist "); 43 return; 44 } 45 Properties[] prop = getPropertyFiles(baseDir); 46 for (int i = 0; i < prop.length; i++) { 47 map.put(prop[i].getProperty("name"), prop[i]); 48 } 49 50 } 51 52 public String[] getNames() { 53 String [] ret = map.keySet().toArray(new String[map.size()]); 54 Arrays.sort(ret); 55 return ret; 56 } 57 58 public Volume getVolume(RenderScript rs, String name) { 59 if (name.equals(Mandelbulb.NAME)) { 60 return Mandelbulb.buildRSVolume(rs,mListener); 61 } 62 if (name.equals(Droid.NAME)) { 63 return Droid.buildRSVolume(rs,mListener); 64 } 65 Properties p = map.get(name); 66 if (p == null) { 67 Log.v(LOGTAG,"Could not find "+name); 68 return null; 69 } 70 String dir = p.getProperty("dir"); 71 Log.v(LOGTAG,"dir ="+dir); 72 73 if ("dicom".equalsIgnoreCase(p.getProperty("format"))) { 74 Log.v(LOGTAG,"processing dicom"); 75 Volume v = LoaderDicom.buildRSVolume(rs, new File(baseDir, dir), mListener); 76 String [] looks = p.getProperty("looks").split(","); 77 for (int j = 0; j < looks.length; j++) { 78 String look_color = p.getProperty(looks[j]+".color"); 79 String look_opacity = p.getProperty(looks[j]+".opacity"); 80 v.addLook(looks[j],look_color,look_opacity); 81 } 82 return v; 83 } else if ("raw".equalsIgnoreCase(p.getProperty("format"))) { 84 Log.v(LOGTAG,"processing dicom"); 85 Volume v = LoaderRaw.buildRSVolume(rs, new File(baseDir, dir), p, mListener); 86 String [] looks = p.getProperty("looks").split(","); 87 for (int j = 0; j < looks.length; j++) { 88 String look_color = p.getProperty(looks[j]+".color"); 89 String look_opacity = p.getProperty(looks[j]+".opacity"); 90 v.addLook(looks[j],look_color,look_opacity); 91 } 92 return v; 93 } 94 Log.v(LOGTAG,"could recognize format"); 95 return null; 96 } 97 98 static Properties[] getPropertyFiles(File dir) { 99 100 File[] f = dir.listFiles(new FilenameFilter() { 101 102 @Override 103 public boolean accept(File dir, String name) { 104 Log.v(LOGTAG, name); 105 return name.endsWith(".prop"); 106 } 107 }); 108 Properties[]ret = new Properties[f.length]; 109 for (int i = 0; i < f.length; i++) { 110 Properties prop = new Properties(); 111 ret[i] = prop; 112 try { 113 prop.load(new FileReader(f[i])); 114 115 } catch (Exception e) { 116 e.printStackTrace(); 117 } 118 } 119 return ret; 120 } 121 122 public void setProgressListener(ProgressListener listener){ 123 mListener = listener; 124 } 125 126 public static interface ProgressListener { 127 public void progress(int n, int total); 128 } 129} 130