File size: 3,522 Bytes
bb654c7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 |
/*
* Copyright (C) 2018 Southern Illinois University Carbondale, SoftSearch Lab
*
* Author: Amiangshu Bosu
*
* Licensed under GNU LESSER GENERAL PUBLIC LICENSE Version 3, 29 June 2007
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.gnu.org/licenses/lgpl.html
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package edu.siu.sentise.model;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Iterator;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.CellType;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
public class SentimentData {
private String text;
private int rating;
public String getText() {
return text;
}
public void setText(String text) {
this.text = text;
}
public int getRating() {
return rating;
}
public void setRating(int rating) {
this.rating = rating;
}
public SentimentData(String text, int rating) {
this.text = text;
this.rating = rating;
}
public static ArrayList<SentimentData> parseSentimentData(String fileName) {
ArrayList<SentimentData> sentimentDataList = new ArrayList<>();
int rowCount=0;
try {
FileInputStream excelFile = new FileInputStream(new File(fileName));
Workbook workbook = new XSSFWorkbook(excelFile);
Sheet datatypeSheet = workbook.getSheetAt(0);
Iterator<Row> iterator = datatypeSheet.iterator();
while (iterator.hasNext()) {
Row currentRow = iterator.next();
rowCount++;
String text = "";
int rating = 0;
try {
Cell textCell=currentRow.getCell(0);
if(textCell.getCellTypeEnum()==CellType.STRING)// first column is text
text = textCell.getStringCellValue();
else if(textCell.getCellTypeEnum()==CellType.NUMERIC)
text = Double.toString(textCell.getNumericCellValue());
else if(textCell.getCellTypeEnum()==CellType.FORMULA)
text = textCell.getCellFormula();
rating = (int) currentRow.getCell(1).getNumericCellValue(); // second column is rating
if (rating <= 1 && rating >= -1) {
SentimentData sentimentData = new SentimentData(text, rating);
sentimentDataList.add(sentimentData);
//System.out.println(sentimentData.toString());
}
else {
System.out.println("Error: "+currentRow+"-> "+rating);
}
} catch (Exception e) {
System.out.println("Error parsing row:"+rowCount);
System.out.println(e.getMessage());
}
}
workbook.close();
} catch (FileNotFoundException e) {
System.out.println("Unable to open oracle file!");
System.out.println(e.getMessage());
System.exit(1);
} catch (IOException e) {
System.out.println("Unable to parse oracle file!");
System.exit(1);
}
return sentimentDataList;
}
public String toString() {
return text+" ["+rating+"]";
}
}
|