Tuesday, May 15, 2012

groovyで画像を暗視カメラ風に変換する

groovyで画像を暗視カメラ風に変換するには、以下のコードを実行します。
@Grab(group='com.jhlabs', module='filters', version='2.0.235')
import java.io.*
import java.awt.*
import java.awt.geom.*
import java.awt.image.*
import javax.imageio.*
import com.jhlabs.composite.*
import com.jhlabs.image.*

// 画像を暗視カメラ風に変換する
def img = ImageIO.read(new File("sf2.jpg"))

// 明るさを80%、コントラストを150% 
def ctf = new ContrastFilter()
ctf.setBrightness(0.8f)
ctf.setContrast(1.5f)
img = ctf.filter(img, null)

def mf = new MedianFilter()
img = mf.filter(img, null)

def gf = new GaussianFilter()
gf.setRadius(3f)
img = gf.filter(img, null)

def grf = new GrayscaleFilter()
img = grf.filter(img, null)

def wr = img.getRaster()

buf = new int[wr.getNumDataElements()]

for(int ly=1;ly<wr.getHeight();ly++){
  for(int lx=0;lx<wr.getWidth();lx++){
    wr.getPixel(lx, ly, buf)
    buf[0] = 0
    buf[1] = (ly%2==0)?buf[1]:(int)(buf[1]/2)
    buf[2] = 0
    wr.setPixel(lx, ly, buf)
  }
}
ImageIO.write(img, "png", new File("nightvision.png"))

元画像
出力画像

動作環境
groovy 1.8.6, JDK7 update4

No comments: