/*
|
* Click nbfs://nbhost/SystemFileSystem/Templates/Licenses/license-default.txt to change this license
|
* Click nbfs://nbhost/SystemFileSystem/Templates/Classes/Class.java to edit this template
|
*/
|
package com.megatim.fdxconvert.util;
|
|
import static com.megatim.typefichier.validator.utilities.Utilities.getCharset;
|
import com.megatim.fdxconvert.pojo.FileToValidateDescription;
|
import java.io.BufferedReader;
|
import java.io.File;
|
import java.io.FileInputStream;
|
import java.io.InputStream;
|
import java.io.InputStreamReader;
|
import java.nio.charset.Charset;
|
import java.util.concurrent.atomic.AtomicBoolean;
|
import java.util.logging.Level;
|
import java.util.logging.Logger;
|
import org.apache.commons.csv.CSVFormat;
|
import org.apache.commons.csv.CSVParser;
|
import org.apache.commons.csv.CSVRecord;
|
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
|
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;
|
|
/**
|
*
|
* @author ASUS
|
*/
|
public class Count {
|
|
public static int countEltsOfTextFile(FileToValidateDescription fileDesc, boolean headerPresent) throws Exception {
|
int nbElts = 0;
|
|
if (fileDesc != null) {
|
Charset charset = getCharset(fileDesc.getFile());
|
|
//Lecture des lignes du fichier et chargement dans une liste
|
try ( FileInputStream fis = new FileInputStream(fileDesc.getFile()); InputStreamReader isr = new InputStreamReader(fis, charset); BufferedReader reader = new BufferedReader(isr)) {
|
|
boolean withHeader = headerPresent;
|
int c = 0;
|
boolean carriageReaded = false; //Indique si on a lu un retour chariot
|
char lastCharacter = '\u0000';
|
|
while ((c = reader.read()) != -1) {
|
|
char characterRead = (char) c;
|
|
if (characterRead == '\r') {
|
|
if (carriageReaded) { //On a lu le retour chariot deux fois consécutives donc on a pas de séparateur de ligne au sens windows
|
|
carriageReaded = false; //On marque que la lecture d'une nouvelle ligne redémarre
|
|
} else { // Première lecture du retour chariot
|
|
carriageReaded = true; // On marque que nous avons lu un retour chariot
|
|
}
|
|
} else {
|
|
if (carriageReaded) { // Le caractère précédemment lu c'est le retour chariot
|
|
if (characterRead == '\n') { // Si le nouveau caractère lu est un saut de ligne c'est que nous venons de lire une nouvelle ligne
|
|
if (!withHeader) {
|
|
if (fileDesc.getColumnDelimiter().equals("\\t")) {
|
nbElts++;
|
|
} else if (fileDesc.getColumnDelimiter().isEmpty()) {
|
nbElts++;
|
} else {
|
nbElts++;
|
}
|
} else {
|
withHeader = false;
|
}
|
carriageReaded = false; //On marque que la lecture d'une nouvelle ligne redémarre
|
|
} else {
|
}
|
} else {
|
if (characterRead == '\n') {
|
if (lastCharacter != '\n') {
|
}
|
} else {
|
|
}
|
}
|
}
|
lastCharacter = characterRead;
|
}
|
}
|
}
|
return nbElts;
|
}
|
|
public static int countEltsOfCsvFile(FileToValidateDescription fileDesc, boolean headerPresent) throws Exception {
|
int nbElts = 0;
|
|
Charset charset = getCharset(fileDesc.getFile());
|
AtomicBoolean withHeader = new AtomicBoolean(headerPresent);
|
|
try ( FileInputStream fis = new FileInputStream(fileDesc.getFile()); InputStreamReader isr = new InputStreamReader(fis, charset); BufferedReader reader = new BufferedReader(isr)) {
|
CSVParser parser = new CSVParser(reader, CSVFormat.DEFAULT.builder()
|
.setSkipHeaderRecord(headerPresent)
|
.setIgnoreSurroundingSpaces(true)
|
.setTrim(true)
|
.setDelimiter(fileDesc.getColumnDelimiter())
|
.setRecordSeparator(fileDesc.getRowDelimiter())
|
.build());
|
|
for (CSVRecord record : parser) {
|
|
if (!withHeader.get()) {
|
nbElts++;
|
} else {
|
withHeader.set(false);
|
}
|
}
|
} catch (Exception e) {
|
Logger.getLogger(ImportData.class.getName()).log(Level.SEVERE, e.getMessage(), e);
|
}
|
return nbElts;
|
}
|
|
public static int countEltsOfparseXlsxFile(File file, int sheetNumber, boolean headerPresent) throws Exception {
|
try ( InputStream is = new FileInputStream(file); Workbook wb = new XSSFWorkbook(is)) {
|
return countEltsOfExcelFile(wb, sheetNumber, headerPresent);
|
}
|
}
|
|
public static int countEltsOfXlsFile(File file, int sheetNumber, boolean headerPresent) throws Exception {
|
try ( InputStream is = new FileInputStream(file); Workbook wb = new HSSFWorkbook(is)) {
|
return countEltsOfExcelFile(wb, sheetNumber, headerPresent);
|
}
|
|
}
|
|
private static int countEltsOfExcelFile(Workbook wb, int sheetNumber, boolean headerPresent) {
|
int nbElts = 0;
|
Sheet sheet = wb.getSheetAt(sheetNumber);
|
AtomicBoolean withHeader = new AtomicBoolean(headerPresent);
|
|
for (Row r : sheet) {
|
if (!withHeader.get()) {
|
nbElts++;
|
} else {
|
withHeader.set(false);
|
}
|
}
|
return nbElts;
|
}
|
}
|
