package edu.btbu.TDSearch.util.file;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileReader;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.nio.ByteBuffer;
import java.nio.CharBuffer;
import java.nio.channels.FileChannel;
import java.nio.charset.Charset;
import java.text.DecimalFormat;
import java.util.ArrayList;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import edu.btbu.TDSearch.util.shape.Face;
import edu.btbu.TDSearch.util.shape.Mesh;
import edu.btbu.TDSearch.util.shape.Point3D;
import edu.btbu.TDSearch.util.statistics.ModelInfo;
/**
* 解析obj文件
* @author 3L5M53X
*
*/
public class ObjParse {
// v float float float
private final static String vetexReg ="v( +[\\d|\\.|\\+|\\-|e]+)( +[\\d|\\.|\\+|\\-|e]+)( +[\\d|\\.|\\+|\\-|e]+)";
//vt float float
private final static String vtReg = "vt( +[\\d|\\.|\\+|\\-|e]+)( +[\\d|\\.|\\+|\\-|e]+)";
//vn float float
private final static String vnReg ="vn( +[\\d|\\.|\\+|\\-|e]+)( +[\\d|\\.|\\+|\\-|e]+)";
// f vertex vertex vertex ...
private final static String faceReg = "f( +[\\d]+)( [\\d]+)( [\\d]+)( [\\d]+)?";
// f vertex/uv vertex/uv vertex/uv ...
private final static String faceReg_v_uv="f( +([\\d]+)/([\\d]+))( ([\\d]+)/([\\d]+))( ([\\d]+)/([\\d]+))( ([\\d]+)/([\\d]+))?";
// f vertex/uv/normal vertex/uv/normal vertex/uv/normal ...
private final static String faceReg_v_uv_n="f( +([\\d]+)/([\\d]+)/([\\d]+))( ([\\d]+)/([\\d]+)/([\\d]+))( ([\\d]+)/([\\d]+)/([\\d]+))( ([\\d]+)/([\\d]+)/([\\d]+))?";
// f vertex//normal vertex//normal vertex//normal ...
private final static String faceReg_v_n="f( +([\\d]+)//([\\d]+))( ([\\d]+)//([\\d]+))( ([\\d]+)//([\\d]+))( ([\\d]+)/\\/([d]+))?";
public static Mesh getMesh(String url) throws IOException{
Mesh mesh = null;
ArrayList<Point3D> vetexList = new ArrayList<Point3D>();
ArrayList<Face> faceList = new ArrayList<Face>();
File file = new File(url);
String fileName= file.getName();
int position = fileName.lastIndexOf(".");
String meshID = fileName.substring(0,position);
if(!Pattern.matches(".+\\.obj$",fileName)){
System.out.println("The file type should be .obj,please check your file Type");
return null;
}
//vetex
CharSequence cs = fromFile(file);
Matcher matcher_v = Pattern.compile(vetexReg).matcher(cs);
while(matcher_v.find()){
Point3D point = new Point3D(Double.parseDouble(matcher_v.group(1)),Double.parseDouble(matcher_v.group(2)),Double.parseDouble(matcher_v.group(3)));
vetexList.add(point);
}
// face
Matcher matcher_f = Pattern.compile(faceReg).matcher(cs);
while(matcher_f.find()){
Point3D a = vetexList.get(Integer.parseInt(matcher_f.group(1).trim())-1);
Point3D b = vetexList.get(Integer.parseInt(matcher_f.group(2).trim())-1);
Point3D c = vetexList.get(Integer.parseInt(matcher_f.group(3).trim())-1);
Face face = new Face(a,b,c);
faceList.add(face);
}
Matcher matcher_fvuv = Pattern.compile(faceReg_v_uv).matcher(cs);
while(matcher_fvuv.find()){
System.out.println(matcher_fvuv.group(1));
}
mesh = new Mesh(meshID,vetexList.size(),faceList.size());
Point3D[] points = new Point3D[vetexList.size()];
vetexList.toArray(points);
mesh.setPointArray(points);
Face[] faces = new Face[faceList.size()];
faceList.toArray(faces);
mesh.setFaceArray(faces);
return mesh;
}
/**
* get the basic info,name \file size \num of vetex \num of face \bpv\bpf
* @param url
* @throws IOException
*/
public static ModelInfo getInfo(File file) throws IOException{
ModelInfo info = new ModelInfo();
info.setF_name(file.getName());
long f_size = file.length();
info.setF_size(file.length());
long v_size = 0;//vetex part size
long p_size = 0;//polygons part size
long vt_size = 0;
long vn_size = 0;
long v_num = 0;
long vt_num = 0;
long nt_num = 0;
long f_num = 0;
//vetex
CharSequence cs = FileIn.fromFile(file);
Matcher matcher_v = Pattern.compile(vetexReg).matcher(cs);
matcher_v.find();
while(matcher_v.find()){
v_size+=matcher_v.group().length()+1;
v_num++;
}
/* Matcher matcher_vt = Pattern.compile(vtReg).matcher(cs);
while(matcher_vt.find()){
v_size+=matcher_vt.group().length()+1;
}
Matcher matcher_vn = Pattern.compile(vnReg).matcher(cs);
while(matcher_vn.find()){
v_size+=matcher_vn.group().length()+1;
}*/
// face
Matcher matcher_f = Pattern.compile(faceReg).matcher(cs);
while(matcher_f.find()){
p_size+=matcher_f.group().length()+1;
f_num++;
}
Matcher matcher_fvuv = Pattern.compile(faceReg_v_uv).matcher(cs);
while(matcher_fvuv.find()){
p_size+=matcher_fvuv.group().length()+1;
f_num++;
/*
System.out.print(matcher_fvuv.group(0)+" ");//f 9/1 5/2 10/3
System.out.print(matcher_fvuv.group(1)+" ");// 9/1
System.out.print(matcher_fvuv.group(2)+" ");//9
System.out.print(matcher_fvuv.group(3)+" ");//1
*/
}
Matcher matcher_f_vn = Pattern.compile(faceReg_v_n).matcher(cs);
while(matcher_f_vn.find()){
p_size+=matcher_f_vn.group().length()+1;
f_num++;
System.out.print(matcher_f_vn.group(0)+" ");//f 998//1455 989//1446 990//1447
System.out.print(matcher_f_vn.group(1)+" ");// 998//1455
System.out.print(matcher_f_vn.group(2)+" ");// 998
System.out.print(matcher_f_vn.group(3)+" ");//1455
System.out.println();//2
}
Matcher matcher_f_v_uv_n = Pattern.compile(faceReg_v_uv_n).matcher(cs);
while(matcher_f_v_uv_n.find()){
p_size+=matcher_f_v_uv_n.group().length()+1;
f_num++;
}
info.setV_num(v_num);
info.setF_num(f_num);
System.out.println(v_size);
System.out.println(p_size);
System.out.println(v_size+p_size);
DecimalFormat format = new DecimalFormat("#.00");
info.setFpv(format.format(1.0*f_num/v_num));
info.setBpf(format.format(8.0*p_size/f_num));
info.setBpv(format.format(8.0*v_size/v_num));
return info;
}
// Converts the contents of a file into a CharSequence
// suitable for use by the regex package.
public static CharSequence fromFile(File filename) throws IOException {
FileInputStream fis = new FileInputStream(filename);
FileChannel fc = fis.getChannel();
long size = fc.size();
// Create a read-only CharBuffer on the file
ByteBuffer bbuf = fc.map(FileChannel.MapMode.READ_ONLY,size);
CharBuffer cbuf = Charset.forName("UTF-8").newDecoder().decode(bbuf);
return cbuf;
/* FileInputStream fis = new FileInputStream(filename);
FileChannel fc = fis.getChannel();
long size = fc.size();
// Create a read-only CharBuffer on the file
ByteBuffer bbuf = fc.map(FileChannel.MapMode.READ_ONLY,size);
CharBuffer cbuf = Charset.forName("ISO-8859-1").newDecoder().decode(bbuf);
return cbuf;*/
}
/**
* @param args
* @throws IOException F:\paper\3d\NTU3D\NTU3D.v1_0-999
*/
public static void main(String[] args) throws IOException {
System.out.println(getInfo(new File("F:\\Work\\数据压缩\\result\\obj\\hand.obj")));
//Y5001_R_hand.obj 51049 799 1606 2.0100125156445556 63.89111389236545 31.78642590286426
//System.out.println("192.168.69.84".hashCode());
}
}
