2011年12月11日日曜日

ディレクトリコピーするツールをSwingで作ってみた

最近Swingの勉強をし始めたので、ディレクトリを別のディレクトリにコピー
するツールをSwingで作ってみました。
コピー中に進捗が分かるように、進捗バー(JProgressBar)をつけてみました。
また、ディレクトリをコピーする処理は別スレッドで処理するようにしました。

デザインは少しいけてませんが、それはまたということで。

・初期画面

・コピーボタン押下後

・コピー完了後






import java.awt.BorderLayout;
import java.awt.Dimension;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.nio.channels.FileChannel;

import javax.swing.BoxLayout;
import javax.swing.JButton;
import javax.swing.JFileChooser;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JPanel;
import javax.swing.JProgressBar;
import javax.swing.SwingWorker;
import javax.swing.Timer;


/**
 * ディレクトリをコピー
 * 処理中は進捗バーを表示
 */
public class DirCopy extends JFrame implements ActionListener {

 JProgressBar progress = new JProgressBar();

 JButton buttonFromCopy = null;
 JButton buttonToCopy = null;
 JButton buttonCopy = null;
 
 int fileSize = 0;
 float currentFileSizePercent = 0;
 
 File fileFrom = null;
 File fileTo = null;
 
 boolean flg1 = false;
 boolean flg2 = false;
 
 Timer time = null;

 
 public static void main(String[] args) {
  DirCopy frame = new DirCopy();
  frame.setBounds(20, 20, 300, 200);
  frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
  frame.setVisible(true);
 }
 
 DirCopy() {
  JLabel label = new JLabel("進捗バー");
  label.setAlignmentX(0.5f);
  
  progress.setPreferredSize(new Dimension(10, 100));
  progress.setStringPainted(true);

  buttonFromCopy = new JButton("コピー元");
  buttonFromCopy.addActionListener(this);
  buttonFromCopy.setActionCommand("FromCopy");
  buttonFromCopy.setAlignmentY(0.5f);

  buttonToCopy = new JButton("コピー先");
  buttonToCopy.setActionCommand("ToCopy");
  buttonToCopy.addActionListener(this);
  buttonToCopy.setAlignmentY(0.5f);
  
  buttonCopy = new JButton("コピー");
  buttonCopy.setEnabled(false);
  buttonCopy.setActionCommand("copy");
  buttonCopy.addActionListener(this);

  JButton buttonClose = new JButton("close");
  buttonClose.setAlignmentX(0.5f);
  buttonClose.setActionCommand("close");
  buttonClose.addActionListener(this);

  JPanel panel1 = new JPanel();
  JPanel panel2 = new JPanel();
  JPanel panel3 = new JPanel();
  panel1.setLayout(new BoxLayout(panel1, BoxLayout.Y_AXIS));
  panel2.setLayout(new BoxLayout(panel2, BoxLayout.LINE_AXIS));
  panel3.setLayout(new BoxLayout(panel3, BoxLayout.Y_AXIS));
  panel1.add(label);
  panel1.add(progress);
  panel2.add(buttonFromCopy);
  panel2.add(buttonToCopy);
  panel2.add(buttonCopy);
  panel3.add(buttonClose);

  getContentPane().add(panel1, BorderLayout.PAGE_START);
  getContentPane().add(panel2, BorderLayout.CENTER);
  getContentPane().add(panel3, BorderLayout.PAGE_END);

 }
 
