【LFW大规模测试的准备1】基于java接口实现LFW数据规范写入写出

java接口定义

package preprocess;

import java.util.List;

public interface PreProcess {

    List<NamePair> getFileNameList(char[] path, int index) ;  
    //输入txt文件路径，及组名，返回图片名称对

    boolean setResult( char[] path, List<ResultPair> resultPair );           
    // 输入ROC点列 ，保存为txt


}

定义了两个类， NamePair是图片对的名称，ResultPair是ROC数据结果的类。

package preprocess;

public class NamePair {
    char[] namePhoto1;
    char[] namePhoto2;
}

package preprocess;

public class ResultPair {
    float truePositiverate;
    float falsePositiverate;
    public ResultPair(){

    }; //构造函数
    public ResultPair(float f1, float f2){
        truePositiverate=f1;
        falsePositiverate=f2;
    };  //构造函数， 可用于直接赋值


}

接口的实现
定义一个Cire的类，来实现接口
1. 首先写了getLineName的一个内部函数， 功能是输入文本内容，输出各图片的全名。 主要用到string的split技术；
2. 具体实现getFileNameList，setResult这个接口的两个方法。 且接口中声明几个方法，需全部实现；
3. 在public static void main (String[] args) 主函数中，定义了编译的入口，并演示了该接口函数的功能效果。

package preprocess;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

public class Cire implements PreProcess {

    private NamePair getLineName( String LineContex){
        NamePair photoName = new NamePair();
        String[] aa=LineContex.split("\\t");
        if(aa.length==3){
            if(Integer.parseInt(aa[1])<10)
            {
                photoName.namePhoto1= (aa[0]+"_000"+aa[1]+".jpg").toCharArray();
            }else{
                photoName.namePhoto1= (aa[0]+"_00"+aa[1]+".jpg").toCharArray();
            }

            if(Integer.parseInt(aa[2])<10)
            {
                photoName.namePhoto2= (aa[0]+"_000"+aa[2]+".jpg").toCharArray();
            }else{
                photoName.namePhoto2= (aa[0]+"_00"+aa[2]+".jpg").toCharArray();
            }
        }else if(aa.length==4){
            if(Integer.parseInt(aa[1])<10)
            {
                photoName.namePhoto1= (aa[0]+"_000"+aa[1]+".jpg").toCharArray();
            }else{
                photoName.namePhoto1= (aa[0]+"_00"+aa[1]+".jpg").toCharArray();
            }

            if(Integer.parseInt(aa[3])<10)
            {
                photoName.namePhoto2= (aa[2]+"_000"+aa[3]+".jpg").toCharArray();
            }else{
                photoName.namePhoto2= (aa[2]+"_00"+aa[3]+".jpg").toCharArray();
            }
        }


        return photoName;
    } // 输入文本内容，输出规范化的图片名称

    public List<NamePair> getFileNameList(char[] path, int index) {
        //读取文件
        List<NamePair> photoNameList = new ArrayList<NamePair>();

        try{

            String encoding="GBK";
            File file= new File(String.valueOf(path));
            if(file.isFile()&& file.exists()){
                InputStreamReader read = new InputStreamReader(new FileInputStream(file),encoding);
                BufferedReader bufferedReader = new BufferedReader(read);
                String lineTxt =null;
                NamePair photoNamePair =new NamePair();  //对象必须new创建，否则容易报指针null的错误
                int i=1;
                while((lineTxt = bufferedReader.readLine())!= null ){

                    //每一行的名字+数字，作为name的输出
                    if(i<=(index*600)&& i>((index-1)*600))
                    {
                        photoNamePair= getLineName( lineTxt);
                        photoNameList.add(photoNamePair);

                        System.out.println(String.valueOf(photoNamePair.namePhoto1)+"    "+String.valueOf(photoNamePair.namePhoto2));

                    }

                    i++;
                }

                read.close();


            }else{
                System.out.println("找不到指定的文件");
            }


        }catch(Exception e){
            System.out.println("读取txt文件失败");
            e.printStackTrace();
        };

        return photoNameList;

    }  //输入txt文件路径，及组名，输出图片名称对 


    public boolean setResult( char[] path, List<ResultPair> resultPair  ) {

        File file= new File(String.valueOf(path));  //要保存的文件路径

        FileWriter out = null;
        try {
            out = new FileWriter( file );
        } catch (IOException e1) {
            // TODO Auto-generated catch block
            e1.printStackTrace();
        }

        for(int i=0;i<resultPair.size();i++){
            float data1= resultPair.get(i).truePositiverate;                    
            float data2= resultPair.get(i).falsePositiverate;
            String data= data1+"   "+data2;
            try {
                out.write(data);
            } catch (IOException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            }

            try {
                out.write("\r\n");
            } catch (IOException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            }
        }

        try {
            out.close();
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }

        return true ;  


    }  // 输入ROC点列， size 是数组的长度，保存为txt


    public static void main (String[] args){
        Cire C =new Cire();
        char[] path="D:\\pairs.txt".toCharArray();  //路径名
        List<NamePair> F= C.getFileNameList(path,10);   //得到对应组的图片名称


        List<ResultPair> resultPair= new ArrayList<ResultPair>();
        ResultPair obj1= new ResultPair(0.9f,0.1f);
        ResultPair obj2= new ResultPair(0.8f,0.2f);
        resultPair.add(obj1);
        resultPair.add(obj2);
        char[] path2="D:\\ROC.txt".toCharArray();  //输出ROC.txt的路径名
        C.setResult( path2, resultPair); //ROC保存到txt文件

    }

}