自动调整iFrame的大小，然后返回到原始高度

Question

我有一个高度为320px的页面，其中一个div隐藏了jQuery。当div显示时，高度为450px。我把它放在一个iFrame中，并希望在显示隐藏的div时调整iFrame的大小。我从this question -

找到了这段代码

jQuery(function($){
  var lastHeight = 0, curHeight = 0, $frame = $('iframe:eq(0)');
  setInterval(function(){
    curHeight = $frame.contents().find('body').height();
    if ( curHeight != lastHeight ) {
      $frame.css('height', (lastHeight = curHeight) + 'px' );
    }
  },500);
});

它可以增加帧的高度，但是我无法弄清楚当div再次被隐藏时如何将其调整回320px。修改此代码的最佳方法是什么，以便iFrame恢复到原始高度320px？