/*
* Copyright 2017 TWO SIGMA OPEN SOURCE, LLC
*
* 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.
*/
package com.twosigma.beaker;
import com.github.lwhite1.tablesaw.api.Table;
import com.twosigma.beaker.chart.xychart.Plot;
import com.twosigma.beaker.chart.xychart.plotitem.XYGraphics;
import com.twosigma.beaker.fileloader.CsvPlotReader;
import com.twosigma.beaker.mimetype.MIMEContainer;
import com.twosigma.beaker.table.TableDisplay;
import com.twosigma.beaker.widgets.DisplayableWidget;
import java.util.Collection;
import java.util.Map;
import static com.twosigma.beaker.mimetype.MIMEContainer.HIDDEN;
import static com.twosigma.beaker.mimetype.MIMEContainer.Text;
public class SerializeToString {
public static MIMEContainer doit(Object input) {
if (input == null) {
return Text("null");
}
return getMimeContainer(input);
}
private static MIMEContainer getMimeContainer(Object input) {
if (input instanceof DisplayableWidget) {
((DisplayableWidget) input).display();
return HIDDEN();
}
TableDisplay table = getTableDisplay(input);
if (table != null) {
table.display();
return HIDDEN();
}
if (input instanceof Table) {
new TableDisplay(new CsvPlotReader().convert((Table) input)).display();
return HIDDEN();
}
if (input instanceof XYGraphics) {
input = new Plot().add((XYGraphics) input);
((DisplayableWidget) input).display();
return HIDDEN();
}
if (input instanceof MIMEContainer) {
return (MIMEContainer) input;
}
return Text(input.toString());
}
public static TableDisplay getTableDisplay(Object input) {
TableDisplay ret = null;
if (input instanceof Map) {
Map map = (Map) input;
ret = new TableDisplay(map);
} else if (input instanceof Collection) {
Collection items = (Collection) input;
if (!items.isEmpty()) {
Object item = items.iterator().next();
if (item instanceof Map) {
ret = new TableDisplay(items);
}
}
} else if (input instanceof Map[]) {
Map[] items = (Map[]) input;
if (items.length > 0) {
ret = new TableDisplay(items);
}
}
return ret;
}
}