 @Override
 public void actionPerformed(ActionEvent e) {

  if (e.getActionCommand().equals("time")) {
   if (currentFileSizePercent < 100) {
    progress.setValue((int)currentFileSizePercent);
    progress.setString("処理中です");
   } else {
    progress.setValue((int)currentFileSizePercent);
    progress.setString("完了");
    time.stop();
   }
  }
  
  // 閉じるボタン押下
  if (e.getActionCommand().equals("close")) {
   System.exit(0);
  }
  
  // コピー元ボタン押下
  if (e.getActionCommand().equals("FromCopy")) {
   JFileChooser fChooser = new JFileChooser();
   //ディレクトリのみ選択可能
   fChooser.setFileSelectionMode(JFileChooser.DIRECTORIES_ONLY);
   int select = fChooser.showOpenDialog(this);
   if(select == JFileChooser.APPROVE_OPTION) {
    fileFrom = fChooser.getSelectedFile();
    fileSize = (int) getFileSize(fileFrom);
    flg1 = true;
    // コピーボタンを活性化
    if (flg2) buttonCopy.setEnabled(true);
   }
  }
  
  // コピー先ボタン押下
  if (e.getActionCommand().equals("ToCopy")) {
   JFileChooser fChooser = new JFileChooser();
   // ディレクトリのみ選択可能
   fChooser.setFileSelectionMode(JFileChooser.DIRECTORIES_ONLY);
   int select = fChooser.showOpenDialog(this);
   if (select == JFileChooser.APPROVE_OPTION) {
    fileTo = fChooser.getSelectedFile();
    flg2 = true;
    // コピーボタンを活性化
    if (flg1) buttonCopy.setEnabled(true); 
   }
  }
  
  // コピーボタン押下
  if(e.getActionCommand().equals("copy")) {
   File dir = new File(fileTo.getPath() + "/" + fileFrom.getName());
   if(dir.isDirectory() && dir.exists()) {
    dirDelete(dir);
   }
   time = new Timer(1000, this);
   time.setActionCommand("time");
   time.start();
   FileCopy fileCopy = new FileCopy();
   // 別スレッドでファイルコピー
   fileCopy.execute();
  }
 }
 
 
 /**
  * ディレクトリ以下のファイルサイズを取得
  * @param dir ディレクトリ
  * @return ファイルサイズ
  */
 private long getFileSize(File dir) {
  long size = 0;
  File[] fileList = dir.listFiles();
  if(fileList != null) {
   for(File file : fileList) {
    if(file.isFile()) {
     size += file.length();
    } else {
     size += getFileSize(file);
    }
   }
  }
  
  return size;
 }
 
 /**
  * @param dirFrom コピー元のディレクトリ
  * @param dirTo コピー先のディレクトリ
  */
 private void fileCopy(File dirFrom, File dirTo) {  
  File[] fromFile = dirFrom.listFiles();
  dirTo = new File(dirTo.getPath() + "/" + dirFrom.getName());
  dirTo.mkdir();

  if(fromFile != null) {
   for(File f : fromFile) {
    // ファイルの場合
    if (f.isFile()) {
     copy(f, dirTo);
    // ディレクトリの場合
    } else {
     fileCopy(f, dirTo);
    }
   }
  }
 }
 
 /**
  * ファイルをディレクトリにコピーする
  * @param file コピーするファイル
  * @param dir コピーするファイルの格納フォルダ
  */
 private void copy(File file, File dir) {
  File copyFile = new File(dir.getPath() + "/" + file.getName());
  FileChannel channelFrom = null;
  FileChannel channelTo = null;
  
  try {
   copyFile.createNewFile();
   channelFrom = new FileInputStream(file).getChannel();
   channelTo = new FileOutputStream(copyFile).getChannel();
   //ファイルをコピー
   channelFrom.transferTo(0, channelFrom.size(), channelTo);
   float currentFileSize = getFileSize(new File(fileTo.getPath() + "/" + fileFrom.getName()));
   //処理の進捗率をファイルサイズから求める
   currentFileSizePercent = (float) (currentFileSize / fileSize) * 100;
  } catch (IOException e) {
   e.printStackTrace();
  } finally {
   try {
    if (channelFrom != null) { channelFrom.close(); }
    if (channelTo != null) { channelTo.close(); }
    } catch (IOException e) { e.printStackTrace(); }
  }
 }
 
