/*******************************************************************************
* Copyright 2012 University of Southern California
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* 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.
*
* This code was developed by the Information Integration Group as part
* of the Karma project at the Information Sciences Institute of the
* University of Southern California. For more information, publications,
* and related projects, please see: http://www.isi.edu/integration
******************************************************************************/
package edu.isi.karma.controller.command.service;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import edu.isi.karma.controller.command.selection.SuperSelection;
import edu.isi.karma.rep.HNode;
import edu.isi.karma.rep.HNode.HNodeType;
import edu.isi.karma.rep.HTable;
import edu.isi.karma.rep.Node;
import edu.isi.karma.rep.RepFactory;
import edu.isi.karma.rep.Row;
import edu.isi.karma.rep.Worksheet;
import edu.isi.karma.rep.sources.Attribute;
import edu.isi.karma.rep.sources.Table;
public class ServiceTableUtil {
private static Logger logger = LoggerFactory.getLogger(ServiceTableUtil.class);
public static void populateEmptyWorksheet(Table table, Worksheet worksheet, RepFactory factory) {
logger.info("Populating an empty worksheet with the service data ...");
List<String> hNodeIdList = new ArrayList<String>();
hNodeIdList = addHeaders(table.getHeaders(), worksheet, factory);
edu.isi.karma.rep.Table dataTable = worksheet.getDataTable();
addRows(table.getValues(), worksheet, factory, hNodeIdList, dataTable);
}
public static void populateWorksheet(Table table, Worksheet worksheet, RepFactory factory, SuperSelection sel) {
logger.info("Populating existing worksheet with the service data ...");
List<String> oldHNodeIdList = new ArrayList<String>(worksheet.getHeaders().getHNodeIds());
List<String> hNodeIdList = new ArrayList<String>();
hNodeIdList = addHeaders(table.getHeaders(), worksheet, factory);
edu.isi.karma.rep.Table dataTable = worksheet.getDataTable();
updateRows(table.getValues(), table.getRowIds(), worksheet, factory, oldHNodeIdList, hNodeIdList, dataTable, sel);
}
private static List<String> addHeaders(List<Attribute> tableHeader, Worksheet worksheet, RepFactory factory) {
HTable headers = worksheet.getHeaders();
List<String> headersList = new ArrayList<String>();
for (int i = 0; i < tableHeader.size(); i++) {
Attribute att = tableHeader.get(i);
HNode hNode = headers.addHNode(att.getName(), HNodeType.Regular, worksheet, factory);
headersList.add(hNode.getId());
// very important
// update the hNodeId of the input/output attributes
att.sethNodeId(hNode.getId());
}
return headersList;
}
private static void addRows(List<List<String>> tableValues, Worksheet worksheet, RepFactory factory,
List<String> hNodeIdList, edu.isi.karma.rep.Table dataTable) {
for (List<String> rowValues : tableValues) {
if (rowValues == null || rowValues.size() == 0)
continue;
Row row = dataTable.addRow(factory);
for (int i = 0; i < rowValues.size(); i++)
row.setValue(hNodeIdList.get(i), rowValues.get(i), factory);
}
}
private static void updateRows(List<List<String>> tableValues, List<String> tableRowIds,
Worksheet worksheet, RepFactory factory,
List<String> oldHNodeIdList, List<String> hNodeIdList,
edu.isi.karma.rep.Table dataTable, SuperSelection sel) {
int rowsCount = dataTable.getNumRows();
List<Row> oldRows = dataTable.getRows(0, rowsCount, sel);
List<HashMap<String, String>> oldRowValues = new ArrayList<HashMap<String,String>>();
List<String> oldRowIds = new ArrayList<String>();
for (Row r : oldRows) {
HashMap<String, String> vals = new HashMap<String, String>();
for (Node n : r.getNodes()) {
vals.put(n.getHNodeId(), n.getValue().asString());
}
oldRowValues.add(vals);
oldRowIds.add(r.getId());
}
HashMap<String, String> currentRow = null;
String currentRowId = "";
int addedRowsCount = 0;
for (int i = 0; i < oldRowValues.size(); i++) {
currentRow = oldRowValues.get(i);
currentRowId = oldRowIds.get(i);
for (int k = 0; k < tableValues.size(); k++) {
List<String> rowValues = tableValues.get(k);
if (rowValues == null || rowValues.size() == 0)
continue;
String tableRowId = tableRowIds.get(k);
if (tableRowId == null || !tableRowId.trim().equalsIgnoreCase(currentRowId.trim())) {
continue;
}
addedRowsCount ++;
Row row = null;
if (addedRowsCount <= rowsCount)
row = dataTable.getRows(addedRowsCount - 1, 1, sel).get(0);
else
row = dataTable.addRow(factory);
// logger.debug(hNodeIdList.size());
for (int j = 0; j < hNodeIdList.size(); j++) {
// the first column in the table is the url column which should not be added to the table
// logger.debug("j:" + j);
// logger.debug(hNodeIdList.get(j));
// logger.debug(rowValues.get(j + 1));
row.setValue(hNodeIdList.get(j), rowValues.get(j), factory);
}
for (String id: oldHNodeIdList) {
row.setValue(id, currentRow.get(id).toString(), factory);
}
}
}
}
}