All Downloads are FREE. Search and download functionalities are using the official Maven repository.
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.viiyue.plugins.excel.metadata.Style Maven / Gradle / Ivy
/**
* Copyright (C) 2021 the original author or authors.
*
* 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.viiyue.plugins.excel.metadata;
import java.awt.Color;
import java.util.Objects;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
import java.util.concurrent.ThreadLocalRandom;
import org.apache.commons.lang3.StringUtils;
import org.apache.poi.hssf.usermodel.HSSFPalette;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.BorderStyle;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.CellStyle;
import org.apache.poi.ss.usermodel.FillPatternType;
import org.apache.poi.ss.usermodel.Font;
import org.apache.poi.ss.usermodel.IndexedColors;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.model.StylesTable;
import org.apache.poi.xssf.usermodel.XSSFCellStyle;
import org.apache.poi.xssf.usermodel.XSSFColor;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.viiyue.plugins.excel.converter.Styleable;
import com.viiyue.plugins.excel.enums.Alignment;
/**
* Cell beautification style
*
* @author tangxbai
* @since 2023/06/19
*/
public class Style {
private static final Logger log = LoggerFactory.getLogger( Styleable.class );
private static final int colorBegin = 8;
private static final int colorBound = IndexedColors.values().length - 1; // 64
private static final int colorSize = colorBound - colorBegin; // 56 colors
private static final ConcurrentMap fonts = new ConcurrentHashMap<>( 8 );
private static final ConcurrentMap colors = new ConcurrentHashMap<>( 8 );
private static final ConcurrentMap styles = new ConcurrentHashMap<>( 8 );
private static final ConcurrentMap beautifys = new ConcurrentHashMap<>( 64 );
private final Workbook wb;
private final CellStyle style;
private final boolean isXssf;
private final boolean isHssf;
private final String cacheKey;
public static final Style of( Workbook wb, String cacheKey ) {
Objects.requireNonNull( wb, "Workbook object cannot be null" );
String bsk = wb.getClass().getSimpleName() + ":" + cacheKey;
Style style = beautifys.get( bsk );
if ( style == null ) {
beautifys.put( bsk, style = new Style( wb, cacheKey ) ); // Avoid instantiating too many objects
}
return style;
}
private Style( Workbook wb, String cacheKey ) {
this.wb = wb;
this.cacheKey = cacheKey;
this.style = newStyle( cacheKey );
this.isHssf = wb instanceof HSSFWorkbook;
this.isXssf = wb instanceof XSSFWorkbook;
}
public boolean is( String cacheKey ) {
return Objects.equals( cacheKey, this.cacheKey ) || StringUtils.startsWith( this.cacheKey, cacheKey );
}
public Style autoWrap() {
style.setWrapText( true );
return this;
}
public Style font( int fontSize ) {
return font( fontSize, false );
}
public Style font( int fontSize, boolean blod ) {
return font( "default", null, fontSize, blod );
}
public Style font( String fontName, int fontSize ) {
return font( fontName, fontSize, false );
}
public Style font( String fontName, int fontSize, boolean blod ) {
return font( fontName, null, fontSize, blod );
}
public Style font( String fontName, String fontColor, int fontSize, boolean blod ) {
style.setFont( newFont( fontName, fontColor, fontSize, blod ) );
return this;
}
public Style fontColor( String fontColor ) {
return font( "default", fontColor, -1, false );
}
public Style bgColor( String bgColor ) {
style.setFillPattern( FillPatternType.SOLID_FOREGROUND );
if ( isHssf ) {
short colorIndex = getColorIndex( bgColor );
style.setFillForegroundColor( colorIndex );
} else if ( isXssf ) {
XSSFCellStyle xssfStyle = ( XSSFCellStyle ) style;
xssfStyle.setFillForegroundColor( newXssfColor( bgColor ) );
}
return this;
}
public Style alignment( Alignment alignment ) {
if ( alignment != null ) {
alignment.setAlignment( style );
}
return this;
}
public Style border( BorderStyle borderStyle ) {
style.setBorderBottom( borderStyle );
style.setBorderLeft( borderStyle );
style.setBorderTop( borderStyle );
style.setBorderRight( borderStyle );
return this;
}
public Style border( BorderStyle borderStyle, String borderColor ) {
border( borderStyle );
if ( isHssf ) {
short colorIndex = getColorIndex( borderColor );
style.setRightBorderColor( colorIndex );
style.setTopBorderColor( colorIndex );
style.setBottomBorderColor( colorIndex );
style.setLeftBorderColor( colorIndex );
} else if ( isXssf ) {
XSSFColor xssfColor = newXssfColor( borderColor );
XSSFCellStyle xssfStyle = ( XSSFCellStyle ) style;
xssfStyle.setRightBorderColor( xssfColor );
xssfStyle.setTopBorderColor( xssfColor );
xssfStyle.setBottomBorderColor( xssfColor );
xssfStyle.setLeftBorderColor( xssfColor );
}
return this;
}
public final void apply( Cell cell ) {
if ( cell != null ) {
cell.setCellStyle( style );
}
}
private CellStyle newStyle( String cacheKey ) {
CellStyle style = styles.get( cacheKey );
if ( style == null ) {
styles.put( cacheKey, style = wb.createCellStyle() );
}
return style;
}
private XSSFColor newXssfColor( String hexColor ) {
XSSFWorkbook xwb = ( XSSFWorkbook ) wb;
StylesTable source = xwb.getStylesSource();
return new XSSFColor( Color.decode( hexColor ), source.getIndexedColors() );
}
private Font newFont( String fontName, String fontColor, int fontSize, boolean blod ) {
if ( fontName == null ) {
fontName = "default";
}
if ( fontColor == null ) {
fontColor = "#";
}
String cacheKey = fontName + ":" + fontSize + ":" + fontColor + ":" + blod;
Font font = fonts.get( cacheKey );
if ( font == null ) {
font = wb.createFont();
if ( !Objects.equals( fontName, "default" ) ) {
font.setFontName( fontName );
}
if ( !Objects.equals( fontColor, "#" ) ) {
font.setColor( getColorIndex( fontColor ) );
}
if ( fontSize != -1 ) {
font.setFontHeightInPoints( ( short ) fontSize );
}
if ( blod ) {
font.setBold( blod );
}
fonts.put( cacheKey, font );
}
return font;
}
private short getColorIndex( String hexColor ) {
if ( colors.size() >= colorSize ) {
log.warn( "Custom color value is out of range, only {} color values are allowed", colorSize );
return IndexedColors.AUTOMATIC.getIndex();
}
Short colorIndex = colors.get( hexColor );
if ( colorIndex == null ) {
int count = 0;
short index = -1;
do {
if ( count ++ >= 10 ) {
index = IndexedColors.AUTOMATIC.getIndex();
break;
}
index = ( short ) ThreadLocalRandom.current().nextInt( colorBegin, colorBound );
} while ( colors.containsValue( index ) );
Color color = Color.decode( hexColor );
byte r = ( byte ) color.getRed();
byte g = ( byte ) color.getGreen();
byte b = ( byte ) color.getBlue();
HSSFPalette palette = ( ( HSSFWorkbook ) wb ).getCustomPalette();
palette.setColorAtIndex( index, r, g, b );
colors.put( hexColor, colorIndex = index );
}
return colorIndex;
}
}