 /**
  * 指定したディレクトリを削除
  * @param file 削除ディレクトリ
  */
 private void dirDelete(File dir) {
  for (File f : dir.listFiles()) {
   if (f.isFile()) {
    f.delete();
   } else {
    dirDelete(f);
    f.delete();
   } 
  }
  dir.delete();
 }
 /**
  * ディレクトリをコピー
  */
 class FileCopy extends SwingWorker<Object, Object> {
  @Override
  public Object doInBackground() {
   fileCopy(fileFrom, fileTo);
   return null;
  }
 }
}

2011年12月7日水曜日

scalaでXMLを操作するサンプル

アテンドのAPIを使ってXML形式で情報を取得し
必要な情報を抽出する処理をscalaで書いてみました。
関数名が「/」の関数があるけど、分かりづらい。。。


import scala.xml._

// XMLを取得
val xml:Elem = XML.load("http://api.atnd.org/events/?keyword=scala")

// ルートノード以下のすべてのノードを取得
for (node:Node <- xml.descendant) {
 // タイトルにscalaが含まれている
 if(node.\("title").text.contains("scala")) {
  println("タイトル:" + node.\("title").text)
  println("場所:" + node.\("place").text)
  println("時間:" + getDate(node.\("started_at").text) +" to " 
  + getDate(node.\("ended_at").text))
  println("URL:" + node.\("event_url").text + "\n")
 }
}

// 日付形式を変換
def getDate(date:String):String = {
 date.replace("-", "/").replace("T", " ").substring(0, 19)
}

2011年12月4日日曜日

scalaでクイックソート

scalaでクイックソートを実装してみました。
一言付け加えますが、このコードはscalaの特徴である関数型をフル活用
したコードになっておりませんので、関数型をフル活用したコードについては
ネットで検索すればいくらでも出てくるのでそちらを参照してください。



import scala.util._
def sort(A:Array[Int], left:Int, right:Int) : Int= {
 if(left<right){
  val p:Int = partition(A, left, right)
  sort(A, left, p-1)
  sort(A, p+1, right)
 }
 1
}
def partition(A:Array[Int], left:Int, right:Int) : Int = {
 val p = left
 var tmp = A(p)
 A(p) = A(right)
 A(right) = tmp
 var store = left
 for(i <- left to right) {
  if(A(i) < A(right)) {
   tmp = A(i)
   A(i) = A(store)
   A(store) = tmp
   store+=1
  }
 }
 tmp = A(store)
 A(store) = A(right)
 A(right) = tmp
 store
}
def randomArray(no:Int):Array[Int] = {
 val r = new Random
 var A = new Array[Int](no)
 for(i <- 0 to A.length-1) A(i) = r.nextInt(no)
 A
}

val A = randomArray(20)
sort(A, 0 ,A.length - 1)
for(i <- 0 to A.length - 1) {
 println(A(i))
}

クイックソート

勉強がてらJavaでクイックソートを実装してみました。
partionメソッドのピボットの選び方や要素数が少ない場合は、挿入ソートに切り替えるなど
改善の余地はありますが、また時間があればやってみます。


import java.util.Random;
public class QuickSort {

 static int NO = 1000000;

 public static void main(String[] args) {
  
  int A[] = makeArray(NO);
  sort(A, 0, A.length - 1);
  for (int b : A) {
    System.out.println(b);
  }
 
 }

 public static void sort(int[] A, int left, int right) {
  
  if (left < right) {
   int p = partition(A, left, right);
   sort(A, left, p - 1);
   sort(A, p + 1, right);
  }
 
 }

 private static int partition(int[] A, int left, int right) {
 
  int p = A.length % (right - left) + left;
  swap(A, right, p);

  int store = left;
  for(int i=left; i<right; ++i) {
   if(A[i] <= A[right]) {
   swap(A, store, i);
   ++store;
   }
  }
  
  swap(A, store, right);
  return store;
 }
 
 private static void swap(int[] A, int a, int b) {
  int tmp = A[a];
  A[a] = A[b];
  A[b] = tmp;
 }
 
 private static int[] makeArray(int no) {
  Random random = new Random();
  int[] A = new int[no];
  for(int i=0; i<no; ++i) {
   A[i] = random.nextInt(no); 
  }
  return A;
 }
}