src/main/java/net/glowstone/block/entity/GlowNoteBlock.java

package net.glowstone.block.entity;

import net.glowstone.GlowChunk;
import net.glowstone.block.GlowBlock;
import net.glowstone.block.GlowBlockState;
import net.glowstone.entity.GlowPlayer;
import org.bukkit.Instrument;
import org.bukkit.Location;
import org.bukkit.Material;
import org.bukkit.Note;
import org.bukkit.block.BlockFace;
import org.bukkit.block.NoteBlock;

/**
 * Represents a noteblock in the world.
 */
public class GlowNoteBlock extends GlowBlockState implements NoteBlock {

    private NoteWrapper wrapper = new NoteWrapper(new Note((byte) 0));

    public GlowNoteBlock(GlowBlock block) {
        super(block);
        if (block.getType() != Material.NOTE_BLOCK) {
            throw new IllegalArgumentException("GlowNoteBlock: expected NOTE_BLOCK, got " + block.getType());
        }
    }

    public Note getNote() {
        return wrapper.note;
    }

    public byte getRawNote() {
        return wrapper.note.getId();
    }

    public void setNote(Note note) {
        this.wrapper.note = note;
    }

    public void setRawNote(byte note) {
        this.wrapper.note = new Note(note);
    }

    public boolean play() {
        return play(instrumentOf(getBlock().getRelative(BlockFace.DOWN).getType()), wrapper.note);
    }

    public boolean play(byte instrument, byte note) {
        if (getBlock().getType() != Material.NOTE_BLOCK) {
            return false;
        }

        Location location = getBlock().getLocation();

        for (GlowPlayer player : getWorld().getRawPlayers()) {
            if (player.canSee(new GlowChunk.Key(getX() >> 4, getZ() >> 4)))
                player.playNote(location, instrument, note);
        }

        return true;
    }

    public boolean play(Instrument instrument, Note note) {
        return play(instrument.getType(), note.getId());
    }

    ////////////////////////////////////////////////////////////////////////////
    // Internals

    private static Instrument instrumentOf(Material mat) {
        // TODO: check more blocks.
        switch (mat) {
            case WOOD:
            case NOTE_BLOCK:
            case WORKBENCH:
            case LOG:
                return Instrument.BASS_GUITAR;
            case SAND:
            case GRAVEL:
            case SOUL_SAND:
                return Instrument.SNARE_DRUM;
            case GLASS:
                return Instrument.STICKS;
            case STONE:
            case OBSIDIAN:
            case NETHERRACK:
            case BRICK:
                return Instrument.BASS_DRUM;
            case DIRT:
            case AIR:
            default:
                return Instrument.PIANO;
        }
    }

    private class NoteWrapper {
        public Note note;

        public NoteWrapper(Note note) {
            this.note = note;
        }
    }

    @Override
    public GlowNoteBlock shallowClone() {
        GlowNoteBlock result = new GlowNoteBlock(getBlock());
        result.wrapper = wrapper;
        return result;
    }

    @Override
    public void destroy() {
        setRawNote((byte) 0);
    }

}